oh-my-adhd 0.2.6 → 0.2.8

package/bin/oh-my-adhd.mjs

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
 import { spawn } from "child_process";
-import { existsSync, mkdirSync, readFileSync, readdirSync, writeFileSync, copyFileSync } from "fs";
+import { existsSync, mkdirSync, readFileSync, readdirSync, writeFileSync, copyFileSync, renameSync } from "fs";
 import { join, dirname } from "path";
 import { homedir } from "os";
 import { fileURLToPath } from "url";
@@ -145,7 +145,9 @@ switch (cmd) {
       });
     }
 
-    writeFileSync(settingsPath, JSON.stringify(settings, null, 2) + "\n", "utf8");
+    const settingsTmp = `${settingsPath}.tmp.${randomUUID()}`;
+    writeFileSync(settingsTmp, JSON.stringify(settings, null, 2) + "\n", { encoding: "utf8", mode: 0o600 });
+    renameSync(settingsTmp, settingsPath);
     console.log(`✓ Hooks registered in ${settingsPath}`);
 
     console.log("\n━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━");

package/dist/mcp/lib/brain.js

@@ -202,8 +202,8 @@ export async function saveCapture(content, threadId) {
         const idx = manifest.findIndex((m) => m.id === tid);
         // Compute signal cache fields from new content
         const stripped = stripGitSuffix(content).trim();
-        const is_open = OPEN_SIGNAL.test(stripped);
-        const is_done = DONE_SIGNAL.test(stripped) && !is_open;
+        const is_open = OPEN_SIGNAL.test(stripped) && !DONE_SIGNAL.test(stripped);
+        const is_done = DONE_SIGNAL.test(stripped);
         const last_action = stripped.replace(/\n+/g, " ").slice(0, 160);
         const next_action = extractFieldBrain(stripped, "다음할것").slice(0, 120);
         const blocker = extractFieldBrain(stripped, "막힌것").slice(0, 120);
@@ -217,6 +217,8 @@ export async function saveCapture(content, threadId) {
         else
             manifest.push(meta);
         await writeManifest(manifest.sort((a, b) => new Date(b.updatedAt).getTime() - new Date(a.updatedAt).getTime()));
+        // Session-scoped dump marker for stop-hook
+        await fs.writeFile(path.join(BRAIN_DIR, ".last-dump"), String(Date.now()), "utf-8").catch(() => { });
     });
     return { capture, threadId: tid, title, skipped };
 }
@@ -240,7 +242,7 @@ export async function getThreads() {
         const scanCaptures = content.split(/\n---\n/).slice(1).filter((p) => p.trim());
         const lastCapture = scanCaptures.at(-1) ?? "";
         const fullText = lastCapture.replace(/^(?:_[^_\n]+_|\*\*[^*\n]+\*\*)\s*/m, "").trim();
-        const is_open = OPEN_SIGNAL.test(fullText);
+        const is_open = OPEN_SIGNAL.test(fullText) && !DONE_SIGNAL.test(fullText);
         return {
             id: tid,
             title,
@@ -251,7 +253,7 @@ export async function getThreads() {
             next_action: extractFieldBrain(fullText, "다음할것").slice(0, 120),
             blocker: extractFieldBrain(fullText, "막힌것").slice(0, 120),
             capture_count: scanCaptures.length,
-            is_done: DONE_SIGNAL.test(fullText) && !OPEN_SIGNAL.test(fullText),
+            is_done: DONE_SIGNAL.test(fullText),
         };
     }));
     results

package/dist/mcp/mcp/tools/wiki-dump.js

@@ -63,8 +63,15 @@ export function registerWikiDump(server) {
                 }
                 catch { /* dead-end check is best-effort, never crash */ }
             }
+            // Dopamine streak: show today's save count when ≥2
+            if (!result.skipped && allThreads.length > 0) {
+                const today = new Date().toISOString().slice(0, 10);
+                const todayCount = allThreads.filter(t => t.updatedAt?.startsWith(today)).length;
+                if (todayCount >= 2)
+                    respLines[0] = `저장됨 ✓ (오늘 ${todayCount}번째 🔥)`;
+            }
             // Only nag if user has NEVER successfully used structured schema — avoids repeated noise
