oh-my-dws 0.0.2 → 0.0.3

package/src/sync.js

@@ -0,0 +1,213 @@
+'use strict';
+
+/**
+ * 自动同步：定时查「钉钉里哪些会话有未读」，只对已归档的会话做增量拉取。
+ *
+ * 为什么这么绕：`+recent-conversations` 一页要十几秒，而 `+unread-chats` 只要 1~2 秒，
+ * 所以用未读列表当「有变化」的信号，再按 openConversationId 映射到本地归档去 pull。
+ */
+
+const fs = require('node:fs');
+const path = require('node:path');
+
+const P = require('./paths');
+const { runDws } = require('./dws');
+const { ensureDir, fmtSecond } = require('./util');
+const { shortNames, archiveKind, conversationId, groupTitle } = require('./archive');
+const A = require('./actions');
+
+const SETTINGS_FILE = path.join(P.DATA_DIR, 'settings.json');
+
+const DEFAULTS = { enabled: true, intervalSec: 60, minIntervalSec: 20, maxIntervalSec: 1800 };
+
+const state = {
+  enabled: DEFAULTS.enabled,
+  intervalSec: DEFAULTS.intervalSec,
+  timer: null,
+  running: false,
+  lastRunAt: '',
+  lastAdded: 0,
+  lastTouched: [],
+  lastError: '',
+  nextRunAt: '',
+};
+
+// ---------------------------------------------------------------- 设置持久化
+
+function readSettings() {
+  try {
+    const j = JSON.parse(fs.readFileSync(SETTINGS_FILE, 'utf8'));
+    const s = j && j.autoSync ? j.autoSync : {};
+    return {
+      enabled: s.enabled === undefined ? DEFAULTS.enabled : Boolean(s.enabled),
+      intervalSec: clampInterval(s.intervalSec),
+    };
+  } catch {
+    return { enabled: DEFAULTS.enabled, intervalSec: DEFAULTS.intervalSec };
+  }
+}
+
+function clampInterval(v) {
+  const n = Number(v);
+  if (!Number.isFinite(n)) return DEFAULTS.intervalSec;
+  return Math.min(DEFAULTS.maxIntervalSec, Math.max(DEFAULTS.minIntervalSec, Math.round(n)));
+}
+
+function writeSettings() {
+  ensureDir(P.DATA_DIR);
+  let all = {};
+  try {
+    all = JSON.parse(fs.readFileSync(SETTINGS_FILE, 'utf8')) || {};
+  } catch {}
+  all.autoSync = { enabled: state.enabled, intervalSec: state.intervalSec };
+  fs.writeFileSync(SETTINGS_FILE, `${JSON.stringify(all, null, 2)}\n`, 'utf8');
+}
+
+// ---------------------------------------------------------------- 同步逻辑
+
+/** 已归档会话里，cid → short 映射 */
+function archiveByCid() {
+  const map = new Map();
+  for (const short of shortNames()) {
+    if (archiveKind(short) === 'mentions') continue;
+    const cid = conversationId(short);
+    if (cid) map.set(cid, short);
+  }
+  return map;
+}
+
+function unreadConversations() {
+  const data = runDws(['chat', '+unread-chats', '--count', '100', '--format', 'json'], { timeout: 30000 });
+  const list = (data && data.conversations) || (data && data.data && data.data.conversations) || [];
+  return list
+    .filter((c) => c && c.conversationId)
+    .map((c) => ({ cid: String(c.conversationId), name: String(c.name || '') }));
+}
+
+/** 跑一轮：有未读且已归档的会话，逐个增量拉取 */
+async function runOnce(reason = 'timer') {
+  if (state.running) return { skipped: 'running' };
+  state.running = true;
+  const touched = [];
+  let added = 0;
+  let error = '';
+  try {
+    let unread = [];
+    try {
+      unread = unreadConversations();
+    } catch (e) {
+      error = e.message || String(e);
+      throw e;
+    }
+    const byCid = archiveByCid();
+    const targets = [];
+    for (const u of unread) {
+      const short = byCid.get(u.cid);
+      if (short && !targets.includes(short)) targets.push(short);
+    }
+    if (targets.length) {
+      const out = await A.withLock(async () => {
+        const detail = [];
+        for (const short of targets) {
+          try {
+            const r = A.pullGroup(short);
+            detail.push({ short, added: r.added || 0 });
+            added += r.added || 0;
+            if (r.added) touched.push(short);
+          } catch (e) {
+            detail.push({ short, error: e.message });
+          }
+        }
+        return detail;
+      });
+      state.lastTouched = out.filter((x) => x.added || x.error);
+    } else {
+      state.lastTouched = [];
+    }
+    state.lastError = '';
+  } catch (e) {
+    state.lastError = error || e.message || String(e);
+    state.lastTouched = [];
+  } finally {
+    state.running = false;
+    state.lastRunAt = fmtSecond(new Date());
+    state.lastAdded = added;
+    state.nextRunAt = state.enabled
+      ? fmtSecond(new Date(Date.now() + state.intervalSec * 1000))
+      : '';
+    if (reason !== 'manual') saveSettingsQuiet();
+  }
+  return {
+    ranAt: state.lastRunAt,
+    added,
+    touched: state.lastTouched,
+    error: state.lastError,
+    unreadCount: state.lastTouched.length,
+  };
+}
+
+function saveSettingsQuiet() {
+  try {
+    writeSettings();
+  } catch {}
+}
+
+// ---------------------------------------------------------------- 定时器
+
+function schedule() {
+  if (state.timer) {
+    clearInterval(state.timer);
+    state.timer = null;
+  }
+  if (!state.enabled) {
+    state.nextRunAt = '';
+    return;
+  }
+  state.timer = setInterval(() => {
+    runOnce('timer').catch(() => {});
+  }, state.intervalSec * 1000);
+  if (state.timer.unref) state.timer.unref();
+  state.nextRunAt = fmtSecond(new Date(Date.now() + state.intervalSec * 1000));
+}
+
+function status() {
+  return {
+    enabled: state.enabled,
+    intervalSec: state.intervalSec,
+    minIntervalSec: DEFAULTS.minIntervalSec,
+    maxIntervalSec: DEFAULTS.maxIntervalSec,
+    running: state.running,
+    lastRunAt: state.lastRunAt,
+    nextRunAt: state.nextRunAt,
+    lastAdded: state.lastAdded,
+    lastTouched: state.lastTouched,
+    lastError: state.lastError,
+  };
+}
+
+function configure({ enabled, intervalSec } = {}) {
+  if (enabled !== undefined) state.enabled = Boolean(enabled);
+  if (intervalSec !== undefined) state.intervalSec = clampInterval(intervalSec);
+  writeSettings();
+  schedule();
+  return status();
+}
+
+/** 服务启动时调用：读配置 + 起定时器（默认开，60s） */
+function start() {
+  const s = readSettings();
+  state.enabled = s.enabled;
+  state.intervalSec = s.intervalSec;
+  schedule();
+  // 启动后先跑一轮，把重启期间漏掉的补上
+  if (state.enabled) setTimeout(() => runOnce('boot').catch(() => {}), 5000).unref?.();
+  return status();
+}
+
+function stop() {
+  if (state.timer) clearInterval(state.timer);
+  state.timer = null;
+  state.nextRunAt = '';
+}
+
+module.exports = { SETTINGS_FILE, start, stop, configure, status, runOnce, archiveByCid };