@satorijs/adapter-lark 3.6.0 → 3.6.2

package/src/message.ts

@@ -1,28 +1,27 @@
-import { Context, h, MessageEncoder } from '@satorijs/core'
+import { Context, Dict, h, MessageEncoder } from '@satorijs/core'
 import { LarkBot } from './bot'
-import { BaseResponse, Lark, MessageContent, MessageType } from './types'
+import { BaseResponse, Lark, MessageContent } from './types'
 import { extractIdType } from './utils'
 
-export interface Addition {
-  file: MessageContent.MediaContents
-  type: MessageType
-}
-
 export class LarkMessageEncoder<C extends Context = Context> extends MessageEncoder<C, LarkBot<C>> {
-  private quote: string | undefined
-  private content = ''
-  private addition: Addition
-  // TODO: currently not used, would be supported in the future
-  private richText: MessageContent.RichText[string]
+  private quote: Dict | undefined
+  private textContent = ''
+  private richContent: MessageContent.RichText.Paragraph[] = []
+  private card: MessageContent.Card | undefined
+  private noteElements: MessageContent.Card.NoteElement.InnerElement[] | undefined
+  private actionElements: MessageContent.Card.Element[] = []
 
   async post(data?: any) {
     try {
       let resp: BaseResponse & { data?: Lark.Message }
-      if (this.quote) {
-        resp = await this.bot.internal.replyImMessage(this.quote, data)
+      if (this.quote?.id) {
+        resp = await this.bot.internal.replyImMessage(this.quote.id, {
+          ...data,
+          reply_in_thread: this.quote.replyInThread,
+        })
       } else {
         data.receive_id = this.channelId
-        resp = await this.bot.internal?.createImMessage(data, {
+        resp = await this.bot.internal.createImMessage(data, {
           receive_id_type: extractIdType(this.channelId),
         })
       }
@@ -46,131 +45,279 @@ export class LarkMessageEncoder<C extends Context = Context> extends MessageEnco
     }
   }
 
-  async flush() {
-    if (this.content === '' && !this.addition && !this.richText) return
-
-    let message: MessageContent.Contents
-    if (this.addition) {
-      message = {
-        ...message,
-        ...this.addition.file,
-      }
+  private flushText(button = false) {
+    if ((this.textContent || !button) && this.actionElements.length) {
+      this.card!.elements.push({ tag: 'action', actions: this.actionElements, layout: 'flow' })
+      this.actionElements = []
     }
-    if (this.richText) {
-      message = { zh_cn: this.richText }
+    if (this.textContent) {
+      this.richContent.push([{ tag: 'md', text: this.textContent }])
+      if (this.noteElements) {
+        this.noteElements.push({ tag: 'plain_text', content: this.textContent })
+      } else if (this.card) {
+        this.card.elements.push({ tag: 'markdown', content: this.textContent })
+      }
+      this.textContent = ''
     }
-    if (this.content) {
-      message = { text: this.content }
+  }
+
+  async flush() {
+    this.flushText()
+    if (!this.card && !this.richContent.length) return
+
+    if (this.card) {
+      this.bot.logger.debug('card', JSON.stringify(this.card.elements))
+      await this.post({
+        msg_type: 'interactive',
+        content: JSON.stringify({
+          elements: this.card.elements,
+        }),
+      })
+    } else {
+      await this.post({
+        msg_type: 'post',
+        content: JSON.stringify({
+          zh_cn: {
+            content: this.richContent,
+          },
+        }),
+      })
     }
-    await this.post({
-      msg_type: this.richText ? 'post' : this.addition ? this.addition.type : 'text',
-      content: JSON.stringify(message),
-    })
 
     // reset cached content
     this.quote = undefined
-    this.content = ''
-    this.addition = undefined
-    this.richText = undefined
+    this.textContent = ''
+    this.richContent = []
+    this.card = undefined
   }
 
-  async sendFile(type: 'img' | 'image' | 'video' | 'audio' | 'file', url: string): Promise<Addition> {
+  async createImage(url: string) {
+    const { filename, type, data } = await this.bot.assetsQuester.file(url)
     const payload = new FormData()
+    payload.append('image', new Blob([data], { type }), filename)
+    payload.append('image_type', 'message')
+    const { data: { image_key } } = await this.bot.internal.createImImage(payload)
+    return image_key
+  }
 
-    const assetKey = type === 'img' || type === 'image' ? 'image' : 'file'
-    const { filename, mime, data } = await this.bot.assetsQuester.file(url)
-    payload.append(assetKey, new Blob([data], { type: mime }), filename)
+  async sendFile(_type: 'video' | 'audio' | 'file', attrs: any) {
+    const url = attrs.src || attrs.url
+    const payload = new FormData()
+    const { filename, type, data } = await this.bot.assetsQuester.file(url)
+    payload.append('file', new Blob([data], { type }), filename)
+    payload.append('file_name', filename)
 
-    if (type === 'img' || type === 'image') {
-      payload.append('image_type', 'message')
-      const { data } = await this.bot.internal.createImImage(payload)
-      return {
-        type: 'image',
-        file: {
-          image_key: data.image_key,
-        },
-      }
+    if (attrs.duration) {
+      payload.append('duration', attrs.duration)
+    }
+
+    if (_type === 'audio') {
+      // FIXME: only support opus
+      payload.append('file_type', 'opus')
+    } else if (_type === 'video') {
+      // FIXME: only support mp4
+      payload.append('file_type', 'mp4')
     } else {
-      let msgType: MessageType = 'file'
-      if (type === 'audio') {
-        // FIXME: only support opus
-        payload.append('file_type', 'opus')
-        msgType = 'audio'
-      } else if (type === 'video') {
-        // FIXME: only support mp4
-        payload.append('file_type', 'mp4')
-        msgType = 'media'
+      const ext = filename.split('.').pop()
+      if (['doc', 'xls', 'ppt', 'pdf'].includes(ext)) {
+        payload.append('file_type', ext)
       } else {
-        const ext = filename.split('.').pop()
-        if (['xls', 'ppt', 'pdf'].includes(ext)) {
-          payload.append('file_type', ext)
-        } else {
-          payload.append('file_type', 'stream')
-        }
+        payload.append('file_type', 'stream')
       }
-      payload.append('file_name', filename)
-      const { data } = await this.bot.internal.createImFile(payload)
-      return {
-        type: msgType,
-        file: {
-          file_key: data.file_key,
+    }
+
+    const { data: { file_key } } = await this.bot.internal.createImFile(payload)
+    await this.post({
+      msg_type: _type === 'video' ? 'media' : _type,
+      content: JSON.stringify({ file_key }),
+    })
+  }
+
+  private createBehavior(attrs: Dict) {
+    const behaviors: MessageContent.Card.ActionBehavior[] = []
+    if (attrs.type === 'link') {
+      behaviors.push({
+        type: 'open_url',
+        default_url: attrs.href,
+      })
+    } else if (attrs.type === 'input') {
+      behaviors.push({
+        type: 'callback',
+        value: {
+          _satori_type: 'command',
+          content: attrs.text,
         },
-      }
+      })
+    } else if (attrs.type === 'action') {
+      // TODO
     }
+    return behaviors.length ? behaviors : undefined
   }
 
   async visit(element: h) {
     const { type, attrs, children } = element
-
-    switch (type) {
-      case 'text':
-        this.content += attrs.content
-        break
-      case 'at': {
-        if (attrs.type === 'all') {
-          this.content += `<at user_id="all">${attrs.name ?? '所有人'}</at>`
-        } else {
-          this.content += `<at user_id="${attrs.id}">${attrs.name}</at>`
-        }
-        break
+    if (type === 'text') {
+      this.textContent += attrs.content
+    } else if (type === 'at') {
+      if (attrs.type === 'all') {
+        this.textContent += `<at user_id="all">${attrs.name ?? '所有人'}</at>`
+      } else {
+        this.textContent += `<at user_id="${attrs.id}">${attrs.name}</at>`
       }
-      case 'a':
-        await this.render(children)
-        if (attrs.href) this.content += ` (${attrs.href})`
-        break
-      case 'p':
-        if (!this.content.endsWith('\n')) this.content += '\n'
+    } else if (type === 'a') {
+      await this.render(children)
+      if (attrs.href) this.textContent += ` (${attrs.href})`
+    } else if (type === 'p') {
+      if (!this.textContent.endsWith('\n')) this.textContent += '\n'
+      await this.render(children)
+      if (!this.textContent.endsWith('\n')) this.textContent += '\n'
+    } else if (type === 'br') {
+      this.textContent += '\n'
+    } else if (type === 'sharp') {
+      // platform does not support sharp
+    } else if (type === 'quote') {
+      await this.flush()
+      this.quote = attrs
+    } else if (type === 'img' || type === 'image') {
+      const image_key = await this.createImage(attrs.src || attrs.url)
+      this.textContent += `![${attrs.alt ?? '图片'}](${image_key})`
+      this.flushText()
+      this.richContent.push([{ tag: 'img', image_key }])
+    } else if (['video', 'audio', 'file'].includes(type)) {
+      await this.flush()
+      await this.sendFile(type as any, attrs)
+    } else if (type === 'figure' || type === 'message') {
+      await this.flush()
+      await this.render(children, true)
+    } else if (type === 'hr') {
+      this.flushText()
+      this.richContent.push([{ tag: 'hr' }])
+      this.card?.elements.push({ tag: 'hr' })
+    } else if (type === 'form') {
+      this.flushText()
+      const length = this.card?.elements.length
+      await this.render(children)
+      if (this.card?.elements.length > length) {
+        const elements = this.card?.elements.slice(length)
+        this.card.elements.push({
+          tag: 'form',
+          name: attrs.name || 'Form',
+          elements,
+        })
+      }
+    } else if (type === 'input') {
+      this.flushText()
+      this.card?.elements.push({
+        tag: 'action',
+        actions: [{
+          tag: 'input',
+          name: attrs.name,
+          width: attrs.width,
+          label: attrs.label && {
+            tag: 'plain_text',
+            content: attrs.label,
+          },
+          placeholder: attrs.placeholder && {
+            tag: 'plain_text',
+            content: attrs.placeholder,
+          },
+          behaviors: this.createBehavior(attrs),
+        }],
+      })
+    } else if (type === 'button') {
+      this.card ??= { elements: [] }
+      this.flushText(true)
+      await this.render(children)
+      this.actionElements.push({
+        tag: 'button',
+        text: {
+          tag: 'plain_text',
+          content: this.textContent,
+        },
+        disabled: attrs.disabled,
+        behaviors: this.createBehavior(attrs),
+      })
+      this.textContent = ''
+    } else if (type === 'button-group') {
+      this.flushText()
+      await this.render(children)
+      this.flushText()
+    } else if (type.startsWith('lark:') || type.startsWith('feishu:')) {
+      const tag = type.slice(type.split(':', 1)[0].length + 1)
+      if (tag === 'share-chat') {
+        await this.flush()
+        await this.post({
+          msg_type: 'share_chat',
+          content: JSON.stringify({ chat_id: attrs.chatId }),
+        })
+      } else if (tag === 'share-user') {
+        await this.flush()
+        await this.post({
+          msg_type: 'share_user',
+          content: JSON.stringify({ user_id: attrs.userId }),
+        })
+      } else if (tag === 'system') {
+        await this.flush()
         await this.render(children)
-        if (!this.content.endsWith('\n')) this.content += '\n'
-        break
-      case 'br':
-        this.content += '\n'
-        break
-      case 'sharp':
-        // platform does not support sharp
-        break
-      case 'quote':
+        await this.post({
+          msg_type: 'system',
+          content: JSON.stringify({
+            type: 'divider',
+            params: { divider_text: { text: this.textContent } },
+            options: { need_rollup: attrs.needRollup },
+          }),
+        })
+        this.textContent = ''
+      } else if (tag === 'card') {
         await this.flush()
-        this.quote = attrs.id
-        break
-      case 'img':
-      case 'image':
-      case 'video':
-      case 'audio':
-      case 'file':
-        if (attrs.src || attrs.url) {
-          await this.flush()
-          this.addition = await this.sendFile(type, attrs.src || attrs.url)
-          await this.flush()
+        this.card = {
+          elements: [],
+          header: attrs.title && {
+            template: attrs.color,
+            ud_icon: attrs.icon && {
+              tag: 'standard_icon',
+              token: attrs.icon,
+            },
+            title: {
+              tag: 'plain_text',
+              content: attrs.title,
+            },
+            subtitle: attrs.subtitle && {
+              tag: 'plain_text',
+              content: attrs.subtitle,
+            },
+          },
         }
-        break
-      case 'figure': // FIXME: treat as message element for now
-      case 'message':
-        await this.flush()
         await this.render(children, true)
-        break
-      default:
+      } else if (tag === 'div') {
+        this.flushText()
         await this.render(children)
+        this.card?.elements.push({
+          tag: 'markdown',
+          text_align: attrs.align,
+          text_size: attrs.size,
+          content: this.textContent,
+        })
+        this.textContent = ''
+      } else if (tag === 'note') {
+        this.flushText()
+        this.noteElements = []
+        await this.render(children)
+        this.flushText()
+        this.card?.elements.push({
+          tag: 'note',
+          elements: this.noteElements,
+        })
+        this.noteElements = undefined
+      } else if (tag === 'icon') {
+        this.flushText()
+        this.noteElements?.push({
+          tag: 'standard_icon',
+          token: attrs.token,
+        })
+      }
+    } else {
+      await this.render(children)
     }
   }
 }

package/src/types/api.ts

@@ -16681,6 +16681,8 @@ export interface ReplyImMessageRequest {
   content: string
   /** 消息类型，包括：text、post、image、file、audio、media、sticker、interactive、share_card、share_user */
   msg_type: string
+  /** 是否以话题形式回复。取值为 true 时将以话题形式回复。注意：如果要回复的消息已经是话题形式的消息，则默认以话题形式进行回复。 */
+  reply_in_thread?: boolean
   /** 由开发者生成的唯一字符串序列，用于回复消息请求去重；持有相同uuid的请求1小时内至多成功执行一次 */
   uuid?: string
 }