-            const hasEverStructured = allThreads.some(t => t.next_action || t.blocker);
+            const hasEverStructured = allThreads.some(t => t.id === result.threadId && (t.next_action || t.blocker));
             if (!isStructured && !result.skipped && !hasEverStructured) {
                 respLines.push("");
                 respLines.push("💡 다음번엔 이 형식으로 쓰면 다음 세션에서 더 잘 복원돼:");

package/dist/mcp/mcp/tools/wiki-recall.js

@@ -38,10 +38,10 @@ export function registerWikiRecall(server) {
                     // Use stored next_action/blocker from manifest if available
                     next_action = t.next_action ?? "";
                     blocker = t.blocker ?? "";
-                    const is_done = t.is_done !== undefined ? t.is_done : (DONE_SIGNAL.test(last_action) && !is_open);
+                    const is_done = t.is_done !== undefined ? t.is_done : DONE_SIGNAL.test(last_action);
                     const gapHours = isNaN(new Date(t.updatedAt).getTime())
                         ? null
-                        : Math.round((Date.now() - new Date(t.updatedAt).getTime()) / 3600000);
+                        : Math.max(0, Math.round((Date.now() - new Date(t.updatedAt).getTime()) / 3600000));
                     return {
                         threadId: t.id,
                         title: t.title,
@@ -63,15 +63,15 @@ export function registerWikiRecall(server) {
                 const captures = content.split(/\n---\n/).slice(1).filter((p) => p.trim());
                 const lastCapture = captures.at(-1) ?? "";
                 const fullText = lastCapture.replace(/^(?:_[^_\n]+_|\*\*[^*\n]+\*\*)\s*/m, "").trim();
-                is_open = OPEN_SIGNAL.test(fullText);
-                const isDone = DONE_SIGNAL.test(fullText) && !is_open;
+                is_open = OPEN_SIGNAL.test(fullText) && !DONE_SIGNAL.test(fullText);
+                const isDone = DONE_SIGNAL.test(fullText);
                 last_action = fullText.replace(/\n+/g, " ").slice(0, 160);
                 next_action = extractFieldBrain(fullText, "다음할것").slice(0, 120);
                 blocker = extractFieldBrain(fullText, "막힌것").slice(0, 120);
                 capture_count = captures.length;
                 const gapHours = isNaN(new Date(t.updatedAt).getTime())
                     ? null
-                    : Math.round((Date.now() - new Date(t.updatedAt).getTime()) / 3600000);
+                    : Math.max(0, Math.round((Date.now() - new Date(t.updatedAt).getTime()) / 3600000));
                 // Back-fill manifest cache so stop hook sees correct is_open next time
                 updateManifestEntry(t.id, {
                     is_open,

package/dist/mcp/mcp/tools/wiki-unstick.js

@@ -1,7 +1,7 @@
 import { z } from "zod";
 import { getThreads, getThread, OPEN_SIGNAL, extractFieldBrain } from "../../lib/brain.js";
 export function registerWikiUnstick(server) {
-    server.tool("wiki_unstick", "막혀서 무력감 느낄 때 호출. 이미 시도해서 안 된 것은 제외하고, 에너지 레벨에 맞는 가장 작은 한 발자국만 알려준다. low=2분, medium=5분, high=15분.", {
+    server.tool("wiki_unstick", "막혀서 무력감 느낄 때 호출. 막힌 컨텍스트와 dead-end 목록을 수집해서 반환한다. **이 툴을 호출한 후 Claude는 반드시** 반환된 컨텍스트를 보고 에너지 레벨에 맞는 구체적인 스텝 하나를 한 줄로 직접 제안해야 한다. 제안은 행동 동사로 시작하고, ⛔ 목록에 있는 것은 절대 포함하지 않는다.", {
         task: z.string().max(2000).optional().describe("막힌 태스크 직접 설명 (없으면 최근 미완료 스레드 자동 감지)"),
         energy: z.enum(["low", "medium", "high"]).optional().default("medium").describe("현재 집중력/에너지 수준 (low=2분짜리, medium=5분짜리, high=15분짜리)"),
     }, async ({ task, energy }) => {
@@ -51,10 +51,11 @@ export function registerWikiUnstick(server) {
                 }
                 // Collect dead-ends from all open threads (not just the chosen one)
                 const allOpenThreads = threads.filter(t => t.is_open && t.id !== chosenThread.id);
-                for (const ot of allOpenThreads.slice(0, 10)) {
-                    const otContent = await getThread(ot.id);
+                const otContents = await Promise.all(allOpenThreads.slice(0, 10).map(ot => getThread(ot.id)));
+                allOpenThreads.slice(0, 10).forEach((ot, i) => {
+                    const otContent = otContents[i];
                     if (!otContent)
-                        continue;
+                        return;
                     const otCaptures = otContent.split(/\n---\n/).slice(1).filter(p => p.trim());
                     const otLast = otCaptures.at(-1) ?? "";
                     const otText = otLast.replace(/^(?:_[^_\n]+_|\*\*[^*\n]+\*\*)\s*/m, "").trim();
@@ -63,7 +64,7 @@ export function registerWikiUnstick(server) {
                     if (otBlocker && !deadEnds.some(d => d.toLowerCase().replace(/\s+/g, " ").includes(otBlockerKey))) {
                         crossThreadBlockers.push(`[${ot.title?.slice(0, 20) ?? "다른 스레드"}] ${otBlocker}`);
                     }
-                }
+                });
             }
             if (!task && !targetTitle && !targetContext) {
                 return { content: [{ type: "text", text: "막힌 대상을 찾지 못했어. 둘 중 하나로 다시 호출해줘:\n" +
@@ -100,7 +101,7 @@ export function registerWikiUnstick(server) {
                 lines.push("", "### 블로커");
                 blockers.forEach((b) => lines.push(`- ${b}`));
             }
-            lines.push("", "---", "", `위 내용을 보고 **지금 당장 할 수 있는 ${taskSize}짜리 스텝 하나**만 알려줘.`, `조건: 행동 동사로 시작 / ${taskDetail} / ⛔ 목록에 있는 것 절대 제안 금지 / 한 줄`);
+            lines.push("", "---", `_제안 기준: ${taskSize}짜리 / ${taskDetail}_`);
             return { content: [{ type: "text", text: lines.join("\n") }] };
         }
         catch (e) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "oh-my-adhd",
-  "version": "0.2.6",
+  "version": "0.2.8",
   "description": "ADHD second brain — zero-friction capture, auto context restore, unstick. MCP-native Claude Code plugin.",
   "author": "Yeachan Heo",
   "repository": {

package/scripts/stop-hook.mjs

@@ -7,11 +7,18 @@ import { homedir } from "os";
 const BRAIN_DIR = process.env.OH_MY_ADHD_DIR ?? join(homedir(), ".oh-my-adhd");
 const MANIFEST = join(BRAIN_DIR, "threads", ".manifest.json");
 const SESSION_START_FILE = join(BRAIN_DIR, ".session-start");
+const LAST_DUMP_FILE = join(BRAIN_DIR, ".last-dump");
 
 try {
   const sessionStartMs = parseInt(readFileSync(SESSION_START_FILE, "utf-8").trim(), 10);
   if (isNaN(sessionStartMs)) process.exit(0); // no marker — don't block
 
+  // Session-scoped check: did wiki_dump run after this session started?
+  try {
+    const lastDumpMs = parseInt(readFileSync(LAST_DUMP_FILE, "utf-8").trim(), 10);
+    if (!isNaN(lastDumpMs) && lastDumpMs > sessionStartMs) process.exit(0);
+  } catch { /* .last-dump missing = no dump this session */ }
+
   let manifest;
   try {
     manifest = JSON.parse(readFileSync(MANIFEST, "utf-8"));
@@ -19,19 +26,18 @@ try {
     process.exit(0); // no manifest — nothing to protect
   }
 
-  // Allow stop if any dump happened after session start
-  const latestDump = manifest.reduce((max, t) => {
-    const ts = new Date(t.updatedAt).getTime();
-    return ts > max ? ts : max;
-  }, 0);
-  if (latestDump > sessionStartMs) process.exit(0);
-
   // Block only if there are open threads worth saving
   const openThreads = manifest.filter(t => t.is_open);
   if (openThreads.length > 0) {
     const top = openThreads[0];
-    const title = (top.title ?? "진행중인 작업").slice(0, 40);
-    const nextHint = top.next_action ? `\n→ 다음할것: ${top.next_action.slice(0, 60)}` : "";
+    // Sanitize: strip control chars and injection-shaped strings before embedding in model-facing text
+    const sanitize = (s, max) => String(s ?? "")
+      .replace(/[\x00-\x1F\x7F]/g, " ")
+      .replace(/[`$<>]/g, "")
+      .replace(/\bignore (all|previous)\b/gi, "[redacted]")
+      .slice(0, max);
+    const title = sanitize(top.title ?? "진행중인 작업", 40);
+    const nextHint = top.next_action ? `\n→ 다음할것: ${sanitize(top.next_action, 60)}` : "";
     process.stdout.write(JSON.stringify({
       decision: "block",
       reason: `저장 없이 끝내려고? "${title}" 스레드가 열려있어.${nextHint}\nwiki_dump로 결정/막힌것/다음할것 저장하고 끝내자.`,