oh-my-dws 0.0.2 → 0.0.3

package/src/ai.js

@@ -0,0 +1,300 @@
+'use strict';
+
+/**
+ * AI 帮忙写回复：任意 OpenAI 兼容接口（/chat/completions）。
+ *
+ * 配置来源，按优先级：
+ *   1. <数据目录>/ai.json（界面上填的，最优先）
+ *   2. 环境变量：AI_BASE_URL / AI_API_KEY / AI_MODEL（也认 OPENAI_*）
+ *   3. pi 自己的 ~/.pi/agent/models.json + settings.json 的默认 provider/model
+ *      —— 这样在本机通常「零配置」就能用
+ */
+
+const fs = require('node:fs');
+const os = require('node:os');
+const path = require('node:path');
+
+const P = require('./paths');
+const { ensureDir } = require('./util');
+
+const CONFIG_FILE = path.join(P.DATA_DIR, 'ai.json');
+
+function readFileConfig() {
+  try {
+    const j = JSON.parse(fs.readFileSync(CONFIG_FILE, 'utf8'));
+    return {
+      baseUrl: String(j.baseUrl || '').trim(),
+      apiKey: String(j.apiKey || '').trim(),
+      model: String(j.model || '').trim(),
+      systemPrompt: String(j.systemPrompt || '').trim(),
+    };
+  } catch {
+    return null;
+  }
+}
+
+/** 从 pi 的配置里推断一个可用端点（只读，不改它） */
+function fromPi() {
+  try {
+    const home = os.homedir();
+    const models = JSON.parse(fs.readFileSync(path.join(home, '.pi/agent/models.json'), 'utf8'));
+    const settings = JSON.parse(fs.readFileSync(path.join(home, '.pi/agent/settings.json'), 'utf8'));
+    const providers = models.providers || {};
+    const wanted = [settings.defaultProvider, ...Object.keys(providers)].filter(Boolean);
+    for (const name of wanted) {
+      const p = providers[name];
+      if (!p || !p.baseUrl || !p.apiKey) continue;
+      if (p.api && !String(p.api).includes('openai')) continue;
+      const list = Array.isArray(p.models) ? p.models : [];
+      const preferred = settings.defaultProvider === name && settings.defaultModel
+        ? settings.defaultModel
+        : list[0] && list[0].id;
+      if (!preferred) continue;
+      return {
+        baseUrl: String(p.baseUrl).replace(/\/+$/, ''),
+        apiKey: String(p.apiKey),
+        model: String(preferred),
+        source: `pi:${name}`,
+      };
+    }
+  } catch {
+    /* pi 没装/格式变了就算了 */
+  }
+  return null;
+}
+
+function resolveConfig() {
+  const file = readFileConfig();
+  if (file && file.baseUrl && file.apiKey && file.model) return { ...file, source: 'ai.json' };
+
+  const env = {
+    baseUrl: String(process.env.AI_BASE_URL || process.env.OPENAI_BASE_URL || '').trim(),
+    apiKey: String(process.env.AI_API_KEY || process.env.OPENAI_API_KEY || '').trim(),
+    model: String(process.env.AI_MODEL || process.env.OPENAI_MODEL || '').trim(),
+  };
+  if (env.baseUrl && env.apiKey && env.model) return { ...env, source: 'env' };
+
+  const pi = fromPi();
+  if (pi) return pi;
+
+  return { baseUrl: '', apiKey: '', model: '', source: '', systemPrompt: '' };
+}
+
+/** 给前端看的配置（apiKey 只回掩码） */
+function publicConfig() {
+  const c = resolveConfig();
+  const usable = Boolean(c.baseUrl && c.apiKey && c.model);
+  return {
+    configured: usable,
+    source: c.source || '',
+    baseUrl: c.baseUrl || '',
+    model: c.model || '',
+    apiKeyMasked: c.apiKey ? `${c.apiKey.slice(0, 4)}…${c.apiKey.slice(-4)}` : '',
+    systemPrompt: c.systemPrompt || '',
+    canSave: true,
+  };
+}
+
+function clearConfig() {
+  try {
+    fs.unlinkSync(CONFIG_FILE);
+  } catch {}
+  return publicConfig();
+}
+
+function saveConfig({ baseUrl, apiKey, model, systemPrompt }) {
+  const cur = readFileConfig() || {};
+  const next = {
+    baseUrl: String(baseUrl || cur.baseUrl || '').trim().replace(/\/+$/, ''),
+    apiKey: String(apiKey || cur.apiKey || '').trim(),
+    model: String(model || cur.model || '').trim(),
+    systemPrompt: String(systemPrompt == null ? cur.systemPrompt || '' : systemPrompt).trim(),
+  };
+  ensureDir(P.DATA_DIR);
+  fs.writeFileSync(CONFIG_FILE, `${JSON.stringify(next, null, 2)}\n`, 'utf8');
+  return publicConfig();
+}
+
+const DEFAULT_SYSTEM = `你在帮用户起草钉钉工作消息的中文回复。
+要求：
+- 只输出消息正文，不要解释、不要加引号、不要用 markdown 标题或列表符号（必要时可用 1. 2. 3. 这种行内编号）
+- 语气跟上下文一致：工作群里直接、简洁、有结论；先给结论或结论性表态，再补一两句必要信息
+- 不编造事实：不知道的信息不要瞎写，可以用「我确认下」「稍后回复」这类说法
+- 长度控制在 1~3 句，除非用户明确要求写长
+- 不要带署名、不要写「收到，感谢」这类空话堆砌`;
+
+function trimText(s, n) {
+  const t = String(s == null ? '' : s).replace(/\s+/g, ' ').trim();
+  return t.length > n ? `${t.slice(0, n)}…` : t;
+}
+
+async function draftReply({ context = [], target = '', hint = '', me = '' } = {}) {
+  const cfg = resolveConfig();
+  if (!cfg.baseUrl || !cfg.apiKey || !cfg.model) {
+    const err = new Error('还没配置 AI：在发消息浮层点「✨ AI」填一下接口地址、模型和 Key（也可以设置 AI_BASE_URL / AI_API_KEY / AI_MODEL）');
+    err.status = 400;
+    throw err;
+  }
+
+  const lines = (Array.isArray(context) ? context : [])
+    .slice(-40)
+    .map((m) => `${m.sender || '?'}（${String(m.time || '').slice(5, 16)}）：${trimText(m.text, 600)}`)
+    .join('\n');
+
+  const userMsg = [
+    target ? `# 会话\n${target}` : '',
+    me ? `# 我是\n${me}` : '',
+    lines ? `# 最近的对话（从旧到新）\n${lines}` : '',
+    hint ? `# 我想表达的意思 / 补充要求\n${hint}` : '',
+    '# 任务\n请以「我」的身份写一条要发到这个会话里的回复。',
+  ].filter(Boolean).join('\n\n');
+
+  const url = `${cfg.baseUrl.replace(/\/+$/, '')}/chat/completions`;
+  const ac = new AbortController();
+  const timer = setTimeout(() => ac.abort(), 60000);
+  let res;
+  try {
+    res = await fetch(url, {
+      method: 'POST',
+      signal: ac.signal,
+      headers: {
+        'Content-Type': 'application/json',
+        Authorization: `Bearer ${cfg.apiKey}`,
+      },
+      body: JSON.stringify({
+        model: cfg.model,
+        messages: [
+          { role: 'system', content: cfg.systemPrompt || DEFAULT_SYSTEM },
+          { role: 'user', content: userMsg },
+        ],
+        temperature: 0.6,
+      }),
+    });
+  } catch (e) {
+    const err = new Error(`连不上 ${url}：${e.name === 'AbortError' ? '超时（60s）' : e.message}`);
+    err.status = 502;
+    throw err;
+  } finally {
+    clearTimeout(timer);
+  }
+
+  const raw = await res.text();
+  if (!res.ok) {
+    const err = new Error(`AI 接口返回 ${res.status}：${trimText(raw, 300)}`);
+    err.status = 502;
+    throw err;
+  }
+  let data;
+  try {
+    data = JSON.parse(raw);
+  } catch {
+    const err = new Error('AI 接口返回的不是 JSON');
+    err.status = 502;
+    throw err;
+  }
+  const text = String(
+    (data.choices && data.choices[0] && data.choices[0].message && data.choices[0].message.content) || ''
+  ).trim();
+  if (!text) {
+    const err = new Error('AI 没返回内容');
+    err.status = 502;
+    throw err;
+  }
+  return { text, model: cfg.model, source: cfg.source };
+}
+
+const SUMMARY_SYSTEM = `你在帮用户快速读懂一段钉钉工作群/单聊记录。
+要求：
+- 用中文，纯文本，允许用「1. 2. 3.」或「- 」开头的短行，不要 markdown 标题符号（#）
+- 结构固定三段，没有内容就写「无」：
+  【结论】一两句话说明这段对话在聊什么、结果是什么
+  【待办】谁需要做什么，没有就写「无」
+  【风险/分歧】没定的事、有争议的点，没有就写「无」
+- 只根据给定消息，不补充任何外部信息，不猜测人名或时间
+- 总长度控制在 200 字左右`;
+
+async function summarize({ messages = [], target = '', rangeLabel = '' } = {}) {
+  const cfg = resolveConfig();
+  if (!cfg.baseUrl || !cfg.apiKey || !cfg.model) {
+    const err = new Error('还没配置 AI：点右上账号胶囊里的「AI 设置」填一下接口地址、模型和 Key');
+    err.status = 400;
+    throw err;
+  }
+  const rows = (Array.isArray(messages) ? messages : []).slice(-300);
+  if (!rows.length) {
+    const err = new Error('当前范围没有消息可总结');
+    err.status = 400;
+    throw err;
+  }
+  const text = rows
+    .map((m) => `${m.sender || '?'}（${String(m.time || '').slice(5, 16)}）：${trimText(m.text, 400)}`)
+    .join('\n');
+
+  const userMsg = [
+    target ? `# 会话\n${target}` : '',
+    rangeLabel ? `# 范围\n${rangeLabel}，共 ${rows.length} 条` : `# 范围\n共 ${rows.length} 条`,
+    `# 消息（从旧到新）\n${text}`,
+    '# 任务\n请按要求总结这段对话。',
+  ].filter(Boolean).join('\n\n');
+
+  const url = `${cfg.baseUrl.replace(/\/+$/, '')}/chat/completions`;
+  const ac = new AbortController();
+  const timer = setTimeout(() => ac.abort(), 90000);
+  let res;
+  try {
+    res = await fetch(url, {
+      method: 'POST',
+      signal: ac.signal,
+      headers: { 'Content-Type': 'application/json', Authorization: `Bearer ${cfg.apiKey}` },
+      body: JSON.stringify({
+        model: cfg.model,
+        messages: [
+          { role: 'system', content: SUMMARY_SYSTEM },
+          { role: 'user', content: userMsg },
+        ],
+        temperature: 0.3,
+      }),
+    });
+  } catch (e) {
+    const err = new Error(`连不上 ${url}：${e.name === 'AbortError' ? '超时（90s）' : e.message}`);
+    err.status = 502;
+    throw err;
+  } finally {
+    clearTimeout(timer);
+  }
+
+  const raw = await res.text();
+  if (!res.ok) {
+    const err = new Error(`AI 接口返回 ${res.status}：${trimText(raw, 300)}`);
+    err.status = 502;
+    throw err;
+  }
+  let data;
+  try {
+    data = JSON.parse(raw);
+  } catch {
+    const err = new Error('AI 接口返回的不是 JSON');
+    err.status = 502;
+    throw err;
+  }
+  const out = String(
+    (data.choices && data.choices[0] && data.choices[0].message && data.choices[0].message.content) || ''
+  ).trim();
+  if (!out) {
+    const err = new Error('AI 没返回内容');
+    err.status = 502;
+    throw err;
+  }
+  return { text: out, count: rows.length, model: cfg.model, source: cfg.source };
+}
+
+module.exports = {
+  CONFIG_FILE,
+  publicConfig,
+  saveConfig,
+  clearConfig,
+  draftReply,
+  summarize,
+  resolveConfig,
+  DEFAULT_SYSTEM,
+};

