@devness/useai-cli 0.4.12 → 0.4.14

package/dist/index.js

@@ -30,7 +30,7 @@ var SYSTEMD_SERVICE_PATH = join(homedir(), ".config", "systemd", "user", "useai-
 var WINDOWS_STARTUP_SCRIPT_PATH = join(process.env["APPDATA"] ?? join(homedir(), "AppData", "Roaming"), "Microsoft", "Windows", "Start Menu", "Programs", "Startup", "useai-daemon.vbs");
 
 // ../shared/dist/constants/version.js
-var VERSION = "0.4.12";
+var VERSION = "0.4.14";
 
 // ../shared/dist/constants/defaults.js
 var DEFAULT_CONFIG = {
@@ -4881,6 +4881,7 @@ import { existsSync as existsSync7, readFileSync as readFileSync3, writeFileSync
 import { join as join3 } from "path";
 import { homedir as homedir3 } from "os";
 var STOP_GUARD_PATH = join3(USEAI_HOOKS_DIR, "stop-guard.js");
+var PROMPT_GUARD_PATH = join3(USEAI_HOOKS_DIR, "prompt-guard.js");
 var CLAUDE_SETTINGS_PATH = join3(homedir3(), ".claude", "settings.json");
 var STOP_GUARD_SCRIPT = `#!/usr/bin/env node
 'use strict';
@@ -4898,17 +4899,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); }
+});
+`;
+var 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); }
 });
 `;
@@ -4928,15 +4947,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"];
@@ -4969,6 +5005,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"];
@@ -4997,6 +5041,11 @@ function removeClaudeCodeHooks() {
       unlinkSync3(STOP_GUARD_PATH);
   } catch {
   }
+  try {
+    if (existsSync7(PROMPT_GUARD_PATH))
+      unlinkSync3(PROMPT_GUARD_PATH);
+  } catch {
+  }
 }
 
 // src/services/tools.ts
@@ -5589,7 +5638,7 @@ async function daemonInstallFlow(tools, explicit) {
   try {
     const hooksInstalled = installClaudeCodeHooks();
     if (hooksInstalled) {
-      console.log(success("\u2713 Claude Code hooks installed (Stop + SessionEnd)"));
+      console.log(success("\u2713 Claude Code hooks installed (UserPromptSubmit + Stop + SessionEnd)"));
     }
   } catch {
     console.log(chalk6.yellow("  \u26A0 Could not install Claude Code hooks"));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@devness/useai-cli",
-  "version": "0.4.12",
+  "version": "0.4.14",
   "description": "CLI tool for useai.dev — stats, sync, publish your AI development workflow",
   "author": "nabeelkausari",
   "license": "MIT",