@pimzino/sgrep 1.3.21 → 1.3.26

package/build/plugins/sgrep/hooks/sgrep_enhance.js

@@ -1,183 +1,183 @@
-#!/usr/bin/env node
-
-const fs = require("fs");
-const path = require("path");
-const { execSync } = require("child_process");
-const os = require("os");
-
-const DEBUG_ENABLED = process.env.SGREP_DEBUG === "1" || process.env.SGREP_DEBUG === "true";
-const DEBUG_LOG_FILE = process.env.SGREP_ENHANCE_LOG || path.join(os.tmpdir(), "sgrep-enhance.log");
-
-// Lightweight reminder injected with every prompt (exploits recency bias)
-const SGREP_REMINDER = `<mandatory>Use sgrep "question" for explanations or sgrep search "query" for code snippets via Bash BEFORE Read/Grep/Glob.</mandatory>`;
-
-function debugLog(message) {
-  if (!DEBUG_ENABLED) return;
-  try {
-    const dir = path.dirname(DEBUG_LOG_FILE);
-    if (!fs.existsSync(dir)) {
-      fs.mkdirSync(dir, { recursive: true });
-    }
-    const stamp = new Date().toISOString();
-    fs.appendFileSync(DEBUG_LOG_FILE, `[${stamp}] ${message}\n`);
-  } catch (e) {
-    // Ignore errors
-  }
-}
-
-function debugLogSection(title, content) {
-  debugLog(`\n========== ${title} ==========`);
-  debugLog(content);
-  debugLog(`========== END ${title} ==========\n`);
-}
-
-function readHookInput() {
-  return new Promise((resolve) => {
-    let data = "";
-    process.stdin.setEncoding("utf8");
-    process.stdin.on("data", (chunk) => {
-      data += chunk;
-    });
-    process.stdin.on("end", () => {
-      if (!data.trim()) {
-        debugLog("No input data received from stdin");
-        resolve(null);
-        return;
-      }
-      try {
-        const parsed = JSON.parse(data);
-        debugLogSection("RAW HOOK INPUT", JSON.stringify(parsed, null, 2));
-        resolve(parsed);
-      } catch (e) {
-        debugLog(`Failed to decode JSON: ${e.message}`);
-        debugLog(`Raw data: ${data}`);
-        resolve(null);
-      }
-    });
-  });
-}
-
-/**
- * Determine if enhancement should proceed.
- * Auto-confirms on all platforms (no user prompting).
- */
-function shouldEnhance(prompt) {
-  debugLog("Auto-confirming enhancement (prompting disabled on all platforms)");
-  return true;
-}
-
-/**
- * Output the hook response with optional enhanced context and mandatory reminder.
- * Always includes the sgrep reminder to exploit recency bias.
- */
-function outputResponse(enhancedContext = null) {
-  const parts = [];
-
-  if (enhancedContext) {
-    parts.push(`[sgrep enhanced context]\n${enhancedContext}`);
-  }
-
-  // Always append the reminder (exploits recency bias - appears fresh with every prompt)
-  parts.push(SGREP_REMINDER);
-
-  const response = {
-    hookSpecificOutput: {
-      hookEventName: "UserPromptSubmit",
-      additionalContext: parts.join("\n\n")
-    }
-  };
-
-  debugLogSection("HOOK RESPONSE", JSON.stringify(response, null, 2));
-  console.log(JSON.stringify(response));
-}
-
-async function main() {
-  debugLog("\n\n====================================================");
-  debugLog("SGREP ENHANCE HOOK TRIGGERED - UserPromptSubmit");
-  debugLog("====================================================");
-
-  const payload = await readHookInput();
-  if (!payload) {
-    debugLog("ERROR: No payload received - exiting");
-    process.exit(0);
-  }
-
-  // Log all payload fields
-  debugLog(`Hook Event: ${payload.hook_event_name || 'unknown'}`);
-  debugLog(`Session ID: ${payload.session_id || 'unknown'}`);
-  debugLog(`CWD: ${payload.cwd || 'unknown'}`);
-  debugLog(`Permission Mode: ${payload.permission_mode || 'unknown'}`);
-
-  const prompt = payload.prompt || "";
-  debugLogSection("ORIGINAL PROMPT", prompt);
-
-  // Skip enhancement for very short prompts or slash commands, but still inject reminder
-  if (prompt.length < 10) {
-    debugLog("SKIP enhancement: Prompt too short (< 10 chars), but injecting reminder");
-    outputResponse(null);
-    debugLog("====================================================\n");
-    process.exit(0);
-  }
-  if (prompt.startsWith("/")) {
-    debugLog("SKIP enhancement: Prompt is a slash command, but injecting reminder");
-    outputResponse(null);
-    debugLog("====================================================\n");
-    process.exit(0);
-  }
-
-  // Check if enhancement should proceed (auto-confirms on all platforms)
-  debugLog("Checking if enhancement should proceed...");
-  const shouldProceed = shouldEnhance(prompt);
-  debugLog(`Enhancement decision: ${shouldProceed ? 'proceeding' : 'skipped'}`);
-
-  if (!shouldProceed) {
-    debugLog("SKIP enhancement: Not proceeding, but injecting reminder");
-    outputResponse(null);
-    debugLog("====================================================\n");
-    process.exit(0);
-  }
-
-  let enhancedContext = null;
-
-  try {
-    // Call sgrep enhance with JSON output
-    debugLog("Calling: sgrep enhance --json \"<prompt>\"");
-
-    // Escape the prompt for shell
-    const escapedPrompt = prompt.replace(/"/g, '\\"').replace(/\n/g, ' ');
-
-    const result = execSync(`sgrep enhance --json "${escapedPrompt}"`, {
-      encoding: "utf8",
-      timeout: 60000, // 60 second timeout
-      stdio: ["pipe", "pipe", "pipe"],
-    });
-
-    debugLogSection("SGREP ENHANCE RAW RESULT", result);
-
-    // Parse the JSON result
-    const parsed = JSON.parse(result);
-    const enhanced = parsed.enhanced || null;
-
-    if (enhanced && enhanced !== prompt) {
-      debugLogSection("ENHANCED PROMPT", enhanced);
-      enhancedContext = enhanced;
-      debugLog("SUCCESS: Enhancement complete");
-    } else {
-      debugLog("SKIP: No enhanced prompt or same as original");
-    }
-  } catch (e) {
-    debugLog(`ERROR: Enhancement failed - ${e.message}`);
-    if (e.stderr) {
-      debugLog(`STDERR: ${e.stderr}`);
-    }
-    // Continue to output reminder even on error
-  }
-
-  // Always output response with reminder (and enhanced context if available)
-  outputResponse(enhancedContext);
-
-  debugLog("====================================================\n");
-  process.exit(0);
-}
-
-main();
+#!/usr/bin/env node
+
+const fs = require("fs");
+const path = require("path");
+const { execSync } = require("child_process");
+const os = require("os");
+
+const DEBUG_ENABLED = process.env.SGREP_DEBUG === "1" || process.env.SGREP_DEBUG === "true";
+const DEBUG_LOG_FILE = process.env.SGREP_ENHANCE_LOG || path.join(os.tmpdir(), "sgrep-enhance.log");
+
+// Lightweight reminder injected with every prompt (exploits recency bias)
+const SGREP_REMINDER = `<mandatory>Use sgrep "question" for explanations or sgrep search "query" for code snippets via Bash BEFORE Read/Grep/Glob.</mandatory>`;
+
+function debugLog(message) {
+  if (!DEBUG_ENABLED) return;
+  try {
+    const dir = path.dirname(DEBUG_LOG_FILE);
+    if (!fs.existsSync(dir)) {
+      fs.mkdirSync(dir, { recursive: true });
+    }
+    const stamp = new Date().toISOString();
+    fs.appendFileSync(DEBUG_LOG_FILE, `[${stamp}] ${message}\n`);
+  } catch (e) {
+    // Ignore errors
+  }
+}
+
+function debugLogSection(title, content) {
+  debugLog(`\n========== ${title} ==========`);
+  debugLog(content);
+  debugLog(`========== END ${title} ==========\n`);
+}
+
+function readHookInput() {
+  return new Promise((resolve) => {
+    let data = "";
+    process.stdin.setEncoding("utf8");
+    process.stdin.on("data", (chunk) => {
+      data += chunk;
+    });
+    process.stdin.on("end", () => {
+      if (!data.trim()) {
+        debugLog("No input data received from stdin");
+        resolve(null);
+        return;
+      }
+      try {
+        const parsed = JSON.parse(data);
+        debugLogSection("RAW HOOK INPUT", JSON.stringify(parsed, null, 2));
+        resolve(parsed);
+      } catch (e) {
+        debugLog(`Failed to decode JSON: ${e.message}`);
+        debugLog(`Raw data: ${data}`);
+        resolve(null);
+      }
+    });
+  });
+}
+
+/**
+ * Determine if enhancement should proceed.
+ * Auto-confirms on all platforms (no user prompting).
+ */
+function shouldEnhance(prompt) {
+  debugLog("Auto-confirming enhancement (prompting disabled on all platforms)");
+  return true;
+}
+
+/**
+ * Output the hook response with optional enhanced context and mandatory reminder.
+ * Always includes the sgrep reminder to exploit recency bias.
+ */
+function outputResponse(enhancedContext = null) {
+  const parts = [];
+
+  if (enhancedContext) {
+    parts.push(`[sgrep enhanced context]\n${enhancedContext}`);
+  }
+
+  // Always append the reminder (exploits recency bias - appears fresh with every prompt)
+  parts.push(SGREP_REMINDER);
+
+  const response = {
+    hookSpecificOutput: {
+      hookEventName: "UserPromptSubmit",
+      additionalContext: parts.join("\n\n")
+    }
+  };
+
+  debugLogSection("HOOK RESPONSE", JSON.stringify(response, null, 2));
+  console.log(JSON.stringify(response));
+}
+
+async function main() {
+  debugLog("\n\n====================================================");
+  debugLog("SGREP ENHANCE HOOK TRIGGERED - UserPromptSubmit");
+  debugLog("====================================================");
+
+  const payload = await readHookInput();
+  if (!payload) {
+    debugLog("ERROR: No payload received - exiting");
+    process.exit(0);
+  }
+
+  // Log all payload fields
+  debugLog(`Hook Event: ${payload.hook_event_name || 'unknown'}`);
+  debugLog(`Session ID: ${payload.session_id || 'unknown'}`);
+  debugLog(`CWD: ${payload.cwd || 'unknown'}`);
+  debugLog(`Permission Mode: ${payload.permission_mode || 'unknown'}`);
+
+  const prompt = payload.prompt || "";
+  debugLogSection("ORIGINAL PROMPT", prompt);
+
+  // Skip enhancement for very short prompts or slash commands, but still inject reminder
+  if (prompt.length < 10) {
+    debugLog("SKIP enhancement: Prompt too short (< 10 chars), but injecting reminder");
+    outputResponse(null);
+    debugLog("====================================================\n");
+    process.exit(0);
+  }
+  if (prompt.startsWith("/")) {
+    debugLog("SKIP enhancement: Prompt is a slash command, but injecting reminder");
+    outputResponse(null);
+    debugLog("====================================================\n");
+    process.exit(0);
+  }
+
+  // Check if enhancement should proceed (auto-confirms on all platforms)
+  debugLog("Checking if enhancement should proceed...");
+  const shouldProceed = shouldEnhance(prompt);
+  debugLog(`Enhancement decision: ${shouldProceed ? 'proceeding' : 'skipped'}`);
+
+  if (!shouldProceed) {
+    debugLog("SKIP enhancement: Not proceeding, but injecting reminder");
+    outputResponse(null);
+    debugLog("====================================================\n");
+    process.exit(0);
+  }
+
+  let enhancedContext = null;
+
+  try {
+    // Call sgrep enhance with JSON output
+    debugLog("Calling: sgrep enhance --json \"<prompt>\"");
+
+    // Escape the prompt for shell
+    const escapedPrompt = prompt.replace(/"/g, '\\"').replace(/\n/g, ' ');
+
+    const result = execSync(`sgrep enhance --json "${escapedPrompt}"`, {
+      encoding: "utf8",
+      timeout: 60000, // 60 second timeout
+      stdio: ["pipe", "pipe", "pipe"],
+    });
+
+    debugLogSection("SGREP ENHANCE RAW RESULT", result);
+
+    // Parse the JSON result
+    const parsed = JSON.parse(result);
+    const enhanced = parsed.enhanced || null;
+
+    if (enhanced && enhanced !== prompt) {
+      debugLogSection("ENHANCED PROMPT", enhanced);
+      enhancedContext = enhanced;
+      debugLog("SUCCESS: Enhancement complete");
+    } else {
+      debugLog("SKIP: No enhanced prompt or same as original");
+    }
+  } catch (e) {
+    debugLog(`ERROR: Enhancement failed - ${e.message}`);
+    if (e.stderr) {
+      debugLog(`STDERR: ${e.stderr}`);
+    }
+    // Continue to output reminder even on error
+  }
+
+  // Always output response with reminder (and enhanced context if available)
+  outputResponse(enhancedContext);
+
+  debugLog("====================================================\n");
+  process.exit(0);
+}
+
+main();