fix: ensure worker is running before executing hook logic in multiple scripts

src/hooks/context-hook.ts

@@ -9,7 +9,7 @@
 import path from "path";
 import { stdin } from "process";
 import { execSync } from "child_process";
-import { getWorkerPort } from "../shared/worker-utils.js";
+import { ensureWorkerRunning, getWorkerPort } from "../shared/worker-utils.js";
 
 export interface SessionStartInput {
   session_id?: string;
@@ -20,36 +20,14 @@ export interface SessionStartInput {
   [key: string]: any;
 }
 
-async function waitForPort(port: number, maxWaitMs: number = 10000): Promise<boolean> {
-  const startTime = Date.now();
-  const pollInterval = 100;
-
-  while (Date.now() - startTime < maxWaitMs) {
-    try {
-      execSync(`curl -s -f -m 1 "http://127.0.0.1:${port}/api/health" > /dev/null 2>&1`, {
-        timeout: 1000,
-      });
-      return true;
-    } catch {
-      await new Promise((resolve) => setTimeout(resolve, pollInterval));
-    }
-  }
-  return false;
-}
-
 async function contextHook(input?: SessionStartInput): Promise<string> {
+  // Ensure worker is running before any other logic
+  await ensureWorkerRunning();
+
   const cwd = input?.cwd ?? process.cwd();
   const project = cwd ? path.basename(cwd) : "unknown-project";
   const port = getWorkerPort();
 
-  // Wait for worker to be available
-  const isAvailable = await waitForPort(port);
-  if (!isAvailable) {
-    throw new Error(
-      `Worker service not available on port ${port} after 10s. Try: npm run worker:restart`
-    );
-  }
-
   const url = `http://127.0.0.1:${port}/api/context/inject?project=${encodeURIComponent(project)}`;
   const result = execSync(`curl -s "${url}"`, { encoding: "utf-8", timeout: 5000 });
   return result.trim();