weaver-work-cli 0.1.2 → 0.1.5

package/dist/shortcuts/yimiaoban/operations/msg-parse.js

@@ -0,0 +1,429 @@
+/**
+ * 消息内容解析核心（msgParser 的 TS 移植，面向 CLI 结构化 data）。
+ *
+ * 只做解析与结构化（忠实还原展示所需字段），不做 markdown 渲染：
+ * - 文本 / 图片 / 文件 / 视频 / @提及 / i18n / 系统消息 / 必达识别
+ * - 媒体段提取（供预览/下载按钮 URL 构造）
+ * - 发送者姓名回退规则
+ */
+export const MSG_TYPES = {
+    1: '文本',
+    2: '图片',
+    3: '语音',
+    4: '文件',
+    5: '位置',
+    6: '图文',
+    7: '视频',
+    8: '事项',
+    9: '系统消息',
+    10: '撤回',
+    11: '抖一抖',
+    12: '屏蔽',
+    14: '通知',
+    15: '自定义',
+    16: '语音通话',
+    17: '聊天记录',
+    18: '视频通话',
+    19: '群公告',
+    20: '分享地址',
+    21: '表情',
+    22: '强提醒',
+    23: '链接',
+    24: '群接龙',
+    25: '+1回复',
+    26: '富文本',
+};
+export function getTypeName(type) {
+    const t = Number(type);
+    return MSG_TYPES[t] || '未知';
+}
+/** 解析消息 JSON 字符串或对象。 */
+export function parseMsg(value) {
+    if (!value)
+        return null;
+    if (typeof value === 'string') {
+        try {
+            return JSON.parse(value);
+        }
+        catch {
+            return null;
+        }
+    }
+    if (typeof value === 'object')
+        return value;
+    return null;
+}
+function normalizeKey(key) {
+    if (key.startsWith('%') && key.endsWith('%'))
+        return key;
+    return `%${key}%`;
+}
+function formatPersonToken(value, nameMap) {
+    const uidRaw = value?.u !== undefined && value.u !== null ? String(value.u)
+        : value?.uid !== undefined && value.uid !== null ? String(value.uid) : '';
+    const cRaw = value?.c !== undefined && value.c !== null ? String(value.c) : '';
+    if (uidRaw && nameMap?.[uidRaw])
+        return nameMap[uidRaw];
+    if (uidRaw)
+        return uidRaw;
+    return cRaw;
+}
+function formatI18nVal(value, nameMap) {
+    if (Array.isArray(value)) {
+        return value.map((item) => formatI18nVal(item, nameMap)).filter(Boolean).join('、');
+    }
+    if (value && typeof value === 'object') {
+        if (value.u !== undefined || value.uid !== undefined)
+            return formatPersonToken(value, nameMap);
+        return '';
+    }
+    if (value === undefined || value === null)
+        return '';
+    return String(value);
+}
+/** %KEY% 占位符替换（kvs/jvs 合并，jvs 优先）。 */
+export function substituteTemplate(template, kvs, jvs, nameMap) {
+    const merged = { ...(kvs || {}), ...(jvs || {}) };
+    let text = String(template || '');
+    for (const [rawKey, value] of Object.entries(merged)) {
+        const key = normalizeKey(rawKey);
+        const replacement = formatI18nVal(value, nameMap);
+        text = text.split(key).join(replacement);
+    }
+    return text;
+}
+/**
+ * 收集消息体中的 @ 提及映射：递归扫描所有 `aus` 条目（`{id: uid, nm: 姓名}`）→ uid → 姓名。
+ * `aus` 由发送端随消息下发，姓名无需再走 hrm 解析；nm 缺失的条目不收集（展示回退 uid）。
+ */
+export function collectMentionMap(msg) {
+    const map = new Map();
+    const walk = (value) => {
+        if (!value || typeof value !== 'object')
+            return;
+        if (Array.isArray(value)) {
+            for (const item of value)
+                walk(item);
+            return;
+        }
+        if (Array.isArray(value.aus)) {
+            for (const item of value.aus) {
+                if (item && item.id !== undefined && item.id !== null && item.nm)
+                    map.set(String(item.id), String(item.nm));
+            }
+        }
+        for (const key of Object.keys(value)) {
+            if (key === 'aus')
+                continue; // 已处理，避免重复遍历
+            walk(value[key]);
+        }
+    };
+    walk(msg);
+    return map;
+}
+/** @提及还原：文本中 :@<uid>; 占位符 → @姓名（aus 优先，nameMap 兜底，最后回退 uid）。 */
+export function replaceMentions(text, mentionMap, nameMap) {
+    let output = String(text || '');
+    if (!output.includes(':@'))
+        return output;
+    output = output.replace(/:@(\d{1,30});/g, (_, uidRaw) => {
+        const uid = uidRaw;
+        const name = mentionMap?.get(uid) || nameMap?.[uid] || uid;
+        return `@${name}`;
+    });
+    return output;
+}
+/** 文本段提取：txt.v → i18n 模板替换 → id18n 翻译 → 回退模板原文。 */
+export function extractTxt(seg, opts = {}) {
+    const txt = seg?.txt;
+    if (!txt)
+        return '';
+    const mentionMap = opts.mentionMap ?? collectMentionMap(seg);
+    if (typeof txt === 'string')
+        return replaceMentions(txt, mentionMap, opts.nameMap);
+    if (typeof txt !== 'object')
+        return '';
+    if (txt.v !== undefined && txt.v !== null && String(txt.v) !== '') {
+        return replaceMentions(String(txt.v), mentionMap, opts.nameMap);
+    }
+    const kvs = txt.kvs || {};
+    const jvs = txt.jvs || {};
+    if (txt.i18n && typeof txt.i18n === 'string' && txt.i18n.includes('%')) {
+        return replaceMentions(substituteTemplate(txt.i18n, kvs, jvs, opts.nameMap), mentionMap, opts.nameMap);
+    }
+    if (txt.id18n !== undefined && txt.id18n !== null) {
+        const translated = opts.i18nMap?.[String(txt.id18n)];
+        if (translated) {
+            return replaceMentions(substituteTemplate(translated, kvs, jvs, opts.nameMap), mentionMap, opts.nameMap);
+        }
+    }
+    if (txt.i18n && typeof txt.i18n === 'string') {
+        return replaceMentions(substituteTemplate(txt.i18n, kvs, jvs, opts.nameMap), mentionMap, opts.nameMap);
+    }
+    return '';
+}
+/** 提取系统消息信息（msg.dt[].sys）。 */
+export function parseSysMsg(msg) {
+    const sys = msg?.dt?.[0]?.sys;
+    if (!sys)
+        return {};
+    return {
+        group: sys.group !== undefined ? String(sys.group) : undefined,
+        type: sys.type !== undefined ? String(sys.type) : undefined,
+        tpl: sys.tpl !== undefined ? String(sys.tpl) : undefined,
+        etName: sys.et?.nm !== undefined && sys.et.nm !== null ? String(sys.et.nm) : undefined,
+        id18n: sys.id18n !== undefined ? String(sys.id18n) : undefined,
+    };
+}
+/** 消息内容解析主入口。 */
+export function parseContent(value, opts = {}) {
+    const msg = parseMsg(value) || {};
+    const type = Number(msg.type || 0);
+    const typeName = getTypeName(type);
+    const dt = Array.isArray(msg.dt) ? msg.dt : [];
+    const data = dt[0] || {};
+    const sys = parseSysMsg(msg);
+    const tf = Number(msg.tf || 0);
+    const must = (tf & 0x01) !== 0;
+    const localOpts = { ...opts, mentionMap: opts.mentionMap ?? collectMentionMap(msg) };
+    let content = '';
+    let fileName = '';
+    switch (type) {
+        case 1:
+            content = extractTxt(data, localOpts);
+            break;
+        case 2: {
+            content = parseSegment(data, localOpts) || '[图片]';
+            fileName = data?.img?.nm ? String(data.img.nm) : '';
+            break;
+        }
+        case 4:
+            fileName = data?.file?.nm ? String(data.file.nm) : '';
+            content = fileName ? `[文件]${fileName}` : '[文件]';
+            break;
+        case 7: {
+            content = parseSegment(data, localOpts) || '[视频]';
+            fileName = data?.media?.nm ? String(data.media.nm)
+                : data?.video?.nm ? String(data.video.nm) : '';
+            break;
+        }
+        case 5:
+            content = data?.location?.name ? String(data.location.name) : '';
+            break;
+        case 8:
+            content = data?.entity?.entityName ? String(data.entity.entityName) : '';
+            break;
+        case 20:
+            content = data?.shareaddr?.name ? String(data.shareaddr.name) : extractTxt(data, localOpts);
+            break;
+        case 26:
+            content = msg.vv ? String(msg.vv) : dt.map((d) => parseSegment(d, localOpts)).join('');
+            break;
+        default:
+            content = extractTxt(data, localOpts);
+    }
+    // 系统消息覆盖：sys.tpl=1 且有实体名 → 用实体名
+    if (sys.tpl === '1' && sys.etName)
+        content = sys.etName;
+    // @ 提及兜底还原：富文本 vv / 文件等路径可能仍带 :@<uid>; 占位符，统一还原为 @姓名
+    if (content && String(content).includes(':@')) {
+        content = replaceMentions(content, localOpts.mentionMap, localOpts.nameMap);
+    }
+    if (must)
+        content = `[必达]${content}`;
+    // title：优先 sys.et.nm，否则 i18n 翻译文案
+    let title = '';
+    if (sys.etName)
+        title = sys.etName;
+    else if (sys.id18n && opts.i18nMap?.[sys.id18n])
+        title = opts.i18nMap[sys.id18n];
+    return { type, typeName, content, title, must, fileName, sys };
+}
+/**
+ * 单段解析：图片/文件/视频/媒体/语音/位置 段 → 展示文本，文本段兜底。
+ * 展示统一为 `[类型]名称` 形式（名称缺失时只保留 `[类型]`），供 agent 直接渲染。
+ */
+function parseSegment(seg, opts = {}) {
+    if (!seg)
+        return '';
+    const name = (value) => (value?.nm !== undefined && value.nm !== null ? String(value.nm) : '');
+    const hasFileId = (value) => Boolean(value) && (value.src !== undefined || value.id !== undefined || value.nm !== undefined);
+    if (hasFileId(seg.img))
+        return `[图片]${name(seg.img)}`;
+    if (hasFileId(seg.file))
+        return `[文件]${name(seg.file)}`;
+    if (hasFileId(seg.video))
+        return `[视频]${name(seg.video)}`;
+    if (hasFileId(seg.media))
+        return `[视频]${name(seg.media)}`;
+    if (hasFileId(seg.voice))
+        return `[语音]${name(seg.voice)}`;
+    if (seg.location?.name)
+        return String(seg.location.name);
+    if (seg.txt)
+        return extractTxt(seg, opts);
+    return '';
+}
+/** 发送者姓名回退规则：自己（authUid）→「我」，其次消息 sname，再次 hrm 姓名，最后 uid。 */
+export function senderName(msg, fromUid, authUid, nameMap = {}) {
+    const uid = String(fromUid ?? '');
+    if (authUid && uid === authUid)
+        return '我';
+    const sname = msg?.sname ? String(msg.sname) : '';
+    if (sname)
+        return sname;
+    if (uid && nameMap[uid])
+        return nameMap[uid];
+    if (uid)
+        return uid;
+    return '';
+}
+/** 提取图片段。 */
+export function extractImgs(msg) {
+    const dt = Array.isArray(msg?.dt) ? msg.dt : [];
+    const items = [];
+    for (const seg of dt) {
+        const img = seg?.img;
+        if (!img)
+            continue;
+        const fileId = img.src !== undefined && img.src !== null ? String(img.src) : img.id !== undefined ? String(img.id) : '';
+        if (!fileId)
+            continue;
+        items.push({
+            fileId,
+            name: img.nm !== undefined ? String(img.nm) : '',
+            w: img.w !== undefined ? String(img.w) : undefined,
+            h: img.h !== undefined ? String(img.h) : undefined,
+            size: img.s !== undefined ? String(img.s) : undefined,
+            mime: img.t !== undefined ? String(img.t) : undefined,
+        });
+    }
+    return items;
+}
+/** 提取视频段。 */
+export function extractVideos(msg) {
+    const dt = Array.isArray(msg?.dt) ? msg.dt : [];
+    const items = [];
+    for (const seg of dt) {
+        const media = seg?.media || seg?.video;
+        if (!media)
+            continue;
+        const fileId = media.src !== undefined && media.src !== null ? String(media.src) : media.id !== undefined ? String(media.id) : '';
+        if (!fileId)
+            continue;
+        items.push({
+            fileId,
+            name: media.nm !== undefined ? String(media.nm) : '',
+            ext: media.ext !== undefined ? String(media.ext) : undefined,
+            w: media.w !== undefined ? String(media.w) : undefined,
+            h: media.h !== undefined ? String(media.h) : undefined,
+            size: media.s !== undefined ? String(media.s) : undefined,
+            mime: media.t !== undefined ? String(media.t) : undefined,
+        });
+    }
+    return items;
+}
+/** 提取文件段。 */
+export function extractFiles(msg) {
+    const dt = Array.isArray(msg?.dt) ? msg.dt : [];
+    const items = [];
+    for (const seg of dt) {
+        const file = seg?.file;
+        if (!file)
+            continue;
+        const fileId = file.id !== undefined && file.id !== null ? String(file.id) : file.src !== undefined ? String(file.src) : '';
+        if (!fileId)
+            continue;
+        items.push({
+            fileId,
+            name: file.nm !== undefined ? String(file.nm) : '',
+            ext: file.ext !== undefined ? String(file.ext) : undefined,
+            size: file.s !== undefined ? String(file.s) : undefined,
+            mime: file.t !== undefined ? String(file.t) : undefined,
+        });
+    }
+    return items;
+}
+function walkCollect(value, kind, out) {
+    if (value === undefined || value === null)
+        return;
+    if (Array.isArray(value)) {
+        for (const item of value)
+            walkCollect(item, kind, out);
+        return;
+    }
+    if (typeof value === 'object') {
+        for (const key of Object.keys(value)) {
+            const child = value[key];
+            if (kind === 'id18n' && key === 'id18n' && child !== undefined && child !== null && /^\d+$/.test(String(child))) {
+                out.add(String(child));
+                continue;
+            }
+            if (kind === 'person') {
+                if ((key === 'u' || key === 'uid') && child !== undefined && child !== null) {
+                    const uid = String(child);
+                    if (/^\d+$/.test(uid))
+                        out.add(uid);
+                    continue;
+                }
+                if (key === 'fd' && child === '@') {
+                    const id = value.id;
+                    if (id !== undefined && id !== null)
+                        out.add(String(id));
+                    continue;
+                }
+                if (key === 'txt' && typeof child === 'string') {
+                    const matches = child.match(/:@(\d+);/g);
+                    if (matches) {
+                        for (const m of matches) {
+                            const uid = m.slice(2, -1);
+                            if (/^\d+$/.test(uid))
+                                out.add(uid);
+                        }
+                    }
+                    continue;
+                }
+            }
+            walkCollect(child, kind, out);
+        }
+        return;
+    }
+    if (kind === 'person' && typeof value === 'string') {
+        const matches = value.match(/:@(\d+);/g);
+        if (matches) {
+            for (const m of matches) {
+                const uid = m.slice(2, -1);
+                if (/^\d+$/.test(uid))
+                    out.add(uid);
+            }
+        }
+    }
+}
+/** 收集消息集合中待翻译 i18n id。 */
+export function collectI18nIds(msgs) {
+    const out = new Set();
+    for (const item of msgs) {
+        const msg = parseMsg(item?.msg);
+        if (msg)
+            walkCollect(msg, 'id18n', out);
+    }
+    return [...out];
+}
+/** 收集消息集合中人员 uid（@提及 + 人员对象 + 占位符）。 */
+export function collectPersonIds(msgs) {
+    const out = new Set();
+    for (const item of msgs) {
+        const msg = parseMsg(item?.msg);
+        if (msg)
+            walkCollect(msg, 'person', out);
+    }
+    return [...out];
+}
+/** 多行换行转 ⏎（展示单行化）。 */
+export function collapseNewlines(text) {
+    return String(text || '').replace(/\r\n+/g, '\n').replace(/\n+/g, '⏎');
+}
+/** 表格 | 转义。 */
+export function escapePipe(text) {
+    return String(text || '').replace(/\|/g, '\\|');
+}