@devness/useai 0.4.12 → 0.4.14

package/dist/index.js

@@ -112,7 +112,7 @@ var VERSION;
 var init_version = __esm({
   "../shared/dist/constants/version.js"() {
     "use strict";
-    VERSION = "0.4.12";
+    VERSION = "0.4.14";
   }
 });
 
@@ -857,15 +857,32 @@ function writeSettings(settings) {
 function installClaudeCodeHooks() {
   mkdirSync3(USEAI_HOOKS_DIR, { recursive: true });
   writeFileSync3(STOP_GUARD_PATH, STOP_GUARD_SCRIPT);
+  writeFileSync3(PROMPT_GUARD_PATH, PROMPT_GUARD_SCRIPT);
   try {
     chmodSync(STOP_GUARD_PATH, "755");
   } catch {
   }
+  try {
+    chmodSync(PROMPT_GUARD_PATH, "755");
+  } catch {
+  }
   const settings = readSettings();
   const hooks = settings["hooks"] ?? {};
+  const promptCmd = `node "${PROMPT_GUARD_PATH}"`;
   const stopCmd = `node "${STOP_GUARD_PATH}"`;
   const sealCmd = `curl -sf -X POST http://127.0.0.1:${DAEMON_PORT}/api/seal-active --max-time 3 2>/dev/null || true`;
   let changed = false;
+  if (!hooks["UserPromptSubmit"])
+    hooks["UserPromptSubmit"] = [];
+  const promptArr = hooks["UserPromptSubmit"];
+  const hasPrompt = promptArr.some((g) => {
+    const inner = g["hooks"];
+    return inner?.some((h) => h["command"]?.includes("prompt-guard"));
+  });
+  if (!hasPrompt) {
+    promptArr.push({ hooks: [{ type: "command", command: promptCmd, timeout: 10 }] });
+    changed = true;
+  }
   if (!hooks["Stop"])
     hooks["Stop"] = [];
   const stopArr = hooks["Stop"];
@@ -898,6 +915,14 @@ function removeClaudeCodeHooks() {
       const settings = readSettings();
       const hooks = settings["hooks"];
       if (hooks) {
+        if (hooks["UserPromptSubmit"]) {
+          hooks["UserPromptSubmit"] = hooks["UserPromptSubmit"].filter((g) => {
+            const inner = g["hooks"];
+            return !inner?.some((h) => h["command"]?.includes("prompt-guard"));
+          });
+          if (hooks["UserPromptSubmit"].length === 0)
+            delete hooks["UserPromptSubmit"];
+        }
         if (hooks["Stop"]) {
           hooks["Stop"] = hooks["Stop"].filter((g) => {
             const inner = g["hooks"];
@@ -926,13 +951,19 @@ function removeClaudeCodeHooks() {
       unlinkSync3(STOP_GUARD_PATH);
   } catch {
   }
+  try {
+    if (existsSync5(PROMPT_GUARD_PATH))
+      unlinkSync3(PROMPT_GUARD_PATH);
+  } catch {
+  }
 }
-var STOP_GUARD_PATH, CLAUDE_SETTINGS_PATH, STOP_GUARD_SCRIPT;
+var STOP_GUARD_PATH, PROMPT_GUARD_PATH, CLAUDE_SETTINGS_PATH, STOP_GUARD_SCRIPT, PROMPT_GUARD_SCRIPT;
 var init_claude_code = __esm({
   "../shared/dist/hooks/claude-code.js"() {
     "use strict";
     init_paths();
     STOP_GUARD_PATH = join3(USEAI_HOOKS_DIR, "stop-guard.js");
+    PROMPT_GUARD_PATH = join3(USEAI_HOOKS_DIR, "prompt-guard.js");
     CLAUDE_SETTINGS_PATH = join3(homedir3(), ".claude", "settings.json");
     STOP_GUARD_SCRIPT = `#!/usr/bin/env node
 'use strict';
@@ -950,17 +981,35 @@ process.stdin.on('end', () => {
     var starts = (transcript.match(/useai_session_start/g) || []).length;
     if (starts === 0) process.exit(0);
     var ends = (transcript.match(/useai_session_end/g) || []).length;
-    if (starts > ends) {
-      process.stdout.write(JSON.stringify({
-        decision: 'block',
-        reason: 'Before ending, call useai_session_end with milestones (generic descriptions of accomplishments), languages used, and files_touched_count.'
-      }));
-    } else {
-      process.stdout.write(JSON.stringify({
-        decision: 'block',
-        reason: 'You did work without UseAI tracking. Call useai_session_start first, then useai_session_end with milestones (generic descriptions of accomplishments), languages used, and files_touched_count.'
-      }));
-    }
+    if (ends >= starts) process.exit(0);
+    process.stdout.write(JSON.stringify({
+      decision: 'block',
+      reason: 'Before ending, call useai_session_end with milestones (generic descriptions of accomplishments), languages used, and files_touched_count.'
+    }));
+  } catch { process.exit(0); }
+});
+`;
+    PROMPT_GUARD_SCRIPT = `#!/usr/bin/env node
+'use strict';
+const { readFileSync } = require('node:fs');
+let input = '';
+process.stdin.setEncoding('utf-8');
+process.stdin.on('data', chunk => input += chunk);
+process.stdin.on('end', () => {
+  try {
+    const hook = JSON.parse(input);
+    if (!hook.transcript_path) process.exit(0);
+    let transcript;
+    try { transcript = readFileSync(hook.transcript_path, 'utf-8'); } catch { process.exit(0); }
+    var starts = (transcript.match(/useai_session_start/g) || []).length;
+    var ends = (transcript.match(/useai_session_end/g) || []).length;
+    if (starts > ends) process.exit(0);
+    process.stdout.write(JSON.stringify({
+      hookSpecificOutput: {
+        hookEventName: 'UserPromptSubmit',
+        additionalContext: 'UseAI: Call useai_session_start with the appropriate task_type before doing any work.'
+      }
+    }));
   } catch { process.exit(0); }
 });
 `;
@@ -1118,8 +1167,7 @@ function registerTools(server2, session2) {
         content: [
           {
             type: "text",
-            text: `useai session started \u2014 tracking ${session2.sessionTaskType} session on ${session2.clientName}.
-Session: ${session2.sessionId.slice(0, 8)} \xB7 Chain: ${record.hash.slice(0, 12)} (${session2.signingAvailable ? "signed" : "unsigned"})`
+            text: `${session2.sessionTaskType} \xB7 ${session2.sessionId.slice(0, 8)}`
           }
         ]
       };
@@ -1889,7 +1937,7 @@ async function daemonInstallFlow(tools, explicit) {
   try {
     const hooksInstalled = installClaudeCodeHooks();
     if (hooksInstalled) {
-      console.log(ok("\u2713 Claude Code hooks installed (Stop + SessionEnd)"));
+      console.log(ok("\u2713 Claude Code hooks installed (UserPromptSubmit + Stop + SessionEnd)"));
     }
   } catch {
     console.log(chalk.yellow("  \u26A0 Could not install Claude Code hooks"));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@devness/useai",
-  "version": "0.4.12",
+  "version": "0.4.14",
   "description": "Track your AI-assisted development workflow. MCP server that records usage metrics across all your AI tools.",
   "keywords": [
     "mcp",