feat(chat): add config maxMessages

koishijs · Apr 18, 2021 · 8b91a7b · 8b91a7b
1 parent 2060ed5
commit 8b91a7b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/packages/plugin-chat/client/chat.vue b/packages/plugin-chat/client/chat.vue
@@ -46,6 +46,7 @@ const divs = ref<Record<string, HTMLElement>>({})
 
 receive('chat', (body) => {
   messages.value.push(body)
+  messages.value.splice(0, messages.value.length - KOISHI_CONFIG.maxMessages)
 })
 
 function isSuccessive({ quote, userId, channelId }: Message, index: number) {

diff --git a/packages/plugin-chat/src/index.ts b/packages/plugin-chat/src/index.ts
@@ -19,15 +19,16 @@ declare module 'koishi-core' {
 }
 
 declare module 'koishi-plugin-webui' {
-  interface ClientConfig {
-    whitelist: string[]
-  }
+  interface ClientConfig extends ClientExtension {}
 }
 
-export interface Config extends ReceiverConfig {
+interface ClientExtension {
   whitelist?: string[]
+  maxMessages?: number
 }
 
+export interface Config extends ReceiverConfig, ClientExtension {}
+
 template.set('chat', {
   send: '[{{ channelName || "私聊" }}] {{ abstract }}',
   receive: '[{{ channelName || "私聊" }}] {{ username }}: {{ abstract }}',
@@ -52,9 +53,14 @@ const builtinWhitelist = [
   'http://c2cpicdw.qpic.cn',
 ]
 
+const defaultOptions: Config = {
+  maxMessages: 1000,
+}
+
 export const name = 'chat'
 
 export function apply(ctx: Context, options: Config = {}) {
+  options = { ...defaultOptions, ...options }
   ctx.plugin(receiver, options)
 
   ctx.on('chat/receive', async (message, session) => {
@@ -71,6 +77,7 @@ export function apply(ctx: Context, options: Config = {}) {
     const whitelist = [...builtinWhitelist, ...options.whitelist || []]
 
     ctx.webui.global.whitelist = whitelist
+    ctx.webui.global.maxMessages = options.maxMessages
     ctx.webui.addEntry(resolve(__dirname, filename))
 
     ctx.webui.addListener('chat', async function ({ id, token, content, platform, selfId, channelId }) {