@react-grab/amp 0.0.82

package/dist/server.js

@@ -0,0 +1,150 @@
+import { execute } from '@sourcegraph/amp-sdk';
+import fkill from 'fkill';
+import { Hono } from 'hono';
+import { cors } from 'hono/cors';
+import { streamSSE } from 'hono/streaming';
+import { serve } from '@hono/node-server';
+import pc from 'picocolors';
+import { pathToFileURL } from 'url';
+
+// src/server.ts
+
+// src/constants.ts
+var DEFAULT_PORT = 9567;
+var VERSION = "0.0.82";
+var sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
+var threadMap = /* @__PURE__ */ new Map();
+var abortControllers = /* @__PURE__ */ new Map();
+var extractTextFromContent = (content) => {
+  return content.filter((item) => item.type === "text" && item.text).map((item) => item.text).join(" ").trim();
+};
+var createServer = () => {
+  const honoApplication = new Hono();
+  honoApplication.use("*", cors());
+  honoApplication.post("/agent", async (context) => {
+    const requestBody = await context.req.json();
+    const { content, prompt, options, sessionId } = requestBody;
+    const existingThread = sessionId ? threadMap.get(sessionId) : void 0;
+    const isFollowUp = Boolean(existingThread);
+    const userPrompt = isFollowUp ? prompt : `${prompt}
+
+${content}`;
+    return streamSSE(context, async (stream) => {
+      const abortController = new AbortController();
+      if (sessionId) {
+        abortControllers.set(sessionId, abortController);
+      }
+      const isAborted = () => abortController.signal.aborted;
+      try {
+        await stream.writeSSE({ data: "Thinking...", event: "status" });
+        const executeOptions = {
+          dangerouslyAllowAll: true
+        };
+        if (options?.cwd) {
+          executeOptions.cwd = options.cwd;
+        } else {
+          executeOptions.cwd = process.cwd();
+        }
+        if (isFollowUp && existingThread) {
+          executeOptions.continue = existingThread.threadId;
+        }
+        let capturedThreadId;
+        for await (const message of execute({
+          prompt: userPrompt,
+          options: executeOptions,
+          signal: abortController.signal
+        })) {
+          if (isAborted()) break;
+          switch (message.type) {
+            case "system":
+              if (message.subtype === "init") {
+                const systemMessage = message;
+                if (systemMessage.thread_id) {
+                  capturedThreadId = systemMessage.thread_id;
+                }
+                await stream.writeSSE({ data: "Session started...", event: "status" });
+              }
+              break;
+            case "assistant": {
+              const messageContent = message.message?.content;
+              if (messageContent && Array.isArray(messageContent)) {
+                const toolUse = messageContent.find((item) => item.type === "tool_use");
+                if (toolUse && "name" in toolUse) {
+                  await stream.writeSSE({
+                    data: `Using ${toolUse.name}...`,
+                    event: "status"
+                  });
+                } else {
+                  const textContent = extractTextFromContent(messageContent);
+                  if (textContent && !isAborted()) {
+                    await stream.writeSSE({ data: textContent, event: "status" });
+                  }
+                }
+              }
+              break;
+            }
+            case "result":
+              if (message.is_error) {
+                await stream.writeSSE({
+                  data: `Error: ${message.error || "Unknown error"}`,
+                  event: "error"
+                });
+              } else {
+                await stream.writeSSE({
+                  data: "Completed successfully",
+                  event: "status"
+                });
+              }
+              break;
+          }
+        }
+        if (sessionId && capturedThreadId && !isAborted()) {
+          threadMap.set(sessionId, { threadId: capturedThreadId });
+        }
+        if (!isAborted()) {
+          await stream.writeSSE({ data: "", event: "done" });
+        }
+      } catch (error) {
+        if (!isAborted()) {
+          const errorMessage = error instanceof Error ? error.message : "Unknown error";
+          await stream.writeSSE({
+            data: `Error: ${errorMessage}`,
+            event: "error"
+          });
+          await stream.writeSSE({ data: "", event: "done" });
+        }
+      } finally {
+        if (sessionId) {
+          abortControllers.delete(sessionId);
+        }
+      }
+    });
+  });
+  honoApplication.post("/abort/:sessionId", (context) => {
+    const { sessionId } = context.req.param();
+    const abortController = abortControllers.get(sessionId);
+    if (abortController) {
+      abortController.abort();
+      abortControllers.delete(sessionId);
+    }
+    return context.json({ status: "ok" });
+  });
+  honoApplication.get("/health", (context) => {
+    return context.json({ status: "ok", provider: "amp" });
+  });
+  return honoApplication;
+};
+var startServer = async (port = DEFAULT_PORT) => {
+  await fkill(`:${port}`, { force: true, silent: true }).catch(() => {
+  });
+  await sleep(100);
+  const honoApplication = createServer();
+  serve({ fetch: honoApplication.fetch, port });
+  console.log(`${pc.magenta("\u269B")} ${pc.bold("React Grab")} ${pc.gray(VERSION)} ${pc.dim("(Amp)")}`);
+  console.log(`- Local:    ${pc.cyan(`http://localhost:${port}`)}`);
+};
+if (import.meta.url === pathToFileURL(process.argv[1]).href) {
+  startServer(DEFAULT_PORT).catch(console.error);
+}
+
+export { createServer, startServer };

package/package.json

@@ -0,0 +1,43 @@
+{
+  "name": "@react-grab/amp",
+  "version": "0.0.82",
+  "type": "module",
+  "bin": {
+    "react-grab-amp": "./dist/cli.cjs"
+  },
+  "exports": {
+    "./client": {
+      "types": "./dist/client.d.ts",
+      "import": "./dist/client.js",
+      "require": "./dist/client.cjs"
+    },
+    "./server": {
+      "types": "./dist/server.d.ts",
+      "import": "./dist/server.js",
+      "require": "./dist/server.cjs"
+    },
+    "./dist/*": "./dist/*.js",
+    "./dist/*.js": "./dist/*.js"
+  },
+  "browser": "dist/client.global.js",
+  "files": [
+    "dist"
+  ],
+  "devDependencies": {
+    "@types/cross-spawn": "^6.0.6",
+    "tsup": "^8.4.0"
+  },
+  "dependencies": {
+    "@hono/node-server": "^1.19.6",
+    "cross-spawn": "^7.0.6",
+    "@sourcegraph/amp-sdk": "0.1.0-20251210081226-g90e3892",
+    "fkill": "^9.0.0",
+    "hono": "^4.0.0",
+    "picocolors": "^1.1.1",
+    "react-grab": "0.0.82"
+  },
+  "scripts": {
+    "dev": "tsup --watch",
+    "build": "rm -rf dist && NODE_ENV=production tsup"
+  }
+}