From 27da2e5e0febbd8cebde8901f6773a0e5babc26c Mon Sep 17 00:00:00 2001
From: Shigma <1700011071@pku.edu.cn>
Date: Mon, 5 Apr 2021 03:06:11 +0800
Subject: [PATCH] feat(chat): support send message

---
 packages/plugin-chat/client/chat.vue          | 28 ++++++++++++-----
 packages/plugin-chat/src/debug.ts             |  2 +-
 packages/plugin-chat/src/index.ts             | 14 +++++++--
 .../client/components/chat-panel.vue          | 30 ++++++++++---------
 .../client/components/message.vue             | 10 +++----
 .../plugin-webui/client/views/sandbox.vue     |  2 +-
 packages/plugin-webui/src/adapter.ts          |  2 +-
 packages/plugin-webui/src/index.ts            |  7 +++++
 packages/plugin-webui/src/server.ts           | 29 ++++++++----------
 9 files changed, 77 insertions(+), 47 deletions(-)

diff --git a/packages/plugin-chat/client/chat.vue b/packages/plugin-chat/client/chat.vue
index 97508e3e8a..dda714d7d3 100644
--- a/packages/plugin-chat/client/chat.vue
+++ b/packages/plugin-chat/client/chat.vue
@@ -1,5 +1,5 @@
 <template>
-  <k-chat-panel class="page-chat" :messages="messages" pinned>
+  <k-chat-panel class="page-chat" :messages="messages" pinned @click="handleClick" @send="handleSend">
     <template #default="{ channelName, username, timestamp, content }">
       <div class="header">
         <span class="channel">{{ channelName || '私聊' }}</span>
@@ -8,24 +8,38 @@
       </div>
       <k-message :content="content"/>
     </template>
+    <template #footer v-if="activeMessage">
+      发送到频道：{{ activeMessage.channelName }}
+    </template>
   </k-chat-panel>
 </template>
 
 <script lang="ts" setup>
 
-import { receive, storage } from '~/client'
-
-interface Message {
-  username: string
-  content: string
-}
+import { receive, storage, send, user } from '~/client'
+import { ref } from 'vue'
+import type { Message } from '../src'
 
+const pinned = ref(true)
+const activeMessage = ref<Message>()
 const messages = storage.create<Message[]>('chat', [])
 
 receive('chat', (body) => {
   messages.value.push(body)
 })
 
