@dtoolkit/dcontext 0.1.5 → 0.2.0

package/README.md

@@ -1,5 +1,5 @@
 <p align="center">
-  <img src="../../logo.png" alt="dtoolkit" />
+  <img src="../../logo.png" alt="dtoolkit" width="420"/>
 </p>
 
 <h1 align="center">@dtoolkit/dcontext</h1>

package/dist/index.js

@@ -9,7 +9,7 @@ import { Command } from "commander";
 import pc from "picocolors";
 
 // src/core/briefing.ts
-import { DBrainClient } from "@dtoolkit/sdk";
+import { DBrainClient, DWorkClient } from "@dtoolkit/sdk";
 function cleanDocContent(content, sectionTitle) {
   const lines = content.split("\n");
   const cleaned = [];
@@ -36,7 +36,7 @@ function truncateFact(fact, max) {
   if (fact.length <= max) return fact;
   return fact.slice(0, max - 1) + "\u2026";
 }
-async function generateBriefing(projectEntity, config) {
+async function generateBriefing(projectEntity, config, cwd) {
   const client = new DBrainClient(config.dbrain.url, config.dbrain.token);
   const safeQuery = `"${projectEntity}"`;
   const [results, documents, health] = await Promise.all([
@@ -94,6 +94,40 @@ async function generateBriefing(projectEntity, config) {
     } catch {
     }
   }
+  const dworkSlugs = cwd ? config.dworkProjects?.[cwd] : void 0;
+  if (config.dwork?.url && dworkSlugs && dworkSlugs.length > 0) {
+    try {
+      const dwork = new DWorkClient(config.dwork.url, config.dwork.token);
+      const allTasks = (await Promise.all(dworkSlugs.map((slug) => dwork.listTasks(slug).catch(() => [])))).flat();
+      const active = allTasks.filter(
+        (t) => t.status === "doing" || t.status === "blocked" || t.status === "todo"
+      );
+      if (active.length > 0) {
+        parts.push("");
+        parts.push(`### Tasks (dwork)`);
+        parts.push(
+          "> Active tasks are listed below. **Do NOT call `get_tasks` or `list_projects` unless the user explicitly asks.** Use `update_task` only when the user requests a status change."
+        );
+        const doing = active.filter((t) => t.status === "doing");
+        const blocked = active.filter((t) => t.status === "blocked");
+        const todo = active.filter((t) => t.status === "todo");
+        for (const group2 of [
+          { label: "doing", items: doing },
+          { label: "blocked", items: blocked },
+          { label: "todo", items: todo }
+        ]) {
+          for (const t of group2.items) {
+            const project = t.project_slug;
+            const meta = [t.priority, t.estimate, t.deadline].filter(Boolean).join(", ");
+            parts.push(
+              `- [${group2.label}] ${t.id} ${project}: ${truncateFact(t.title, 100)}${meta ? ` (${meta})` : ""}`
+            );
+          }
+        }
+      }
+    } catch {
+    }
+  }
   let briefing = parts.join("\n");
   if (briefing.length > config.briefing.maxChars) {
     briefing = briefing.slice(0, config.briefing.maxChars - 3) + "\u2026";
@@ -113,6 +147,7 @@ var DEFAULT_CONFIG = {
     token: ""
   },
   projects: {},
+  dworkProjects: {},
   briefing: {
     maxFacts: 15,
     includeIdentity: true,
@@ -395,7 +430,7 @@ async function handleSessionStart(target, cwd, config) {
     process.stdout.write("{}");
     return;
   }
-  const briefing = await generateBriefing(projectEntity, config);
+  const briefing = await generateBriefing(projectEntity, config, cwd);
   if (!briefing) {
     process.stdout.write("{}");
     return;
@@ -421,7 +456,7 @@ async function handlePreCompact(sessionId, target, cwd, config) {
 
 // src/commands/init.ts
 import * as p from "@clack/prompts";
-import { DBrainClient as DBrainClient3 } from "@dtoolkit/sdk";
+import { DBrainClient as DBrainClient3, DWorkClient as DWorkClient2 } from "@dtoolkit/sdk";
 import { Command as Command3 } from "commander";
 import pc2 from "picocolors";
 function createInitCommand() {
@@ -459,11 +494,18 @@ async function runInit() {
     }
   ];
   if (config.initialized && config.dbrain.url) {
-    options.unshift({
-      value: "remap",
-      label: "Keep dbrain, remap this directory",
-      description: `Change entity mapping for ${process.cwd()}`
-    });
+    options.unshift(
+      {
+        value: "keep",
+        label: "Keep current config",
+        description: "Only update dwork connection"
+      },
+      {
+        value: "remap",
+        label: "Keep dbrain, remap this directory",
+        description: `Change entity mapping for ${process.cwd()}`
+      }
+    );
   }
   const mode = await p.select({
     message: config.initialized ? "What do you want to change?" : "How do you want to connect to dbrain?",
@@ -473,7 +515,21 @@ async function runInit() {
     p.cancel("Init cancelled.");
     process.exit(0);
   }
-  if (mode === "remap") {
+  if (mode === "keep") {
+    await detectDwork(config);
+    config.initialized = true;
+    await saveConfig(config);
+    p.note(
+      [
+        `dbrain:  ${pc2.green(config.dbrain.url)}`,
+        config.dwork ? `dwork:   ${pc2.green(config.dwork.url)}` : "",
+        `Config:  ${pc2.green(getDataDir() + "/config.json")}`
+      ].filter(Boolean).join("\n"),
+      "Configuration"
+    );
+    p.outro("Done");
+    return;
+  } else if (mode === "remap") {
     await finishInit(config);
   } else if (mode === "local") {
     await startLocalDbrain(config);
@@ -635,11 +691,13 @@ async function finishInit(config) {
   } catch (err) {
     p.log.warn(`Could not list entities: ${err.message}`);
   }
+  await detectDwork(config);
   config.initialized = true;
   await saveConfig(config);
   p.note(
     [
       `dbrain:  ${pc2.green(config.dbrain.url)}`,
+      config.dwork ? `dwork:   ${pc2.green(config.dwork.url)}` : "",
       entityName ? `Project: ${pc2.green(process.cwd())} \u2192 ${pc2.green(entityName)}` : "",
       `Config:  ${pc2.green(getDataDir() + "/config.json")}`
     ].filter(Boolean).join("\n"),
@@ -647,6 +705,87 @@ async function finishInit(config) {
   );
   p.outro(`Next: ${pc2.cyan("dcontext install claude")}`);
 }
+async function detectDwork(config) {
+  if (config.dwork?.url) {
+    try {
+      const client = new DWorkClient2(config.dwork.url, config.dwork.token);
+      const overview = await client.overview();
+      p.log.info(
+        `dwork: ${pc2.green(config.dwork.url)} \u2014 ${overview.totalProjects} projects, ${overview.totalTasks} tasks`
+      );
+      const keep = await p.confirm({
+        message: "Keep current dwork connection?",
+        initialValue: true
+      });
+      if (!p.isCancel(keep) && keep) return;
+    } catch {
+      p.log.warn(`dwork: ${pc2.red(config.dwork.url)} (unreachable)`);
+    }
+  }
+  const connect = await p.confirm({
+    message: "Connect to a dwork server? (project management)",
+    initialValue: !!config.dwork?.url
+  });
+  if (p.isCancel(connect) || !connect) {
+    config.dwork = void 0;
+    return;
+  }
+  const answers = await p.group(
+    {
+      url: () => p.text({
+        message: "dwork URL",
+        initialValue: config.dwork?.url || "http://localhost:7881",
+        validate: (v) => !v ? "URL is required" : void 0
+      }),
+      token: () => p.text({
+        message: "dwork token (empty if none)",
+        initialValue: config.dwork?.token || ""
+      })
+    },
+    {
+      onCancel: () => {
+        p.cancel("Init cancelled.");
+        process.exit(0);
+      }
+    }
+  );
+  const s = p.spinner();
+  s.start("Connecting to dwork");
+  let projects = [];
+  try {
+    const client = new DWorkClient2(answers.url, answers.token);
+    const listed = await client.listProjects();
+    projects = listed.map((pr) => ({ slug: pr.slug, name: pr.name }));
+    s.stop(`Connected \u2014 ${pc2.green(String(projects.length))} projects`);
+    config.dwork = { url: answers.url, token: answers.token };
+  } catch (err) {
+    s.stop(pc2.red("Connection failed"));
+    p.log.error(err.message);
+    p.log.warn("Skipping dwork. You can add it later with dcontext init.");
+    config.dwork = void 0;
+    return;
+  }
+  if (projects.length > 0) {
+    const cwd = process.cwd();
+    const current = config.dworkProjects[cwd] || [];
+    const selected = await p.multiselect({
+      message: `Map ${pc2.dim(cwd)} to dwork projects:`,
+      options: projects.map((pr) => ({
+        value: pr.slug,
+        label: `${pr.name} (${pr.slug})`,
+        selected: current.includes(pr.slug)
+      })),
+      required: false
+    });
+    if (!p.isCancel(selected)) {
+      if (selected.length > 0) {
+        config.dworkProjects[cwd] = selected;
+      } else {
+        config.dworkProjects[cwd] = [];
+      }
+    }
+  }
+}
 async function requireInit() {
   const config = await loadConfig();
   if (!config.initialized) {
@@ -693,7 +832,7 @@ async function runInstall(targetName) {
 }
 
 // src/commands/status.ts
-import { DBrainClient as DBrainClient4 } from "@dtoolkit/sdk";
+import { DBrainClient as DBrainClient4, DWorkClient as DWorkClient3 } from "@dtoolkit/sdk";
 import { Command as Command5 } from "commander";
 import pc4 from "picocolors";
 function createStatusCommand() {
@@ -706,99 +845,162 @@ function createStatusCommand() {
     }
   });
 }
+function section(title) {
+  console.log(`  ${pc4.bold(pc4.blue(title))}`);
+}
+function row(label, value) {
+  console.log(`    ${pc4.dim(label.padEnd(12))} ${value}`);
+}
+function sep() {
+  console.log();
+}
 async function runStatus() {
   const config = await loadConfig();
   const stats = await loadStats();
-  console.log(pc4.bold("dcontext status"));
-  console.log();
-  console.log(pc4.bold("dbrain"));
   const { url, token } = config.dbrain;
+  console.log();
+  console.log(`  ${pc4.bold("dcontext")} ${pc4.dim("status")}`);
+  console.log(`  ${pc4.dim("\u2500".repeat(40))}`);
+  sep();
+  section("Services");
+  sep();
   if (url) {
     try {
       const client = new DBrainClient4(url, token);
       const health = await client.health();
-      console.log(`  ${pc4.green(url)} \u2014 ${health.entities} entities, ${health.facts} facts`);
       const brainType = health.brainType ?? "personal";
-      console.log(`  Type: ${brainType === "shared" ? pc4.green(brainType) : pc4.blue(brainType)}`);
+      row(
+        "dbrain",
+        `${pc4.green("\u25CF")} ${url} ${pc4.dim("\u2014")} ${health.entities} entities, ${health.facts} facts ${pc4.dim(`(${brainType})`)}`
+      );
       if (health.connectedBrains && health.connectedBrains > 0) {
-        console.log(`  Connected: ${pc4.green(String(health.connectedBrains))} brain(s)`);
         try {
           const conns = await client.listConnections();
           for (const c of conns) {
-            const s = c.online ? pc4.green("online") : pc4.red("offline");
-            console.log(`    ${pc4.bold(c.name)} ${s}`);
+            const s = c.online ? pc4.green("\u25CF") : pc4.red("\u25CF");
+            row("", `${s} ${c.name}`);
           }
         } catch {
         }
       }
     } catch {
-      console.log(`  ${pc4.red(url)} (unreachable)`);
+      row("dbrain", `${pc4.red("\u25CF")} ${url} ${pc4.dim("(unreachable)")}`);
     }
   } else {
-    console.log(`  ${pc4.red("not configured")} \u2014 run ${pc4.cyan("dcontext init")}`);
+    row("dbrain", `${pc4.red("\u25CF")} ${pc4.dim("not configured")}`);
   }
-  console.log();
-  console.log(pc4.bold("Targets"));
+  if (config.dwork?.url) {
+    try {
+      const dworkClient = new DWorkClient3(config.dwork.url, config.dwork.token);
+      const overview = await dworkClient.overview();
+      row(
+        "dwork",
+        `${pc4.green("\u25CF")} ${config.dwork.url} ${pc4.dim("\u2014")} ${overview.totalProjects} projects, ${overview.totalTasks} tasks`
+      );
+    } catch {
+      row("dwork", `${pc4.red("\u25CF")} ${config.dwork.url} ${pc4.dim("(unreachable)")}`);
+    }
+  } else {
+    row("dwork", `${pc4.dim("\u2014")} ${pc4.dim("not configured")}`);
+  }
+  sep();
+  section("Targets");
+  sep();
   for (const name of ["claude", "gemini", "opencode"]) {
     const target = resolveTarget(name);
     if (!target) continue;
     const installed = await target.isInstalled();
     const cfg = config.targets[name];
     if (installed) {
-      const since = cfg?.installedAt ? ` (since ${cfg.installedAt.split("T")[0]})` : "";
-      console.log(`  ${name}: ${pc4.green("installed")}${pc4.dim(since)}`);
+      const since = cfg?.installedAt ? pc4.dim(` since ${cfg.installedAt.split("T")[0]}`) : "";
+      row(name, `${pc4.green("\u25CF")} installed${since}`);
     } else {
-      console.log(`  ${name}: ${pc4.dim("not installed")}`);
+      row(name, `${pc4.dim("\u25CB")} ${pc4.dim("not installed")}`);
     }
   }
-  console.log();
+  sep();
   const cwd = process.cwd();
   const entity = config.projects[cwd];
-  console.log(pc4.bold("Context") + pc4.dim(` (${cwd})`));
+  const dworkSlugs = config.dworkProjects?.[cwd];
+  section("Context");
+  row("cwd", pc4.dim(cwd));
+  sep();
   if (entity) {
-    console.log(`  Entity: ${pc4.blue(entity)}`);
+    row("entity", pc4.blue(entity));
     if (url) {
       try {
         const client = new DBrainClient4(url, token);
         const results = await client.search(`"${entity}"`, { limit: 5 });
         if (results.length > 0) {
-          console.log(`  Facts:  ${pc4.green(String(results.length))}+ matching`);
+          row("facts", `${pc4.green(String(results.length))}+ matching`);
           for (const r of results.slice(0, 3)) {
-            console.log(`    ${pc4.dim("\xB7")} ${r.fact.fact.slice(0, 80)}`);
+            console.log(`    ${pc4.dim("             \xB7 " + r.fact.fact.slice(0, 72))}`);
           }
           if (results.length > 3) {
-            console.log(`    ${pc4.dim(`... and ${results.length - 3} more`)}`);
+            console.log(`    ${pc4.dim(`             \u2026 and ${results.length - 3} more`)}`);
           }
         } else {
-          console.log(`  Facts:  ${pc4.dim("none found")}`);
+          row("facts", pc4.dim("none"));
         }
       } catch {
-        console.log(`  Facts:  ${pc4.dim("could not query")}`);
+        row("facts", pc4.dim("could not query"));
       }
     }
   } else {
-    console.log(
-      `  ${pc4.dim("no entity mapped")} \u2014 run ${pc4.cyan("dcontext init")} in this directory`
-    );
+    row("entity", `${pc4.dim("not mapped")} \u2014 run ${pc4.cyan("dcontext init")}`);
   }
-  console.log();
+  if (dworkSlugs && dworkSlugs.length > 0) {
+    row("dwork", dworkSlugs.map((s) => pc4.blue(s)).join(pc4.dim(", ")));
+    if (config.dwork?.url) {
+      try {
+        const dworkClient = new DWorkClient3(config.dwork.url, config.dwork.token);
+        let totalTasks = 0;
+        let totalDocs = 0;
+        for (const slug of dworkSlugs) {
+          const tasks = await dworkClient.listTasks(slug).catch(() => []);
+          const docs = await dworkClient.listDocs(slug).catch(() => []);
+          totalTasks += tasks.length;
+          totalDocs += docs.length;
+        }
+        row("tasks", String(totalTasks));
+        row("docs", String(totalDocs));
+      } catch {
+      }
+    }
+  }
+  sep();
   const projects = Object.entries(config.projects).filter(([path]) => path !== cwd);
   if (projects.length > 0) {
-    console.log(pc4.bold("Other projects"));
+    section("Other projects");
+    sep();
     for (const [path, ent] of projects) {
-      console.log(`  ${pc4.dim(path)} \u2192 ${pc4.blue(ent)}`);
+      const short = path.replace(/^\/Users\/[^/]+/, "~");
+      row(ent, pc4.dim(short));
     }
-    console.log();
+    sep();
   }
-  console.log(pc4.bold("Stats"));
-  console.log(
-    `  Briefings:    ${stats.briefings.total}${stats.briefings.lastAt ? pc4.dim(` (last: ${stats.briefings.lastAt.split("T")[0]})`) : ""}`
-  );
-  console.log(
-    `  Extractions:  ${stats.extractions.total}${stats.extractions.messagesSaved ? ` (${stats.extractions.messagesSaved} messages)` : ""}${stats.extractions.lastAt ? pc4.dim(` (last: ${stats.extractions.lastAt.split("T")[0]})`) : ""}`
-  );
+  section("Stats");
+  sep();
+  const lastBriefing = stats.briefings.lastAt ? pc4.dim(` last ${stats.briefings.lastAt.split("T")[0]}`) : "";
+  row("briefings", `${stats.briefings.total}${lastBriefing}`);
+  const msgCount = stats.extractions.messagesSaved ? ` ${pc4.dim(`(${stats.extractions.messagesSaved} msgs)`)}` : "";
+  const lastExtraction = stats.extractions.lastAt ? pc4.dim(` last ${stats.extractions.lastAt.split("T")[0]}`) : "";
+  row("extractions", `${stats.extractions.total}${msgCount}${lastExtraction}`);
+  if (config.dwork?.url) {
+    try {
+      const dworkClient = new DWorkClient3(config.dwork.url, config.dwork.token);
+      const overview = await dworkClient.overview();
+      const activeTasks = overview.totalTasks;
+      row(
+        "tasks",
+        `${activeTasks} ${pc4.dim(`across ${overview.totalProjects} projects, ${overview.totalDocs} docs`)}`
+      );
+    } catch {
+    }
+  }
+  sep();
+  console.log(`  ${pc4.dim(`Config: ${getDataDir()}/config.json`)}`);
   console.log();
-  console.log(pc4.dim(`Config: ${getDataDir()}/config.json`));
 }
 
 // src/commands/uninstall.ts
@@ -849,7 +1051,7 @@ var description = `${pc6.green(banner)}
 
 ${pc6.green("dbrain hooks for AI coding CLIs")}
 ${pc6.dim("Part of the dtoolkit suite")}`;
-program.name("dcontext").description(description).version("0.1.5");
+program.name("dcontext").description(description).version("0.2.0");
 var guarded = (cmd) => {
   cmd.hook("preAction", async () => {
     await requireInit();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dtoolkit/dcontext",
-  "version": "0.1.5",
+  "version": "0.2.0",
   "description": "dbrain hooks for AI coding CLIs",
   "type": "module",
   "license": "MIT",
@@ -40,12 +40,12 @@
     "@clack/prompts": "^1.3.0",
     "commander": "^13.0.0",
     "picocolors": "^1.1.0",
-    "@dtoolkit/adapter-claude": "1.3.2",
-    "@dtoolkit/adapter-gemini": "1.3.2",
     "@dtoolkit/core": "0.5.0",
     "@dtoolkit/adapter-codex": "1.3.2",
-    "@dtoolkit/sdk": "0.5.0",
-    "@dtoolkit/adapter-opencode": "1.3.2"
+    "@dtoolkit/sdk": "0.5.1",
+    "@dtoolkit/adapter-claude": "1.3.2",
+    "@dtoolkit/adapter-opencode": "1.3.2",
+    "@dtoolkit/adapter-gemini": "1.3.2"
   },
   "devDependencies": {
     "@types/node": "^22.0.0",