dsh-trivium 0.4.9 → 0.4.13

package/lib/purge.js

@@ -0,0 +1,122 @@
+/** Wipe plugin files on real uninstall. Never run this on dsh web restart. */
+
+import { existsSync, readdirSync, unlinkSync } from "node:fs";
+import { basename, join, resolve } from "node:path";
+import { listedWorkspaces, settingsFilePath } from "./settings.js";
+import { closeAll, listOpenWorkspaces } from "./store.js";
+
+export function isTriviumArtifactName(name) {
+  const n = String(name || "");
+  return (
+    n === "trivium-pending.json" ||
+    n === "trivium.jsonl" ||
+    n.startsWith("trivium.jsonl") ||
+    n === "trivium.tdb" ||
+    n.startsWith("trivium.tdb")
+  );
+}
+
+/**
+ * `dsh plugin remove` / `pnpm remove` should wipe.
+ * `pnpm add` / update / ci reinstall the same package and must keep memory.
+ */
+export function shouldWipeOnUninstall(env = process.env, argv = process.argv) {
+  if (argv.includes("--force-purge")) return true;
+  const event = String(env.npm_lifecycle_event || "").toLowerCase();
+  if (event && event !== "preuninstall" && event !== "uninstall") return false;
+
+  const cmd = String(env.npm_command || "").toLowerCase();
+  if (cmd === "remove" || cmd === "uninstall") return true;
+  if (cmd === "install" || cmd === "add" || cmd === "update" || cmd === "upgrade" || cmd === "ci" || cmd === "dedupe") {
+    return false;
+  }
+
+  const cooked = String(env.npm_config_argv || "").toLowerCase();
+  if (/\b(remove|uninstall)\b/.test(cooked) && !/\b(install|add|update|upgrade)\b/.test(cooked)) return true;
+  if (/\b(install|add|update|upgrade|ci)\b/.test(cooked)) return false;
+
+  // Unknown command: keep memory. Upgrades must never delete the store.
+  return false;
+}
+
+export function collectPurgeCwds(extraCwds = []) {
+  const seen = new Set();
+  const out = [];
+  const add = (cwd) => {
+    const key = String(cwd || "").trim();
+    if (!key) return;
+    const resolved = resolve(key);
+    if (seen.has(resolved)) return;
+    seen.add(resolved);
+    out.push(resolved);
+  };
+  for (const cwd of listedWorkspaces()) add(cwd);
+  for (const row of listOpenWorkspaces()) add(row.cwd);
+  for (const cwd of extraCwds) add(cwd);
+  return out;
+}
+
+function unlinkQuiet(path, files, errors) {
+  try {
+    unlinkSync(path);
+    files.push(path);
+  } catch (err) {
+    if (err && err.code !== "ENOENT") errors.push({ path, message: String(err.message || err) });
+  }
+}
+
+export function purgeWorkspaceArtifacts(cwd, files, errors) {
+  const dir = join(resolve(cwd), ".dsh");
+  if (!existsSync(dir)) return;
+  let names = [];
+  try {
+    names = readdirSync(dir);
+  } catch (err) {
+    errors.push({ path: dir, message: String(err.message || err) });
+    return;
+  }
+  for (const name of names) {
+    if (!isTriviumArtifactName(name)) continue;
+    unlinkQuiet(join(dir, name), files, errors);
+  }
+}
+
+/** Delete known workspace traces and ~/.dsh/trivium.json. Leaves other .dsh files alone. */
+export function purgeAllTraces({ extraCwds = [], close = closeAll } = {}) {
+  const settingsFile = settingsFilePath();
+  const workspaces = collectPurgeCwds(extraCwds);
+  try {
+    close();
+  } catch {
+    // ignore
+  }
+  const files = [];
+  const errors = [];
+  for (const cwd of workspaces) purgeWorkspaceArtifacts(cwd, files, errors);
+  unlinkQuiet(settingsFile, files, errors);
+  return { ok: errors.length === 0, workspaces, files, errors, settingsFile };
+}
+
+function shouldRunCli() {
+  return basename(String(process.argv[1] || "")) === "purge.js";
+}
+
+export function runUninstallPurge(env = process.env, argv = process.argv, log = console) {
+  if (!shouldWipeOnUninstall(env, argv)) {
+    log.info?.("[dsh-trivium] skip data wipe (install/update, not remove)");
+    return { skipped: true };
+  }
+  const result = purgeAllTraces();
+  const n = result.files.length;
+  log.info?.(`[dsh-trivium] removed ${n} plugin file${n === 1 ? "" : "s"} from ${result.workspaces.length} workspace(s)`);
+  if (result.errors.length) {
+    log.warn?.("[dsh-trivium] some files could not be deleted (stop dsh web and delete leftovers):");
+    for (const err of result.errors) log.warn?.(`  ${err.path}: ${err.message}`);
+  }
+  return result;
+}
+
+if (shouldRunCli()) {
+  const result = runUninstallPurge();
+  if (result?.errors?.length) process.exitCode = 1;
+}

