@dcrays/dcgchat 0.1.8 → 0.1.9

package/src/bot.ts

@@ -1,272 +1,272 @@
-import path from "node:path";
-import type { ClawdbotConfig, RuntimeEnv } from "openclaw/plugin-sdk";
-import { createReplyPrefixContext } from "openclaw/plugin-sdk";
-import type { InboundMessage, OutboundReply } from "./types.js";
-import { getDcgchatRuntime } from "./runtime.js";
-import { resolveAccount } from "./channel.js";
-import { setMsgParams } from "./tool.js";
-
-type MediaInfo = {
-  path: string;
-  contentType: string;
-  placeholder: string;
-};
-
-async function resolveMediaFromUrls(
-  fileUrls: string[],
-  maxBytes: number,
-  log?: (msg: string) => void,
-): Promise<MediaInfo[]> {
-  const core = getDcgchatRuntime();
-  const out: MediaInfo[] = [];
-
-  log?.(`dcgchat media: starting resolve for ${fileUrls.length} file(s): ${JSON.stringify(fileUrls)}`);
-
-  for (let i = 0; i < fileUrls.length; i++) {
-    const url = fileUrls[i];
-    log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] fetching ${url}`);
-    try {
-      const response = await fetch(url);
-      log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] fetch response status=${response.status}, content-type=${response.headers.get("content-type")}, content-length=${response.headers.get("content-length")}`);
-      if (!response.ok) {
-        log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] fetch failed with HTTP ${response.status}, skipping`);
-        continue;
-      }
-      const buffer = Buffer.from(await response.arrayBuffer());
-      log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] downloaded buffer size=${buffer.length} bytes`);
-
-      let contentType = response.headers.get("content-type") || "";
-      if (!contentType) {
-        log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] no content-type header, detecting mime...`);
-        contentType = await core.media.detectMime({ buffer });
-      }
-      log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] resolved contentType=${contentType}`);
-
-      const fileName = path.basename(new URL(url).pathname) || "file";
-      log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] fileName=${fileName}, saving to disk (maxBytes=${maxBytes})...`);
-
-      const saved = await core.channel.media.saveMediaBuffer(
-        buffer,
-        contentType,
-        "inbound",
-        maxBytes,
-        fileName,
-      );
-
-      const isImage = contentType.startsWith("image/");
-      out.push({
-        path: saved.path,
-        contentType: saved.contentType,
-        placeholder: isImage ? "<media:image>" : "<media:file>",
-      });
-
-      log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] saved to ${saved.path} (contentType=${saved.contentType}, isImage=${isImage})`);
-    } catch (err) {
-      log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] FAILED to process ${url}: ${String(err)}`);
-    }
-  }
-
-  log?.(`dcgchat media: resolve complete, ${out.length}/${fileUrls.length} file(s) succeeded`);
-
-  return out;
-}
-
-function buildMediaPayload(mediaList: MediaInfo[]): {
-  MediaPath?: string;
-  MediaType?: string;
-  MediaUrl?: string;
-  MediaPaths?: string[];
-  MediaUrls?: string[];
-  MediaTypes?: string[];
-} {
-  if (mediaList.length === 0) return {};
-  const first = mediaList[0];
-  const mediaPaths = mediaList.map((m) => m.path);
-  const mediaTypes = mediaList.map((m) => m.contentType).filter(Boolean);
-  return {
-    MediaPath: first?.path,
-    MediaType: first?.contentType,
-    MediaUrl: first?.path,
-    MediaPaths: mediaPaths.length > 0 ? mediaPaths : undefined,
-    MediaUrls: mediaPaths.length > 0 ? mediaPaths : undefined,
-    MediaTypes: mediaTypes.length > 0 ? mediaTypes : undefined,
-  };
-}
-
-/**
- * 处理一条用户消息，调用 Agent 并返回回复
- */
-export async function handleDcgchatMessage(params: {
-  cfg: ClawdbotConfig;
-  msg: InboundMessage;
-  accountId: string;
-  runtime?: RuntimeEnv;
-  onChunk: (reply: OutboundReply) => void;
-}): Promise<void> {
-  const { cfg, msg, accountId, runtime } = params;
-  const log = runtime?.log ?? console.log;
-  const error = runtime?.error ?? console.error;
-
-  const account = resolveAccount(cfg, accountId);
-  const userId = msg._userId.toString();
-  // const text = msg.text?.trim();
-  const text = msg.content.text?.trim();
-
-  if (!userId || !text) {
-    params.onChunk({
-      messageType: "openclaw_bot_chat",
-      _userId: msg._userId,
-      source: "client",
-      content: {
-        bot_token: msg.content.bot_token,
-        session_id: msg.content.session_id,
-        message_id: msg.content.message_id,
-        response: "[错误] 消息格式不正确",
-      },
-    });
-    return;
-  }
-
-  try {
-    const core = getDcgchatRuntime();
-
-    const route = core.channel.routing.resolveAgentRoute({
-      cfg,
-      channel: "dcgchat",
-      accountId: account.accountId,
-      peer: { kind: "direct", id: userId },
-    });
-
-    // 处理用户上传的文件
-    const fileUrls = msg.content.file_urls ?? [];
-    log(`dcgchat[${accountId}]: incoming message from user=${userId}, text="${text?.slice(0, 80)}", file_urls count=${fileUrls.length}`);
-    let mediaPayload: Record<string, unknown> = {};
-    if (fileUrls.length > 0) {
-      log(`dcgchat[${accountId}]: processing ${fileUrls.length} file(s): ${JSON.stringify(fileUrls)}`);
-      const mediaMaxBytes = 30 * 1024 * 1024;
-      const mediaList = await resolveMediaFromUrls(fileUrls, mediaMaxBytes, log);
-      mediaPayload = buildMediaPayload(mediaList);
-      log(`dcgchat[${accountId}]: media resolved ${mediaList.length}/${fileUrls.length} file(s), payload=${JSON.stringify(mediaPayload)}`);
-    }
-
-    const envelopeOptions = core.channel.reply.resolveEnvelopeFormatOptions(cfg);
-    // const messageBody = `${userId}: ${text}`;
-    const messageBody = text;
-    const bodyFormatted = core.channel.reply.formatAgentEnvelope({
-      channel: "DCG Chat",
-      from: userId,
-      timestamp: new Date(),
-      envelope: envelopeOptions,
-      body: messageBody,
-    });
-
-    const ctxPayload = core.channel.reply.finalizeInboundContext({
-      Body: bodyFormatted,
-      RawBody: text,
-      CommandBody: text,
-      From: userId,
-      To: userId,
-      SessionKey: route.sessionKey,
-      AccountId: msg.content.session_id,
-      ChatType: "direct",
-      SenderName: userId,
-      SenderId: userId,
-      Provider: "dcgchat" as const,
-      Surface: "dcgchat" as const,
-      MessageSid: msg.content.message_id,
-      Timestamp: Date.now(),
-      WasMentioned: true,
-      CommandAuthorized: true,
-      OriginatingChannel: "dcgchat" as const,
-      OriginatingTo: `user:${userId}`,
-      ...mediaPayload,
-    });
-
-    log(`dcgchat[${accountId}]: ctxPayload=${JSON.stringify(ctxPayload)}`);
-
-    const prefixContext = createReplyPrefixContext({ cfg, agentId: route.agentId });
-
-    const { dispatcher, replyOptions, markDispatchIdle } =
-      core.channel.reply.createReplyDispatcherWithTyping({
-        responsePrefix: prefixContext.responsePrefix,
-        responsePrefixContextProvider: prefixContext.responsePrefixContextProvider,
-        humanDelay: core.channel.reply.resolveHumanDelayConfig(cfg, route.agentId),
-        onReplyStart: async () => {},
-        deliver: async (payload) => {
-          log(`dcgchat[${accountId}][deliver]: received chunk, text length=${payload.text?.length || 0}`);
-          const t = payload.text?.trim();
-          if (t) {
-            log(`dcgchat[${accountId}][deliver]: sending chunk to user ${msg._userId}, text="${t.slice(0, 50)}..."`);
-            params.onChunk({
-              messageType: "openclaw_bot_chat",
-              _userId: msg._userId,
-              source: "client",
-              content: {
-                bot_token: msg.content.bot_token,
-                session_id: msg.content.session_id,
-                message_id: msg.content.message_id,
-                response: t,
-                state: 'chunk',
-              },
-            });
-            log(`dcgchat[${accountId}][deliver]: chunk sent successfully`);
-          } else {
-            log(`dcgchat[${accountId}][deliver]: skipping empty chunk`);
-          }
-        },
-        onError: (err, info) => {
-          error(`dcgchat[${accountId}] ${info.kind} reply failed: ${String(err)}`);
-        },
-        onIdle: () => {},
-      });
-
-    log(`dcgchat[${accountId}]: dispatching to agent (session=${route.sessionKey})`);
-
-    await core.channel.reply.dispatchReplyFromConfig({
-      ctx: ctxPayload,
-      cfg,
-      dispatcher,
-      replyOptions: {
-        ...replyOptions,
-        onModelSelected: prefixContext.onModelSelected,
-      },
-    });
-
-    log(`dcgchat[${accountId}]: dispatch complete, sending final state`);
-    params.onChunk({
-      messageType: "openclaw_bot_chat",
-      _userId: msg._userId,
-      source: "client",
-      content: {
-        bot_token: msg.content.bot_token,
-        session_id: msg.content.session_id,
-        message_id: msg.content.message_id,
-        response: '',
-        state: 'final',
-      },
-    });
-    setMsgParams({
-      sessionId: '',
-      messageId: ''
-    });
-    log(`dcgchat[${accountId}]: final state sent`);
-
-    markDispatchIdle();
-    log(`dcgchat[${accountId}]: message handling complete`);
-
-  } catch (err) {
-    error(`dcgchat[${accountId}]: handle message failed: ${String(err)}`);
-    params.onChunk({
-      messageType: "openclaw_bot_chat",
-      _userId: msg._userId,
-      source: "client",
-      content: {
-        bot_token: msg.content.bot_token,
-        session_id: msg.content.session_id,
-        message_id: msg.content.message_id,
-        response: `[错误] ${err instanceof Error ? err.message : String(err)}`,
-        state: 'final',
-      },
-    });
-  }
-}
+import path from "node:path";
+import type { ClawdbotConfig, RuntimeEnv } from "openclaw/plugin-sdk";
+import { createReplyPrefixContext } from "openclaw/plugin-sdk";
+import type { InboundMessage, OutboundReply } from "./types.js";
+import { getDcgchatRuntime } from "./runtime.js";
+import { resolveAccount } from "./channel.js";
+import { setMsgStatus } from "./tool.js";
+
+type MediaInfo = {
+  path: string;
+  contentType: string;
+  placeholder: string;
+};
+
+async function resolveMediaFromUrls(
+  fileUrls: string[],
+  maxBytes: number,
+  log?: (msg: string) => void,
+): Promise<MediaInfo[]> {
+  const core = getDcgchatRuntime();
+  const out: MediaInfo[] = [];
+
+  log?.(`dcgchat media: starting resolve for ${fileUrls.length} file(s): ${JSON.stringify(fileUrls)}`);
+
+  for (let i = 0; i < fileUrls.length; i++) {
+    const url = fileUrls[i];
+    log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] fetching ${url}`);
+    try {
+      const response = await fetch(url);
+      log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] fetch response status=${response.status}, content-type=${response.headers.get("content-type")}, content-length=${response.headers.get("content-length")}`);
+      if (!response.ok) {
+        log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] fetch failed with HTTP ${response.status}, skipping`);
+        continue;
+      }
+      const buffer = Buffer.from(await response.arrayBuffer());
+      log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] downloaded buffer size=${buffer.length} bytes`);
+
+      let contentType = response.headers.get("content-type") || "";
+      if (!contentType) {
+        log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] no content-type header, detecting mime...`);
+        // @ts-ignore
+        contentType = await core.media.detectMime({ buffer });
+      }
+      log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] resolved contentType=${contentType}`);
+
+      const fileName = path.basename(new URL(url).pathname) || "file";
+      log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] fileName=${fileName}, saving to disk (maxBytes=${maxBytes})...`);
+
+      const saved = await core.channel.media.saveMediaBuffer(
+        buffer,
+        contentType,
+        "inbound",
+        maxBytes,
+        fileName,
+      );
+
+      const isImage = contentType.startsWith("image/");
+      out.push({
+        path: saved.path,
+        // @ts-ignore
+        contentType: saved.contentType,
+        placeholder: isImage ? "<media:image>" : "<media:file>",
+      });
+
+      log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] saved to ${saved.path} (contentType=${saved.contentType}, isImage=${isImage})`);
+    } catch (err) {
+      log?.(`dcgchat media: [${i + 1}/${fileUrls.length}] FAILED to process ${url}: ${String(err)}`);
+    }
+  }
+
+  log?.(`dcgchat media: resolve complete, ${out.length}/${fileUrls.length} file(s) succeeded`);
+
+  return out;
+}
+
+function buildMediaPayload(mediaList: MediaInfo[]): {
+  MediaPath?: string;
+  MediaType?: string;
+  MediaUrl?: string;
+  MediaPaths?: string[];
+  MediaUrls?: string[];
+  MediaTypes?: string[];
+} {
+  if (mediaList.length === 0) return {};
+  const first = mediaList[0];
+  const mediaPaths = mediaList.map((m) => m.path);
+  const mediaTypes = mediaList.map((m) => m.contentType).filter(Boolean);
+  return {
+    MediaPath: first?.path,
+    MediaType: first?.contentType,
+    MediaUrl: first?.path,
+    MediaPaths: mediaPaths.length > 0 ? mediaPaths : undefined,
+    MediaUrls: mediaPaths.length > 0 ? mediaPaths : undefined,
+    MediaTypes: mediaTypes.length > 0 ? mediaTypes : undefined,
+  };
+}
+
+/**
+ * 处理一条用户消息，调用 Agent 并返回回复
+ */
+export async function handleDcgchatMessage(params: {
+  cfg: ClawdbotConfig;
+  msg: InboundMessage;
+  accountId: string;
+  runtime?: RuntimeEnv;
+  onChunk: (reply: OutboundReply) => void;
+}): Promise<void> {
+  const { cfg, msg, accountId, runtime } = params;
+  const log = runtime?.log ?? console.log;
+  const error = runtime?.error ?? console.error;
+
+  const account = resolveAccount(cfg, accountId);
+  const userId = msg._userId.toString();
+  // const text = msg.text?.trim();
+  const text = msg.content.text?.trim();
+
+  if (!userId || !text) {
+    params.onChunk({
+      messageType: "openclaw_bot_chat",
+      _userId: msg._userId,
+      source: "client",
+      // @ts-ignore
+      content: {
+        bot_token: msg.content.bot_token,
+        session_id: msg.content.session_id,
+        message_id: msg.content.message_id,
+        response: "[错误] 消息格式不正确",
+      },
+    });
+    return;
+  }
+
+  try {
+    const core = getDcgchatRuntime();
+
+    const route = core.channel.routing.resolveAgentRoute({
+      cfg,
+      channel: "dcgchat",
+      accountId: account.accountId,
+      peer: { kind: "direct", id: userId },
+    });
+
+    // 处理用户上传的文件
+    const fileUrls = msg.content.file_urls ?? [];
+    log(`dcgchat[${accountId}]: incoming message from user=${userId}, text="${text?.slice(0, 80)}", file_urls count=${fileUrls.length}`);
+    let mediaPayload: Record<string, unknown> = {};
+    if (fileUrls.length > 0) {
+      log(`dcgchat[${accountId}]: processing ${fileUrls.length} file(s): ${JSON.stringify(fileUrls)}`);
+      const mediaMaxBytes = 30 * 1024 * 1024;
+      const mediaList = await resolveMediaFromUrls(fileUrls, mediaMaxBytes, log);
+      mediaPayload = buildMediaPayload(mediaList);
+      log(`dcgchat[${accountId}]: media resolved ${mediaList.length}/${fileUrls.length} file(s), payload=${JSON.stringify(mediaPayload)}`);
+    }
+
+    const envelopeOptions = core.channel.reply.resolveEnvelopeFormatOptions(cfg);
+    // const messageBody = `${userId}: ${text}`;
+    const messageBody = text;
+    const bodyFormatted = core.channel.reply.formatAgentEnvelope({
+      channel: "DCG Chat",
+      from: userId,
+      timestamp: new Date(),
+      envelope: envelopeOptions,
+      body: messageBody,
+    });
+
+    const ctxPayload = core.channel.reply.finalizeInboundContext({
+      Body: bodyFormatted,
+      RawBody: text,
+      CommandBody: text,
+      From: userId,
+      To: userId,
+      SessionKey: route.sessionKey,
+      AccountId: msg.content.session_id,
+      ChatType: "direct",
+      SenderName: userId,
+      SenderId: userId,
+      Provider: "dcgchat" as const,
+      Surface: "dcgchat" as const,
+      MessageSid: msg.content.message_id,
+      Timestamp: Date.now(),
+      WasMentioned: true,
+      CommandAuthorized: true,
+      OriginatingChannel: "dcgchat" as const,
+      OriginatingTo: `user:${userId}`,
+      ...mediaPayload,
+    });
+
+    log(`dcgchat[${accountId}]: ctxPayload=${JSON.stringify(ctxPayload)}`);
+
+    const prefixContext = createReplyPrefixContext({ cfg, agentId: route.agentId });
+
+    const { dispatcher, replyOptions, markDispatchIdle } =
+      core.channel.reply.createReplyDispatcherWithTyping({
+        responsePrefix: prefixContext.responsePrefix,
+        responsePrefixContextProvider: prefixContext.responsePrefixContextProvider,
+        humanDelay: core.channel.reply.resolveHumanDelayConfig(cfg, route.agentId),
+        onReplyStart: async () => {},
+        deliver: async (payload) => {
+          log(`dcgchat[${accountId}][deliver]: received chunk, text length=${payload.text?.length || 0}`);
+          const t = payload.text?.trim();
+          if (t) {
+            log(`dcgchat[${accountId}][deliver]: sending chunk to user ${msg._userId}, text="${t.slice(0, 50)}..."`);
+            params.onChunk({
+              messageType: "openclaw_bot_chat",
+              _userId: msg._userId,
+              source: "client",
+              content: {
+                bot_token: msg.content.bot_token,
+                session_id: msg.content.session_id,
+                message_id: msg.content.message_id,
+                response: t,
+                state: 'chunk',
+              },
+            });
+            log(`dcgchat[${accountId}][deliver]: chunk sent successfully`);
+          } else {
+            log(`dcgchat[${accountId}][deliver]: skipping empty chunk`);
+          }
+        },
+        onError: (err, info) => {
+          error(`dcgchat[${accountId}] ${info.kind} reply failed: ${String(err)}`);
+        },
+        onIdle: () => {},
+      });
+
+    log(`dcgchat[${accountId}]: dispatching to agent (session=${route.sessionKey})`);
+
+    await core.channel.reply.dispatchReplyFromConfig({
+      ctx: ctxPayload,
+      cfg,
+      dispatcher,
+      replyOptions: {
+        ...replyOptions,
+        onModelSelected: prefixContext.onModelSelected,
+      },
+    });
+
+    log(`dcgchat[${accountId}]: dispatch complete, sending final state`);
+    params.onChunk({
+      messageType: "openclaw_bot_chat",
+      _userId: msg._userId,
+      source: "client",
+      content: {
+        bot_token: msg.content.bot_token,
+        session_id: msg.content.session_id,
+        message_id: msg.content.message_id,
+        response: '',
+        state: 'final',
+      },
+    });
+    setMsgStatus('finished');
+    log(`dcgchat[${accountId}]: final state sent`);
+
+    markDispatchIdle();
+    log(`dcgchat[${accountId}]: message handling complete`);
+
+  } catch (err) {
+    error(`dcgchat[${accountId}]: handle message failed: ${String(err)}`);
+    params.onChunk({
+      messageType: "openclaw_bot_chat",
+      _userId: msg._userId,
+      source: "client",
+      content: {
+        bot_token: msg.content.bot_token,
+        session_id: msg.content.session_id,
+        message_id: msg.content.message_id,
+        response: `[错误] ${err instanceof Error ? err.message : String(err)}`,
+        state: 'final',
+      },
+    });
+  }
+}