@react-grab/cursor 0.0.78 → 0.0.80

package/dist/client.js

@@ -55,19 +55,33 @@ async function* streamSSE(stream, signal) {
   }
 }
 async function* streamFromServer(serverUrl, context, signal) {
-  const response = await fetch(`${serverUrl}/agent`, {
-    method: "POST",
-    headers: { "Content-Type": "application/json" },
-    body: JSON.stringify(context),
-    signal
-  });
-  if (!response.ok) {
-    throw new Error(`Server error: ${response.status}`);
-  }
-  if (!response.body) {
-    throw new Error("No response body");
+  const sessionId = context.sessionId;
+  const handleAbort = () => {
+    if (sessionId) {
+      fetch(`${serverUrl}/abort/${sessionId}`, { method: "POST" }).catch(
+        () => {
+        }
+      );
+    }
+  };
+  signal.addEventListener("abort", handleAbort);
+  try {
+    const response = await fetch(`${serverUrl}/agent`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify(context),
+      signal
+    });
+    if (!response.ok) {
+      throw new Error(`Server error: ${response.status}`);
+    }
+    if (!response.body) {
+      throw new Error("No response body");
+    }
+    yield* streamSSE(response.body, signal);
+  } finally {
+    signal.removeEventListener("abort", handleAbort);
   }
-  yield* streamSSE(response.body, signal);
 }
 var createCursorAgentProvider = (providerOptions = {}) => {
   const { serverUrl = DEFAULT_SERVER_URL, getOptions } = providerOptions;
@@ -103,6 +117,7 @@ var createCursorAgentProvider = (providerOptions = {}) => {
       yield* streamFromServer(serverUrl, mergedContext, signal);
     },
     supportsResume: true,
+    supportsFollowUp: true,
     checkConnection: async () => {
       const now = Date.now();
       if (connectionCache && now - connectionCache.timestamp < CONNECTION_CHECK_TTL_MS) {
@@ -118,9 +133,9 @@ var createCursorAgentProvider = (providerOptions = {}) => {
         return false;
       }
     },
-    undo: async () => {
+    abort: async (sessionId) => {
       try {
-        await fetch(`${serverUrl}/undo`, { method: "POST" });
+        await fetch(`${serverUrl}/abort/${sessionId}`, { method: "POST" });
       } catch {
       }
     }
@@ -129,19 +144,26 @@ var createCursorAgentProvider = (providerOptions = {}) => {
 var attachAgent = async () => {
   if (typeof window === "undefined") return;
   const provider = createCursorAgentProvider();
+  const attach = (api2) => {
+    api2.setAgent({ provider, storage: sessionStorage });
+  };
   const api = window.__REACT_GRAB__;
   if (api) {
-    api.setAgent({ provider, storage: sessionStorage });
+    attach(api);
     return;
   }
   window.addEventListener(
     "react-grab:init",
     (event) => {
       const customEvent = event;
-      customEvent.detail.setAgent({ provider, storage: sessionStorage });
+      attach(customEvent.detail);
     },
     { once: true }
   );
+  const apiAfterListener = window.__REACT_GRAB__;
+  if (apiAfterListener) {
+    attach(apiAfterListener);
+  }
 };
 attachAgent();