amai 0.0.5 → 0.0.7

package/dist/lib/daemon-entry.js

@@ -1,11 +1,10 @@
 #!/usr/bin/env node
 import WebSocket from 'ws';
-import { EventEmitter } from 'events';
 import { z } from 'zod';
 import { readFile, writeFile, stat, access, readdir, glob, unlink, mkdir } from 'fs/promises';
-import path9 from 'path';
-import fs3, { readdirSync } from 'fs';
-import os2 from 'os';
+import path11 from 'path';
+import fs4, { readdirSync } from 'fs';
+import os3 from 'os';
 import { randomUUID, createHash } from 'crypto';
 import { exec, spawn } from 'child_process';
 import { promisify } from 'util';
@@ -15,12 +14,12 @@ import { serve } from '@hono/node-server';
 import { cors } from 'hono/cors';
 
 var DEFAULT_SERVER_URL = "wss://ama-production-a628.up.railway.app";
-var AMA_DIR = path9.join(os2.homedir(), ".amai");
-var CODE_DIR = path9.join(AMA_DIR, "code");
-var STORAGE_DIR = path9.join(AMA_DIR, "storage");
+var AMA_DIR = path11.join(os3.homedir(), ".amai");
+var CODE_DIR = path11.join(AMA_DIR, "code");
+var STORAGE_DIR = path11.join(AMA_DIR, "storage");
 
 // src/lib/project-registry.ts
