@seanmozeik/tripwire 0.6.6 → 0.7.0

package/src/cli.ts

@@ -1,49 +1,35 @@
 #!/usr/bin/env bun
-// `tripwire test '<command>'` — pipe a synthetic event through the
-// Dispatcher and pretty-print the decision. Indispensable for tuning
-// Rules without going through Claude Code.
-//
-// `tripwire install <target>` — install tripwire hooks for AI agents.
-//
-// Usage:
-//   Bun src/cli.ts test 'rm -rf /etc'
-//   Bun src/cli.ts test --tool=Read --path=.env
-//   Bun src/cli.ts test --post --tool=Bash --stdout='ghp_<token>'
-//   Bun src/cli.ts install claude
-//   Bun src/cli.ts install codex
-//   Bun src/cli.ts install pi
-//   Bun src/cli.ts install all
-
-// oxlint-disable-next-line unicorn/import-style
-import { dirname } from 'node:path';
+// CLI commands for testing policy and installing agent integrations.
+// Usage examples:
+//   Bun src/main.ts test 'rm -rf /etc'
+//   Bun src/main.ts test --tool=Read --path=.env
+//   Bun src/main.ts test --post --tool=Bash --stdout='ghp_<token>'
+//   Bun src/main.ts install claude
+//   Bun src/main.ts install codex
+//   Bun src/main.ts install cursor
+//   Bun src/main.ts install pi
+//   Bun src/main.ts install all
 
 import { BunServices } from '@effect/platform-bun';
-import { file } from 'bun';
 import { Effect, Option } from 'effect';
 import { Argument, Command, Flag } from 'effect/unstable/cli';
 
 import pkg from '../package.json' with { type: 'json' };
-import { installAll, installClaude, installCodex, installPi } from './lib/install';
-
-// Resolve tripwire-hook path at runtime using process.argv
-// This works in both script mode (bun run) and compiled/bundled mode
-const runtimeSelf = (): string => {
-  const isBunCli = /\/bun(?<ext>\.exe)?$/.test(process.argv[0] ?? '');
-  return isBunCli ? process.argv[1]! : process.argv[0]!;
-};
+import {
+  installAll,
+  installClaude,
+  installCodex,
+  installCursor,
+  installOhMyPi,
+  installPi,
+} from './lib/install';
 
