oh-my-dws 0.0.2 → 0.0.3

package/src/contacts.js

@@ -0,0 +1,563 @@
+'use strict';
+
+/**
+ * 常用联系人：**只针对真实单聊**（钉钉里的 direct 会话），不是群成员。
+ *
+ * 数据来源两条，合并去重（按单聊 openConversationId）：
+ *   1. dws chat +recent-conversations  → 当前账号真实的单聊会话（cid + 对方姓名 + 最近消息时间）
+ *   2. 本地已归档的单聊（_原始数据/<简称>_消息.json 里 kind=dm 的那批）
+ *
+ * 名单存在 <数据目录>/常用联系人.json：{ cid, name, openId?, addedAt }
+ *   - 有 cid：点开就能归档 / 打开这个单聊，历史 + 发消息都走现成逻辑
+ *   - 只有 openId（手动添加）：没有会话 id，退化成「跨群时间线 + 直接按 openDingTalkId 发单聊」
+ */
+
+const fs = require('node:fs');
+const path = require('node:path');
+const { execFile } = require('node:child_process');
+
+const P = require('./paths');
+const { ensureDir, fmtSecond } = require('./util');
+const {
+  shortNames,
+  loadMessages,
+  groupTitle,
+  archiveKind,
+  participants,
+} = require('./archive');
+
+const CONTACTS_FILE = path.join(P.DATA_DIR, '常用联系人.json');
+
+// ---------------------------------------------------------------- 名单存取
+
+function readContacts() {
+  try {
+    const raw = JSON.parse(fs.readFileSync(CONTACTS_FILE, 'utf8'));
+    const list = Array.isArray(raw) ? raw : raw && Array.isArray(raw.contacts) ? raw.contacts : [];
+    return list
+      .filter((c) => c && (c.cid || c.openId || c.openDingtalkId))
+      .map((c) => ({
+        cid: String(c.cid || c.conversationId || ''),
+        openId: String(c.openId || c.openDingtalkId || ''),
+        name: String(c.name || ''),
+        pinned: Boolean(c.pinned),
+        avatar: String(c.avatar || ''),
+        pinnedAt: String(c.pinnedAt || ''),
+        addedAt: String(c.addedAt || ''),
+      }));
+  } catch {
+    return [];
+  }
+}
+
+function writeContacts(list) {
+  ensureDir(P.DATA_DIR);
+  fs.writeFileSync(
+    CONTACTS_FILE,
+    `${JSON.stringify({ version: 2, contacts: list }, null, 2)}\n`,
+    'utf8'
+  );
+}
+
+const keyOf = (c) => c.cid || `open:${c.openId}`;
+
+// ---------------------------------------------------------------- 单聊来源
+
+/** 本地已归档的单聊：cid → { cid, openId, name, short, count, lastTime } */
+function archivedDms() {
+  const out = new Map();
+  for (const short of shortNames()) {
+    if (archiveKind(short) !== 'dm') continue;
+    const doc = loadMessages(short);
+    const msgs = doc.messages || [];
+    const parts = participants(msgs);
+    const openId = String(doc.peerOpenId || (parts.length === 1 ? parts[0][0] : '') || '');
+    const cid = String(doc.conversationId || '');
+    const item = {
+      cid,
+      openId,
+      name: groupTitle(short) || (parts.length === 1 ? parts[0][1] : '') || short,
+      short,
+      count: msgs.length,
+      lastTime: msgs.length ? String(msgs[msgs.length - 1].createTime || '') : '',
+      preview: msgs.length ? previewOf(msgs[msgs.length - 1].text) : '',
+    };
+    if (cid) out.set(`cid:${cid}`, item);
+    if (openId) out.set(`open:${openId}`, item);
+  }
+  return out;
+}
+
+const DM_CACHE_FILE = path.join(P.DATA_DIR, '_运行', 'dm-cache.json');
+const DM_TTL = 30 * 60 * 1000; // 30 分钟
+
+/**
+ * 钉钉单聊列表缓存。钉钉这个接口一页要十几秒，所以：
+ *   - 本地永远读缓存（内存/磁盘），请求立刻返回
+ *   - 缓存过期就起一个后台任务去刷新，不阻塞 HTTP
+ */
+let LIVE = { list: [], at: 0, running: false, error: '' };
+
+(function loadDiskCache() {
+  try {
+    const j = JSON.parse(fs.readFileSync(DM_CACHE_FILE, 'utf8'));
+    if (j && Array.isArray(j.list)) {
+      LIVE.list = j.list;
+      LIVE.at = Number(j.at) || 0;
+    }
+  } catch {}
+})();
+
+function saveDiskCache() {
+  try {
+    ensureDir(path.dirname(DM_CACHE_FILE));
+    fs.writeFileSync(DM_CACHE_FILE, `${JSON.stringify({ at: LIVE.at, list: LIVE.list }, null, 2)}\n`, 'utf8');
+  } catch {}
+}
+
+function runDwsAsync(args, { timeout = 180000 } = {}) {
+  return new Promise((resolve, reject) => {
+    execFile('dws', args, { timeout, maxBuffer: 64 * 1024 * 1024, encoding: 'utf8' }, (err, stdout) => {
+      if (err) return reject(err);
+      const out = String(stdout || '').trim();
+      if (!out.startsWith('{')) return reject(new Error('dws 返回不是 JSON'));
+      try {
+        resolve(JSON.parse(out));
+      } catch (e) {
+        reject(e);
+      }
+    });
+  });
+}
+
+async function fetchLiveDms() {
+  const now = new Date();
+  const start = new Date(now.getTime() - 365 * 86400000);
+  const data = await runDwsAsync([
+    'chat', '+recent-conversations',
+    '--start', fmtSecond(start), '--end', fmtSecond(now),
+    '--limit', '100', '--format', 'json',
+  ]);
+  const list = (data && data.data && data.data.conversations) || (data && data.conversations) || [];
+  return list
+    .filter((c) => String(c.type || '') === 'direct' && c.conversationId)
+    .map((c) => ({
+      cid: String(c.conversationId),
+      name: String(c.name || '(未命名单聊)'),
+      lastTime: String(c.latestMessageTime || '').replace('T', ' ').slice(0, 16),
+    }));
+}
+
+// ---------------------------------------------------------------- 最后一条消息预览
+
+const PREVIEW_FILE = path.join(P.DATA_DIR, '_运行', 'dm-preview.json');
+const PREVIEW_TTL = 10 * 60 * 1000;
+
+/** cid -> { text, time, at } */
+let PREVIEWS = {};
+let PREVIEW_RUNNING = false;
+
+(function loadPreviewCache() {
+  try {
+    const j = JSON.parse(fs.readFileSync(PREVIEW_FILE, 'utf8'));
+    if (j && typeof j === 'object') PREVIEWS = j;
+  } catch {}
+})();
+
+function savePreviewCache() {
+  try {
+    ensureDir(path.dirname(PREVIEW_FILE));
+    fs.writeFileSync(PREVIEW_FILE, `${JSON.stringify(PREVIEWS, null, 2)}\n`, 'utf8');
+  } catch {}
+}
+
+/** 把消息正文压成一行预览 */
+function previewOf(text) {
+  const t = String(text || '')
+    .replace(/\r/g, '')
+    .replace(/\n+/g, ' ')
+    .replace(/\[文件\]/g, '[文件]')
+    .trim();
+  return t.length > 60 ? `${t.slice(0, 60)}…` : t;
+}
+
+async function fetchLastMessage(cid) {
+  const data = await runDwsAsync(
+    ['chat', '+chat-messages', '--group', cid, '--order', 'desc', '--limit', '1',
+      '--no-reactions', '--format', 'json'],
+    { timeout: 60000 }
+  );
+  const msgs = data.messages || [];
+  const m = msgs[0];
+  if (!m) return null;
+  return { text: previewOf(m.text), time: String(m.createTime || '').slice(0, 16), at: Date.now() };
+}
+
+/** 后台补齐常用联系人的「最后一条消息」，不阻塞接口 */
+function pumpPreviews() {
+  if (PREVIEW_RUNNING) return;
+  const pool = dmPool(false);
+  const need = readContacts()
+    .filter((c) => c.cid)
+    .map((c) => {
+      const hit = pool.get(`cid:${c.cid}`);
+      const cached = PREVIEWS[c.cid];
+      const fresh = cached && Date.now() - Number(cached.at || 0) < PREVIEW_TTL;
+      return { cid: c.cid, archived: Boolean(hit && hit.short), fresh };
+    })
+    .filter((x) => !x.fresh);
+  if (!need.length) return;
+  PREVIEW_RUNNING = true;
+  (async () => {
+    try {
+      for (const item of need) {
+        try {
+          const got = await fetchLastMessage(item.cid);
+          if (got) PREVIEWS[item.cid] = got;
+          else PREVIEWS[item.cid] = { text: '', time: '', at: Date.now() };
+        } catch {
+          PREVIEWS[item.cid] = { text: '', time: '', at: Date.now(), error: true };
+        }
+      }
+      savePreviewCache();
+    } finally {
+      PREVIEW_RUNNING = false;
+    }
+  })();
+}
+
+/** 拿缓存里的单聊列表；过期就后台刷新，返回当前状态（绝不阻塞） */
+function liveDms(force = false) {
+  const stale = Date.now() - LIVE.at > DM_TTL;
+  if ((stale || force) && !LIVE.running) {
+    LIVE.running = true;
+    LIVE.error = '';
+    fetchLiveDms()
+      .then((list) => {
+        LIVE = { list, at: Date.now(), running: false, error: '' };
+        saveDiskCache();
+      })
+      .catch((e) => {
+        LIVE.running = false;
+        LIVE.error = e.message || String(e);
+      });
+  }
+  return {
+    list: LIVE.list,
+    updatedAt: LIVE.at,
+    refreshing: LIVE.running,
+    error: LIVE.error,
+  };
+}
+
+/** 单聊候选池：真实单聊 + 已归档单聊，按 cid 去重，已归档的带上条数 */
+function dmPool(includeLive = true) {
+  const arch = archivedDms();
+  const map = new Map(); // key -> { cid, openId, name, short, count, lastTime, archived }
+  for (const item of arch.values()) {
+    const key = item.cid ? `cid:${item.cid}` : `open:${item.openId}`;
+    map.set(key, { ...item, archived: true });
+  }
+  if (!includeLive) return map;
+  for (const conv of liveDms().list) {
+    const key = `cid:${conv.cid}`;
+    const hit = map.get(key);
+    if (hit) {
+      if (!hit.name || hit.name === conv.name) hit.name = conv.name || hit.name;
+      if (!hit.lastTime) hit.lastTime = conv.lastTime;
+    } else {
+      map.set(key, { cid: conv.cid, openId: '', name: conv.name, short: '', count: 0, lastTime: conv.lastTime, archived: false });
+    }
+  }
+  return map;
+}
+
+// ---------------------------------------------------------------- 对外接口
+
+/** 常用联系人（带单聊状态：是否已归档、多少条、最近时间） */
+function listContacts() {
+  const pool = dmPool(false); // 侧栏只看本地归档，不碰网络
+  pumpPreviews();             // 缺预览的在后台慢慢补
+  const rows = readContacts().map((c) => {
+    const hit = (c.cid && pool.get(`cid:${c.cid}`)) || (c.openId && pool.get(`open:${c.openId}`)) || null;
+    const pv = c.cid ? PREVIEWS[c.cid] : null;
+    const archTime = (hit && hit.lastTime) || '';
+    // 归档里的最后一条 / 实时拉到的最后一条，取更新的那个
+    const useLive = pv && pv.text && String(pv.time || '') >= archTime;
+    return {
+      ...c,
+      name: c.name || (hit && hit.name) || c.cid || c.openId,
+      openId: c.openId || (hit && hit.openId) || '',
+      hasDm: Boolean(hit && hit.cid),
+      short: hit ? hit.short : '',
+      msgCount: hit ? hit.count : 0,
+      lastTime: useLive ? String(pv.time || '') : archTime,
+      preview: useLive ? String(pv.text || '') : (hit ? String(hit.preview || '') : ''),
+      previewPending: !useLive && !(hit && hit.preview) && Boolean(c.cid),
+    };
+  });
+
+  return rows; // 顺序 = 名单里的顺序（可拖动调整；置顶只是把条目前移 + 标记）
+}
+
+/** 可选单聊池（只列真实单聊 / 已归档单聊，不列群成员） */
+function candidates(query = '', force = false) {
+  const q = String(query || '').trim().toLowerCase();
+  const saved = new Set(readContacts().map(keyOf));
+  let rows = [...dmPool().values()].map((p) => ({
+    cid: p.cid,
+    openId: p.openId,
+    name: p.name,
+    msgCount: p.count,
+    lastTime: p.lastTime,
+    short: p.short,
+    archived: p.archived,
+    saved: saved.has(p.cid || `open:${p.openId}`),
+  }));
+  if (q) {
+    rows = rows.filter(
+      (r) =>
+        r.name.toLowerCase().includes(q) ||
+        String(r.cid || '').toLowerCase().includes(q) ||
+        String(r.openId || '').toLowerCase().includes(q)
+    );
+  }
+  rows.sort((a, b) => String(b.lastTime || '').localeCompare(String(a.lastTime || '')));
+  const live = liveDms(Boolean(force));
+  return {
+    total: rows.length,
+    candidates: rows.slice(0, 300),
+    dmCache: {
+      updatedAt: live.updatedAt,
+      refreshing: live.refreshing,
+      error: live.error,
+      count: live.list.length,
+    },
+  };
+}
+
+function addContact({ cid, openId, name }) {
+  const theCid = String(cid || '').trim();
+  const theOpen = String(openId || '').trim();
+  const nm = String(name || '').trim();
+  if (!theCid && !theOpen) {
+    const err = new Error('需要一个单聊会话 ID（openConversationId）或 openDingTalkId');
+    err.status = 400;
+    throw err;
+  }
+  const list = readContacts();
+  const hit = list.find((c) => (theCid ? c.cid === theCid : c.openId === theOpen));
+  if (hit) {
+    if (nm) hit.name = nm;
+    if (theCid) hit.cid = theCid;
+    if (theOpen) hit.openId = theOpen;
+  } else {
+    list.unshift({ cid: theCid, openId: theOpen, name: nm, addedAt: new Date().toISOString() });
+  }
+  writeContacts(list);
+  return { ok: true, total: list.length, contact: list.find((c) => (theCid ? c.cid === theCid : c.openId === theOpen)) };
+}
+
+function pinContact({ cid, openId, pinned = true }) {
+  const theCid = String(cid || '').trim();
+  const theOpen = String(openId || '').trim();
+  if (!theCid && !theOpen) {
+    const err = new Error('缺少 cid 或 openDingTalkId');
+    err.status = 400;
+    throw err;
+  }
+  const list = readContacts();
+  const hit = list.find((c) => (theCid ? c.cid === theCid : c.openId === theOpen));
+  if (!hit) {
+    const err = new Error('这条不在常用联系人里');
+    err.status = 404;
+    throw err;
+  }
+  hit.pinned = Boolean(pinned);
+  hit.pinnedAt = hit.pinned ? new Date().toISOString() : '';
+  writeContacts(list);
+  return { ok: true, contact: hit };
+}
+
+/** 拖动排序：按传入的 key 顺序重排名单（key = cid|open:<openId>） */
+function reorderContacts(keys = []) {
+  const list = readContacts();
+  const keyOfRow = (c) => c.cid || `open:${c.openId}`;
+  const map = new Map(list.map((c) => [keyOfRow(c), c]));
+  const next = [];
+  for (const k of keys) {
+    const hit = map.get(String(k));
+    if (hit) {
+      next.push(hit);
+      map.delete(String(k));
+    }
+  }
+  for (const rest of map.values()) next.push(rest); // 没传到的保持在后
+  writeContacts(next);
+  return { ok: true, total: next.length, order: next.map(keyOfRow) };
+}
+
+const AVATAR_DIR = path.join(P.MEDIA_DIR, '头像');
+const AVATAR_EXT = { 'image/png': 'png', 'image/jpeg': 'jpg', 'image/webp': 'webp', 'image/gif': 'gif' };
+
+/** 直接存一个图片链接（http/https），前端按 <img src> 显示 */
+function setAvatarUrl({ cid, openId, url = '' }) {
+  const theCid = String(cid || '').trim();
+  const theOpen = String(openId || '').trim();
+  const u = String(url || '').trim();
+  if (!/^https?:\/\/[^\s]+$/i.test(u) && !/^data:image\//i.test(u)) {
+    const err = new Error('只支持 http(s) 图片链接');
+    err.status = 400;
+    throw err;
+  }
+  const list = readContacts();
+  const hit = list.find((c) => (theCid ? c.cid === theCid : c.openId === theOpen));
+  if (!hit) {
+    const err = new Error('这条不在常用联系人里');
+    err.status = 404;
+    throw err;
+  }
+  hit.avatar = u;
+  writeContacts(list);
+  return { ok: true, contact: hit };
+}
+
+/** 自定义头像：前端传 dataURL，落盘到 _媒体/头像/，联系人里只存 /api/avatar/<文件> */
+function setAvatar({ cid, openId, dataUrl = '', clear = false }) {
+  const theCid = String(cid || '').trim();
+  const theOpen = String(openId || '').trim();
+  if (!theCid && !theOpen) {
+    const err = new Error('缺少 cid 或 openDingTalkId');
+    err.status = 400;
+    throw err;
+  }
+  const list = readContacts();
+  const hit = list.find((c) => (theCid ? c.cid === theCid : c.openId === theOpen));
+  if (!hit) {
+    const err = new Error('这条不在常用联系人里');
+    err.status = 404;
+    throw err;
+  }
+  if (clear) {
+    hit.avatar = '';
+    writeContacts(list);
+    return { ok: true, contact: hit };
+  }
+  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 = AVATAR_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;
+  }
+  ensureDir(AVATAR_DIR);
+  const safe = (theCid || theOpen).replace(/[^A-Za-z0-9_-]/g, '').slice(0, 32) || 'contact';
+  const file = `${safe}-${Date.now().toString(36)}.${ext}`;
+  // 先清掉这个人的旧头像（别把刚写的文件也删了）
+  for (const old of fs.readdirSync(AVATAR_DIR)) {
+    if (old !== file && old.startsWith(`${safe}-`)) {
+      try {
+        fs.unlinkSync(path.join(AVATAR_DIR, old));
+      } catch {}
+    }
+  }
+  fs.writeFileSync(path.join(AVATAR_DIR, file), buf);
+  hit.avatar = `/api/avatar/${file}`;
+  writeContacts(list);
+  return { ok: true, contact: hit };
+}
+
+/** 读头像文件（防路径穿越） */
+function avatarFile(name) {
+  const base = path.basename(String(name || ''));
+  if (!/^[A-Za-z0-9_-]+\.(png|jpg|jpeg|webp|gif)$/i.test(base)) return '';
+  const file = path.join(AVATAR_DIR, base);
+  return fs.existsSync(file) ? file : '';
+}
+
+function removeContact({ cid, openId }) {
+  const theCid = String(cid || '').trim();
+  const theOpen = String(openId || '').trim();
+  const list = readContacts();
+  const next = list.filter((c) => (theCid ? c.cid !== theCid : c.openId !== theOpen));
+  writeContacts(next);
+  return { ok: true, removed: list.length - next.length, total: next.length };
+}
+
+/** 只有 openId、没有单聊会话的手动联系人：退化成跨群时间线 */
+function personMessages(openId) {
+  const oid = String(openId || '').trim();
+  if (!oid) {
+    const err = new Error('缺少 openDingTalkId');
+    err.status = 400;
+    throw err;
+  }
+  const messages = [];
+  const counts = new Map();
+  let name = '';
+  for (const short of shortNames()) {
+    if (archiveKind(short) === 'mentions') continue;
+    for (const m of loadMessages(short).messages || []) {
+      if (String(m.senderId || '') !== oid) continue;
+      if (!name && m.sender) name = m.sender;
+      messages.push({
+        conversationId: m.conversationId || '',
+        messageId: m.messageId || '',
+        sender: m.sender || '',
+        senderId: m.senderId || '',
+        text: m.text || '',
+        createTime: m.createTime || '',
+        groupShort: short,
+        groupName: groupTitle(short),
+      });
+      counts.set(short, (counts.get(short) || 0) + 1);
+    }
+  }
+  messages.sort((a, b) => {
+    const t = String(a.createTime || '').localeCompare(String(b.createTime || ''));
+    return t !== 0 ? t : String(a.messageId || '').localeCompare(String(b.messageId || ''));
+  });
+  const groups = [...counts.entries()]
+    .map(([short, count]) => ({ short, name: groupTitle(short), count }))
+    .sort((a, b) => b.count - a.count);
+  const saved = readContacts().find((c) => c.openId === oid);
+  return {
+    openId: oid,
+    name: (saved && saved.name) || name || oid,
+    messages,
+    groups,
+    count: messages.length,
+  };
+}
+
+module.exports = {
+  CONTACTS_FILE,
+  readContacts,
+  listContacts,
+  addContact,
+  removeContact,
+  pinContact,
+  reorderContacts,
+  setAvatar,
+  setAvatarUrl,
+  avatarFile,
+  candidates,
+  personMessages,
+  dmPool,
+};