package/lib/server.js

@@ -1,8 +1,10 @@
 import { existsSync } from "node:fs";
 import {
+  generateSidecarFromDb,
   archiveNode,
   backfillEmbeddings,
   dbPathFor,
+  deleteGitSidecars,
   deleteNode,
   exportGraph,
   getActiveCwd,
@@ -15,7 +17,7 @@ import {
   updateNode,
 } from "./store.js";
 import { embeddingPublicStatus } from "./embed.js";
-import { chipsEnabledOf, recallModeOf, sessionLayerEnabledOf, writeUiSettings } from "./settings.js";
+import { chipsEnabledOf, configPanelOpenOf, gitSidecarEnabledOf, pluginLocaleOf, recallModeOf, sessionLayerEnabledOf, writeUiSettings } from "./settings.js";
 import { discoverWorkbuddyFiles, discoverExternalFiles, exportMarkdown, importWorkbuddy, importExternal } from "./markdown.js";
 import { checkNpmUpdate, installedVersion } from "./update.js";
 import { writeGateMessage } from "./hygiene.js";
@@ -116,6 +118,9 @@ async function handle(req, res, { liveOptions, status }) {
       extractEnabled: opts.extractEnabled !== false,
       chipsEnabled: chipsEnabledOf(opts),
       sessionLayerEnabled: opts.sessionLayerEnabled === true,
+      gitSidecarEnabled: gitSidecarEnabledOf(opts),
+      pluginLocale: pluginLocaleOf(opts),
+      configPanelOpen: configPanelOpenOf(opts),
       writeApproval: !!opts.writeApproval,
       mapTokenBudget: opts.mapTokenBudget || 400,
       workspaces: listOpenWorkspaces(),
@@ -138,6 +143,29 @@ async function handle(req, res, { liveOptions, status }) {
     return;
   }
 
+  if (path === "/api/dsh-trivium/sidecar/generate" && method === "POST") {
+    if (!gitSidecarEnabledOf(liveOptions())) {
+      sendJson(res, 400, { ok: false, message: "git sidecar off" });
+      return;
+    }
+    const { cwd, db } = await dbForRequest(url);
+    if (!db) {
+      sendJson(res, 400, { ok: false, message: "no workspace" });
+      return;
+    }
+    sendJson(res, 200, generateSidecarFromDb(cwd, db));
+    return;
+  }
+
+  if (path === "/api/dsh-trivium/sidecar/delete" && method === "POST") {
+    if (gitSidecarEnabledOf(liveOptions())) {
+      sendJson(res, 400, { ok: false, message: "turn sidecar off first" });
+      return;
+    }
+    sendJson(res, 200, deleteGitSidecars());
+    return;
+  }
+
   if (path === "/api/dsh-trivium/export" && method === "GET") {
     const { cwd, db } = await dbForRequest(url);
     if (!db) {

package/lib/settings.js

@@ -1,19 +1,68 @@
 import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
-import { homedir } from "node:os";
-import { join } from "node:path";
+import { homedir, tmpdir } from "node:os";
+import { dirname, join, relative, resolve, sep } from "node:path";
 import { looksHttpUrl } from "./embed.js";
 
 export const SETTINGS_FILE = join(homedir(), ".dsh", "trivium.json");
 
+/** Tests may set DSH_TRIVIUM_SETTINGS to a temp file so we never touch the live home config. */
+export function settingsFilePath() {
+  const override = String(process.env.DSH_TRIVIUM_SETTINGS || "").trim();
+  return override ? resolve(override) : SETTINGS_FILE;
+}
+
+function persistSettings(next) {
+  const file = settingsFilePath();
+  mkdirSync(dirname(file), { recursive: true });
+  writeFileSync(file, JSON.stringify(next, null, 2), "utf8");
+  return next;
+}
+
 export function readUiSettings() {
   try {
-    const raw = JSON.parse(readFileSync(SETTINGS_FILE, "utf8"));
+    const raw = JSON.parse(readFileSync(settingsFilePath(), "utf8"));
     return raw && typeof raw === "object" ? raw : {};
   } catch {
     return {};
   }
 }
 
+export function listedWorkspaces() {
+  const raw = readUiSettings().workspaces;
+  if (!Array.isArray(raw)) return [];
+  const seen = new Set();
+  const out = [];
+  for (const item of raw) {
+    const cwd = String(item || "").trim();
+    if (!cwd) continue;
+    const key = resolve(cwd);
+    if (seen.has(key)) continue;
+    seen.add(key);
+    out.push(key);
+  }
+  return out;
+}
+
+function isInsideDir(parent, child) {
+  const rel = relative(resolve(parent), resolve(child));
+  return rel === "" || (rel !== ".." && !rel.startsWith(".." + sep));
+}
+
+/** Remember a workspace so uninstall can delete its .tdb. No-op if already listed. */
+export function rememberWorkspace(cwd) {
+  const key = String(cwd || "").trim();
+  if (!key) return listedWorkspaces();
+  const resolved = resolve(key);
+  if (!process.env.DSH_TRIVIUM_SETTINGS && isInsideDir(tmpdir(), resolved)) {
+    return listedWorkspaces();
+  }
+  const prev = readUiSettings();
+  const workspaces = listedWorkspaces();
+  if (workspaces.includes(resolved)) return workspaces;
+  persistSettings({ ...prev, workspaces: [...workspaces, resolved] });
+  return [...workspaces, resolved];
+}
+
 export function recallModeOf(opts = {}) {
   const mode = opts.recallMode;
   if (mode === "auto" || mode === "anchor" || mode === "off") return mode;
@@ -33,6 +82,44 @@ export function sessionLayerEnabledOf(opts = {}) {
   return chipsEnabledOf(opts) && opts.sessionLayerEnabled === true;
 }
 
+/** Git JSONL sidecar write. Missing key = off. Import still runs when this is off. */
+export function gitSidecarEnabledOf(opts = {}) {
+  return opts.gitSidecarEnabled === true;
+}
+
+/** Plugin UI language. Missing key = follow the DSH host. */
+export function pluginLocaleOf(opts = {}) {
+  const v = String(opts.pluginLocale || "").toLowerCase();
+  if (v === "zh" || v === "en" || v === "host") return v;
+  return "host";
+}
+
+/** Settings fold. Missing key = collapsed. */
+export function configPanelOpenOf(opts = {}) {
+  return opts.configPanelOpen === true;
+}
+
+export function sidecarHashOf(cwd) {
+  const raw = readUiSettings().sidecarHashByCwd;
+  if (!raw || typeof raw !== "object") return "";
+  const key = resolve(String(cwd || ""));
+  return String(raw[key] || "");
+}
+
+export function writeSidecarHash(cwd, hash) {
+  const key = resolve(String(cwd || ""));
+  if (!process.env.DSH_TRIVIUM_SETTINGS && isInsideDir(tmpdir(), key)) {
+    return readUiSettings();
+  }
+  const prev = readUiSettings();
+  const sidecarHashByCwd = {
+    ...(prev.sidecarHashByCwd && typeof prev.sidecarHashByCwd === "object" ? prev.sidecarHashByCwd : {}),
+  };
+  if (!hash) delete sidecarHashByCwd[key];
+  else sidecarHashByCwd[key] = String(hash);
+  return persistSettings({ ...prev, sidecarHashByCwd });
+}
+
 export function applyRecallMode(target, mode) {
   const next = mode === "auto" || mode === "anchor" ? mode : "off";
   target.recallMode = next;
@@ -42,7 +129,6 @@ export function applyRecallMode(target, mode) {
 }
 
 export function writeUiSettings(patch) {
-  mkdirSync(join(homedir(), ".dsh"), { recursive: true });
   const prev = readUiSettings();
   const next = { ...prev };
   let mode = recallModeOf(prev);
@@ -62,6 +148,12 @@ export function writeUiSettings(patch) {
   if (typeof patch.extractEnabled === "boolean") next.extractEnabled = patch.extractEnabled;
   if (typeof patch.chipsEnabled === "boolean") next.chipsEnabled = patch.chipsEnabled;
   if (typeof patch.sessionLayerEnabled === "boolean") next.sessionLayerEnabled = patch.sessionLayerEnabled;
+  if (typeof patch.gitSidecarEnabled === "boolean") next.gitSidecarEnabled = patch.gitSidecarEnabled;
+  if (typeof patch.configPanelOpen === "boolean") next.configPanelOpen = patch.configPanelOpen;
+  if (typeof patch.pluginLocale === "string") {
+    const loc = String(patch.pluginLocale || "").toLowerCase();
+    next.pluginLocale = loc === "zh" || loc === "en" ? loc : "host";
+  }
   if (typeof patch.embeddingEnabled === "boolean") next.embeddingEnabled = patch.embeddingEnabled;
   if (typeof patch.embeddingUrl === "string") {
     const url = patch.embeddingUrl.trim();
@@ -76,8 +168,14 @@ export function writeUiSettings(patch) {
   } else if (prev.pinsBySession && typeof prev.pinsBySession === "object") {
     next.pinsBySession = prev.pinsBySession;
   }
-  writeFileSync(SETTINGS_FILE, JSON.stringify(next, null, 2), "utf8");
-  return next;
+  if (Array.isArray(patch.workspaces)) next.workspaces = patch.workspaces;
+  else if (Array.isArray(prev.workspaces)) next.workspaces = prev.workspaces;
+  if (patch.sidecarHashByCwd && typeof patch.sidecarHashByCwd === "object") {
+    next.sidecarHashByCwd = patch.sidecarHashByCwd;
+  } else if (prev.sidecarHashByCwd && typeof prev.sidecarHashByCwd === "object") {
+    next.sidecarHashByCwd = prev.sidecarHashByCwd;
+  }
+  return persistSettings(next);
 }
 
 export function pinsOf(sessionId) {
@@ -96,11 +194,10 @@ export function writePins(sessionId, ids) {
   if (!key) return nextIds;
   if (!nextIds.length) delete pinsBySession[key];
   else pinsBySession[key] = nextIds;
-  mkdirSync(join(homedir(), ".dsh"), { recursive: true });
-  writeFileSync(SETTINGS_FILE, JSON.stringify({ ...prev, pinsBySession }, null, 2), "utf8");
+  persistSettings({ ...prev, pinsBySession });
   return nextIds;
 }
 
 export function settingsFileExists() {
-  return existsSync(SETTINGS_FILE);
+  return existsSync(settingsFilePath());
 }