@lasso-ai/cli 1.0.29 → 1.1.1

package/dist/cli/bridge.js

@@ -181,6 +181,34 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
     let activeAgentController = null;
     const taskControllers = new Map();
     let localAgents = new Set();
+    const pageIgnoredDirs = new Set(["node_modules", ".git", ".next", "dist", "build", ".turbo", "out", "coverage", ".vercel", ".cache"]);
+    const resolveInsideProject = (root, relative) => {
+        const target = node_path_1.default.resolve(root, relative && relative !== "." ? relative : "");
+        return target === root || target.startsWith(`${root}${node_path_1.default.sep}`) ? target : null;
+    };
+    // Directories are returned first so the overlay can render an explorer-style
+    // tree that mirrors how VS Code orders folders above files.
+    const pageEntries = (root, relative) => {
+        const directory = resolveInsideProject(root, relative);
+        if (!directory)
+            return [];
+        let entries = [];
+        try {
+            for (const entry of node_fs_1.default.readdirSync(directory, { withFileTypes: true })) {
+                if (entry.name.startsWith(".") || (entry.isDirectory() && pageIgnoredDirs.has(entry.name)))
+                    continue;
+                entries.push({
+                    name: entry.name,
+                    path: relative && relative !== "." ? node_path_1.default.join(relative, entry.name) : entry.name,
+                    type: entry.isDirectory() ? "directory" : "file",
+                });
+            }
+        }
+        catch {
+            // Folders that cannot be read are treated as empty.
+        }
+        return entries.sort((a, b) => a.type === b.type ? a.name.localeCompare(b.name) : a.type === "directory" ? -1 : 1);
+    };
     bridgeServer.on("request", (req, res) => {
         if (req.method !== "POST" || req.url?.split("?", 1)[0] !== "/__lasso/bridge/restart")
             return;
@@ -284,6 +312,10 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
                 if (!controller.signal.aborted && socket.readyState === socket.OPEN) {
                     socket.send(JSON.stringify({ type: "agent_status", taskId: request.taskId, status: "working", message, detail }));
                 }
+            }, (prompt) => {
+                if (!controller.signal.aborted && socket.readyState === socket.OPEN) {
+                    socket.send(JSON.stringify({ type: "agent_prompt", taskId: request.taskId, prompt }));
+                }
             })
                 .then((proposal) => {
                 if (controller.signal.aborted || socket.readyState !== socket.OPEN)
@@ -319,6 +351,69 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
             else {
                 console.log("[lasso] received from overlay:", msg);
             }
+            if (msg.type === "list_page_folders") {
+                const requested = msg.path || ".";
+                const root = node_path_1.default.resolve(cwd);
+                const directory = resolveInsideProject(root, requested);
+                if (!directory) {
+                    socket.send(JSON.stringify({ type: "page_folders", path: requested, entries: [], error: "That folder is outside the current project." }));
+                    return;
+                }
+                socket.send(JSON.stringify({ type: "page_folders", path: requested, entries: pageEntries(root, requested), project: node_path_1.default.basename(root) }));
+                return;
+            }
+            if (msg.type === "create_page_folder") {
+                const root = node_path_1.default.resolve(cwd);
+                const name = node_path_1.default.basename(msg.name || "");
+                const parent = resolveInsideProject(root, msg.parent || ".");
+                if (!parent || !name || name !== msg.name || name === "." || name === "..") {
+                    socket.send(JSON.stringify({ type: "page_folder_created", path: msg.name || "", error: "Enter a folder name without slashes or dots." }));
+                    return;
+                }
+                const target = node_path_1.default.resolve(parent, name);
+                if (!target.startsWith(`${root}${node_path_1.default.sep}`)) {
+                    socket.send(JSON.stringify({ type: "page_folder_created", path: name, error: "Choose a folder inside the current project." }));
+                    return;
+                }
+                try {
+                    node_fs_1.default.mkdirSync(target);
+                    socket.send(JSON.stringify({ type: "page_folder_created", path: node_path_1.default.relative(root, target) }));
+                }
+                catch (error) {
+                    socket.send(JSON.stringify({
+                        type: "page_folder_created",
+                        path: node_path_1.default.relative(root, target),
+                        error: error instanceof Error && error.message.includes("EEXIST") ? "A folder with that name already exists." : "The folder could not be created.",
+                    }));
+                }
+                return;
+            }
+            if (msg.type === "create_page") {
+                const root = node_path_1.default.resolve(cwd);
+                const folder = resolveInsideProject(root, msg.folder || ".");
+                // A page name may contain "/" so the overlay can create nested routes in
+                // one step, but every segment is validated and resolved inside the project.
+                const segments = (msg.fileName || "").split("/").map((segment) => segment.trim()).filter(Boolean);
+                const fileName = segments.pop() || "";
+                if (!folder || !fileName || segments.some((segment) => segment === "." || segment === "..") || fileName === "." || fileName === "..") {
+                    socket.send(JSON.stringify({ type: "page_created", path: msg.fileName || "", error: "Choose a valid project folder and file name." }));
+                    return;
+                }
+                const target = node_path_1.default.resolve(folder, ...segments, fileName);
+                if (!target.startsWith(`${root}${node_path_1.default.sep}`)) {
+                    socket.send(JSON.stringify({ type: "page_created", path: msg.fileName, error: "Choose a valid project folder and file name." }));
+                    return;
+                }
+                try {
+                    node_fs_1.default.mkdirSync(node_path_1.default.dirname(target), { recursive: true });
+                    node_fs_1.default.writeFileSync(target, msg.content.slice(0, 200000), { encoding: "utf8", flag: "wx" });
+                    socket.send(JSON.stringify({ type: "page_created", path: node_path_1.default.relative(root, target) }));
+                }
+                catch (error) {
+                    socket.send(JSON.stringify({ type: "page_created", path: node_path_1.default.relative(root, target), error: error instanceof Error && error.message.includes("EEXIST") ? "A file with that name already exists." : "The page could not be created." }));
+                }
+                return;
+            }
             if (msg.type === "ask") {
                 const localProvider = msg.provider === "cli";
                 const cliProvider = msg.model.startsWith("claude-code:") ? "claude-code" : msg.model.startsWith("opencode:") ? "opencode" : "codex";
@@ -337,9 +432,12 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
                     ? { provider: cliProvider, model: msg.model }
                     : { ...agentConfig, provider: (msg.provider || agentConfig.provider), model: msg.model };
                 socket.send(JSON.stringify({ type: "agent_status", taskId: msg.taskId, status: "working", message: "Working…" }));
-                void (0, agent_1.answerQuestion)(cwd, { question: msg.question, context: msg.context, element: msg.element, messages: msg.messages }, selectedConfig, controller.signal, (message, detail) => {
+                void (0, agent_1.answerQuestion)(cwd, { taskId: msg.taskId, question: msg.question, context: msg.context, element: msg.element, messages: msg.messages }, selectedConfig, controller.signal, (message, detail) => {
                     if (!controller.signal.aborted && socket.readyState === socket.OPEN)
                         socket.send(JSON.stringify({ type: "agent_status", taskId: msg.taskId, status: "working", message, detail }));
+                }, (prompt) => {
+                    if (!controller.signal.aborted && socket.readyState === socket.OPEN)
+                        socket.send(JSON.stringify({ type: "agent_prompt", taskId: msg.taskId, prompt }));
                 }).then((answer) => {
                     if (!controller.signal.aborted)
                         socket.send(JSON.stringify({ type: "assistant_message", taskId: msg.taskId, message: answer }));
@@ -370,6 +468,9 @@ function startBridge(cwd = process.cwd(), collabConfig = null, bridgePort = expo
                 reviewRefreshAttempts.set(msg.taskId, 0);
                 runEditReview(msg, selectedConfig);
             }
+            else if (msg.type === "agent_prompt_response") {
+                (0, agent_1.respondToAgentPrompt)(msg.taskId, msg.response);
+            }
             else if (msg.type === "stop") {
                 if (msg.taskId)
                     taskControllers.get(msg.taskId)?.abort();