@dcrays/dcgchat-test 0.2.26 → 0.2.28

package/src/tool.ts

@@ -1,140 +1,136 @@
+import type { OpenClawPluginApi } from 'openclaw/plugin-sdk'
+import { getMsgParams, getMsgStatus, getWsConnection } from './utils/global.js'
+import { dcgLogger } from './utils/log.js'
+import { isWsOpen } from './transport.js'
 
-import type { OpenClawPluginApi } from "openclaw/plugin-sdk";
-import { getWsConnection } from "./connection.js";
-
-let msgParams = {} as {
-  userId: number;
-  token: string;
-  sessionId: string;
-  messageId: string;
-  domainId: string;
-  appId: string;
-  botId: string;
-  agentId: string;
-}
-let msgStatus: 'running' | 'finished' | '' = '';
-export function setMsgParams(params: any) {
-  msgParams = params;
-}
-
-export function getMsgParams() {
-  return msgParams;
-}
-
-export function setMsgStatus(status: 'running' | 'finished' | '') {
-  msgStatus = status;
-}
-
-export function getMsgStatus() {
-  return msgStatus;
-}
-let toolCallId = '';
-let toolName = '';
-type PluginHookName = "before_model_resolve" | "before_prompt_build" | "before_agent_start" | "llm_input" | "llm_output" | "agent_end" | "before_compaction" | "after_compaction" | "before_reset" | "message_received" | "message_sending" | "message_sent" | "before_tool_call" | "after_tool_call" | "tool_result_persist" | "before_message_write" | "session_start" | "session_end" | "subagent_spawning" | "subagent_delivery_target" | "subagent_spawned" | "subagent_ended" | "gateway_start" | "gateway_stop";
+let toolCallId = ''
+let toolName = ''
+type PluginHookName =
+  | 'before_model_resolve'
+  | 'before_prompt_build'
+  | 'before_agent_start'
+  | 'llm_input'
+  | 'llm_output'
+  | 'agent_end'
+  | 'before_compaction'
+  | 'after_compaction'
+  | 'before_reset'
+  | 'message_received'
+  | 'message_sending'
+  | 'message_sent'
+  | 'before_tool_call'
+  | 'after_tool_call'
+  | 'tool_result_persist'
+  | 'before_message_write'
+  | 'session_start'
+  | 'session_end'
+  | 'subagent_spawning'
+  | 'subagent_delivery_target'
+  | 'subagent_spawned'
+  | 'subagent_ended'
+  | 'gateway_start'
+  | 'gateway_stop'
 const eventList = [
-  {event: 'message_received', message: ''},
+  { event: 'message_received', message: '' },
   // {event: 'before_model_resolve', message: ''},
   // {event: 'before_prompt_build', message: '正在查阅背景资料，构建思考逻辑'},
   // {event: 'before_agent_start', message: '书灵墨宝已就位，准备开始执行任务'},
-  {event: 'subagent_spawning', message: ''},
-  {event: 'subagent_spawned', message: ''},
-  {event: 'subagent_delivery_target', message: ''},
+  { event: 'subagent_spawning', message: '' },
+  { event: 'subagent_spawned', message: '' },
+  { event: 'subagent_delivery_target', message: '' },
   // {event: 'llm_input', message: ''},
-  {event: 'llm_output', message: ''},
+  { event: 'llm_output', message: '' },
   // {event: 'agent_end', message: '核心任务已处理完毕...'},
-  {event: 'subagent_ended', message: ''},
+  { event: 'subagent_ended', message: '' },
   // {event: 'before_message_write', message: '正在将本次对话存入记忆库...'},
   // {event: 'message_sending', message: ''},
   // {event: 'message_send', message: ''},
-  {event: 'before_tool_call', message: ''},
-  {event: 'after_tool_call', message: ''},
-];
+  { event: 'before_tool_call', message: '' },
+  { event: 'after_tool_call', message: '' }
+]
 
 function sendToolCallMessage(text: string, toolCallId: string, isCover: number) {
   const ws = getWsConnection()
-  const params = getMsgParams();
-  const status = getMsgStatus();
-  if (ws?.readyState === WebSocket.OPEN && status === 'running') {
-    ws.send(JSON.stringify({
-      messageType: "openclaw_bot_chat",
-      _userId: params?.userId,
-      source: "client",
-      content: {
-        bot_token: params?.token,
-        domain_id: params?.domainId,
-        app_id: params?.appId,
-        bot_id: params?.botId,
-        agent_id: params?.agentId,
-        response: 'all_finished',
-        session_id:params?.sessionId,
-        message_id: params?.messageId || Date.now().toString()
-      },
-    }))
-    ws.send(JSON.stringify({
-      messageType: "openclaw_bot_chat",
-      _userId: params?.userId,
-      source: "client",
-      content: {
-        bot_token: params?.token,
-        domain_id: params?.domainId,
-        app_id: params?.appId,
-        bot_id: params?.botId,
-        agent_id: params?.agentId,
-        tool_call_id: toolCallId,
-        is_cover: isCover,
-        thinking_content: text,
-        response: '',
-        session_id:params?.sessionId,
-        message_id: params?.messageId || Date.now().toString()
-      },
-    }));
-    ws.send(JSON.stringify({
-      messageType: "openclaw_bot_chat",
-      _userId: params?.userId,
-      source: "client",
-      content: {
-        bot_token: params?.token,
-        domain_id: params?.domainId,
-        app_id: params?.appId,
-        bot_id: params?.botId,
-        agent_id: params?.agentId,
-        response: 'all_finished',
-        session_id:params?.sessionId,
-        message_id: params?.messageId || Date.now().toString()
-      },
-    }))
+  const params = getMsgParams()
+  if (isWsOpen()) {
+    ws?.send(
+      JSON.stringify({
+        messageType: 'openclaw_bot_chat',
+        _userId: params?.userId,
+        source: 'client',
+        is_finish: -1,
+        content: {
+          is_finish: -1,
+          bot_token: params?.token,
+          domain_id: params?.domainId,
+          app_id: params?.appId,
+          bot_id: params?.botId,
+          agent_id: params?.agentId,
+          tool_call_id: toolCallId,
+          is_cover: isCover,
+          thinking_content: text,
+          response: '',
+          session_id: params?.sessionId,
+          message_id: params?.messageId || Date.now().toString()
+        }
+      })
+    )
   }
 }
 
 export function monitoringToolMessage(api: OpenClawPluginApi) {
-    for (const item of eventList) {
-      api.on(item.event as PluginHookName, (event: any) => {
-        console.log("🚀 ~ 工具调用结果： ~ event:", item.event)
+  for (const item of eventList) {
+    api.on(item.event as PluginHookName, (event: any) => {
+      const status = getMsgStatus()
+      if (status === 'running') {
+        dcgLogger(`工具调用结果： ~ event:${item.event} ${JSON.stringify(event)}`)
         if (['after_tool_call', 'before_tool_call'].includes(item.event)) {
-          const { result: _result, ...rest } = event;
+          const { result: _result, ...rest } = event
           const text = JSON.stringify({
             type: item.event,
             specialIdentification: 'dcgchat_tool_call_special_identification',
             callId: event.toolCallId || event.runId || Date.now().toString(),
             ...rest,
             status: item.event === 'after_tool_call' ? 'finished' : 'running'
-          });
+          })
           sendToolCallMessage(text, event.toolCallId || event.runId || Date.now().toString(), item.event === 'after_tool_call' ? 1 : 0)
         } else if (item.event) {
+          if (item.event === 'llm_output') {
+            if (event.lastAssistant?.errorMessage === '429-账户额度耗尽') {
+              const ws = getWsConnection()
+              const params = getMsgParams()
+              if (isWsOpen()) {
+                ws?.send(
+                  JSON.stringify({
+                    messageType: 'openclaw_bot_cost',
+                    _userId: params?.userId,
+                    source: 'client',
+                    content: {
+                      bot_token: params?.token,
+                      domain_id: params?.domainId,
+                      app_id: params?.appId,
+                      bot_id: params?.botId,
+                      agent_id: params?.agentId,
+                      response: '您的余额不足，请充值后继续使用',
+                      session_id: params?.sessionId,
+                      message_id: params?.messageId || Date.now().toString()
+                    }
+                  })
+                )
+              }
+              return
+            }
+          }
           const text = JSON.stringify({
-              type: item.event,
-              specialIdentification: 'dcgchat_tool_call_special_identification',
-              toolName: '',
-              callId: event.runId || Date.now().toString(),
-              params: item.message
-            });
-            sendToolCallMessage(text, event.runId || Date.now().toString(), 0)
-          console.log('消息事件：', item.event, item.message)
-        } 
-    //   log?.(`dcgchat[${params?.sessionId}]:11111111 tool message to ${params?.sessionId}, ${JSON.stringify(rest)}`);
-    // } else {
-    //   log?.(`dcgchat[${params?.sessionId}]:22222222 tool ${status} message to ${ws?.readyState}, ${JSON.stringify(rest)}`);
-    // }
-      });
-    }
-}
+            type: item.event,
+            specialIdentification: 'dcgchat_tool_call_special_identification',
+            toolName: '',
+            callId: event.runId || Date.now().toString(),
+            params: item.message
+          })
+          sendToolCallMessage(text, event.runId || Date.now().toString(), 0)
+        }
+      }
+    })
+  }
+}