package/src/archive.js

@@ -112,13 +112,21 @@ function groupSummary(short) {
   if (data.kind === 'mentions') {
     const counts = {};
     for (const m of msgs) {
-      if (m.groupShort) counts[m.groupShort] = (counts[m.groupShort] || 0) + 1;
+      const key = m.groupShort || `cid:${m.groupCid || ''}`;
+      counts[key] = (counts[key] || 0) + 1;
     }
-    data.sources = (doc.sources || []).map((s) => ({
-      short: s.short || '',
-      name: s.name || '',
-      count: counts[s.short || ''] || 0,
-    }));
+    data.since = doc.since || '';
+    data.sources = (doc.sources || []).map((s) => {
+      const key = s.short || `cid:${s.cid || ''}`;
+      return {
+        short: s.short || '',
+        cid: s.cid || '',
+        name: s.name || '',
+        archived: Boolean(s.short),
+        count: counts[key] || 0,
+      };
+    });
+    data.sourceCount = data.sources.length; // 含未归档来源
   }
   return data;
 }
@@ -149,7 +157,7 @@ function writeGroupInfoMd(short, msgsDoc, mem = {}) {
       `# ${name}`, '',
       '| 项 | 值 |', '|---|---|',
       '| 类型 | @我 聚合视图 |',
-      `| 汇总范围 | 仅本工具已归档的 ${srcs.length} 个群 |`,
+      `| 汇总范围 | ${srcs.length} 个群（含未归档，点开可自动归档） |`,
       `| 消息跨度 | ${spanS} ~ ${spanE}（${msgs.length} 条） |`,
       `| 最近汇总 | ${pulled} |`, '',
       '## 来源群', '',
@@ -157,7 +165,8 @@ function writeGroupInfoMd(short, msgsDoc, mem = {}) {
     ];
     srcs.forEach((s, i) => {
       const n = parts[s.name || ''] || 0;
-      lines.push(`| ${i + 1} | ${s.name || ''} | ${n} | \`${s.cid || ''}\` |`);
+      const tag = s.short ? '' : '（未归档）';
+      lines.push(`| ${i + 1} | ${s.name || ''}${tag} | ${n} | \`${s.cid || ''}\` |`);
     });
     lines.push('', '> 数据源变化（新增/删除归档群）后，点一次「拉取最新消息」重新汇总。', '');
     writeText(infoFile, lines.join('\n'));

package/src/auth.js

@@ -124,4 +124,36 @@ function logout() {
   return { ok: true, output: text.slice(0, 500) };
 }
 
-module.exports = { authProfile, startLogin, loginRunning, logout };
+/**
+ * 我自己的 openDingTalkId（用来判断「这条是不是我自己发的」，不计未读）。
+ * 取法：auth status 拿 userId + userName → contact user search 按名字查，按 userId 对上那条。
+ * 结果缓存，进程内只查一次（查不到就下次再试）。
+ */
+let MY_OPEN_ID = null;
+let MY_OPEN_TRIED = 0;
+
+function myOpenId() {
+  if (MY_OPEN_ID) return MY_OPEN_ID;
+  if (Date.now() - MY_OPEN_TRIED < 60000) return '';
+  MY_OPEN_TRIED = Date.now();
+  try {
+    const st = runDws(['auth', 'status', '-f', 'json'], { timeout: 20000 });
+    const userId = String(st.user_id || '');
+    const userName = String(st.user_name || '');
+    if (!userName) return '';
+    const res = runDws(['contact', 'user', 'search', '--query', userName, '--format', 'json'], { timeout: 20000 });
+    const rows = res.result || [];
+    const hit = userId
+      ? rows.find((r) => String(r.userId || '') === userId)
+      : rows.length === 1
+        ? rows[0]
+        : null;
+    const oid = hit && String(hit.openDingTalkId || '');
+    if (oid) MY_OPEN_ID = oid;
+    return MY_OPEN_ID || '';
+  } catch {
+    return '';
+  }
+}
+
+module.exports = { authProfile, startLogin, loginRunning, logout, myOpenId };

package/src/avatars.js

@@ -0,0 +1,143 @@
+'use strict';
+
+/**
+ * 全局头像库：openDingTalkId → 本地头像文件 / 外链。
+ *
+ * 钉钉 CLI 不返回人员头像 URL（通讯录、成员、好友、会话信息都没有 avatar 字段），
+ * 所以这里让用户自己挂一次，之后**同一个人的联系人条目和所有消息头像都变真图**。
+ *
+ * 存：
+ *   <数据目录>/_媒体/头像/avatars.json   { "<openId>": { file|url, name, at } }
+ *   <数据目录>/_媒体/头像/<openId>.png   上传的本地图片
+ */
+
+const fs = require('node:fs');
+const path = require('node:path');
+
+const P = require('./paths');
+const { ensureDir } = require('./util');
+
+const DIR = path.join(P.MEDIA_DIR, '头像');
+const INDEX = path.join(DIR, 'avatars.json');
+const EXT = { 'image/png': 'png', 'image/jpeg': 'jpg', 'image/webp': 'webp', 'image/gif': 'gif' };
+
+function readIndex() {
+  try {
+    const j = JSON.parse(fs.readFileSync(INDEX, 'utf8'));
+    return j && typeof j === 'object' ? j : {};
+  } catch {
+    return {};
+  }
+}
+
+function writeIndex(idx) {
+  ensureDir(DIR);
+  fs.writeFileSync(INDEX, `${JSON.stringify(idx, null, 2)}\n`, 'utf8');
+}
+
+const safeKey = (openId) => String(openId || '').replace(/[^A-Za-z0-9_-]/g, '').slice(0, 64);
+
+/** 给前端用的 { openId: 可访问 url } */
+function list() {
+  const idx = readIndex();
+  const out = {};
+  for (const [openId, v] of Object.entries(idx)) {
+    if (!v) continue;
+    if (v.url) out[openId] = v.url;
+    else if (v.file && fs.existsSync(path.join(DIR, v.file))) {
+      out[openId] = `/api/avatar-file/${encodeURIComponent(openId)}`;
+    }
+  }
+  return out;
+}
+
+function remove(openId) {
+  const key = String(openId || '').trim();
+  const idx = readIndex();
+  const hit = idx[key];
+  if (hit && hit.file) {
+    try {
+      fs.unlinkSync(path.join(DIR, hit.file));
+    } catch {}
+  }
+  delete idx[key];
+  writeIndex(idx);
+  return { ok: true, avatars: list() };
+}
+
+/** dataUrl（本地图片）或 http(s) 链接 */
+function set({ openId, name = '', dataUrl = '', url = '', clear = false }) {
+  const key = String(openId || '').trim();
+  if (!key) {
+    const err = new Error('缺少 openDingTalkId');
+    err.status = 400;
+    throw err;
+  }
+  if (clear) return remove(key);
+
+  const idx = readIndex();
+  const prev = idx[key];
+
+  if (url) {
+    if (!/^https?:\/\/\S+$/i.test(url)) {
+      const err = new Error('只支持 http(s) 图片链接');
+      err.status = 400;
+      throw err;
+    }
+    if (prev && prev.file) {
+      try {
+        fs.unlinkSync(path.join(DIR, prev.file));
+      } catch {}
+    }
+    idx[key] = { url, name: String(name || ''), at: new Date().toISOString() };
+    writeIndex(idx);
+    return { ok: true, url, avatars: list() };
+  }
+
+  const m = String(dataUrl).match(/^data:(image\/[a-z+]+);base64,([A-Za-z0-9+/=\s]+)$/i);
+  if (!m) {
+    const err = new Error('只支持图片 dataURL（png/jpeg/webp/gif）');
+    err.status = 400;
+    throw err;
+  }
+  const ext = EXT[m[1].toLowerCase()];
+  if (!ext) {
+    const err = new Error(`不支持的图片类型：${m[1]}`);
+    err.status = 400;
+    throw err;
+  }
+  const buf = Buffer.from(m[2].replace(/\s/g, ''), 'base64');
+  if (!buf.length) {
+    const err = new Error('图片内容为空');
+    err.status = 400;
+    throw err;
+  }
+  if (buf.length > 3 * 1024 * 1024) {
+    const err = new Error('头像太大（>3MB），先压缩一下');
+    err.status = 400;
+    throw err;
+  }
+  if (prev && prev.file) {
+    try {
+      fs.unlinkSync(path.join(DIR, prev.file));
+    } catch {}
+  }
+  const file = `${safeKey(key) || 'user'}.${ext}`;
+  ensureDir(DIR);
+  fs.writeFileSync(path.join(DIR, file), buf);
+  idx[key] = { file, name: String(name || ''), at: new Date().toISOString() };
+  writeIndex(idx);
+  return { ok: true, url: `/api/avatar-file/${encodeURIComponent(key)}`, avatars: list() };
+}
+
+/** 读头像文件（按 openId） */
+function fileOf(openId) {
+  const key = String(openId || '').trim();
+  const idx = readIndex();
+  const hit = idx[key];
+  if (!hit || !hit.file) return '';
+  const file = path.join(DIR, path.basename(hit.file));
+  return fs.existsSync(file) ? file : '';
+}
+
+module.exports = { DIR, INDEX, list, set, remove, fileOf };