@devness/useai 0.3.3 → 0.3.5

package/dist/index.js

@@ -11,7 +11,7 @@ var __export = (target, all) => {
 
 // src/tools.ts
 import { execSync } from "child_process";
-import { existsSync as existsSync3, readFileSync as readFileSync2, writeFileSync as writeFileSync2, mkdirSync as mkdirSync2 } from "fs";
+import { existsSync as existsSync3, readFileSync as readFileSync2, writeFileSync as writeFileSync2, mkdirSync as mkdirSync2, unlinkSync } from "fs";
 import { dirname, join as join3 } from "path";
 import { homedir as homedir2 } from "os";
 import { parse as parseToml, stringify as stringifyToml } from "smol-toml";
@@ -200,6 +200,54 @@ function removeYaml(configPath) {
     writeYamlFile(configPath, config);
   }
 }
+function hasInstructionsBlock(filePath) {
+  if (!existsSync3(filePath)) return false;
+  try {
+    return readFileSync2(filePath, "utf-8").includes(INSTRUCTIONS_START);
+  } catch {
+    return false;
+  }
+}
+function injectInstructions(config) {
+  mkdirSync2(dirname(config.path), { recursive: true });
+  if (config.method === "create") {
+    writeFileSync2(config.path, USEAI_INSTRUCTIONS + "\n");
+    return;
+  }
+  if (hasInstructionsBlock(config.path)) return;
+  let existing = "";
+  if (existsSync3(config.path)) {
+    existing = readFileSync2(config.path, "utf-8");
+  }
+  const separator = existing && !existing.endsWith("\n") ? "\n\n" : existing ? "\n" : "";
+  writeFileSync2(config.path, existing + separator + USEAI_INSTRUCTIONS_BLOCK + "\n");
+}
+function removeInstructions(config) {
+  if (config.method === "create") {
+    if (existsSync3(config.path)) {
+      try {
+        unlinkSync(config.path);
+      } catch {
+      }
+    }
+    return;
+  }
+  if (!existsSync3(config.path)) return;
+  try {
+    const content = readFileSync2(config.path, "utf-8");
+    const escaped = (s) => s.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+    const regex = new RegExp(
+      `\\n?${escaped(INSTRUCTIONS_START)}[\\s\\S]*?${escaped(INSTRUCTIONS_END)}\\n?`
+    );
+    const cleaned = content.replace(regex, "").trim();
+    if (cleaned) {
+      writeFileSync2(config.path, cleaned + "\n");
+    } else {
+      unlinkSync(config.path);
+    }
+  } catch {
+  }
+}
 function createTool(def) {
   const handler = formatHandlers[def.configFormat];
   return {
@@ -209,8 +257,14 @@ function createTool(def) {
     getConfigPath: () => def.configPath,
     detect: def.detect,
     isConfigured: () => handler.isConfigured(def.configPath),
-    install: () => handler.install(def.configPath),
-    remove: () => handler.remove(def.configPath)
+    install: () => {
+      handler.install(def.configPath);
+      if (def.instructions) injectInstructions(def.instructions);
+    },
+    remove: () => {
+      handler.remove(def.configPath);
+      if (def.instructions) removeInstructions(def.instructions);
+    }
   };
 }
 function matchesTool(tool, query) {
@@ -234,7 +288,7 @@ function resolveTools(names) {
   }
   return { matched, unmatched };
 }
-var MCP_ENTRY, home, formatHandlers, appSupport, AI_TOOLS;
+var MCP_ENTRY, home, INSTRUCTIONS_START, INSTRUCTIONS_END, USEAI_INSTRUCTIONS, USEAI_INSTRUCTIONS_BLOCK, formatHandlers, appSupport, AI_TOOLS;
 var init_tools = __esm({
   "src/tools.ts"() {
     "use strict";
@@ -243,6 +297,17 @@ var init_tools = __esm({
       args: ["-y", "@devness/useai"]
     };
     home = homedir2();
+    INSTRUCTIONS_START = "<!-- useai:start -->";
+    INSTRUCTIONS_END = "<!-- useai:end -->";
+    USEAI_INSTRUCTIONS = [
+      "## UseAI Session Tracking",
+      "- At the START of every conversation, call `useai_session_start` with the appropriate task_type.",
+      "- For LONG sessions (15+ min), call `useai_heartbeat` periodically.",
+      "- At the END of every conversation, call `useai_session_end` with languages used, files_touched_count, and milestones (generic descriptions only \u2014 no project names, file paths, or company names)."
+    ].join("\n");
+    USEAI_INSTRUCTIONS_BLOCK = `${INSTRUCTIONS_START}
+${USEAI_INSTRUCTIONS}
+${INSTRUCTIONS_END}`;
     formatHandlers = {
       standard: { isConfigured: isConfiguredStandard, install: installStandard, remove: removeStandard },
       vscode: { isConfigured: isConfiguredVscode, install: installVscode, remove: removeVscode },
@@ -257,7 +322,8 @@ var init_tools = __esm({
         name: "Claude Code",
         configFormat: "standard",
         configPath: join3(home, ".claude.json"),
-        detect: () => hasBinary("claude") || existsSync3(join3(home, ".claude.json"))
+        detect: () => hasBinary("claude") || existsSync3(join3(home, ".claude.json")),
+        instructions: { method: "append", path: join3(home, ".claude", "CLAUDE.md") }
       }),
       createTool({
         id: "claude-desktop",
@@ -278,28 +344,32 @@ var init_tools = __esm({
         name: "Windsurf",
         configFormat: "standard",
         configPath: join3(home, ".codeium", "windsurf", "mcp_config.json"),
-        detect: () => existsSync3(join3(home, ".codeium", "windsurf"))
+        detect: () => existsSync3(join3(home, ".codeium", "windsurf")),
+        instructions: { method: "append", path: join3(home, ".codeium", "windsurf", "memories", "global_rules.md") }
       }),
       createTool({
         id: "vscode",
         name: "VS Code",
         configFormat: "vscode",
         configPath: join3(appSupport, "Code", "User", "mcp.json"),
-        detect: () => existsSync3(join3(appSupport, "Code"))
+        detect: () => existsSync3(join3(appSupport, "Code")),
+        instructions: { method: "create", path: join3(appSupport, "Code", "User", "prompts", "useai.instructions.md") }
       }),
       createTool({
         id: "vscode-insiders",
         name: "VS Code Insiders",
         configFormat: "vscode",
         configPath: join3(appSupport, "Code - Insiders", "User", "mcp.json"),
-        detect: () => existsSync3(join3(appSupport, "Code - Insiders"))
+        detect: () => existsSync3(join3(appSupport, "Code - Insiders")),
+        instructions: { method: "create", path: join3(appSupport, "Code - Insiders", "User", "prompts", "useai.instructions.md") }
       }),
       createTool({
         id: "gemini-cli",
         name: "Gemini CLI",
         configFormat: "standard",
         configPath: join3(home, ".gemini", "settings.json"),
-        detect: () => hasBinary("gemini")
+        detect: () => hasBinary("gemini"),
+        instructions: { method: "append", path: join3(home, ".gemini", "GEMINI.md") }
       }),
       createTool({
         id: "zed",
@@ -323,7 +393,8 @@ var init_tools = __esm({
         ),
         detect: () => existsSync3(
           join3(appSupport, "Code", "User", "globalStorage", "saoudrizwan.claude-dev")
-        )
+        ),
+        instructions: { method: "create", path: join3(home, "Documents", "Cline", "Rules", "useai.md") }
       }),
       createTool({
         id: "roo-code",
@@ -340,7 +411,8 @@ var init_tools = __esm({
         ),
         detect: () => existsSync3(
           join3(appSupport, "Code", "User", "globalStorage", "rooveterinaryinc.roo-cline")
-        )
+        ),
+        instructions: { method: "create", path: join3(home, ".roo", "rules", "useai.md") }
       }),
       createTool({
         id: "amazon-q-cli",
@@ -361,21 +433,24 @@ var init_tools = __esm({
         name: "Codex",
         configFormat: "toml",
         configPath: join3(home, ".codex", "config.toml"),
-        detect: () => hasBinary("codex") || existsSync3(join3(home, ".codex")) || existsSync3("/Applications/Codex.app")
+        detect: () => hasBinary("codex") || existsSync3(join3(home, ".codex")) || existsSync3("/Applications/Codex.app"),
+        instructions: { method: "append", path: join3(home, ".codex", "AGENTS.md") }
       }),
       createTool({
         id: "goose",
         name: "Goose",
         configFormat: "yaml",
         configPath: join3(home, ".config", "goose", "config.yaml"),
-        detect: () => existsSync3(join3(home, ".config", "goose"))
+        detect: () => existsSync3(join3(home, ".config", "goose")),
+        instructions: { method: "append", path: join3(home, ".config", "goose", ".goosehints") }
       }),
       createTool({
         id: "opencode",
         name: "OpenCode",
         configFormat: "standard",
         configPath: join3(home, ".config", "opencode", "opencode.json"),
-        detect: () => hasBinary("opencode") || existsSync3(join3(home, ".config", "opencode"))
+        detect: () => hasBinary("opencode") || existsSync3(join3(home, ".config", "opencode")),
+        instructions: { method: "append", path: join3(home, ".config", "opencode", "AGENTS.md") }
       }),
       createTool({
         id: "junie",
@@ -438,17 +513,16 @@ function showStatus(tools) {
 }
 async function installFlow(tools, autoYes, explicit) {
   if (explicit) {
-    const toInstall2 = tools.filter((t) => !t.isConfigured());
-    const alreadyDone = tools.filter((t) => t.isConfigured());
-    for (const tool of alreadyDone) {
-      console.log(dim(`${tool.name} is already configured.`));
-    }
-    if (toInstall2.length === 0) return;
     console.log();
-    for (const tool of toInstall2) {
+    for (const tool of tools) {
       try {
+        const wasConfigured = tool.isConfigured();
         tool.install();
-        console.log(ok(`\u2713 ${tool.name.padEnd(18)} \u2192 ${chalk.dim(shortenPath(tool.getConfigPath()))}`));
+        if (wasConfigured) {
+          console.log(ok(`\u2713 ${tool.name.padEnd(18)}   ${chalk.dim("(updated)")}`));
+        } else {
+          console.log(ok(`\u2713 ${tool.name.padEnd(18)} \u2192 ${chalk.dim(shortenPath(tool.getConfigPath()))}`));
+        }
       } catch (e) {
         console.log(err(`\u2717 ${tool.name.padEnd(18)} \u2014 ${e.message}`));
       }
@@ -481,14 +555,20 @@ async function installFlow(tools, autoYes, explicit) {
   if (autoYes) {
     toInstall = unconfigured;
   } else {
-    const selected = await checkbox({
-      message: "Select tools to configure:",
-      choices: unconfigured.map((t) => ({
-        name: t.name,
-        value: t.id,
-        checked: true
-      }))
-    });
+    let selected;
+    try {
+      selected = await checkbox({
+        message: "Select tools to configure:",
+        choices: unconfigured.map((t) => ({
+          name: t.name,
+          value: t.id,
+          checked: true
+        }))
+      });
+    } catch {
+      console.log("\n");
+      return;
+    }
     toInstall = unconfigured.filter((t) => selected.includes(t.id));
   }
   if (toInstall.length === 0) {
@@ -506,6 +586,12 @@ async function installFlow(tools, autoYes, explicit) {
       console.log(err(`\u2717 ${tool.name.padEnd(18)} \u2014 ${e.message}`));
     }
   }
+  for (const tool of alreadyConfigured) {
+    try {
+      tool.install();
+    } catch {
+    }
+  }
   console.log(`
   Done! UseAI MCP server configured in ${chalk.bold(String(toInstall.length))} tool${toInstall.length === 1 ? "" : "s"}.
 `);
@@ -560,14 +646,20 @@ async function removeFlow(tools, autoYes, explicit) {
   if (autoYes) {
     toRemove = configured;
   } else {
-    const selected = await checkbox({
-      message: "Select tools to remove UseAI from:",
-      choices: configured.map((t) => ({
-        name: t.name,
-        value: t.id,
-        checked: true
-      }))
-    });
+    let selected;
+    try {
+      selected = await checkbox({
+        message: "Select tools to remove UseAI from:",
+        choices: configured.map((t) => ({
+          name: t.name,
+          value: t.id,
+          checked: true
+        }))
+      });
+    } catch {
+      console.log("\n");
+      return;
+    }
     toRemove = configured.filter((t) => selected.includes(t.id));
   }
   if (toRemove.length === 0) {

package/package.json

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