package/src/transport.ts

@@ -0,0 +1,108 @@
+import { getWsConnection } from './utils/global.js'
+import { dcgLogger } from './utils/log.js'
+
+export type DcgchatMsgContext = {
+  userId: number
+  botToken: string
+  domainId?: string
+  appId?: string
+  botId?: string
+  agentId?: string
+  sessionId: string
+  messageId: string
+}
+
+export function createMsgContext(msg: {
+  _userId: number
+  content: {
+    bot_token: string
+    domain_id?: string
+    app_id?: string
+    bot_id?: string
+    agent_id?: string
+    session_id: string
+    message_id: string
+  }
+}): DcgchatMsgContext {
+  return {
+    userId: msg._userId,
+    botToken: msg.content.bot_token,
+    domainId: msg.content.domain_id,
+    appId: msg.content.app_id,
+    botId: msg.content.bot_id,
+    agentId: msg.content.agent_id,
+    sessionId: msg.content.session_id,
+    messageId: msg.content.message_id
+  }
+}
+
+function buildContent(ctx: DcgchatMsgContext, extra: Record<string, unknown>) {
+  return {
+    bot_token: ctx.botToken,
+    domain_id: ctx.domainId,
+    app_id: ctx.appId,
+    bot_id: ctx.botId,
+    agent_id: ctx.agentId,
+    session_id: ctx.sessionId,
+    message_id: ctx.messageId || Date.now().toString(),
+    ...extra
+  }
+}
+
+function buildEnvelope(ctx: DcgchatMsgContext, extra: Record<string, unknown>) {
+  return {
+    messageType: 'openclaw_bot_chat' as const,
+    _userId: ctx.userId,
+    source: 'client' as const,
+    content: buildContent(ctx, extra)
+  }
+}
+
+export function isWsOpen(): boolean {
+  const isOpen = getWsConnection()?.readyState === WebSocket.OPEN
+  if (!isOpen) {
+    dcgLogger(`socket not ready ${getWsConnection()?.readyState}`, 'error')
+  }
+  return isOpen
+}
+
+/**
+ * Content is stringified separately (double-encoded) to match the
+ * dcgchat wire protocol used by the chat stream path.
+ */
+export function wsSend(ctx: DcgchatMsgContext, extra: Record<string, unknown>): boolean {
+  const ws = getWsConnection()
+  if (ws?.readyState !== WebSocket.OPEN) return false
+  const envelope = buildEnvelope(ctx, extra)
+  ws.send(JSON.stringify({ ...envelope, content: JSON.stringify(envelope.content) }))
+  return true
+}
+
+/**
+ * Content stays as a nested object (single-encoded).
+ * Matches the legacy wire format used by media and outbound-pipeline messages.
+ */
+export function wsSendRaw(ctx: DcgchatMsgContext, extra: Record<string, unknown>): boolean {
+  const ws = getWsConnection()
+  if (isWsOpen()) {
+    ws?.send(JSON.stringify(buildEnvelope(ctx, extra)))
+  }
+  return true
+}
+
+export function sendChunk(ctx: DcgchatMsgContext, text: string): boolean {
+  return wsSend(ctx, { response: text, state: 'chunk' })
+}
+
+export function sendFinal(ctx: DcgchatMsgContext): boolean {
+  dcgLogger(` message handling complete`)
+  return wsSend(ctx, { response: '', state: 'final' })
+}
+
+export function sendText(ctx: DcgchatMsgContext, text: string): boolean {
+  return wsSend(ctx, { response: text })
+}
+
+export function sendError(ctx: DcgchatMsgContext, errorMsg: string): boolean {
+  return wsSend(ctx, { response: `[错误] ${errorMsg}`, state: 'final' })
+}

