@xgjktech/xg_cwork_im 1.0.8 → 1.0.9

package/src/inbound-media-local.ts

@@ -0,0 +1,159 @@
+/**
+ * 入站附件：可选落盘到当前 Agent workspace 下指定子目录（相对路径）。
+ *
+ * 目录约定：{subdir}/{userId}/{YYYY-MM-DD}/{stem}_{HHmmssmmm}{ext}
+ * 时间取写入时的本地时区。
+ */
+
+import { mkdir, writeFile } from "node:fs/promises";
+import path from "node:path";
+import { pathToFileURL } from "node:url";
+import type { OpenClawConfig } from "openclaw/plugin-sdk";
+import type { Log } from "./auth.js";
+import { downloadMediaBuffer, resolveMaxAttachmentBytes } from "./resource-file.js";
+import type { MsgFileVO, XgImConfig } from "./types.js";
+
+/** 校验：非空、相对路径、不含 `..` 段 */
+export function normalizeInboundWorkspaceSubdir(subdir: string): string | null {
+    const t = subdir.trim().replace(/\\/g, "/");
+    if (!t || t === "." || t === "..") return null;
+    const parts = t.split("/").filter(Boolean);
+    for (const p of parts) {
+        if (p === "..") return null;
+    }
+    if (path.isAbsolute(t)) return null;
+    return parts.join("/");
+}
+
+export function resolveAgentWorkspaceDirFromRuntime(rt: unknown, cfg: OpenClawConfig): string | undefined {
+    const agent = (rt as { agent?: { resolveAgentWorkspaceDir?: (c: OpenClawConfig) => unknown } })?.agent;
+    const fn = agent?.resolveAgentWorkspaceDir;
+    if (typeof fn !== "function") return undefined;
+    try {
+        const out = fn(cfg);
+        if (typeof out === "string" && out.trim()) return path.resolve(out.trim());
+    } catch {
+        /* ignore */
+    }
+    return undefined;
+}
+
+export async function ensureAgentWorkspaceFromRuntime(rt: unknown, cfg: OpenClawConfig): Promise<void> {
+    const agent = (rt as { agent?: { ensureAgentWorkspace?: (c: OpenClawConfig) => Promise<void> } })?.agent;
+    const fn = agent?.ensureAgentWorkspace;
+    if (typeof fn !== "function") return;
+    try {
+        await fn(cfg);
+    } catch {
+        /* 落盘前仍会 mkdir，不阻断 */
+    }
+}
+
+function sanitizeUserIdForPath(id: string): string {
+    const t = id.trim();
+    if (/^-?\d+$/.test(t) && t.length <= 32) return t;
+    const safe = t.replace(/[^\w.-]/g, "_").replace(/_+/g, "_").replace(/^_|_$/g, "");
+    return safe.slice(0, 64) || "unknown";
+}
+
+function formatDateFolderLocal(d: Date): string {
+    const y = d.getFullYear();
+    const m = String(d.getMonth() + 1).padStart(2, "0");
+    const day = String(d.getDate()).padStart(2, "0");
+    return `${y}-${m}-${day}`;
+}
+
+/** `HHmmss` + 三位毫秒，如 `120459237` */
+export function formatInboundTimeSuffix(d: Date): string {
+    const h = String(d.getHours()).padStart(2, "0");
+    const min = String(d.getMinutes()).padStart(2, "0");
+    const s = String(d.getSeconds()).padStart(2, "0");
+    const ms = String(d.getMilliseconds()).padStart(3, "0");
+    return `${h}${min}${s}${ms}`;
+}
+
+const MAX_STEM_LEN = 120;
+
+function sanitizeStem(stem: string): string {
+    const s = stem
+        .replace(/[<>:"/\\|?*\u0000-\u001f]/g, "_")
+        .replace(/_+/g, "_")
+        .replace(/^\.+|\.+$/g, "")
+        .trim();
+    const out = s || "file";
+    return out.length > MAX_STEM_LEN ? out.slice(0, MAX_STEM_LEN) : out;
+}
+
+/** `report_120459237.pdf` */
+export function buildInboundStoredFilename(originalFilename: string, at: Date): string {
+    const base = path.basename(originalFilename) || "file";
+    const ext = path.extname(base);
+    const stem = ext ? base.slice(0, -ext.length) : base;
+    const suffix = formatInboundTimeSuffix(at);
+    return `${sanitizeStem(stem)}_${suffix}${ext.toLowerCase()}`;
+}
+
+/**
+ * 将入站附件下载到 workspace 下子目录；失败项保留原 URL。
+ */
+export async function saveInboundFilesToWorkspace(args: {
+    rt: unknown;
+    cfg: OpenClawConfig;
+    config: XgImConfig;
+    workspaceSubdir: string;
+    senderUserId: string | undefined;
+    fileItems: MsgFileVO[];
+    log?: Log;
+}): Promise<MsgFileVO[]> {
+    const { rt, cfg, config, workspaceSubdir, senderUserId, fileItems, log } = args;
+    const normalized = normalizeInboundWorkspaceSubdir(workspaceSubdir);
+    if (!normalized || fileItems.length === 0) return fileItems;
+
+    const workspaceRoot = resolveAgentWorkspaceDirFromRuntime(rt, cfg);
+    if (!workspaceRoot) {
+        log?.warn?.(
+            "[cwork_im:inbound-local] resolveAgentWorkspaceDir unavailable, skip inbound download; keep remote URLs",
+        );
+        return fileItems;
+    }
+
+    await ensureAgentWorkspaceFromRuntime(rt, cfg);
+
+    const maxBytes = resolveMaxAttachmentBytes(config);
+    const userSeg = sanitizeUserIdForPath(senderUserId ?? "unknown");
+    const relParts = normalized.split("/");
+
+    const out: MsgFileVO[] = [];
+    for (const item of fileItems) {
+        const srcUrl = item.url?.trim();
+        if (!srcUrl) {
+            out.push(item);
+            continue;
+        }
+        const now = new Date();
+        const dateFolder = formatDateFolderLocal(now);
+        const localName = buildInboundStoredFilename(item.name?.trim() || "file", now);
+        const dir = path.join(workspaceRoot, ...relParts, userSeg, dateFolder);
+        const absPath = path.join(dir, localName);
+
+        try {
+            await mkdir(dir, { recursive: true });
+            const { buffer } = await downloadMediaBuffer(srcUrl, maxBytes, log);
+            await writeFile(absPath, buffer);
+            const fileUrl = pathToFileURL(absPath).href;
+            log?.info?.(`[cwork_im:inbound-local] saved ${localName} bytes=${buffer.length} -> ${absPath}`);
+            out.push({
+                ...item,
+                url: fileUrl,
+                size: buffer.length,
+                name: localName,
+            });
+        } catch (e: unknown) {
+            log?.warn?.(
+                `[cwork_im:inbound-local] download/save failed, keep original URL: ${String(e)} ref=${srcUrl.slice(0, 80)}`,
+            );
+            out.push(item);
+        }
+    }
+    return out;
+}