+function handleClick(message: Message) {
+  activeMessage.value = message
+}
+
+function handleSend(content: string) {
+  if (!activeMessage.value) return
+  pinned.value = false
+  const { platform, selfId, channelId } = activeMessage.value
+  const { token, id } = user.value
+  send('chat', { token, id, content, platform, selfId, channelId })
+}
+
 function formatDateTime(timestamp: number) {
   const date = new Date(timestamp)
   const now = new Date()
diff --git a/packages/plugin-chat/src/debug.ts b/packages/plugin-chat/src/debug.ts
index ddc6e97a04..74350838ac 100644
--- a/packages/plugin-chat/src/debug.ts
+++ b/packages/plugin-chat/src/debug.ts
@@ -27,7 +27,7 @@ const cqTypes = {
   card: '卡片消息',
 }
 
-interface Message {
+export interface Message {
   avatar?: string
   content?: string
   abstract?: string
diff --git a/packages/plugin-chat/src/index.ts b/packages/plugin-chat/src/index.ts
index 5828970ef6..7ac278e432 100644
--- a/packages/plugin-chat/src/index.ts
+++ b/packages/plugin-chat/src/index.ts
@@ -1,8 +1,11 @@
 import { Context } from 'koishi-core'
-import {} from 'koishi-plugin-webui'
+// eslint-disable-next-line @typescript-eslint/no-unused-vars
+import { SocketChannel } from 'koishi-plugin-webui'
 import { resolve } from 'path'
 import debug, { DebugConfig } from './debug'
 
+export * from './debug'
+
 export interface Config extends DebugConfig {}
 
 export const name = 'chat'
@@ -10,9 +13,16 @@ export const name = 'chat'
 export function apply(ctx: Context, options: Config = {}) {
   ctx.plugin(debug, options)
 
-  ctx.with(['koishi-plugin-webui'], (ctx) => {
+  ctx.with(['koishi-plugin-webui'] as const, (ctx, { SocketChannel }) => {
     const { devMode } = ctx.webui.config
     const filename = devMode ? '../client/index.ts' : '../dist/index.js'
     ctx.webui.addEntry(resolve(__dirname, filename))
+
+    SocketChannel.prototype['$chat'] = async function (this: SocketChannel, { id, token, content, platform, selfId, channelId }) {
+      const user = await this.validate(id, token, ['name'])
+      if (!user) return
+      content = await this.app.transformAssets(content)
+      this.app.bots[`${platform}:${selfId}`]?.sendMessage(channelId, content)
+    }
   })
 }
diff --git a/packages/plugin-webui/client/components/chat-panel.vue b/packages/plugin-webui/client/components/chat-panel.vue
index 7bb8471de2..6ccea8b74e 100644
--- a/packages/plugin-webui/client/components/chat-panel.vue
+++ b/packages/plugin-webui/client/components/chat-panel.vue
@@ -1,11 +1,14 @@
 <template>
   <k-card class="k-chat-panel">
     <div class="k-chat-body" ref="body">
-      <div class="k-chat-message" v-for="(message, index) in messages" :key="index">
+      <div class="k-chat-message" v-for="(message, index) in messages" :key="index" @click="$emit('click', message)">
         <slot v-bind="message"/>
       </div>
     </div>
-    <k-input v-model="text" @enter="onEnter" @paste="onPaste"></k-input>
+    <div class="k-chat-footer">
+      <slot name="footer"/>
+      <k-input v-model="text" @enter="onEnter" @paste="onPaste"/>
+    </div>
   </k-card>
 </template>
 
@@ -14,7 +17,7 @@
 import { ref, watch, defineProps, onMounted, nextTick, defineEmit } from 'vue'
 import { segment } from '~/client'
 
-const emit = defineEmit(['enter'])
+const emit = defineEmit(['send', 'click'])
 const props = defineProps<{ messages: any[], pinned?: boolean }>()
 
 const text = ref('')
@@ -35,7 +38,7 @@ watch(props.messages, () => {
 
 function onEnter() {
   if (!text.value) return
-  emit('enter', text.value)
+  emit('send', text.value)
   text.value = ''
 }
 
@@ -46,7 +49,7 @@ async function onPaste(event: ClipboardEvent) {
     const file = item.getAsFile()
     const reader  = new FileReader()
     reader.addEventListener('load', function () {
-      emit('enter', segment.image('base64://' + reader.result.slice(22)))
+      emit('send', segment.image('base64://' + reader.result.slice(22)))
     }, false)
     reader.readAsDataURL(file)
   }
@@ -62,12 +65,13 @@ $padding: 1.5rem;
   height: 100%;
   position: relative;
 
+  .k-card-body {
+    display: flex;
+    flex-direction: column;
+    height: -webkit-fill-available;
+  }
+
   .k-chat-body {
-    position: absolute;
-    top: $padding;
-    left: $padding;
-    right: $padding;
-    bottom: 2rem + $padding * 2;
     overflow-x: visible;
     overflow-y: auto;
   }
@@ -83,10 +87,8 @@ $padding: 1.5rem;
   }
 
   .k-input {
-    position: absolute;
-    bottom: $padding;
-    left: $padding;
-    right: $padding;
+    padding-top: 1rem;
+    width: -webkit-fill-available;
   }
 }
 
diff --git a/packages/plugin-webui/client/components/message.vue b/packages/plugin-webui/client/components/message.vue
index d8301be634..7b14e752b2 100644
--- a/packages/plugin-webui/client/components/message.vue
+++ b/packages/plugin-webui/client/components/message.vue
@@ -2,7 +2,8 @@
   <div class="k-message">
     <template v-for="({ type, data }, index) in segment.parse(content)" :key="index">
       <span v-if="type === 'text'">{{ data.content }}</span>
-      <img v-else-if="type === 'image'" :src="normalizeUrl(data)"/>
+      <span v-else-if="type === 'quote'">[引用回复]</span>
+      <img v-else-if="type === 'image'" :src="normalizeUrl(data.url)"/>
     </template>
   </div>
 </template>
@@ -16,10 +17,9 @@ defineProps<{
   content: string
 }>()
 
-function normalizeUrl(data: any) {
-  if (!data.url) return console.log(data), ''
-  if (data.url.startsWith('base64://')) return data.url
-  return KOISHI_CONFIG.endpoint + '/assets/' + encodeURIComponent(data.url)
+function normalizeUrl(url: string) {
+  if (!KOISHI_CONFIG.whitelist.some(prefix => url.startsWith(prefix))) return url
+  return KOISHI_CONFIG.endpoint + '/assets/' + encodeURIComponent(url)
 }
 
 </script>
diff --git a/packages/plugin-webui/client/views/sandbox.vue b/packages/plugin-webui/client/views/sandbox.vue
index 1bde2d9acd..e36ac6486a 100644
--- a/packages/plugin-webui/client/views/sandbox.vue
+++ b/packages/plugin-webui/client/views/sandbox.vue
@@ -1,5 +1,5 @@
 <template>
-  <k-chat-panel class="sandbox" :messages="messages" @enter="sendSandbox" :pinned="pinned">
+  <k-chat-panel class="sandbox" :messages="messages" @send="sendSandbox" :pinned="pinned">
     <template #default="{ from, content }">
       <k-message :class="from" :content="content"/>
     </template>
diff --git a/packages/plugin-webui/src/adapter.ts b/packages/plugin-webui/src/adapter.ts
index cb4a122d57..a7d97ce698 100644
--- a/packages/plugin-webui/src/adapter.ts
+++ b/packages/plugin-webui/src/adapter.ts
@@ -7,7 +7,7 @@ const states: Record<string, [string, number, SocketChannel]> = {}
 
 const TOKEN_TIMEOUT = Time.minute * 10
 
-class SocketChannel {
+export class SocketChannel {
   readonly app: App
   readonly id = Random.uuid()
 
diff --git a/packages/plugin-webui/src/index.ts b/packages/plugin-webui/src/index.ts
index d7d7a47d6e..f93fcba7ae 100644
--- a/packages/plugin-webui/src/index.ts
+++ b/packages/plugin-webui/src/index.ts
@@ -88,16 +88,23 @@ const defaultConfig: Config = {
   apiPath: '/status',
   uiPath: '/console',
   selfUrl: '',
+  whitelist: [],
   title: 'Koishi 控制台',
   expiration: Time.week,
   tickInterval: Time.second * 5,
   refreshInterval: Time.hour,
 }
 
+const builtinWhitelist = [
+  'http://gchat.qpic.cn/',
+  'http://c2cpicdw.qpic.cn',
+]
+
 export const name = 'status'
 
 export function apply(ctx: Context, config: Config = {}) {
   config = Object.assign(defaultConfig, config)
+  config.whitelist.push(...builtinWhitelist)
 
   ctx.webui = new WebServer(ctx, config)
 
diff --git a/packages/plugin-webui/src/server.ts b/packages/plugin-webui/src/server.ts
index 144f281497..bd95210fbf 100644
--- a/packages/plugin-webui/src/server.ts
+++ b/packages/plugin-webui/src/server.ts
@@ -10,18 +10,20 @@ import type PluginVue from '@vitejs/plugin-vue'
 
 Context.delegate('webui')
 
-export interface Config extends WebAdapter.Config, Profile.Config, Meta.Config, Registry.Config, Statistics.Config {
+interface BaseConfig {
   title?: string
-  selfUrl?: string
-  uiPath?: string
   devMode?: boolean
+  uiPath?: string
+  whitelist?: string[]
+}
+
+export interface Config extends BaseConfig, WebAdapter.Config, Profile.Config, Meta.Config, Registry.Config, Statistics.Config {
+  title?: string
+  selfUrl?: string
 }
 
-export interface ClientConfig {
-  title: string
-  uiPath: string
+export interface ClientConfig extends Required<BaseConfig> {
   endpoint: string
-  devMode: boolean
   extensions: string[]
 }
 
@@ -43,11 +45,6 @@ export class WebServer {
   private vite: Vite.ViteDevServer
   private readonly [Context.current]: Context
 
-  static whitelist = [
-    'http://gchat.qpic.cn/',
-    'http://c2cpicdw.qpic.cn',
-  ]
-
   constructor(private ctx: Context, public config: Config) {
     this.root = resolve(__dirname, '..', config.devMode ? 'client' : 'dist')
     this.sources = {
@@ -75,7 +72,7 @@ export class WebServer {
   }
 
   private async start() {
-    const { uiPath, apiPath } = this.config
+    const { uiPath, apiPath, whitelist } = this.config
     await Promise.all([this.createVite(), this.createAdapter()])
 
     this.ctx.router.get(uiPath + '(/.+)*', async (ctx) => {
@@ -106,7 +103,7 @@ export class WebServer {
     })
 
     this.ctx.router.get(apiPath + '/assets/:url', async (ctx) => {
-      if (!WebServer.whitelist.some(prefix => ctx.params.url.startsWith(prefix))) {
+      if (!whitelist.some(prefix => ctx.params.url.startsWith(prefix))) {
         console.log(ctx.params.url)
         return ctx.status = 403
       }
@@ -117,12 +114,12 @@ export class WebServer {
 
   private async transformHtml(template: string) {
     if (this.vite) template = await this.vite.transformIndexHtml(this.config.uiPath, template)
-    const { apiPath, uiPath, devMode, selfUrl, title } = this.config
+    const { apiPath, uiPath, devMode, selfUrl, title, whitelist } = this.config
     const endpoint = selfUrl + apiPath
     const extensions = Object.entries(this.entries).map(([name, filename]) => {
       return this.config.devMode ? '/vite/@fs' + filename : `./${name}`
     })
-    const global: ClientConfig = { title, uiPath, endpoint, devMode, extensions }
+    const global: ClientConfig = { title, uiPath, endpoint, devMode, extensions, whitelist }
     const headInjection = `<script>KOISHI_CONFIG = ${JSON.stringify(global)}</script>`
     return template.replace('</title>', '</title>' + headInjection)
   }