omnirush 0.8.6 → 0.9.0

package/assets/extensions/omnirush/yolo-lib.ts

@@ -0,0 +1,80 @@
+// Approval mode state (guarded by default, yolo on request), shared by
+// /yolo, /mode, the guarded-mode gate, /status and spawn_agents.
+//
+// The launcher resolves yolo for the run (--yolo / --no-yolo, OMNIRUSH_YOLO,
+// then "yolo" in <state dir>/settings.json), adds the core's --approve so the
+// project folder is trusted, and hands the result over in OMNIRUSH_YOLO
+// ("1"/"0"), OMNIRUSH_YOLO_SOURCE and OMNIRUSH_YOLO_TRUSTED. /yolo on|off
+// changes process.env for the rest of the session (sub-agents inherit it)
+// and saves the setting for the next runs.
+
+import { mkdirSync, readFileSync, renameSync, writeFileSync } from "node:fs";
+import path from "node:path";
+
+export const YOLO_BANNER = "yolo mode: no prompts, project trusted";
+/** Footer status key: shows "guarded" or "yolo". */
+export const MODE_STATUS_KEY = "omnirush-mode";
+export const SETTINGS_FILE = "settings.json";
+
+export function yoloActive(env: NodeJS.ProcessEnv = process.env): boolean {
+  return (env.OMNIRUSH_YOLO || "").trim() === "1";
+}
+
+export type ApprovalMode = "guarded" | "yolo";
+
+export function currentMode(env: NodeJS.ProcessEnv = process.env): ApprovalMode {
+  return yoloActive(env) ? "yolo" : "guarded";
+}
+
+/** Switch the running session (sub-agents started from now on inherit it). */
+export function setSessionYolo(on: boolean, env: NodeJS.ProcessEnv = process.env): void {
+  env.OMNIRUSH_YOLO = on ? "1" : "0";
+  env.OMNIRUSH_YOLO_SOURCE = "session";
+  if (on) env.OMNIRUSH_YOLO_TRUSTED = "1";
+}
+
+/** The banner for this run (an explicit --no-approve keeps the project untrusted). */
+export function yoloBanner(env: NodeJS.ProcessEnv = process.env): string {
+  return (env.OMNIRUSH_YOLO_TRUSTED || "").trim() === "0"
+    ? "yolo mode: no prompts (project not trusted: --no-approve)"
+    : YOLO_BANNER;
+}
+
+function readAll(dir: string): Record<string, any> {
+  try {
+    const parsed = JSON.parse(readFileSync(path.join(dir, SETTINGS_FILE), "utf8"));
+    return parsed && typeof parsed === "object" && !Array.isArray(parsed) ? parsed : {};
+  } catch {
+    return {};
+  }
+}
+
+export function loadYoloSetting(dir: string): boolean {
+  return readAll(dir).yolo === true;
+}
+
+/** Keeps the file's other keys; atomic (temp file + rename), 0600. */
+export function saveYoloSetting(dir: string, on: boolean): void {
+  const all = readAll(dir);
+  all.yolo = on;
+  mkdirSync(dir, { recursive: true, mode: 0o700 });
+  const file = path.join(dir, SETTINGS_FILE);
+  const tmp = `${file}.${process.pid}.tmp`;
+  writeFileSync(tmp, `${JSON.stringify(all, null, 2)}\n`, { mode: 0o600 });
+  renameSync(tmp, file);
+}
+
+const SOURCE_LABEL: Record<string, string> = {
+  flag: "--yolo",
+  env: "OMNIRUSH_YOLO",
+  setting: "saved with /yolo on",
+  session: "/yolo in this session",
+};
+
+/** The /status line. */
+export function formatModeStatus(env: NodeJS.ProcessEnv = process.env): string {
+  if (!yoloActive(env)) return "mode: guarded — git writes, destructive commands and sudo ask first";
+  const label = SOURCE_LABEL[(env.OMNIRUSH_YOLO_SOURCE || "").trim()];
+  const trusted = (env.OMNIRUSH_YOLO_TRUSTED || "").trim() !== "0";
+  return `mode: yolo — no prompts, ${trusted ? "project trusted" : "project not trusted (--no-approve)"}${label ? ` (${label})` : ""}`;
+}

package/assets/extensions/omnirush/yolo.ts