-const dispatchBin = async (): Promise<string> => {
-  const cliPath = runtimeSelf();
-  const cliDir = dirname(cliPath);
-  // Try tripwire-hook in same directory first (installed scenario)
-  const installedPath = `${cliDir}/tripwire-hook`;
-  try {
-    await file(installedPath).text();
-    return installedPath;
-  } catch {
-    // Fallback to development scenario: tripwire-hook in ../dist relative to CLI
-    return `${cliDir}/tripwire.js`;
+const hookCommand = (): string[] => {
+  const isBunCli = /\/bun(?<ext>\.exe)?$/.test(process.execPath);
+  if (isBunCli) {
+    return [process.execPath, new URL('main.ts', import.meta.url).pathname, '--tripwire-hook'];
   }
+  return [process.execPath, '--tripwire-hook'];
 };
 
 interface BuiltEvent {
@@ -60,7 +46,7 @@ const buildToolInput = (
   command: string | undefined,
   path: string | undefined,
   content: string | undefined,
-): unknown => {
+) => {
   if (tool === 'Bash') {
     return { command: command ?? '' };
   }
@@ -73,7 +59,7 @@ const buildToolInput = (
   if (tool === 'Edit' || tool === 'MultiEdit') {
     return { file_path: path ?? '', old_string: '', new_string: content ?? '' };
   }
-  return undefined;
+  return null;
 };
 
 interface EventParams {
@@ -89,12 +75,13 @@ interface EventParams {
 const buildEvent = (params: EventParams): BuiltEvent => {
   const { tool, post, command, path, stdout, stderr, content } = params;
   const eventName = post ? 'PostToolUse' : 'PreToolUse';
+  const toolInput = buildToolInput(tool, command, path, content);
   const event: BuiltEvent = {
     hook_event_name: eventName,
     tool_name: tool,
     cwd: process.cwd(),
     session_id: 'tripwire-cli-test',
-    tool_input: buildToolInput(tool, command, path, content),
+    ...(toolInput !== null && { tool_input: toolInput }),
   };
   if (post) {
     event.tool_response =
@@ -103,6 +90,14 @@ const buildEvent = (params: EventParams): BuiltEvent => {
   return event;
 };
 
+const prettyJson = (output: string): string => {
+  try {
+    return JSON.stringify(JSON.parse(output) as unknown, null, 2);
+  } catch {
+    return output;
+  }
+};
+
 const runTest = (config: {
   readonly command: string | undefined;
   readonly content: string | undefined;
@@ -112,11 +107,10 @@ const runTest = (config: {
   readonly stdout: string | undefined;
   readonly tool: string;
 }): Effect.Effect<void> =>
-  Effect.gen(function* () {
+  Effect.sync(() => {
     const { command, content, path, post, stderr, stdout, tool } = config;
     const event = buildEvent({ tool, post, command, path, stdout, stderr, content });
-    const bin = yield* Effect.promise(() => dispatchBin());
-    const result = Bun.spawnSync([bin], {
+    const result = Bun.spawnSync(hookCommand(), {
       stdin: new TextEncoder().encode(JSON.stringify(event)),
       timeout: 10_000,
       stdout: 'pipe',
@@ -128,12 +122,7 @@ const runTest = (config: {
       process.exit(1);
     }
     const output = new TextDecoder().decode(result.stdout);
-    try {
-      const parsed = JSON.parse(output) as unknown;
-      console.log(JSON.stringify(parsed, null, 2));
-    } catch {
-      console.log(output);
-    }
+    console.log(prettyJson(output));
   });
 
 const testCommand = Command.make(
@@ -151,7 +140,10 @@ const testCommand = Command.make(
       Flag.optional,
       Flag.withDescription('File path for Read/Write/Edit tools'),
     ),
-    post: Flag.boolean('post').pipe(Flag.withDescription('Test PostToolUse instead of PreToolUse')),
+    post: Flag.boolean('post').pipe(
+      Flag.withDefault(false),
+      Flag.withDescription('Test PostToolUse instead of PreToolUse'),
+    ),
     stderr: Flag.string('stderr').pipe(
       Flag.optional,
       Flag.withDescription('Stderr for PostToolUse Bash'),
@@ -178,10 +170,10 @@ const testCommand = Command.make(
 ).pipe(Command.withDescription('Test a synthetic hook event'));
 
 const runInstall = (target: string): Effect.Effect<void> =>
-  Effect.gen(function* () {
-    if (!['claude', 'codex', 'pi', 'all'].includes(target)) {
+  Effect.gen(function* runInstallEffect() {
+    if (!['claude', 'codex', 'cursor', 'pi', 'oh-my-pi', 'omp', 'all'].includes(target)) {
       console.error(`error: unknown target "${target}"`);
-      console.error('Valid targets: claude, codex, pi, all');
+      console.error('Valid targets: claude, codex, cursor, pi, oh-my-pi, all');
       process.exit(1);
     }
 
@@ -206,6 +198,17 @@ const runInstall = (target: string): Effect.Effect<void> =>
         results = [{ target: 'pi', result }];
         break;
       }
+      case 'oh-my-pi':
+      case 'omp': {
+        const result = yield* Effect.promise(() => installOhMyPi());
+        results = [{ target: 'oh-my-pi', result }];
+        break;
+      }
+      case 'cursor': {
+        const result = yield* Effect.promise(() => installCursor());
+        results = [{ target: 'cursor', result }];
+        break;
+      }
       case 'all': {
         const installResults = yield* Effect.promise(() => installAll());
         results = installResults.map((r) => ({ target: r.target, result: r }));
@@ -237,7 +240,7 @@ const installCommand = Command.make(
   'install',
   {
     target: Argument.string('target').pipe(
-      Argument.withDescription('Target agent (claude, codex, pi, or all)'),
+      Argument.withDescription('Target agent (claude, codex, cursor, pi, oh-my-pi, or all)'),
     ),
   },
   ({ target }) => runInstall(target),
@@ -250,7 +253,7 @@ const app = Command.make('tripwire').pipe(
 
 const program = Command.run(app, { version: pkg.version });
 
-const main = async (): Promise<void> => {
+const runCli = async (): Promise<void> => {
   try {
     await Effect.runPromise(program.pipe(Effect.provide(BunServices.layer)));
   } catch (error) {
@@ -260,5 +263,9 @@ const main = async (): Promise<void> => {
   }
 };
 
-// oxlint-disable-next-line no-void, unicorn/prefer-top-level-await
-void main();
+if (import.meta.main) {
+  // oxlint-disable-next-line no-void -- runCli reports failures through process.exitCode.
+  void runCli();
+}
+
+export { runCli };