package/src/types.ts

@@ -2,27 +2,27 @@
  * 插件配置（channels.dcgchat 下的字段）
  */
 export type DcgchatConfig = {
-  enabled?: boolean;
+  enabled?: boolean
   /** 后端 WebSocket 地址，例如 ws://localhost:8080/openclaw/ws */
-  wsUrl?: string;
+  wsUrl?: string
   /** 连接认证 token */
-  botToken?: string;
+  botToken?: string
   /** 用户标识 */
-  userId?: string;
-  domainId?: string;
-  appId?: string;
-};
+  userId?: string
+  domainId?: string
+  appId?: string
+}
 
 export type ResolvedDcgchatAccount = {
-  accountId: string;
-  enabled: boolean;
-  configured: boolean;
-  wsUrl: string;
-  botToken: string;
-  userId: string;
-  domainId?: string;
-  appId?: string;
-};
+  accountId: string
+  enabled: boolean
+  configured: boolean
+  wsUrl: string
+  botToken: string
+  userId: string
+  domainId?: string
+  appId?: string
+}
 
 /**
  * 下行消息：后端 → OpenClaw（用户发的消息）
@@ -33,25 +33,25 @@ export type ResolvedDcgchatAccount = {
 //   text: string;
 // };
 export type InboundMessage = {
-  messageType: string; // "openclaw_bot_chat",
-  _userId: number;
-  source: string; // 'server',
+  messageType: string // "openclaw_bot_chat",
+  _userId: number
+  source: string // 'server',
   // content: string;
   content: {
-    bot_token: string;
-    domain_id?: string;
-    app_id?: string;
-    bot_id?: string;
-    agent_id?: string;
-    session_id: string;
-    message_id: string;
-    text: string;
+    bot_token: string
+    domain_id?: string
+    app_id?: string
+    bot_id?: string
+    agent_id?: string
+    session_id: string
+    message_id: string
+    text: string
     files?: {
-      url: string;
-      name: string;
-    }[];
-  };
-};
+      url: string
+      name: string
+    }[]
+  }
+}
 
 //  {"_userId":40,"content":"{\"bot_token\":\"sk_b7f8a3e1c5d24e6f8a1b3c4d5e6f7a8b\",\"session_id\":\"1\",\"message_id\":\"1\",\"text\":\"你好\"}","messageType":"openclaw_bot_chat","msgId":398599,"source":"server","title":"OPENCLAW机器人对话"}
 
@@ -65,51 +65,62 @@ export type InboundMessage = {
 // };
 
 export type OutboundReply = {
-  messageType: string; // "openclaw_bot_chat",
-  _userId: number; // 100
-  source: string; // 'client',
+  messageType: string // "openclaw_bot_chat",
+  _userId: number // 100
+  source: string // 'client',
   // content: string;
   content: {
-    bot_token: string; // ""
-    session_id: string; // ""
-    message_id: string; // ""
-    response: string; // ""
-    state: string; // final, chunk
-    domain_id?: string;
-    app_id?: string;
-    bot_id?: string;
-    agent_id?: string;
-    files?: {url: string, name: string}[];
-  };
-};
+    bot_token: string // ""
+    session_id: string // ""
+    message_id: string // ""
+    response: string // ""
+    state: string // final, chunk
+    domain_id?: string
+    app_id?: string
+    bot_id?: string
+    agent_id?: string
+    files?: { url: string; name: string }[]
+  }
+}
 
 export interface IResponse<T = unknown> {
   /** 响应状态码 */
