@shakecodeslikecray/whiterose 1.0.0 → 1.0.2

package/dist/cli/index.js

@@ -1,10 +1,11 @@
 #!/usr/bin/env node
-import { existsSync, mkdirSync, writeFileSync, readdirSync, readFileSync, statSync, rmSync, mkdtempSync, realpathSync } from 'fs';
-import { join, isAbsolute, resolve, basename, relative, dirname } from 'path';
+import { readFileSync, existsSync, mkdirSync, writeFileSync, readdirSync, statSync, rmSync, mkdtempSync, realpathSync } from 'fs';
+import { join, dirname, isAbsolute, resolve, basename, relative } from 'path';
 import chalk3 from 'chalk';
 import * as readline from 'readline';
 import { Command } from 'commander';
 import * as p3 from '@clack/prompts';
+import { fileURLToPath } from 'url';
 import { execa } from 'execa';
 import { homedir, tmpdir } from 'os';
 import fg3 from 'fast-glob';
@@ -467,13 +468,56 @@ var AiderExecutor = class {
     }
   }
 };
+var OLLAMA_TIMEOUT = 6e5;
+var DEFAULT_MODEL = "codellama";
+var OllamaExecutor = class {
+  name = "ollama";
+  model;
+  constructor(model = DEFAULT_MODEL) {
+    this.model = model;
+  }
+  async isAvailable() {
+    return isProviderAvailable("ollama");
+  }
+  async runPrompt(prompt, options) {
+    const ollamaCommand = getProviderCommand("ollama");
+    try {
+      const { stdout, stderr } = await execa(
+        ollamaCommand,
+        ["run", this.model, prompt],
+        {
+          cwd: options.cwd,
+          timeout: options.timeout || OLLAMA_TIMEOUT,
+          env: {
+            ...process.env,
+            NO_COLOR: "1"
+          },
+          reject: false
+        }
+      );
+      return {
+        output: stdout || "",
+        error: stderr || void 0
+      };
+    } catch (error) {
+      if (error.message?.includes("ENOENT")) {
+        throw new Error("Ollama not found. Install from: https://ollama.ai");
+      }
+      if (error.message?.includes("model") && error.message?.includes("not found")) {
+        throw new Error(`Ollama model '${this.model}' not found. Run: ollama pull ${this.model}`);
+      }
+      throw error;
+    }
+  }
+};
 
 // src/providers/executors/index.ts
 var executors = {
   "claude-code": () => new ClaudeCodeExecutor(),
   "codex": () => new CodexExecutor(),
   "gemini": () => new GeminiExecutor(),
-  "aider": () => new AiderExecutor()
+  "aider": () => new AiderExecutor(),
+  "ollama": () => new OllamaExecutor()
 };
 function getExecutor(name) {
   const factory = executors[name];
@@ -2658,7 +2702,10 @@ async function getChangedFiles(cwd, config, options) {
   };
   if (existsSync(cachePath)) {
     try {
-      cachedState = JSON.parse(readFileSync(cachePath, "utf-8"));
+      const parsed = JSON.parse(readFileSync(cachePath, "utf-8"));
+      if (Array.isArray(parsed.fileHashes)) {
+        cachedState = parsed;
+      }
     } catch {
     }
   }
@@ -2888,7 +2935,7 @@ function parseListItems(content) {
   return items;
 }
 function parseFeatureSection(section) {
-  const titleMatch = section.match(/###\s+([^\[\n]+)(?:\s*\[([^\]]+)\])?/);
+  const titleMatch = section.match(/###\s+([^[\n]+)(?:\s*\[([^\]]+)\])?/);
   if (!titleMatch) return null;
   const name = titleMatch[1].trim();
   const badge = titleMatch[2]?.trim().toLowerCase();
@@ -5427,7 +5474,7 @@ var Dashboard = ({ bugs, onSelectCategory }) => {
       color: categoryColors[category] || "white"
     });
   }
-  useInput((input, key) => {
+  useInput((_input, key) => {
     if (key.upArrow) {
       setSelectedIndex((i) => Math.max(0, i - 1));
     } else if (key.downArrow) {
@@ -6092,7 +6139,7 @@ async function createFixBranch(branchName, bug, cwd = process.cwd()) {
 }
 async function commitFix(bug, cwd = process.cwd()) {
   try {
-    await execa("git", ["add", bug.file], { cwd, timeout: GIT_TIMEOUT });
+    await execa("git", ["add", "--", bug.file], { cwd, timeout: GIT_TIMEOUT });
     const { stdout: diff } = await execa("git", ["diff", "--cached", "--name-only"], {
       cwd,
       timeout: GIT_TIMEOUT
@@ -6493,7 +6540,7 @@ async function startFixTUI(bugs, config, options, cwd) {
     const handleExit = () => {
       resolve6();
     };
-    const { unmount, waitUntilExit } = render(
+    const { waitUntilExit } = render(
       /* @__PURE__ */ jsx(
         App,
         {
@@ -6505,9 +6552,7 @@ async function startFixTUI(bugs, config, options, cwd) {
         }
       )
     );
-    waitUntilExit().then(() => {
-      resolve6();
-    });
+    waitUntilExit().then(resolve6).catch(() => resolve6());
   });
 }
 
@@ -6958,7 +7003,7 @@ function mapSarifLevel(level) {
       return "medium";
   }
 }
-async function refreshCommand(options) {
+async function refreshCommand(_options) {
   const cwd = process.cwd();
   const whiterosePath = join(cwd, ".whiterose");
   if (!existsSync(whiterosePath)) {
@@ -7211,6 +7256,9 @@ async function clearCommand(options) {
 }
 
 // src/cli/index.ts
+var __filename$1 = fileURLToPath(import.meta.url);
+var __dirname$1 = dirname(__filename$1);
+var pkg = JSON.parse(readFileSync(join(__dirname$1, "../../package.json"), "utf-8"));
 process.setMaxListeners(50);
 var BANNER = `
 ${chalk3.red("\u2588\u2588\u2557    \u2588\u2588\u2557\u2588\u2588\u2557  \u2588\u2588\u2557\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2557  \u2588\u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2588\u2557")}
@@ -7223,7 +7271,7 @@ ${chalk3.red(" \u255A\u2550\u2550\u255D\u255A\u2550\u2550\u255D \u255A\u2550\u25
 ${chalk3.dim(`  "I've been staring at your code for a long time."`)}
 `;
 var program = new Command();
-program.name("whiterose").description("AI-powered bug hunter that uses your existing LLM subscription").version("0.2.7").hook("preAction", () => {
+program.name("whiterose").description("AI-powered bug hunter that uses your existing LLM subscription").version(pkg.version).hook("preAction", () => {
   const args = process.argv.slice(2);
   if (!args.includes("--help") && !args.includes("-h") && args.length > 0) {
     console.log(BANNER);