@punkcode/cli 0.1.9 → 0.1.11

package/dist/cli.js

@@ -320,6 +320,10 @@ function runClaude(options, callbacks) {
   };
 }
 
+// src/commands/connect.ts
+import fs3 from "fs";
+import os3 from "os";
+
 // src/lib/device-info.ts
 import os from "os";
 import path from "path";
@@ -345,7 +349,15 @@ function getOrCreateDeviceId() {
   fs.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2) + "\n", "utf-8");
   return id;
 }
-function collectDeviceInfo(deviceId, customName, customTags) {
+function getDefaultWorkingDirectory() {
+  try {
+    const config = JSON.parse(fs.readFileSync(CONFIG_FILE, "utf-8"));
+    if (config.defaultWorkingDirectory) return config.defaultWorkingDirectory;
+  } catch {
+  }
+  return path.join(os.homedir(), "punk");
+}
+function collectDeviceInfo(deviceId, customName, customTags, defaultCwd) {
   if (customName) {
     saveConfigField("deviceName", customName);
   }
@@ -361,7 +373,7 @@ function collectDeviceInfo(deviceId, customName, customTags) {
     arch: process.arch,
     username: os.userInfo().username,
     timezone: Intl.DateTimeFormat().resolvedOptions().timeZone,
-    defaultWorkingDirectory: process.cwd(),
+    defaultWorkingDirectory: defaultCwd || getDefaultWorkingDirectory(),
     model: getModel(),
     cpuModel: cpus.length > 0 ? cpus[0].model : "Unknown",
     memoryGB: Math.round(os.totalmem() / 1024 ** 3),
@@ -893,6 +905,84 @@ function escapeRegex(str) {
   return str.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
 }
 
+// src/lib/directory-discovery.ts
+import { query as query2 } from "@anthropic-ai/claude-agent-sdk";
+async function findProjectDirectory(description, searchRoot, rejections) {
+  let prompt2 = `Find up to 3 project directories on this machine that best match: "${description}"
+
+Search starting from ${searchRoot}. Rank by relevance. Include a brief reason for each match.`;
+  if (rejections?.length) {
+    prompt2 += "\n\nThe user rejected these previous suggestions:";
+    for (const r of rejections) {
+      prompt2 += `
+- ${r.path}${r.feedback ? ` (user said: "${r.feedback}")` : ""}`;
+    }
+    prompt2 += "\n\nFind different matches based on their feedback.";
+  }
+  const q = query2({
+    prompt: prompt2,
+    options: {
+      systemPrompt: {
+        type: "preset",
+        preset: "claude_code",
+        append: "IMPORTANT: When searching for directories, always try listing likely parent directories with ls FIRST (e.g. ls ~/github, ls ~/projects). Only use find as a last resort, and always with -maxdepth 3. Never scan the entire home directory."
+      },
+      permissionMode: "bypassPermissions",
+      persistSession: false,
+      cwd: searchRoot,
+      outputFormat: {
+        type: "json_schema",
+        schema: {
+          type: "object",
+          properties: {
+            suggestions: {
+              type: "array",
+              items: {
+                type: "object",
+                properties: {
+                  path: { type: "string", description: "Absolute path to the project directory" },
+                  name: { type: "string", description: "Human-readable project name" },
+                  reason: { type: "string", description: 'Brief reason why this matches (e.g. "Direct match under github folder")' }
+                },
+                required: ["path", "name", "reason"]
+              }
+            }
+          },
+          required: ["suggestions"]
+        }
+      }
+    }
+  });
+  try {
+    for await (const msg of q) {
+      if (msg.type === "result") {
+        const resultMsg = msg;
+        if (resultMsg.subtype === "success") {
+          const structured = resultMsg.structured_output;
+          if (structured?.suggestions?.length) {
+            logger.info({ count: structured.suggestions.length }, "Project directories found (structured)");
+            return structured.suggestions;
+          }
+          const result = resultMsg.result?.trim();
+          if (result && result !== "null" && result.startsWith("/")) {
+            const path3 = result.split("\n")[0].trim();
+            const name = path3.split("/").pop() ?? path3;
+            logger.info({ path: path3, name }, "Project directory found (text fallback)");
+            return [{ path: path3, name, reason: "Best match" }];
+          }
+          logger.info("No matching directories found");
+          return [];
+        }
+        logger.warn({ subtype: resultMsg.subtype }, "Directory search query failed");
+        return [];
+      }
+    }
+  } finally {
+    q.close();
+  }
+  return [];
+}
+
 // src/lib/auth.ts
 import fs2 from "fs";
 import path2 from "path";
@@ -1037,6 +1127,8 @@ async function connect(server, options) {
   }
   logger.info("All checks passed");
   const deviceId = options.deviceId || getOrCreateDeviceId();
+  const defaultCwd = options.cwd || getDefaultWorkingDirectory();
+  fs3.mkdirSync(defaultCwd, { recursive: true });
   const url = buildUrl(server);
   let idToken;
   if (options.token) {
@@ -1053,7 +1145,9 @@ async function connect(server, options) {
   const socket = io(url, {
     path: "/socket.io",
     transports: ["websocket"],
-    auth: { token: idToken },
+    auth: (cb) => {
+      refreshIdToken().then((token) => cb({ token })).catch(() => cb({ token: idToken }));
+    },
     reconnection: true,
     reconnectionAttempts: Infinity,
     reconnectionDelay: 1e3,
@@ -1069,7 +1163,7 @@ async function connect(server, options) {
   }, 3e4);
   socket.on("connect", () => {
     logger.info("Connected");
-    const deviceInfo = collectDeviceInfo(deviceId, options.name, options.tag);
+    const deviceInfo = collectDeviceInfo(deviceId, options.name, options.tag, defaultCwd);
     socket.emit("register", deviceInfo, (response) => {
       if (response.success) {
         logger.info({ deviceId }, "Registered");
@@ -1088,12 +1182,12 @@ async function connect(server, options) {
   });
   socket.on("list-sessions", async (msg) => {
     if (msg.type === "list-sessions") {
-      handleListSessions(socket, msg);
+      handleListSessions(socket, msg, defaultCwd);
     }
   });
   socket.on("get-context", (msg) => {
     if (msg.type === "get-context") {
-      handleGetContext(socket, msg);
+      handleGetContext(socket, msg, defaultCwd);
     }
   });
   socket.on("get-commands", (msg) => {
@@ -1101,6 +1195,11 @@ async function connect(server, options) {
       handleGetCommands(socket, msg);
     }
   });
+  socket.on("find-project", (msg) => {
+    if (msg.type === "find-project") {
+      handleFindProject(socket, msg, defaultCwd);
+    }
+  });
   socket.on("cancel", (msg) => {
     handleCancel(msg.id, activeSessions);
   });
@@ -1117,14 +1216,10 @@ async function connect(server, options) {
   });
   socket.io.on("reconnect_attempt", () => {
     logger.info("Reconnecting...");
-    refreshIdToken().then((token) => {
-      socket.auth = { token };
-    }).catch(() => {
-    });
   });
   socket.on("reconnect", (attemptNumber) => {
     logger.info({ attemptNumber }, "Reconnected");
-    socket.emit("register", collectDeviceInfo(deviceId, options.name, options.tag));
+    socket.emit("register", collectDeviceInfo(deviceId, options.name, options.tag, defaultCwd));
   });
   socket.on("connect_error", (err) => {
     const { reason, ...detail } = formatConnectionError(err);
@@ -1270,9 +1365,9 @@ function handleCancel(id, activeSessions) {
     logger.info({ sessionId: id }, "Session cancelled");
   }
 }
-async function handleListSessions(socket, msg) {
+async function handleListSessions(socket, msg, defaultCwd) {
   const { id } = msg;
-  const workingDirectory = msg.workingDirectory ?? process.cwd();
+  const workingDirectory = msg.workingDirectory ?? defaultCwd;
   logger.info("Listing sessions...");
   const sessions = await listSessions(workingDirectory);
   send(socket, "response", { type: "sessions_list", sessions, requestId: id });
@@ -1305,8 +1400,24 @@ async function handleGetCommands(socket, msg) {
     log2.error({ err }, "Commands error");
   }
 }
-async function handleGetContext(socket, msg) {
-  const { id, sessionId, workingDirectory } = msg;
+async function handleFindProject(socket, msg, _defaultCwd) {
+  const { id, description, rootDirectory, rejections } = msg;
+  const searchRoot = rootDirectory ?? os3.homedir();
+  const log2 = createChildLogger({ requestId: id });
+  log2.info({ description, searchRoot }, "Finding project directory...");
+  try {
+    const suggestions = await findProjectDirectory(description, searchRoot, rejections);
+    send(socket, "response", { type: "project_suggestions", suggestions, requestId: id });
+    log2.info({ count: suggestions.length }, "Project suggestions sent");
+  } catch (err) {
+    const message = err instanceof Error ? err.message : String(err);
+    send(socket, "response", { type: "error", message, requestId: id });
+    log2.error({ err }, "Find project error");
+  }
+}
+async function handleGetContext(socket, msg, defaultCwd) {
+  const { id, sessionId } = msg;
+  const workingDirectory = msg.workingDirectory ?? defaultCwd;
   const log2 = createChildLogger({ sessionId });
   log2.info("Getting context...");
   try {
@@ -1392,7 +1503,7 @@ function logout() {
 
 // src/commands/index.ts
 function registerCommands(program2) {
-  program2.command("connect").argument("[server]", "Backend server URL", "https://api.punkcode.dev").description("Connect to backend server").option("-t, --token <token>", "Authentication token").option("-d, --device-id <deviceId>", "Device identifier (defaults to hostname)").option("-n, --name <name>", "Custom device display name").option("--tag <tag>", "Device tag (repeatable, e.g. --tag home --tag mac --tag docker)", (val, acc) => [...acc, val], []).action(connect);
+  program2.command("connect").argument("[server]", "Backend server URL", "https://api.punkcode.dev").description("Connect to backend server").option("-t, --token <token>", "Authentication token").option("-d, --device-id <deviceId>", "Device identifier (defaults to hostname)").option("-n, --name <name>", "Custom device display name").option("--tag <tag>", "Device tag (repeatable, e.g. --tag home --tag mac --tag docker)", (val, acc) => [...acc, val], []).option("--cwd <directory>", "Working directory for sessions (default: ~/punk)").action(connect);
   program2.command("login").description("Log in with your email and password").action(login);
   program2.command("logout").description("Log out and clear stored credentials").action(logout);
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@punkcode/cli",
-  "version": "0.1.9",
+  "version": "0.1.11",
   "description": "Control Claude Code from your phone",
   "type": "module",
   "bin": {