-  code?: number | string;
+  code?: number | string
   /** 响应数据 */
-  data?: T;
+  data?: T
   /** 响应消息 */
-  message?: string;
+  message?: string
 }
 
 export interface IStsToken {
-  bucket: string;
-  endPoint: string;
-  expiration: string;
-  ossFileKey: string;
-  policy: string;
-  region: string;
-  signature: string;
-  sourceFileName: string;
-  stsEndPoint: string;
-  tempAccessKeyId: string;
-  tempAccessKeySecret: string;
-  tempSecurityToken: string;
-  uploadDir: string;
-  protocol: string;
+  bucket: string
+  endPoint: string
+  expiration: string
+  ossFileKey: string
+  policy: string
+  region: string
+  signature: string
+  sourceFileName: string
+  stsEndPoint: string
+  tempAccessKeyId: string
+  tempAccessKeySecret: string
+  tempSecurityToken: string
+  uploadDir: string
+  protocol: string
 }
 
 export interface IStsTokenReq {
   sourceFileName: string
   isPrivate: number
 }
+
+export interface IMsgParams {
+  userId: number
+  token: string
+  sessionId: string
+  messageId: string
+  domainId: string
+  appId: string
+  botId: string
+  agentId: string
+}

package/src/utils/constant.ts

@@ -0,0 +1,4 @@
+export const ENV: 'production' | 'test' | 'develop' = 'test'
+
+
+export const emptyToolText = ['/new', '/search', '/stop', '/abort', '/queue interrupt']

