@react-grab/ami 0.0.88 → 0.0.89

package/dist/cli.cjs

@@ -101,7 +101,7 @@ var require_picocolors = __commonJS({
 
 // src/cli.ts
 var import_picocolors = __toESM(require_picocolors());
-var VERSION = "0.0.88";
+var VERSION = "0.0.89";
 console.log(
   `${import_picocolors.default.magenta("\u273F")} ${import_picocolors.default.bold("React Grab")} ${import_picocolors.default.gray(VERSION)} ${import_picocolors.default.dim("(Ami)")}`
 );

package/dist/cli.js

@@ -99,7 +99,7 @@ var require_picocolors = __commonJS({
 
 // src/cli.ts
 var import_picocolors = __toESM(require_picocolors());
-var VERSION = "0.0.88";
+var VERSION = "0.0.89";
 console.log(
   `${import_picocolors.default.magenta("\u273F")} ${import_picocolors.default.bold("React Grab")} ${import_picocolors.default.gray(VERSION)} ${import_picocolors.default.dim("(Ami)")}`
 );

package/dist/client.cjs

@@ -17364,11 +17364,60 @@ var q = async (e) => {
   return { success: o2.success, patches: t2.patches, error: o2.error };
 };
 
+// ../utils/dist/client.js
+var CONNECTION_CHECK_TTL_MS = 5e3;
+var STORAGE_KEY = "react-grab:agent-sessions";
+var isRecord = (value) => typeof value === "object" && value !== null;
+var getStoredAgentContext = (storage, sessionId, storageKey = STORAGE_KEY) => {
+  const rawSessions = storage.getItem(storageKey);
+  if (!rawSessions) throw new Error("No sessions to resume");
+  let parsed;
+  try {
+    parsed = JSON.parse(rawSessions);
+  } catch {
+    throw new Error("Failed to parse stored sessions");
+  }
+  if (!isRecord(parsed)) throw new Error("Invalid stored sessions");
+  const storedSession = parsed[sessionId];
+  if (!isRecord(storedSession)) {
+    throw new Error(`Session ${sessionId} not found`);
+  }
+  const context = storedSession.context;
+  if (!isRecord(context)) throw new Error(`Session ${sessionId} is invalid`);
+  const content = context.content;
+  const prompt = context.prompt;
+  if (typeof content !== "string" || typeof prompt !== "string") {
+    throw new Error(`Session ${sessionId} is invalid`);
+  }
+  const options = context.options;
+  const storedSessionId = context.sessionId;
+  return {
+    content,
+    prompt,
+    options,
+    sessionId: typeof storedSessionId === "string" ? storedSessionId : void 0
+  };
+};
+var createCachedConnectionChecker = (checkConnection, ttlMs = CONNECTION_CHECK_TTL_MS) => {
+  let cache = null;
+  return async () => {
+    const now = Date.now();
+    if (cache && now - cache.timestamp < ttlMs) return cache.result;
+    try {
+      const result = await checkConnection();
+      cache = { result, timestamp: now };
+      return result;
+    } catch {
+      cache = { result: false, timestamp: now };
+      return false;
+    }
+  };
+};
+
 // src/constants.ts
 var COMPLETED_STATUS = "COMPLETED_STATUS";
 
 // src/client.ts
-var STORAGE_KEY = "react-grab:agent-sessions";
 var TOKEN_STORAGE_KEY = "react-grab:ami-token";
 var BRIDGE_TOKEN_STORAGE_KEY = "react-grab:ami-bridge-token";
 var loadCachedAuth = () => {
@@ -17493,9 +17542,8 @@ ${context.content}`;
       return { status: COMPLETED_STATUS, messages, chatId };
   }
 };
-var CONNECTION_CHECK_TTL_MS = 5e3;
+var isReactGrabApi = (value) => typeof value === "object" && value !== null && "setAgent" in value;
 var createAmiAgentProvider = (projectId) => {
-  let connectionCache = null;
   let lastAgentMessages = null;
   const sessionData = /* @__PURE__ */ new Map();
   const getLatestProjectId = async (token) => {
@@ -17617,16 +17665,10 @@ var createAmiAgentProvider = (projectId) => {
     },
     resume: async function* (sessionId, signal, storage) {
       const startTime = Date.now();
-      const savedSessions = storage.getItem(STORAGE_KEY);
-      if (!savedSessions) {
-        throw new Error("No sessions to resume");
-      }
-      const sessionsObject = JSON.parse(savedSessions);
-      const session = sessionsObject[sessionId];
-      if (!session) {
-        throw new Error(`Session ${sessionId} not found`);
-      }
-      const context = session.context;
+      const storedContext = getStoredAgentContext(storage, sessionId);
+      const context = {
+        content: storedContext.content,
+        prompt: storedContext.prompt};
       const auth = await getOrCreateAuth();
       yield "Resuming...";
       const statusQueue = [];
@@ -17724,21 +17766,10 @@ var createAmiAgentProvider = (projectId) => {
     },
     supportsResume: true,
     supportsFollowUp: true,
-    checkConnection: async () => {
-      const now = Date.now();
-      if (connectionCache && now - connectionCache.timestamp < CONNECTION_CHECK_TTL_MS) {
-        return connectionCache.result;
-      }
-      try {
-        const response = await fetch(l, { method: "HEAD" });
-        const result = response.ok;
-        connectionCache = { result, timestamp: now };
-        return result;
-      } catch {
-        connectionCache = { result: false, timestamp: now };
-        return false;
-      }
-    },
+    checkConnection: createCachedConnectionChecker(async () => {
+      const response = await fetch(l, { method: "HEAD" });
+      return response.ok;
+    }, CONNECTION_CHECK_TTL_MS),
     undo: async () => {
       if (!lastAgentMessages) return;
       try {
@@ -17755,24 +17786,25 @@ var createAmiAgentProvider = (projectId) => {
 var attachAgent = async () => {
   if (typeof window === "undefined") return;
   const provider = createAmiAgentProvider();
-  const attach = (api2) => {
-    api2.setAgent({ provider, storage: sessionStorage });
+  const attach = (api) => {
+    api.setAgent({ provider, storage: sessionStorage });
   };
-  const api = window.__REACT_GRAB__;
-  if (api) {
-    attach(api);
+  const existingApi = window.__REACT_GRAB__;
+  if (isReactGrabApi(existingApi)) {
+    attach(existingApi);
     return;
   }
   window.addEventListener(
     "react-grab:init",
     (event) => {
-      const customEvent = event;
-      attach(customEvent.detail);
+      if (!(event instanceof CustomEvent)) return;
+      if (!isReactGrabApi(event.detail)) return;
+      attach(event.detail);
     },
     { once: true }
   );
   const apiAfterListener = window.__REACT_GRAB__;
-  if (apiAfterListener) {
+  if (isReactGrabApi(apiAfterListener)) {
     attach(apiAfterListener);
   }
 };