clisponsor 1.0.4 → 1.0.6

package/bin/clisponsor.mjs

@@ -15,6 +15,7 @@ const DEFAULT_SERVE_BASE_URL =
 const DEFAULT_BACKEND_BASE_URL = process.env.CLISPONSOR_BACKEND_BASE_URL || "https://backend.clisponsor.com";
 const HOOK_VERSION = "1.0.0";
 const NETWORK_TIMEOUT_MS = 3000;
+const ANTIGRAVITY_EVENTS = ["PreInvocation", "UserPromptSubmit", "PostInvocation", "Stop"];
 
 function argValue(name) {
   const prefix = `${name}=`;
@@ -130,6 +131,23 @@ function chmodExecutable(file) {
 }
 
 function commandExists(command) {
+  const paths = String(process.env.PATH || "").split(path.delimiter).filter(Boolean);
+  const extensions = process.platform === "win32" ? String(process.env.PATHEXT || ".EXE;.CMD;.BAT").split(";") : [""];
+  for (const directory of paths) {
+    for (const extension of extensions) {
+      const candidate = path.join(directory, command + extension.toLowerCase());
+      try {
+        fs.accessSync(candidate, fs.constants.X_OK);
+        return true;
+      } catch {}
+      if (extension) {
+        try {
+          fs.accessSync(path.join(directory, command + extension.toUpperCase()), fs.constants.X_OK);
+          return true;
+        } catch {}
+      }
+    }
+  }
   try {
     execFileSync(command, ["--version"], { stdio: "ignore" });
     return true;
@@ -140,15 +158,15 @@ function commandExists(command) {
 
 function isClisponsorCommand(value) {
   return (
-	    typeof value === "string" &&
-	    (value.includes("clisponsor_claude_hook.mjs") ||
-	      value.includes("clisponsor_gemini_hook.mjs") ||
-	      value.includes("clisponsor_antigravity_hook.mjs") ||
-	      value.includes(`${path.sep}.clisponsor${path.sep}`) ||
-	      value.includes("/.clisponsor/") ||
-	      value.includes("\\.clisponsor\\"))
-	  );
-	}
+    typeof value === "string" &&
+    (value.includes("clisponsor_claude_hook.mjs") ||
+      value.includes("clisponsor_gemini_hook.mjs") ||
+      value.includes("clisponsor_antigravity_hook.mjs") ||
+      value.includes(`${path.sep}.clisponsor${path.sep}`) ||
+      value.includes("/.clisponsor/") ||
+      value.includes("\\.clisponsor\\"))
+  );
+}
 
 function isClisponsorHookEntry(entry) {
   if (!isPlainObject(entry)) return false;
@@ -245,6 +263,29 @@ function removeClaudeCommandHooks(settings) {
   return changed;
 }
 
+function setAntigravityCommandHooks(settings, hookPath) {
+  removeClaudeCommandHooks(settings);
+  settings.clisponsor = Object.fromEntries(
+    ANTIGRAVITY_EVENTS.map((eventName) => [
+      eventName,
+      [
+        {
+          type: "command",
+          command: `node ${JSON.stringify(hookPath)} ${eventName}`,
+          timeout: 5,
+        },
+      ],
+    ]),
+  );
+}
+
+function removeAntigravityCommandHooks(settings) {
+  const removedLegacyHooks = removeClaudeCommandHooks(settings);
+  const removedNativeHooks = Object.hasOwn(settings, "clisponsor");
+  if (removedNativeHooks) delete settings.clisponsor;
+  return removedLegacyHooks || removedNativeHooks;
+}
+
 async function registerDevice() {
   const email = positionalArg(3);
   if (!email) {
@@ -361,13 +402,15 @@ function installClaude() {
   console.log("Claude Code CLI hook installed.");
 }
 
-function agentHookSource(client) {
+function agentHookSource(client, options = {}) {
+  const outputMode = options.outputMode || "systemMessage";
   return `#!/usr/bin/env node
 import fs from "node:fs";
 import crypto from "node:crypto";
 const cfg = JSON.parse(fs.readFileSync(${JSON.stringify(CONFIG_PATH)}, "utf8"));
 const event = process.argv[2] || "BeforeAgent";
-const placements = { SessionStart: "StartSession", BeforeAgent: "StartTurn", AfterAgent: "EndTurn", StartTurn: "StartTurn" };
+const outputMode = ${JSON.stringify(outputMode)};
+const placements = { SessionStart: "StartSession", PreInvocation: "StartSession", BeforeAgent: "StartTurn", UserPromptSubmit: "StartTurn", PreToolUse: "StartTurn", AfterAgent: "EndTurn", PostInvocation: "EndTurn", Stop: "EndTurn", StartTurn: "StartTurn" };
 const serveBaseUrl = cfg.serveBaseUrl || cfg.apiBaseUrl;
 function sponsoredLine(line) {
   return "[Sponsored] " + line;
@@ -395,7 +438,13 @@ try {
   });
   if (res.ok) {
     const ad = await res.json();
-    if (ad.display_line) console.log(JSON.stringify({ systemMessage: sponsoredLine(ad.display_line) }));
+    if (outputMode === "antigravity") {
+      const payload = { decision: "allow" };
+      if (ad.display_line) payload.systemMessage = sponsoredLine(ad.display_line);
+      console.log(JSON.stringify(payload));
+    } else if (ad.display_line) {
+      console.log(JSON.stringify({ systemMessage: sponsoredLine(ad.display_line) }));
+    }
   }
 } catch {
   process.exit(0);
@@ -433,13 +482,11 @@ function installAntigravity() {
   const antigravityDir = path.join(CONFIG_DIR, "antigravity");
   fs.mkdirSync(antigravityDir, { recursive: true });
   const hookPath = path.join(antigravityDir, "clisponsor_antigravity_hook.mjs");
-  fs.writeFileSync(hookPath, agentHookSource("Antigravity"), { mode: 0o755 });
+  fs.writeFileSync(hookPath, agentHookSource("Antigravity", { outputMode: "antigravity" }), { mode: 0o755 });
 
   const hooksPath = path.join(HOME, ".gemini", "config", "hooks.json");
   const hooksConfig = readEditableJson(hooksPath, {});
-  addGeminiCommandHook(hooksConfig, "SessionStart", "startup", `node ${JSON.stringify(hookPath)} SessionStart`);
-  addGeminiCommandHook(hooksConfig, "BeforeAgent", "*", `node ${JSON.stringify(hookPath)} BeforeAgent`);
-  addGeminiCommandHook(hooksConfig, "AfterAgent", "*", `node ${JSON.stringify(hookPath)} AfterAgent`);
+  setAntigravityCommandHooks(hooksConfig, hookPath);
   writeJson(hooksPath, hooksConfig);
   console.log(`Updated ${hooksPath}`);
   console.log("Antigravity CLI hook installed.");
@@ -502,7 +549,7 @@ function uninstallGemini() {
 function uninstallAntigravity() {
   const hooksPath = path.join(HOME, ".gemini", "config", "hooks.json");
   const hooksConfig = readEditableJson(hooksPath, {});
-  if (removeClaudeCommandHooks(hooksConfig)) {
+  if (removeAntigravityCommandHooks(hooksConfig)) {
     writeJson(hooksPath, hooksConfig);
     console.log(`Removed CLIsponsor hooks from ${hooksPath}`);
   } else {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "clisponsor",
-  "version": "1.0.4",
+  "version": "1.0.6",
   "description": "CLIsponsor installer for Codex, Claude Code, and Gemini sponsored CLI placements.",
   "type": "module",
   "engines": {