package/src/utils/global.ts

@@ -0,0 +1,112 @@
+/** socket connection */
+import type WebSocket from 'ws'
+
+let ws: WebSocket | null = null
+
+export function setWsConnection(next: WebSocket | null) {
+  ws = next
+}
+
+export function getWsConnection(): WebSocket | null {
+  return ws
+}
+
+// OpenClawConfig
+let config: OpenClawConfig | null = null
+
+export function setOpenClawConfig(next: OpenClawConfig | null) {
+  config = next
+}
+
+export function getOpenClawConfig(): OpenClawConfig | null {
+  return config
+}
+
+import type { ChannelLogSink, OpenClawConfig, PluginRuntime } from 'openclaw/plugin-sdk'
+import { dcgLogger } from './log.js'
+import { IMsgParams } from '../types.js'
+
+const path = require('path')
+const fs = require('fs')
+const os = require('os')
+
+function getWorkspacePath() {
+  const workspacePath = path.join(os.homedir(), '.openclaw', 'workspace')
+  if (fs.existsSync(workspacePath)) {
+    return workspacePath
+  }
+  return null
+}
+
+let runtime: PluginRuntime | null = null
+let workspaceDir: string = getWorkspacePath()
+
+export function setWorkspaceDir(dir?: string) {
+  if (dir) {
+    workspaceDir = dir
+  }
+}
+export function getWorkspaceDir(): string {
+  if (!workspaceDir) {
+    dcgLogger?.('Workspace directory not initialized', 'error')
+  }
+  return workspaceDir
+}
+
+export function setDcgchatRuntime(next: PluginRuntime) {
+  runtime = next
+}
+
+export function getDcgchatRuntime(): PluginRuntime {
+  if (!runtime) {
+    dcgLogger?.('runtime not initialized', 'error')
+  }
+  return runtime as PluginRuntime
+}
+
+let msgParams = {} as IMsgParams
+export function setMsgParams(params: any) {
+  msgParams = params
+}
+export function getMsgParams() {
+  return msgParams
+}
+
+let msgStatus: 'running' | 'finished' | '' = ''
+export function setMsgStatus(status: 'running' | 'finished' | '') {
+  msgStatus = status
+}
+export function getMsgStatus() {
+  return msgStatus
+}
+
+const getMediaKey = (url: string) => url.split(/[\\/]/).pop() ?? url
+
+/** 已发送媒体去重：外层 messageId → 内层该会话下已发送的媒体 key（文件名） */
+const sentMediaKeysBySession = new Map<string, Set<string>>()
+
+function getSessionMediaSet(messageId: string): Set<string> {
+  let set = sentMediaKeysBySession.get(messageId)
+  if (!set) {
+    set = new Set<string>()
+    sentMediaKeysBySession.set(messageId, set)
+  }
+  return set
+}
+
+export function addSentMediaKey(messageId: string, url: string) {
+  getSessionMediaSet(messageId).add(getMediaKey(url))
+}
+
+export function hasSentMediaKey(messageId: string, url: string): boolean {
+  return sentMediaKeysBySession.get(messageId)?.has(getMediaKey(url)) ?? false
+}
+
+/** 不传 messageId 时清空全部会话；传入则只清空该会话 */
+export function clearSentMediaKeys(messageId?: string) {
+  if (messageId) {
+    sentMediaKeysBySession.delete(messageId)
+  } else {
+    sentMediaKeysBySession.clear()
+  }
+}

package/src/utils/log.ts

@@ -0,0 +1,15 @@
+import { RuntimeEnv } from 'openclaw/plugin-sdk'
+
+let logger: RuntimeEnv | null = null
+
+export function setLogger(next: RuntimeEnv | null) {
+  logger = next
+}
+
+export function dcgLogger(message: string, type: 'log' | 'error' = 'log'): void {
+  if (logger) {
+    logger[type](`书灵墨宝🚀 ~ [${new Date().toISOString()}] ${message}`)
+  } else {
+    console[type](`书灵墨宝🚀 ~ ${new Date().toISOString()} [${"dcgchat-test"}]: ${message}`)
+  }
+}