@@ -0,0 +1,85 @@
+/**
+ * Yolo mode: no prompts, project trusted.
+ *
+ *   omnirush --yolo | OMNIRUSH_YOLO=1 | /yolo on (saved for the next runs)
+ *
+ * The only confirmation an omnirush session shows is the core's "Trust
+ * project folder?" prompt (the core itself has no tool-permission prompts,
+ * and the omnirush extensions show no confirmation dialogs: finished
+ * background commands and sub-agents are delivered and stopped without
+ * asking). Yolo mode answers it:
+ *   - at launch the launcher passes the core's --approve (trusted for the
+ *     run, every folder the session switches to included);
+ *   - after /yolo on in a running session, the project_trust handler below
+ *     trusts any folder the session switches to (not remembered);
+ *   - sub-agents get --approve too (agents-lib.ts), since a headless child
+ *     would otherwise ignore the project's settings.
+ * Default behaviour (yolo off) is unchanged: the handler stays undecided.
+ */
+
+import { omniDir } from "./auth";
+import {
+  MODE_STATUS_KEY,
+  currentMode,
+  formatModeStatus,
+  saveYoloSetting,
+  setSessionYolo,
+  yoloActive,
+  yoloBanner,
+} from "./yolo-lib";
+
+function showFooter(ctx: any): void {
+  try {
+    ctx?.ui?.setStatus?.(MODE_STATUS_KEY, currentMode());
+  } catch {
+    /* no UI */
+  }
+}
+
+export default function (pi: any) {
+  pi.on("project_trust", () => (yoloActive() ? { trusted: "yes", remember: false } : { trusted: "undecided" }));
+
+  pi.on("session_start", (_event: any, ctx: any) => {
+    if (!yoloActive()) return;
+    showFooter(ctx);
+    if (ctx?.hasUI) ctx.ui?.notify?.(yoloBanner(), "warning");
+  });
+
+  pi.registerCommand("yolo", {
+    description: "Yolo mode: no prompts, project trusted — /yolo [on|off|status]",
+    getArgumentCompletions: (prefix: string) => {
+      const typed = prefix.trim();
+      const items = ["on", "off", "status"]
+        .filter((value) => value.startsWith(typed) && value !== typed)
+        .map((value) => ({ value, label: value }));
+      return items.length ? items : null;
+    },
+    handler: async (args: string, ctx: any) => {
+      const sub = String(args ?? "").trim().toLowerCase();
+      const notify = (text: string, level: "info" | "warning" | "error" = "info") => ctx?.ui?.notify?.(text, level);
+      if (sub === "" || sub === "status") {
+        notify(`${formatModeStatus()}\n/yolo on|off saves it for the next runs; --yolo or OMNIRUSH_YOLO=1 for one run.`);
+        return;
+      }
+      if (sub !== "on" && sub !== "off") {
+        notify("Usage: /yolo [on|off|status]", "error");
+        return;
+      }
+      const on = sub === "on";
+      try {
+        saveYoloSetting(omniDir(), on);
+      } catch (error: any) {
+        notify(`/yolo: could not save the setting (${error?.message ?? error})`, "error");
+        return;
+      }
+      setSessionYolo(on);
+      showFooter(ctx);
+      notify(
+        on
+          ? `${yoloBanner()} (saved). Folders this session switches to and sub-agents are trusted now; this folder's project settings load from the next start if they were not trusted.`
+          : "yolo mode off (saved): guarded mode — git writes, destructive commands and sudo ask first; the project trust prompt is back from the next start.",
+        on ? "warning" : "info",
+      );
+    },
+  });
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "omnirush",
-  "version": "0.8.6",
+  "version": "0.9.0",
   "description": "Omnirush \u2014 free daily tokens for the most powerful coding model on earth.",
   "license": "Apache-2.0",
   "type": "module",
@@ -15,7 +15,7 @@
     "test:bun": "if [ -x .runtime/bin/bun ]; then .runtime/bin/bun test --timeout 300000 test/; else bun test --timeout 300000 test/; fi",
     "build:mcp": "node scripts/build-mcp-bundle.js",
     "build:capture-vendor": "node scripts/build-capture-vendor.js",
-    "postinstall": "node scripts/patch-pi-branding.js && node scripts/postinstall.js"
+    "postinstall": "node scripts/brand-engine.js && node scripts/postinstall.js"
   },
   "dependencies": {
     "@earendil-works/pi-coding-agent": "0.86.1"
@@ -25,6 +25,7 @@
     "url": "git+https://github.com/omnirush-ai/omnirush-cli.git"
   },
   "keywords": [
+    "omnirush",
     "ai",
     "coding-agent",
     "cli"
@@ -40,5 +41,8 @@
     "esbuild": "^0.28.2",
     "minimatch": "^10.2.6",
     "zod": "^4.6.5"
+  },
+  "optionalDependencies": {
+    "@picovoice/pvrecorder-node": "1.2.9"
   }
-}
+}