package/src/layout.js

@@ -0,0 +1,63 @@
+'use strict';
+
+/**
+ * 左侧会话列表的布局：手拖顺序 + 置顶。
+ * 存 <数据目录>/布局.json —— {"order": ["群简称", ...], "pinned": ["群简称", ...]}
+ * 只记「简称」，群删了/改了名会自动从列表里消失（排序时忽略未知项）。
+ */
+
+const fs = require('node:fs');
+const path = require('node:path');
+
+const P = require('./paths');
+const { ensureDir } = require('./util');
+
+const LAYOUT_FILE = path.join(P.DATA_DIR, '布局.json');
+
+function readLayout() {
+  try {
+    const j = JSON.parse(fs.readFileSync(LAYOUT_FILE, 'utf8'));
+    const arr = (v) => (Array.isArray(v) ? v.filter((x) => typeof x === 'string' && x) : []);
+    return { order: arr(j.order), pinned: arr(j.pinned) };
+  } catch {
+    return { order: [], pinned: [] };
+  }
+}
+
+function writeLayout(layout) {
+  ensureDir(P.DATA_DIR);
+  const out = {
+    order: Array.isArray(layout.order) ? [...new Set(layout.order.filter(Boolean))] : [],
+    pinned: Array.isArray(layout.pinned) ? [...new Set(layout.pinned.filter(Boolean))] : [],
+  };
+  fs.writeFileSync(LAYOUT_FILE, `${JSON.stringify(out, null, 2)}\n`, 'utf8');
+  return out;
+}
+
+/** 拖动排序：keys = 拖完之后的完整可视顺序（会话简称数组） */
+function reorderGroups({ keys = [] } = {}) {
+  const layout = readLayout();
+  const incoming = (Array.isArray(keys) ? keys : []).filter((k) => typeof k === 'string' && k);
+  const known = new Set(incoming);
+  // 没出现在可视列表里的（例如被筛选掉了）按原顺序接在后面，别丢
+  const tail = layout.order.filter((k) => !known.has(k));
+  layout.order = [...incoming, ...tail];
+  return { ok: true, layout: writeLayout(layout) };
+}
+
+function setPinned({ short, pinned = true } = {}) {
+  const name = String(short || '').trim();
+  if (!name) {
+    const err = new Error('缺少会话简称');
+    err.status = 400;
+    throw err;
+  }
+  const layout = readLayout();
+  const set = new Set(layout.pinned);
+  if (pinned) set.add(name);
+  else set.delete(name);
+  layout.pinned = [...set];
+  return { ok: true, layout: writeLayout(layout) };
+}
+
+module.exports = { LAYOUT_FILE, readLayout, writeLayout, reorderGroups, setPinned };