@workclaw/openclaw-workclaw 1.0.202 → 1.0.331

package/dist/packages/plugin/src/gateway/message-context.js

@@ -0,0 +1,410 @@
+import fs from "node:fs/promises";
+import http from "node:http";
+import https from "node:https";
+import os from "node:os";
+import path from "node:path";
+import logger from "../logging.js";
+const OPENCLAW_WORKCLAW_CHANNEL = "openclaw-workclaw";
+function parseWorkClawMessage(data, log) {
+  let input = {};
+  try {
+    input = data ? JSON.parse(data) : {};
+  } catch {
+    logger.info(`解析消息失败: ${data}`);
+    return null;
+  }
+  if (typeof input === "string" && input.toLowerCase() === "ping") {
+    return {
+      type: "ping",
+      pongData: input
+    };
+  }
+  const envelope = input && typeof input === "object" ? input : { data: input };
+  const topic = envelope?.metadata?.topic;
+  const type = String(envelope?.type || "").trim().toUpperCase();
+  const topicLower = String(topic || "").trim().toLowerCase();
+  if (type === "INIT_AGENT") {
+    let parsedData = envelope?.data;
+    if (typeof parsedData === "string") {
+      try {
+        parsedData = JSON.parse(parsedData);
+      } catch (e) {
+      }
+    }
+    return {
+      type: "init_agent",
+      eventData: parsedData
+    };
+  }
+  if (type === "SYSTEM") {
+    if (topicLower === "ping") {
+      return {
+        type: "ping",
+        pongData: envelope?.data
+      };
+    }
+    if (topicLower === "disconnect") {
+      return {
+        type: "disconnect"
+      };
+    }
+    return {
+      type: "ignored"
+    };
+  }
+  if (type === "EVENT") {
+    const eventType = String(envelope?.metadata?.event || "").trim().toUpperCase();
+    let parsedData = envelope?.data;
+    if (typeof parsedData === "string") {
+      try {
+        parsedData = JSON.parse(parsedData);
+      } catch (e) {
+      }
+    }
+    if (eventType === "AGENT_CREATED") {
+      return {
+        type: "agent_created",
+        eventData: parsedData
+      };
+    }
+    if (eventType === "AGENT_UPDATED") {
+      return {
+        type: "agent_updated",
+        eventData: parsedData
+      };
+    }
+    if (eventType === "AGENT_DELETED") {
+      return {
+        type: "agent_deleted",
+        eventData: parsedData
+      };
+    }
+    if (eventType === "TOOLS_LIST") {
+      return {
+        type: "tools_list",
+        eventData: parsedData
+      };
+    }
+    if (eventType === "SKILLS_LIST") {
+      return {
+        type: "skills_list",
+        eventData: parsedData
+      };
+    }
+    if (eventType === "INIT_AGENT") {
+      return {
+        type: "init_agent",
+        eventData: parsedData
+      };
+    }
+    if (eventType === "SKILL_DO_REMIND") {
+      const eventTopic = String(parsedData?.dotype || "").toLowerCase();
+      return {
+        type: "skills_event",
+        eventData: {
+          topic: `skills/${eventTopic}`,
+          requestId: "",
+          callbackUrl: "",
+          authToken: "",
+          data: parsedData
+        }
+      };
+    }
+    if (eventType === "CRON_REMIND") {
+      const eventTopic = String(parsedData?.dotype || "").toLowerCase();
+      return {
+        type: "cron_task_event",
+        eventData: {
+          dotype: `${eventTopic}`,
+          data: parsedData
+        }
+      };
+    }
+    return {
+      type: "ignored"
+    };
+  }
+  if (type !== "CALLBACK" || String(topic || "").trim().toUpperCase() !== "AGENT_MESSAGE") {
+    return {
+      type: "ignored"
+    };
+  }
+  let payload = envelope?.data || {};
+  if (typeof payload === "string") {
+    try {
+      payload = JSON.parse(payload);
+    } catch (e) {
+    }
+  }
+  const text = typeof payload?.message?.content === "string" ? payload.message.content : typeof payload?.message === "string" ? payload.message : "";
+  function isImageFile(url) {
+    const imageExtensions = [".jpg", ".jpeg", ".png", ".gif", ".webp", ".bmp", ".svg", ".ico"];
+    const lowerUrl = url.toLowerCase();
+    return imageExtensions.some((ext) => lowerUrl.endsWith(ext));
+  }
+  const fileUrlList = Array.isArray(payload?.fileUrl) ? payload.fileUrl : [];
+  const attachments = fileUrlList.map((url) => ({
+    url,
+    fileId: payload?.fileId ? String(payload.fileId) : "",
+    type: isImageFile(url) ? "image" : "file"
+  }));
+  return {
+    type: "agent_message",
+    message: {
+      text,
+      userId: String(payload?.userId ?? ""),
+      openConversationId: String(payload?.conversationId ?? ""),
+      messageId: String(payload?.msgId ?? ""),
+      taskId: payload?.taskId ? String(payload.taskId) : void 0,
+      agentId: payload?.agentId,
+      attachments: attachments.length > 0 ? attachments : void 0
+    }
+  };
+}
+function detectMediaType(url) {
+  const lower = url.toLowerCase();
+  if (lower.endsWith(".jpg") || lower.endsWith(".jpeg") || lower.endsWith(".png") || lower.endsWith(".gif") || lower.endsWith(".webp") || lower.endsWith(".bmp") || lower.endsWith(".svg") || lower.endsWith(".ico")) {
+    return "image";
+  }
+  if (lower.endsWith(".mp3") || lower.endsWith(".wav") || lower.endsWith(".amr") || lower.endsWith(".m4a") || lower.endsWith(".ogg")) {
+    return "audio";
+  }
+  if (lower.endsWith(".mp4") || lower.endsWith(".avi") || lower.endsWith(".mov") || lower.endsWith(".mkv") || lower.endsWith(".webm")) {
+    return "video";
+  }
+  if (lower.endsWith(".pdf"))
+    return "application/pdf";
+  if (lower.endsWith(".docx"))
+    return "application/vnd.openxmlformats-officedocument.wordprocessingml.document";
+  if (lower.endsWith(".xlsx"))
+    return "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";
+  if (lower.endsWith(".pptx"))
+    return "application/vnd.openxmlformats-officedocument.presentationml.presentation";
+  if (lower.endsWith(".doc"))
+    return "application/msword";
+  if (lower.endsWith(".xls"))
+    return "application/vnd.ms-excel";
+  if (lower.endsWith(".ppt"))
+    return "application/vnd.ms-powerpoint";
+  if (lower.endsWith(".txt"))
+    return "text/plain";
+  if (lower.endsWith(".zip"))
+    return "application/zip";
+  if (lower.endsWith(".rar"))
+    return "application/x-rar-compressed";
+  return "application/octet-stream";
+}
+function getMediaTypeLabel(url) {
+  const lower = url.toLowerCase();
+  if (lower.endsWith(".pdf"))
+    return "PDF";
+  if (lower.endsWith(".docx"))
+    return "Word";
+  if (lower.endsWith(".doc"))
+    return "Word";
+  if (lower.endsWith(".xlsx"))
+    return "Excel";
+  if (lower.endsWith(".xls"))
+    return "Excel";
+  if (lower.endsWith(".pptx"))
+    return "PowerPoint";
+  if (lower.endsWith(".ppt"))
+    return "PowerPoint";
+  if (lower.endsWith(".txt"))
+    return "文本";
+  if (lower.endsWith(".jpg") || lower.endsWith(".jpeg"))
+    return "JPG图片";
+  if (lower.endsWith(".png"))
+    return "PNG图片";
+  if (lower.endsWith(".gif"))
+    return "GIF图片";
+  if (lower.endsWith(".webp"))
+    return "WebP图片";
+  if (lower.endsWith(".mp3"))
+    return "音频";
+  if (lower.endsWith(".wav"))
+    return "音频";
+  if (lower.endsWith(".mp4"))
+    return "视频";
+  if (lower.endsWith(".zip") || lower.endsWith(".rar"))
+    return "压缩包";
+  return "文件";
+}
+async function downloadFileToLocal(url, accountId, peerId, log) {
+  const downloadsDir = path.join(os.homedir(), ".openclaw", "media", "workclaw", "downloads", accountId, peerId);
+  try {
+    await fs.mkdir(downloadsDir, { recursive: true });
+  } catch {
+  }
+  const urlObj = new URL(url);
+  const fileName = decodeURIComponent(path.basename(urlObj.pathname)) || "file";
+  const ext = path.extname(fileName);
+  const stem = path.basename(fileName, ext);
+  const localFileName = `${stem}_${Date.now()}${ext}`;
+  const localFilePath = path.join(downloadsDir, localFileName);
+  return new Promise((resolve) => {
+    const client = url.startsWith("https:") ? https : http;
+    const req = client.get(url, (res) => {
+      if (res.statusCode !== 200) {
+        logger.warn(`下载文件: HTTP ${res.statusCode} - ${url}`);
+        resolve(null);
+        return;
+      }
+      const chunks = [];
+      res.on("data", (chunk) => chunks.push(chunk));
+      res.on("end", async () => {
+        try {
+          const buffer = Buffer.concat(chunks);
+          await fs.writeFile(localFilePath, buffer);
+          logger.info(`下载文件: 已保存 ${buffer.length} 字节 -> ${localFilePath}`);
+          resolve(localFilePath);
+        } catch (err) {
+          logger.error(`下载文件: 写入失败: ${String(err)}`);
+          resolve(null);
+        }
+      });
+      res.on("error", (err) => {
+        logger.error(`下载文件: 读取错误: ${String(err)}`);
+        resolve(null);
+      });
+    });
+    req.on("error", (err) => {
+      logger.error(`下载文件: 请求错误: ${String(err)}`);
+      resolve(null);
+    });
+    req.setTimeout(3e4, () => {
+      req.destroy();
+      logger.warn(`下载文件: ${url} 请求超时`);
+      resolve(null);
+    });
+  });
+}
+async function buildInboundContext(message, accountId, config, agentId, pluginRuntime, recordSession = true, log) {
+  let existingSessionKey;
+  try {
+    const rawPayload = JSON.parse(message.rawPayload);
+    existingSessionKey = rawPayload.sessionKey;
+  } catch {
+  }
+  const peerId = message.from;
+  const computedSessionKey = existingSessionKey || (agentId && agentId !== "main" ? `agent:${agentId}:openclaw-workclaw:${accountId}:direct:${peerId}` : void 0);
+  const sessionKey = computedSessionKey || (agentId && agentId !== "main" ? `agent:${agentId}:openclaw-workclaw:${accountId}:direct:${peerId}` : void 0);
+  const result = {
+    Body: message.text,
+    BodyForAgent: message.text,
+    RawBody: JSON.stringify(message),
+    From: message.from,
+    To: message.to,
+    ChatType: message.chatType,
+    Provider: OPENCLAW_WORKCLAW_CHANNEL,
+    Surface: OPENCLAW_WORKCLAW_CHANNEL,
+    Timestamp: message.timestamp,
+    AccountId: accountId,
+    AgentId: agentId,
+    MessageSid: message.messageId,
+    OriginatingChannel: OPENCLAW_WORKCLAW_CHANNEL,
+    OriginatingTo: message.to
+  };
+  if (sessionKey) {
+    result.SessionKey = sessionKey;
+  }
+  if (!pluginRuntime) {
+    return result;
+  }
+  const storePath = pluginRuntime.channel.session.resolveStorePath(config.session?.store, { agentId });
+  logger.info(`[会话] 存储路径=${storePath}, 智能体ID=${agentId}, 会话Key=${sessionKey}`);
+  const envelopeOptions = pluginRuntime.channel.reply.resolveEnvelopeFormatOptions(config);
+  const previousTimestamp = pluginRuntime.channel.session.readSessionUpdatedAt({
+    storePath,
+    sessionKey
+  });
+  logger.info(`[会话] 上次更新时间=${previousTimestamp}`);
+  const mediaUrls = [];
+  const mediaTypes = [];
+  const bodyText = message.text;
+  if (message.attachments && message.attachments.length > 0) {
+    logger.info(`[附件] 正在下载 ${message.attachments.length} 个文件...`);
+    const downloadResults = await Promise.all(
+      message.attachments.map(async (att) => {
+        const localPath = await downloadFileToLocal(att.url, accountId, message.from);
+        const mediaType = detectMediaType(att.url);
+        const typeLabel = getMediaTypeLabel(att.url);
+        return { att, localPath, mediaType, typeLabel };
+      })
+    );
+    for (const { att, localPath, mediaType } of downloadResults) {
+      if (localPath) {
+        const homeDir = os.homedir().replace(/\\/g, "/").trim();
+        const normalizedLocalPath = localPath.replace(/\\/g, "/");
+        const relativePath = normalizedLocalPath.startsWith(`${homeDir}/.openclaw/`) ? `/${normalizedLocalPath.slice(homeDir.length + 1)}` : normalizedLocalPath;
+        mediaUrls.push(relativePath);
+        mediaTypes.push(mediaType);
+        logger.info(`[附件] 已下载: ${relativePath}`);
+      } else {
+        logger.warn(`[附件] 下载失败: ${att.url}`);
+      }
+    }
+    logger.info(`[附件] 已下载 ${mediaUrls.length}/${message.attachments.length} 个文件`);
+  }
+  const fromLabel = message.to;
+  const body = pluginRuntime.channel.reply.formatInboundEnvelope({
+    channel: OPENCLAW_WORKCLAW_CHANNEL,
+    from: fromLabel,
+    timestamp: message.timestamp,
+    body: bodyText,
+    chatType: message.chatType,
+    sender: { name: message.to, id: message.to },
+    previousTimestamp,
+    envelope: envelopeOptions
+  });
+  const finalized = pluginRuntime.channel.reply.finalizeInboundContext({
+    Body: body,
+    BodyForAgent: bodyText,
+    RawBody: JSON.stringify(message),
+    CommandBody: message.text,
+    From: message.from,
+    To: message.to,
+    SessionKey: sessionKey,
+    AccountId: accountId,
+    ChatType: message.chatType,
+    ConversationLabel: fromLabel,
+    Provider: OPENCLAW_WORKCLAW_CHANNEL,
+    Surface: "openclaw-workclaw_dm",
+    MessageSid: message.messageId,
+    Timestamp: message.timestamp,
+    OriginatingChannel: OPENCLAW_WORKCLAW_CHANNEL,
+    OriginatingTo: message.to,
+    // 附件媒体信息（供 skill 读取本地文件）
+    MediaPath: mediaUrls[0] || void 0,
+    MediaUrl: mediaUrls[0] || void 0,
+    MediaUrls: mediaUrls.length > 0 ? mediaUrls : void 0,
+    MediaTypes: mediaTypes.length > 0 ? mediaTypes : void 0
+  });
+  if (sessionKey && recordSession) {
+    try {
+      await pluginRuntime.channel.session.recordInboundSession({
+        storePath,
+        sessionKey: finalized.SessionKey || sessionKey,
+        ctx: finalized,
+        updateLastRoute: {
+          sessionKey,
+          channel: OPENCLAW_WORKCLAW_CHANNEL,
+          to: message.to,
+          accountId
+        },
+        onRecordError: (err) => {
+          logger.error(`[会话] 记录入站会话失败: ${String(err)}`);
+        }
+      });
+      logger.info(`[会话] 已记录会话: key=${sessionKey}`);
+    } catch (err) {
+      logger.warn(`[会话] 记录入站会话错误: ${String(err)}`);
+    }
+  }
+  return finalized;
+}
+export {
+  OPENCLAW_WORKCLAW_CHANNEL,
+  buildInboundContext,
+  parseWorkClawMessage
+};