@ycodium-ai/agent-antigravity 0.2.3122 → 0.2.3130

package/dist/index.js

@@ -28742,6 +28742,18 @@ function acpAgentVersion(initializeResult) {
 }
 var MAX_VERSION_OUTPUT_BYTES = 64 * 1024;
 
+// src/clock.ts
+var realAgentClock = {
+  now: () => Date.now(),
+  delay(ms, callback) {
+    const timer = setTimeout(callback, ms);
+    timer.unref?.();
+    return () => {
+      clearTimeout(timer);
+    };
+  }
+};
+
 // src/profile.ts
 import * as NodeCrypto2 from "node:crypto";
 import * as NodeFSP from "node:fs/promises";
@@ -29000,36 +29012,36 @@ async function* watchAgentStderr(source, onLoginPrompt, onBrowserHelperLine) {
 }
 
 // src/agentProcess.ts
-async function waitForExit(handle, timeoutMs) {
-  let timer;
+async function waitForExit(handle, timeoutMs, clock) {
+  let cancel = () => {
+  };
   const exited = await Promise.race([
     handle.exited.then(
       () => true,
       () => true
     ),
     new Promise((resolve6) => {
-      timer = setTimeout(() => resolve6(false), timeoutMs);
-      timer.unref?.();
+      cancel = clock.delay(timeoutMs, () => resolve6(false));
     })
   ]);
-  clearTimeout(timer);
+  cancel();
   return exited;
 }
-function stopAgentProcess(handle, exitWaitMs, log) {
+function stopAgentProcess(handle, exitWaitMs, log, clock = realAgentClock) {
   let stopping = stoppingByHandle.get(handle);
   if (stopping === void 0) {
-    stopping = terminate(handle, exitWaitMs, log);
+    stopping = terminate(handle, exitWaitMs, log, clock);
     stoppingByHandle.set(handle, stopping);
   }
   return stopping;
 }
 var stoppingByHandle = /* @__PURE__ */ new WeakMap();
-async function terminate(handle, exitWaitMs, log) {
+async function terminate(handle, exitWaitMs, log, clock) {
   await handle.kill("SIGTERM").catch(() => void 0);
-  if (await waitForExit(handle, exitWaitMs)) return;
+  if (await waitForExit(handle, exitWaitMs, clock)) return;
   log.warn("the agent process ignored SIGTERM; killing it", { pid: handle.pid, exitWaitMs });
   await handle.kill("SIGKILL").catch(() => void 0);
-  if (await waitForExit(handle, exitWaitMs)) return;
+  if (await waitForExit(handle, exitWaitMs, clock)) return;
   log.error("the agent process is still running after SIGKILL; leaving it behind", {
     pid: handle.pid,
     exitWaitMs
@@ -29062,6 +29074,7 @@ async function openAgentProcess(input2) {
     await removeRuntimeTempDir(directory);
     throw error62;
   }
+  const clock = input2.clock ?? realAgentClock;
   const { readable, writable } = nodeStreamsFromHandle({
     stdout: filterAgentStdout(handle.stdout, { onLoginPrompt: input2.onLoginPrompt }),
     stderr: watchAgentStderr(handle.stderr, input2.onLoginPrompt, input2.onBrowserHelperLine),
@@ -29073,22 +29086,21 @@ async function openAgentProcess(input2) {
     const pending = connection.request(method, params);
     if (timeoutMs === void 0) return pending;
     return new Promise((resolve6, reject) => {
-      const timer = setTimeout(() => {
+      const cancel = clock.delay(timeoutMs, () => {
         reject(
           new AntigravityAuthError(
             "agent-timeout",
             `Antigravity did not answer '${method}' in time. It may still be starting up; try again.`
           )
         );
-      }, timeoutMs);
-      timer.unref?.();
+      });
       pending.then(
         (value) => {
-          clearTimeout(timer);
+          cancel();
           resolve6(value);
         },
         (error62) => {
-          clearTimeout(timer);
+          cancel();
           reject(error62);
         }
       );
@@ -29098,7 +29110,7 @@ async function openAgentProcess(input2) {
   const close = () => {
     closing ??= (async () => {
       await connection.close().catch(() => void 0);
-      await stopAgentProcess(handle, input2.exitWaitMs, input2.host.log);
+      await stopAgentProcess(handle, input2.exitWaitMs, input2.host.log, clock);
       await removeRuntimeTempDir(directory);
     })();
     return closing;
@@ -30054,6 +30066,7 @@ function createAntigravitySession({
   platform: platform2,
   processEnv,
   exitWaitMs,
+  clock,
   userHome,
   dirs,
   events,
@@ -30135,7 +30148,7 @@ function createAntigravitySession({
         if (sessionReady) {
           void noteSession(() => login.noteSessionSignedOut());
           host.log.warn("antigravity asked to sign in during a session; stopping the process");
-          void stopAgentProcess(handle, exitWaitMs, host.log);
+          void stopAgentProcess(handle, exitWaitMs, host.log, clock);
         }
       };
       const { readable, writable, stderrTail } = nodeStreamsFromHandle2({
@@ -30282,7 +30295,7 @@ function createAntigravitySession({
         return session;
       } catch (error62) {
         await executor.dispose().catch(() => void 0);
-        await stopAgentProcess(handle, exitWaitMs, host.log);
+        await stopAgentProcess(handle, exitWaitMs, host.log, clock);
         await forwarding.catch(() => void 0);
         await removeRuntimeTempDir(runtimeTemp);
         if (live.signInRequired || isSignInRequiredError(error62)) {
@@ -30876,6 +30889,7 @@ function createLoginController(deps) {
   const { host } = deps;
   const authDeadlineMs = deps.authDeadlineMs ?? AUTH_DEADLINE_MS;
   const startupMs = deps.startupTimeoutMs ?? AGENT_STARTUP_TIMEOUT_MS;
+  const clock = deps.clock ?? realAgentClock;
   const forward = deps.forward ?? forwardCallback;
   const markerFile = signedInMarkerFile(deps.dirs.profile);
   const browserCommand = buildBrowserCommand({ nodePath: deps.nodePath, platform: deps.platform });
@@ -30944,6 +30958,7 @@ function createLoginController(deps) {
         platform: deps.platform,
         processEnv: deps.processEnv,
         exitWaitMs: deps.exitWaitMs,
+        clock,
         clientInfo: deps.clientInfo,
         browserCommand: hooks.browser ? browserCommand : void 0,
         onLoginPrompt: hooks.onLoginPrompt,
@@ -31039,13 +31054,12 @@ function createLoginController(deps) {
   };
   const runSignIn = async (target) => {
     const guard = (promise2) => Promise.race([promise2, target.interrupted]);
-    const deadline = setTimeout(() => {
+    const cancelDeadline = clock.delay(authDeadlineMs, () => {
       abortFlow(target, {
         kind: "failed",
         failure: failure2("expired", "Google sign-in expired. Start sign-in again.")
       });
-    }, authDeadlineMs);
-    deadline.unref?.();
+    });
     const opening = open4({
       browser: true,
       onLoginPrompt: (detail) => receiveAuthorizationUrl(target, detail),
@@ -31089,7 +31103,7 @@ function createLoginController(deps) {
         });
       }
     } finally {
-      clearTimeout(deadline);
+      cancelDeadline();
       const agent = await opening.catch(() => void 0);
       if (agent !== void 0) await release(agent);
       if (refreshAfter) requestRefresh();
@@ -31116,7 +31130,7 @@ function createLoginController(deps) {
     phase = "starting";
     authorizationUrl = null;
     error62 = null;
-    expiresAt = new Date(Date.now() + authDeadlineMs).toISOString();
+    expiresAt = new Date(clock.now() + authDeadlineMs).toISOString();
     publish();
     target.done = track(runSignIn(target));
     return snapshot();
@@ -31812,6 +31826,7 @@ var TextGenerationRejection = class extends Error {
 };
 function createAntigravityTextGeneration(deps) {
   const timeoutMs = deps.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+  const clock = deps.clock ?? realAgentClock;
   const running = /* @__PURE__ */ new Set();
   const runJson = async (request2) => {
     const { operation, signal } = request2;
@@ -31839,7 +31854,8 @@ function createAntigravityTextGeneration(deps) {
     const closeAll = async () => {
       stopped = true;
       await executor?.dispose().catch(() => void 0);
-      if (handle !== void 0) await stopAgentProcess(handle, deps.exitWaitMs, deps.host.log);
+      if (handle !== void 0)
+        await stopAgentProcess(handle, deps.exitWaitMs, deps.host.log, clock);
       await pump?.catch(() => void 0);
       if (runtimeTemp !== void 0) await removeRuntimeTempDir(runtimeTemp);
       if (workspace !== void 0) {
@@ -32133,6 +32149,7 @@ async function createAntigravityInstance(config2, host, deps) {
   const platform2 = deps.platform ?? HOST_PLATFORM;
   const processEnv = deps.processEnv ?? process.env;
   const exitWaitMs = deps.exitWaitMs ?? DEFAULT_EXIT_WAIT_MS;
+  const clock = deps.clock ?? realAgentClock;
   const userHome = deps.userHome ?? resolveAntigravityUserHome(platform2, { ...processEnv, ...recipe.environment });
   const dirs = resolveAntigravityDirectories({
     dataDir: deps.dataDir,
@@ -32165,7 +32182,7 @@ async function createAntigravityInstance(config2, host, deps) {
   const stopBooked = async (booked) => {
     for (const stopped of booked.background.finish()) events.push(stopped);
     await booked.executor.dispose().catch(() => void 0);
-    await stopAgentProcess(booked.handle, exitWaitMs, host.log);
+    await stopAgentProcess(booked.handle, exitWaitMs, host.log, clock);
     await booked.forwarding.catch(() => void 0);
     if (!await removeRuntimeTempDir(booked.runtimeTemp)) {
       host.log.warn("could not remove a runtime temp directory; it is swept on the next start", {
@@ -32181,6 +32198,7 @@ async function createAntigravityInstance(config2, host, deps) {
     platform: platform2,
     processEnv,
     exitWaitMs,
+    clock,
     clientInfo: CLIENT_INFO,
     acquireExecutable: () => deps.runtime.acquire(resolveInput),
     ...deps.textGenerationTimeoutMs === void 0 ? {} : { timeoutMs: deps.textGenerationTimeoutMs }
@@ -32200,6 +32218,7 @@ async function createAntigravityInstance(config2, host, deps) {
     acquireExecutable: () => deps.runtime.acquire(resolveInput),
     stopSessions: stopAllSessions,
     exitWaitMs,
+    clock,
     clientInfo: CLIENT_INFO,
     nodePath: deps.nodePath ?? process.execPath,
     ...deps.authDeadlineMs === void 0 ? {} : { authDeadlineMs: deps.authDeadlineMs },
@@ -32278,6 +32297,7 @@ async function createAntigravityInstance(config2, host, deps) {
     platform: platform2,
     processEnv,
     exitWaitMs,
+    clock,
     userHome,
     dirs,
     events,
@@ -33478,7 +33498,7 @@ var executors = [
 ];
 var index_default = definePlugin({
   id: "agent-antigravity",
-  apiVersion: 13,
+  apiVersion: 14,
   executors,
   activate(ctx) {
     const runtime = createRuntimeManager({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ycodium-ai/agent-antigravity",
-  "version": "0.2.3122",
+  "version": "0.2.3130",
   "description": "Google Antigravity executor: runs the Antigravity agent over Agent Client Protocol stdio.",
   "keywords": [
     "ycodium",
@@ -31,8 +31,8 @@
     "typescript": "~6.0.3",
     "vite-plus": "0.2.2",
     "yauzl": "^3.4.0",
-    "ycodium-acp-executor": "0.2.3122",
-    "@ycodium-ai/plugin-api": "0.2.3122",
+    "@ycodium-ai/plugin-api": "0.2.3130",
+    "ycodium-acp-executor": "0.2.3130",
     "ycodium-executor-testing": "0.0.0"
   },
   "ycodium": {