-var REGISTRY_FILE = path9.join(AMA_DIR, "projects.json");
+var REGISTRY_FILE = path11.join(AMA_DIR, "projects.json");
 var ProjectRegistry = class {
   projects = /* @__PURE__ */ new Map();
   constructor() {
@@ -28,14 +27,14 @@ var ProjectRegistry = class {
   }
   load() {
     try {
-      if (fs3.existsSync(REGISTRY_FILE)) {
-        const data = fs3.readFileSync(REGISTRY_FILE, "utf8");
+      if (fs4.existsSync(REGISTRY_FILE)) {
+        const data = fs4.readFileSync(REGISTRY_FILE, "utf8");
         const parsed = JSON.parse(data);
         if (!Array.isArray(parsed)) {
           console.error("Invalid project registry format: expected array, got", typeof parsed);
           const backupFile = REGISTRY_FILE + ".backup." + Date.now();
-          fs3.copyFileSync(REGISTRY_FILE, backupFile);
-          fs3.unlinkSync(REGISTRY_FILE);
+          fs4.copyFileSync(REGISTRY_FILE, backupFile);
+          fs4.unlinkSync(REGISTRY_FILE);
           return;
         }
         const projects = parsed;
@@ -48,11 +47,11 @@ var ProjectRegistry = class {
       }
     } catch (error) {
       console.error("Failed to load project registry:", error);
-      if (fs3.existsSync(REGISTRY_FILE)) {
+      if (fs4.existsSync(REGISTRY_FILE)) {
         try {
           const backupFile = REGISTRY_FILE + ".backup." + Date.now();
-          fs3.copyFileSync(REGISTRY_FILE, backupFile);
-          fs3.unlinkSync(REGISTRY_FILE);
+          fs4.copyFileSync(REGISTRY_FILE, backupFile);
+          fs4.unlinkSync(REGISTRY_FILE);
           console.log("Corrupted registry file backed up and removed. Starting fresh.");
         } catch (backupError) {
         }
@@ -61,21 +60,21 @@ var ProjectRegistry = class {
   }
   save() {
     try {
-      if (!fs3.existsSync(AMA_DIR)) {
-        fs3.mkdirSync(AMA_DIR, { recursive: true });
+      if (!fs4.existsSync(AMA_DIR)) {
+        fs4.mkdirSync(AMA_DIR, { recursive: true });
       }
       const projects = Array.from(this.projects.values());
-      fs3.writeFileSync(REGISTRY_FILE, JSON.stringify(projects, null, 2), "utf8");
+      fs4.writeFileSync(REGISTRY_FILE, JSON.stringify(projects, null, 2), "utf8");
     } catch (error) {
       console.error("Failed to save project registry:", error);
     }
   }
   register(projectId, cwd, name) {
-    const normalizedCwd = path9.normalize(path9.resolve(cwd));
+    const normalizedCwd = path11.normalize(path11.resolve(cwd));
     this.projects.set(projectId, {
       id: projectId,
       cwd: normalizedCwd,
-      name: name || path9.basename(normalizedCwd),
+      name: name || path11.basename(normalizedCwd),
       active: true
     });
     this.save();
@@ -105,9 +104,9 @@ var ProjectRegistry = class {
 var projectRegistry = new ProjectRegistry();
 function isPathWithinProject(filePath, projectCwd) {
   try {
-    const resolved = path9.resolve(projectCwd, filePath);
-    const normalized = path9.normalize(resolved);
-    const normalizedCwd = path9.normalize(projectCwd);
+    const resolved = path11.resolve(projectCwd, filePath);
+    const normalized = path11.normalize(resolved);
+    const normalizedCwd = path11.normalize(projectCwd);
     return normalized.startsWith(normalizedCwd);
   } catch {
     return false;
@@ -123,7 +122,7 @@ function validatePath(filePath, projectCwd) {
     };
   }
   try {
-    const resolvedPath = path9.resolve(projectCwd, filePath);
+    const resolvedPath = path11.resolve(projectCwd, filePath);
     if (!isPathWithinProject(filePath, projectCwd)) {
       return {
         valid: false,
@@ -142,7 +141,7 @@ function validatePath(filePath, projectCwd) {
   }
 }
 function resolveProjectPath(filePath, projectCwd) {
-  return path9.resolve(projectCwd, filePath);
+  return path11.resolve(projectCwd, filePath);
 }
 
 // src/tools/read-file.ts
@@ -264,7 +263,7 @@ var read_file = async function(input, projectCwd) {
         };
       }
     } else {
-      const absolute_file_path = path9.resolve(relative_file_path);
+      const absolute_file_path = path11.resolve(relative_file_path);
       try {
         const fileStats = await stat(absolute_file_path);
         if (!fileStats.isFile()) {
@@ -858,20 +857,20 @@ var editFiles = async function(input, projectCwd) {
     }
     const basePath = projectCwd || process.cwd();
     const filePath = resolveProjectPath(target_file, basePath);
-    const dirPath = path9.dirname(filePath);
+    const dirPath = path11.dirname(filePath);
     await mkdir(dirPath, { recursive: true });
     let isNewFile = providedNewFile;
     let existingContent = "";
     if (isNewFile === void 0) {
       try {
-        existingContent = await fs3.promises.readFile(filePath, "utf-8");
+        existingContent = await fs4.promises.readFile(filePath, "utf-8");
         isNewFile = false;
       } catch (error) {
         isNewFile = true;
       }
     } else if (!isNewFile) {
       try {
-        existingContent = await fs3.promises.readFile(filePath, "utf-8");
+        existingContent = await fs4.promises.readFile(filePath, "utf-8");
       } catch (error) {
         isNewFile = true;
       }
@@ -884,7 +883,7 @@ var editFiles = async function(input, projectCwd) {
       content
     );
     try {
-      await fs3.promises.writeFile(filePath, content);
+      await fs4.promises.writeFile(filePath, content);
     } catch (writeError) {
       checkpointStore.removeCheckpoint(checkpointId);
       throw writeError;
@@ -1008,7 +1007,7 @@ var grepTool = async function(input, projectCwd) {
   try {
     const { includePattern, excludePattern: excludePattern2, caseSensitive } = options || {};
     const searchDir = projectCwd || process.cwd();
-    if (projectCwd && !path9.isAbsolute(projectCwd)) {
+    if (projectCwd && !path11.isAbsolute(projectCwd)) {
       return {
         success: false,
         message: "Invalid project directory",
@@ -1200,8 +1199,8 @@ var list = async function(input, projectCwd) {
     const walk = async (currentDir, depth) => {
       const entries = await readdir(currentDir, { withFileTypes: true });
       for (const entry of entries) {
-        const entryAbsolutePath = path9.join(currentDir, entry.name);
-        const entryRelativePath = path9.relative(absolutePath, entryAbsolutePath) || ".";
+        const entryAbsolutePath = path11.join(currentDir, entry.name);
+        const entryRelativePath = path11.relative(absolutePath, entryAbsolutePath) || ".";
         if (entry.isDirectory()) {
           const isExcluded = entry.name.match(excludePattern);
           if (includeDirectoriesNormalized && matchPattern(entry.name) && !isExcluded) {
@@ -1251,213 +1250,9 @@ var list = async function(input, projectCwd) {
     };
   }
 };
-var ignoreFiles = ["node_modules", ".git", ".next", ".env", ".env.local", ".env.development.local", ".env.test.local", ".env.production.local"];
-var getContext = (dir, base = dir, allFiles = []) => {
-  const filePath = readdirSync(dir, { withFileTypes: true });
-  for (const file of filePath) {
-    if (ignoreFiles.includes(file.name)) continue;
-    const fullPath = path9.join(dir, file.name);
-    if (file.isDirectory()) {
-      getContext(fullPath, base, allFiles);
-    } else {
-      allFiles.push(path9.relative(base, fullPath));
-    }
-  }
-  return allFiles;
-};
-var HOME = os2.homedir();
-var IDE_PROJECTS_PATHS = {
-  vscode: path9.join(HOME, ".vscode", "projects"),
-  cursor: path9.join(HOME, ".cursor", "projects"),
-  claude: path9.join(HOME, ".claude", "projects")
-};
-function getWorkspaceStoragePath(ide) {
-  const platform = os2.platform();
-  const appName = "Cursor" ;
-  if (platform === "darwin") {
-    return path9.join(HOME, "Library", "Application Support", appName, "User", "workspaceStorage");
-  } else if (platform === "win32") {
-    return path9.join(process.env.APPDATA || "", appName, "User", "workspaceStorage");
-  } else {
-    return path9.join(HOME, ".config", appName, "User", "workspaceStorage");
-  }
-}
-function scanWorkspaceStorage(ide) {
-  const projects = [];
-  const storagePath = getWorkspaceStoragePath();
-  if (!fs3.existsSync(storagePath)) {
-    return projects;
-  }
-  try {
-    const workspaces = fs3.readdirSync(storagePath);
-    for (const workspace of workspaces) {
-      const workspaceJsonPath = path9.join(storagePath, workspace, "workspace.json");
-      if (fs3.existsSync(workspaceJsonPath)) {
-        try {
-          const content = fs3.readFileSync(workspaceJsonPath, "utf-8");
-          const data = JSON.parse(content);
-          if (data.folder && typeof data.folder === "string") {
-            let projectPath = data.folder;
-            if (projectPath.startsWith("file://")) {
-              projectPath = projectPath.replace("file://", "");
-              projectPath = decodeURIComponent(projectPath);
-            }
-            if (fs3.existsSync(projectPath) && fs3.statSync(projectPath).isDirectory()) {
-              projects.push({
-                name: path9.basename(projectPath),
-                path: projectPath,
-                type: ide
-              });
-            }
-          }
-        } catch (err) {
-          console.debug(`Error reading workspace.json at ${workspaceJsonPath}: ${err}`);
-        }
-      }
-    }
-  } catch (err) {
-    console.debug(`Error scanning workspaceStorage for ${ide}: ${err}`);
-  }
-  return projects;
-}
-var scanIdeProjects = async () => {
-  try {
-    const allProjects = [];
-    const seenPaths = /* @__PURE__ */ new Set();
-    const addProject = (projectPath, ide) => {
-      try {
-        const resolvedPath = fs3.realpathSync(projectPath);
-        if (fs3.existsSync(resolvedPath) && fs3.statSync(resolvedPath).isDirectory() && !seenPaths.has(resolvedPath)) {
-          const isIdeProjectsDir = Object.values(IDE_PROJECTS_PATHS).some((ideDir) => {
-            try {
-              return fs3.realpathSync(ideDir) === resolvedPath;
-            } catch {
-              return false;
-            }
-          });
-          if (!isIdeProjectsDir) {
-            seenPaths.add(resolvedPath);
-            allProjects.push({
-              name: path9.basename(resolvedPath),
-              path: resolvedPath,
-              type: ide
-            });
-          }
-        }
-      } catch {
-      }
-    };
-    const cursorProjects = scanWorkspaceStorage("cursor");
-    for (const project of cursorProjects) {
-      addProject(project.path, "cursor");
-    }
-    for (const [ide, dirPath] of Object.entries(IDE_PROJECTS_PATHS)) {
-      if (ide === "cursor") continue;
-      if (fs3.existsSync(dirPath)) {
-        const projects = fs3.readdirSync(dirPath);
-        projects.forEach((project) => {
-          const projectPath = path9.join(dirPath, project);
-          try {
-            const stats = fs3.lstatSync(projectPath);
-            let actualPath = null;
-            if (stats.isSymbolicLink()) {
-              actualPath = fs3.realpathSync(projectPath);
-            } else if (stats.isFile()) {
-              try {
-                let content = fs3.readFileSync(projectPath, "utf-8").trim();
-                if (content.startsWith("~/") || content === "~") {
-                  content = content.replace(/^~/, HOME);
-                }
-                const resolvedContent = path9.isAbsolute(content) ? content : path9.resolve(path9.dirname(projectPath), content);
-                if (fs3.existsSync(resolvedContent) && fs3.statSync(resolvedContent).isDirectory()) {
-                  actualPath = fs3.realpathSync(resolvedContent);
-                }
-              } catch {
-                return;
-              }
-            } else if (stats.isDirectory()) {
-              actualPath = fs3.realpathSync(projectPath);
-            }
-            if (actualPath) {
-              addProject(actualPath, ide);
-            }
-          } catch {
-          }
-        });
-      }
-    }
-    return allProjects;
-  } catch (error) {
-    console.debug(`Error scanning IDE projects: ${error}`);
-    return [];
-  }
-};
-var wsConnection = null;
-var startHttpServer = (connection) => {
-  if (connection) {
-    wsConnection = connection;
-  }
+var startHttpServer = () => {
   const app = new Hono();
   app.use(cors());
-  app.post("/daemon/status", (c) => {
-    const status = wsConnection ? getConnectionStatus(wsConnection) : "closed";
-    return c.json({ connected: status === "open" });
-  });
-  app.get("/daemon/status/stream", (c) => {
-    const encoder = new TextEncoder();
-    const stream = new ReadableStream({
-      start(controller) {
-        const initialStatus = wsConnection ? getConnectionStatus(wsConnection) : "closed";
-        controller.enqueue(encoder.encode(`data: ${JSON.stringify({ connected: initialStatus === "open" })}
-
-`));
-        const statusHandler = (data) => {
-          try {
-            controller.enqueue(encoder.encode(`data: ${JSON.stringify(data)}
-
-`));
-          } catch {
-          }
-        };
-        statusEmitter.on("status", statusHandler);
-        const heartbeatInterval = setInterval(() => {
-          try {
-            const currentStatus = wsConnection ? getConnectionStatus(wsConnection) : "closed";
-            controller.enqueue(encoder.encode(`data: ${JSON.stringify({ connected: currentStatus === "open" })}
-
-`));
-          } catch {
-          }
-        }, 15e3);
-        c.req.raw.signal.addEventListener("abort", () => {
-          statusEmitter.off("status", statusHandler);
-          clearInterval(heartbeatInterval);
-        });
-      }
-    });
-    return new Response(stream, {
-      headers: {
-        "Content-Type": "text/event-stream",
-        "Cache-Control": "no-cache",
-        "Connection": "keep-alive"
-      }
-    });
-  });
-  app.get("context", async (c) => {
-    const context = getContext(process.cwd());
-    return c.body(JSON.stringify(context));
-  });
-  app.get("/ide-projects", async (c) => {
-    try {
-      const projects = await scanIdeProjects();
-      if (!projects) {
-        return c.json({ error: "No projects found" }, 500);
-      }
-      return c.json({ projects });
-    } catch (error) {
-      return c.json({ error: "Failed to scan IDE projects" }, 500);
-    }
-  });
   app.post("/projects/register", async (c) => {
     try {
       const { projectId, cwd, name } = await c.req.json();
@@ -1486,14 +1281,14 @@ var startHttpServer = (connection) => {
       }
       let resolved;
       if (projectCwd) {
-        resolved = path9.isAbsolute(filePath) ? filePath : path9.resolve(projectCwd, filePath);
-        const normalizedResolved = path9.normalize(resolved);
-        const normalizedCwd = path9.normalize(projectCwd);
+        resolved = path11.isAbsolute(filePath) ? filePath : path11.resolve(projectCwd, filePath);
+        const normalizedResolved = path11.normalize(resolved);
+        const normalizedCwd = path11.normalize(projectCwd);
         if (!normalizedResolved.startsWith(normalizedCwd)) {
           return c.json({ error: "Path is outside project directory" }, 403);
         }
       } else {
-        resolved = path9.isAbsolute(filePath) ? filePath : path9.join(process.cwd(), filePath);
+        resolved = path11.isAbsolute(filePath) ? filePath : path11.join(process.cwd(), filePath);
       }
       let currentContent;
       try {
@@ -1580,9 +1375,9 @@ var startHttpServer = (connection) => {
       }
       let resolved;
       if (projectCwd) {
-        resolved = path9.isAbsolute(filePath || checkpoint.filePath) ? filePath || checkpoint.filePath : path9.resolve(projectCwd, filePath || checkpoint.filePath);
-        const normalizedResolved = path9.normalize(resolved);
-        const normalizedCwd = path9.normalize(projectCwd);
+        resolved = path11.isAbsolute(filePath || checkpoint.filePath) ? filePath || checkpoint.filePath : path11.resolve(projectCwd, filePath || checkpoint.filePath);
+        const normalizedResolved = path11.normalize(resolved);
+        const normalizedCwd = path11.normalize(projectCwd);
         if (!normalizedResolved.startsWith(normalizedCwd)) {
           return c.json({ error: "Path is outside project directory" }, 403);
         }
@@ -1644,16 +1439,30 @@ var startHttpServer = (connection) => {
   });
   serve({ fetch: app.fetch, port: 3456 });
 };
-var CREDENTIALS_DIR = path9.join(os2.homedir(), ".amai");
-var CREDENTIALS_PATH = path9.join(CREDENTIALS_DIR, "credentials.json");
+var CREDENTIALS_DIR = path11.join(os3.homedir(), ".amai");
+var CREDENTIALS_PATH = path11.join(CREDENTIALS_DIR, "credentials.json");
 function getTokens() {
-  if (!fs3.existsSync(CREDENTIALS_PATH)) {
+  if (!fs4.existsSync(CREDENTIALS_PATH)) {
     return null;
   }
-  const raw = fs3.readFileSync(CREDENTIALS_PATH, "utf8");
+  const raw = fs4.readFileSync(CREDENTIALS_PATH, "utf8");
   const data = JSON.parse(raw);
   return data;
 }
+var getUserId = () => {
+  try {
+    if (!fs4.existsSync(CREDENTIALS_PATH)) {
+      return;
+    }
+    const raw = fs4.readFileSync(CREDENTIALS_PATH, "utf8");
+    const data = JSON.parse(raw);
+    return {
+      userId: data.user.id
+    };
+  } catch {
+    throw new Error("Error while getting userId");
+  }
+};
 var ExplanationSchema = z.object({
   explanation: z.string().describe("One sentence explanation as to why this tool is being used")
 });
@@ -1741,9 +1550,343 @@ var runTerminalCommand = async (input, projectCwd) => {
     };
   }
 };
+var ignoreFiles = ["node_modules", ".git", ".next", ".env", ".env.local", ".env.development.local", ".env.test.local", ".env.production.local"];
+var getContext = (dir, base = dir, allFiles = []) => {
+  const filePath = readdirSync(dir, { withFileTypes: true });
+  for (const file of filePath) {
+    if (ignoreFiles.includes(file.name)) continue;
+    const fullPath = path11.join(dir, file.name);
+    if (file.isDirectory()) {
+      getContext(fullPath, base, allFiles);
+    } else {
+      allFiles.push(path11.relative(base, fullPath));
+    }
+  }
+  return allFiles;
+};
+var HOME = os3.homedir();
+var IDE_PROJECTS_PATHS = {
+  vscode: path11.join(HOME, ".vscode", "projects"),
+  cursor: path11.join(HOME, ".cursor", "projects"),
+  claude: path11.join(HOME, ".claude", "projects")
+};
+function getWorkspaceStoragePath(ide) {
+  const platform = os3.platform();
+  const appName = "Cursor" ;
+  if (platform === "darwin") {
+    return path11.join(HOME, "Library", "Application Support", appName, "User", "workspaceStorage");
+  } else if (platform === "win32") {
+    return path11.join(process.env.APPDATA || "", appName, "User", "workspaceStorage");
+  } else {
+    return path11.join(HOME, ".config", appName, "User", "workspaceStorage");
+  }
+}
+function scanWorkspaceStorage(ide) {
+  const projects = [];
+  const storagePath = getWorkspaceStoragePath();
+  if (!fs4.existsSync(storagePath)) {
+    return projects;
+  }
+  try {
+    const workspaces = fs4.readdirSync(storagePath);
+    for (const workspace of workspaces) {
+      const workspaceJsonPath = path11.join(storagePath, workspace, "workspace.json");
+      if (fs4.existsSync(workspaceJsonPath)) {
+        try {
+          const content = fs4.readFileSync(workspaceJsonPath, "utf-8");
+          const data = JSON.parse(content);
+          if (data.folder && typeof data.folder === "string") {
+            let projectPath = data.folder;
+            if (projectPath.startsWith("file://")) {
+              projectPath = projectPath.replace("file://", "");
+              projectPath = decodeURIComponent(projectPath);
+            }
+            if (fs4.existsSync(projectPath) && fs4.statSync(projectPath).isDirectory()) {
+              projects.push({
+                name: path11.basename(projectPath),
+                path: projectPath,
+                type: ide
+              });
+            }
+          }
+        } catch (err) {
+          console.debug(`Error reading workspace.json at ${workspaceJsonPath}: ${err}`);
+        }
+      }
+    }
+  } catch (err) {
+    console.debug(`Error scanning workspaceStorage for ${ide}: ${err}`);
+  }
+  return projects;
+}
+var scanIdeProjects = async () => {
+  try {
+    const allProjects = [];
+    const seenPaths = /* @__PURE__ */ new Set();
+    const addProject = (projectPath, ide) => {
+      try {
+        const resolvedPath = fs4.realpathSync(projectPath);
+        if (fs4.existsSync(resolvedPath) && fs4.statSync(resolvedPath).isDirectory() && !seenPaths.has(resolvedPath)) {
+          const isIdeProjectsDir = Object.values(IDE_PROJECTS_PATHS).some((ideDir) => {
+            try {
+              return fs4.realpathSync(ideDir) === resolvedPath;
+            } catch {
+              return false;
+            }
+          });
+          if (!isIdeProjectsDir) {
+            seenPaths.add(resolvedPath);
+            allProjects.push({
+              name: path11.basename(resolvedPath),
+              path: resolvedPath,
+              type: ide
+            });
+          }
+        }
+      } catch {
+      }
+    };
+    const cursorProjects = scanWorkspaceStorage("cursor");
+    for (const project of cursorProjects) {
+      addProject(project.path, "cursor");
+    }
+    for (const [ide, dirPath] of Object.entries(IDE_PROJECTS_PATHS)) {
+      if (ide === "cursor") continue;
+      if (fs4.existsSync(dirPath)) {
+        const projects = fs4.readdirSync(dirPath);
+        projects.forEach((project) => {
+          const projectPath = path11.join(dirPath, project);
+          try {
+            const stats = fs4.lstatSync(projectPath);
+            let actualPath = null;
+            if (stats.isSymbolicLink()) {
+              actualPath = fs4.realpathSync(projectPath);
+            } else if (stats.isFile()) {
+              try {
+                let content = fs4.readFileSync(projectPath, "utf-8").trim();
+                if (content.startsWith("~/") || content === "~") {
+                  content = content.replace(/^~/, HOME);
+                }
+                const resolvedContent = path11.isAbsolute(content) ? content : path11.resolve(path11.dirname(projectPath), content);
+                if (fs4.existsSync(resolvedContent) && fs4.statSync(resolvedContent).isDirectory()) {
+                  actualPath = fs4.realpathSync(resolvedContent);
+                }
+              } catch {
+                return;
+              }
+            } else if (stats.isDirectory()) {
+              actualPath = fs4.realpathSync(projectPath);
+            }
+            if (actualPath) {
+              addProject(actualPath, ide);
+            }
+          } catch {
+          }
+        });
+      }
+    }
+    return allProjects;
+  } catch (error) {
+    console.debug(`Error scanning IDE projects: ${error}`);
+    return [];
+  }
+};
+
+// src/lib/rpc-handlers.ts
+var rpcHandlers = {
+  "daemon:get_workspace_folders": async () => {
+    const projects = projectRegistry.list();
+    return {
+      folders: projects.map((p) => ({
+        id: p.id,
+        cwd: p.cwd,
+        name: p.name,
+        active: p.active
+      }))
+    };
+  },
+  "daemon:get_environment": async ({ gitRepositoryUrl }) => {
+    const projects = projectRegistry.list();
+    if (projects.length === 0) {
+      const error = {
+        _tag: "ProjectUnlinkedError",
+        message: `Getting a local project by git repository URL "${gitRepositoryUrl}" returned an unlinked project. Please link it by running \`amai link <project name> <path to project directory>\``
+      };
+      throw error;
+    }
+    return {
+      project: projects[0],
+      env: {
+        platform: process.platform,
+        arch: process.arch,
+        nodeVersion: process.version
+      }
+    };
+  },
+  "daemon:get_context": async ({ cwd }) => {
+    try {
+      const files = getContext(cwd);
+      return { files, cwd };
+    } catch (error) {
+      const rpcError = {
+        _tag: "ContextError",
+        message: error.message || "Failed to get context"
+      };
+      throw rpcError;
+    }
+  },
+  "daemon:get_ide_projects": async () => {
+    const projects = await scanIdeProjects();
+    return { projects };
+  },
+  "daemon:register_project": async ({ projectId, cwd, name }) => {
+    if (!projectId || !cwd) {
+      const error = {
+        _tag: "ValidationError",
+        message: "projectId and cwd are required"
+      };
+      throw error;
+    }
+    projectRegistry.register(projectId, cwd, name);
+    return { success: true, projectId, cwd };
+  },
+  "daemon:unregister_project": async ({ projectId }) => {
+    if (!projectId) {
+      const error = {
+        _tag: "ValidationError",
+        message: "projectId is required"
+      };
+      throw error;
+    }
+    projectRegistry.unregister(projectId);
+    return { success: true, projectId };
+  },
+  "daemon:get_project": async ({ projectId }) => {
+    const project = projectRegistry.getProject(projectId);
+    if (!project) {
+      const error = {
+        _tag: "ProjectNotFoundError",
+        message: `Project not found: ${projectId}`
+      };
+      throw error;
+    }
+    return { project };
+  },
+  "daemon:list_projects": async () => {
+    const projects = projectRegistry.list();
+    return { projects };
+  },
+  "daemon:status": async () => {
+    return {
+      connected: true,
+      timestamp: Date.now(),
+      platform: process.platform,
+      arch: process.arch
+    };
+  }
+};
+var reconnectTimeout = null;
+var connectToUserStreams = async (serverUrl) => {
+  const userId = getUserId();
+  if (!userId?.userId) {
+    throw new Error("User ID not found");
+  }
+  const params = new URLSearchParams({
+    userId: userId.userId,
+    type: "cli"
+  });
+  const tokens = getTokens();
+  if (!tokens) {
+    throw new Error("No tokens found");
+  }
+  const wsUrl = `${serverUrl}/api/v1/user-streams?${params.toString()}`;
+  const ws = new WebSocket(wsUrl, {
+    headers: {
+      Authorization: `Bearer ${tokens.access_token}`
+    }
+  });
+  ws.on("open", () => {
+    console.log(pc2.green("CLI connected to user-streams"));
+    if (reconnectTimeout) {
+      clearTimeout(reconnectTimeout);
+      reconnectTimeout = null;
+    }
+  });
+  ws.on("message", async (event) => {
+    try {
+      const message = JSON.parse(event.toString());
+      if (message._tag === "rpc_call") {
+        const { requestId, method, input } = message;
+        console.log(pc2.gray(`RPC call: ${method}`));
+        const handler = rpcHandlers[method];
+        if (!handler) {
+          ws.send(JSON.stringify({
+            _tag: "rpc_result",
+            requestId,
+            data: {
+              _tag: "UnknownMethodError",
+              message: `Unknown RPC method: ${method}`
+            }
+          }));
+          console.log(pc2.yellow(`Unknown RPC method: ${method}`));
+          return;
+        }
+        try {
+          const result = await handler(input || {});
+          ws.send(JSON.stringify({
+            _tag: "rpc_result",
+            requestId,
+            data: result
+          }));
+          console.log(pc2.green(`RPC completed: ${method}`));
+        } catch (error) {
+          const rpcError = error._tag ? error : {
+            _tag: "RpcError",
+            message: error.message || String(error)
+          };
+          ws.send(JSON.stringify({
+            _tag: "rpc_result",
+            requestId,
+            data: rpcError
+          }));
+          console.log(pc2.red(`RPC failed: ${method} - ${rpcError.message}`));
+        }
+        return;
+      }
+      if (message.type === "presence_request") {
+        if (message.status === "connected") {
+          ws.send(JSON.stringify({
+            type: "presence_request",
+            status: "connected"
+          }));
+        }
+        if (message.status === "disconnected") {
+          ws.send(JSON.stringify({
+            type: "presence_request",
+            status: "disconnected"
+          }));
+        }
+      }
+    } catch (parseError) {
+      console.error(pc2.red(`Failed to parse message: ${parseError}`));
+    }
+  });
+  ws.on("close", (code, reason) => {
+    console.log(pc2.yellow(`CLI disconnected from user-streams (code: ${code})`));
+    console.log(pc2.gray("Reconnecting in 5 seconds..."));
+    reconnectTimeout = setTimeout(() => {
+      connectToUserStreams(serverUrl).catch((err) => {
+        console.error(pc2.red(`Reconnection failed: ${err.message}`));
+      });
+    }, 5e3);
+  });
+  ws.on("error", (error) => {
+    console.error(pc2.red(`User streams WebSocket error: ${error.message}`));
+  });
+  return ws;
+};
 
 // src/server.ts
-var statusEmitter = new EventEmitter();
 var toolExecutors = {
   editFile: editFiles,
   deleteFile,
@@ -1754,10 +1897,7 @@ var toolExecutors = {
   stringReplace: apply_patch,
   runTerminalCommand
 };
-function getConnectionStatus(ws) {
-  return ws.readyState === WebSocket.CONNECTING ? "connecting" : ws.readyState === WebSocket.OPEN ? "open" : ws.readyState === WebSocket.CLOSING ? "closing" : "closed";
-}
-function connectToServer2(serverUrl = DEFAULT_SERVER_URL) {
+function connectToServer(serverUrl = DEFAULT_SERVER_URL) {
   const tokens = getTokens();
   if (!tokens) {
     throw new Error("No tokens found");
@@ -1770,7 +1910,6 @@ function connectToServer2(serverUrl = DEFAULT_SERVER_URL) {
   });
   ws.on("open", () => {
     console.log(pc2.green("Connected to server agent streams"));
-    statusEmitter.emit("status", { connected: true });
   });
   ws.on("message", async (data) => {
     const message = JSON.parse(data.toString());
@@ -1800,12 +1939,10 @@ function connectToServer2(serverUrl = DEFAULT_SERVER_URL) {
   });
   ws.on("close", () => {
     console.log(pc2.red("Disconnected from server. Reconnecting in 5s..."));
-    statusEmitter.emit("status", { connected: false });
-    setTimeout(() => connectToServer2(serverUrl), 5e3);
+    setTimeout(() => connectToServer(serverUrl), 5e3);
   });
   ws.on("error", (error) => {
     console.error(pc2.red(`WebSocket error: ${error.message}`));
-    statusEmitter.emit("status", { connected: false });
   });
   return ws;
 }
@@ -1813,8 +1950,9 @@ async function main() {
   const serverUrl = DEFAULT_SERVER_URL;
   console.log(pc2.green("Starting local amai..."));
   console.log(pc2.gray(`Connecting to server at ${serverUrl}`));
-  const connection = connectToServer2(serverUrl);
-  startHttpServer(connection);
+  connectToServer(serverUrl);
+  await connectToUserStreams(serverUrl);
+  startHttpServer();
 }
 var execAsync2 = promisify(exec);
 var CODE_SERVER_VERSION = "4.96.4";
@@ -1843,27 +1981,27 @@ function getDownloadUrl() {
 }
 function getCodeServerDir() {
   const { platform, arch } = getPlatformInfo();
-  return path9.join(CODE_DIR, `code-server-${CODE_SERVER_VERSION}-${platform}-${arch}`);
+  return path11.join(CODE_DIR, `code-server-${CODE_SERVER_VERSION}-${platform}-${arch}`);
 }
 function getCodeServerBin() {
-  return path9.join(getCodeServerDir(), "bin", "code-server");
+  return path11.join(getCodeServerDir(), "bin", "code-server");
 }
 function isCodeServerInstalled() {
   const binPath = getCodeServerBin();
-  return fs3.existsSync(binPath);
+  return fs4.existsSync(binPath);
 }
 async function installCodeServer() {
   const { ext } = getPlatformInfo();
   const downloadUrl = getDownloadUrl();
-  const tarballPath = path9.join(AMA_DIR, `code-server.${ext}`);
-  if (!fs3.existsSync(AMA_DIR)) {
-    fs3.mkdirSync(AMA_DIR, { recursive: true });
+  const tarballPath = path11.join(AMA_DIR, `code-server.${ext}`);
+  if (!fs4.existsSync(AMA_DIR)) {
+    fs4.mkdirSync(AMA_DIR, { recursive: true });
   }
-  if (!fs3.existsSync(CODE_DIR)) {
-    fs3.mkdirSync(CODE_DIR, { recursive: true });
+  if (!fs4.existsSync(CODE_DIR)) {
+    fs4.mkdirSync(CODE_DIR, { recursive: true });
   }
-  if (!fs3.existsSync(STORAGE_DIR)) {
-    fs3.mkdirSync(STORAGE_DIR, { recursive: true });
+  if (!fs4.existsSync(STORAGE_DIR)) {
+    fs4.mkdirSync(STORAGE_DIR, { recursive: true });
   }
   console.log(pc2.cyan(`Downloading code-server v${CODE_SERVER_VERSION}...`));
   console.log(pc2.gray(downloadUrl));
@@ -1872,13 +2010,13 @@ async function installCodeServer() {
     throw new Error(`Failed to download code-server: ${response.statusText}`);
   }
   const buffer = await response.arrayBuffer();
-  await fs3.promises.writeFile(tarballPath, Buffer.from(buffer));
+  await fs4.promises.writeFile(tarballPath, Buffer.from(buffer));
   console.log(pc2.cyan("Extracting code-server..."));
   await execAsync2(`tar -xzf ${tarballPath} -C ${CODE_DIR}`);
-  await fs3.promises.unlink(tarballPath);
+  await fs4.promises.unlink(tarballPath);
   const binPath = getCodeServerBin();
-  if (fs3.existsSync(binPath)) {
-    await fs3.promises.chmod(binPath, 493);
+  if (fs4.existsSync(binPath)) {
+    await fs4.promises.chmod(binPath, 493);
   }
   console.log(pc2.green("\u2713 code-server installed successfully"));
 }
@@ -1903,19 +2041,19 @@ async function killExistingCodeServer() {
 async function startCodeServer(cwd) {
   const binPath = getCodeServerBin();
   const workDir = cwd || process.cwd();
-  if (!fs3.existsSync(binPath)) {
+  if (!fs4.existsSync(binPath)) {
     throw new Error("code-server is not installed. Run installCodeServer() first.");
   }
   await killExistingCodeServer();
-  const workspaceStoragePath = path9.join(STORAGE_DIR, "User", "workspaceStorage");
-  path9.join(STORAGE_DIR, "User", "globalStorage");
+  const workspaceStoragePath = path11.join(STORAGE_DIR, "User", "workspaceStorage");
+  path11.join(STORAGE_DIR, "User", "globalStorage");
   try {
-    if (fs3.existsSync(workspaceStoragePath)) {
-      await fs3.promises.rm(workspaceStoragePath, { recursive: true, force: true });
+    if (fs4.existsSync(workspaceStoragePath)) {
+      await fs4.promises.rm(workspaceStoragePath, { recursive: true, force: true });
     }
-    const stateDbPath = path9.join(STORAGE_DIR, "User", "globalStorage", "state.vscdb");
-    if (fs3.existsSync(stateDbPath)) {
-      await fs3.promises.unlink(stateDbPath);
+    const stateDbPath = path11.join(STORAGE_DIR, "User", "globalStorage", "state.vscdb");
+    if (fs4.existsSync(stateDbPath)) {
+      await fs4.promises.unlink(stateDbPath);
     }
   } catch {
   }
@@ -1955,7 +2093,7 @@ if (process.env.AMA_DAEMON === "1") {
       }
       if (isCodeServerInstalled()) {
         try {
-          const projectDir = process.cwd() || os2.homedir();
+          const projectDir = process.cwd() || os3.homedir();
           await startCodeServer(projectDir);
         } catch (error) {
           console.error(pc2.red(`Failed to start code-server: ${error.message}`));