@punkcode/cli 0.1.4 → 0.1.6

package/dist/cli.js

@@ -92,6 +92,7 @@ function runClaude(options, callbacks) {
     };
   }
   (async () => {
+    let sentCompactSummary = false;
     try {
       for await (const message of q) {
         switch (message.type) {
@@ -118,7 +119,19 @@ function runClaude(options, callbacks) {
           }
           case "user": {
             const userContent = message.message?.content;
-            if (Array.isArray(userContent)) {
+            if (typeof userContent === "string") {
+              const match = userContent.match(/<local-command-stdout>([\s\S]*?)<\/local-command-stdout>/);
+              if (match) {
+                if (sentCompactSummary && match[1].trim() === "Compacted") {
+                  sentCompactSummary = false;
+                } else {
+                  callbacks.onSlashCommandOutput?.(match[1]);
+                }
+              } else if (userContent.startsWith("This session is being continued")) {
+                sentCompactSummary = true;
+                callbacks.onSlashCommandOutput?.(userContent);
+              }
+            } else if (Array.isArray(userContent)) {
               for (const block of userContent) {
                 if (typeof block === "object" && block !== null && "type" in block && block.type === "tool_result") {
                   const tr = block;
@@ -132,10 +145,35 @@ function runClaude(options, callbacks) {
             }
             break;
           }
-          case "result": {
-            callbacks.onResult(message.session_id);
+          case "system": {
+            const sys = message;
+            if (sys.subtype === "init" && callbacks.onSessionCreated) {
+              callbacks.onSessionCreated({
+                sessionId: sys.session_id ?? "",
+                tools: sys.tools ?? [],
+                slashCommands: (sys.slash_commands ?? []).map((cmd) => ({ name: cmd, description: "" })),
+                skills: sys.skills ?? [],
+                mcpServers: sys.mcp_servers ?? [],
+                model: sys.model ?? "",
+                cwd: sys.cwd ?? "",
+                claudeCodeVersion: sys.claude_code_version ?? "",
+                permissionMode: sys.permissionMode ?? "default"
+              });
+            }
             break;
           }
+          case "tool_use_summary": {
+            const summary = message.summary;
+            if (summary) {
+              callbacks.onSlashCommandOutput?.(summary);
+            }
+            break;
+          }
+          case "result": {
+            const resultText = message.subtype === "success" ? message.result : void 0;
+            callbacks.onResult(message.session_id, resultText);
+            return;
+          }
         }
       }
     } catch (err) {
@@ -180,11 +218,14 @@ function getOrCreateDeviceId() {
   fs.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2) + "\n", "utf-8");
   return id;
 }
-function collectDeviceInfo(deviceId) {
+function collectDeviceInfo(deviceId, customName) {
+  if (customName) {
+    saveConfigField("deviceName", customName);
+  }
   const cpus = os.cpus();
   return {
     deviceId,
-    name: getDeviceName(),
+    name: customName || getDeviceName(),
     platform: process.platform,
     arch: process.arch,
     username: os.userInfo().username,
@@ -198,6 +239,11 @@ function collectDeviceInfo(deviceId) {
   };
 }
 function getDeviceName() {
+  try {
+    const config = JSON.parse(fs.readFileSync(CONFIG_FILE, "utf-8"));
+    if (config.deviceName) return config.deviceName;
+  } catch {
+  }
   if (process.platform === "darwin") {
     try {
       const { stdout } = execaSync("scutil", ["--get", "ComputerName"], { timeout: 3e3 });
@@ -208,6 +254,16 @@ function getDeviceName() {
   }
   return os.hostname();
 }
+function saveConfigField(key, value) {
+  fs.mkdirSync(PUNK_DIR, { recursive: true });
+  let config = {};
+  try {
+    config = JSON.parse(fs.readFileSync(CONFIG_FILE, "utf-8"));
+  } catch {
+  }
+  config[key] = value;
+  fs.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2) + "\n", "utf-8");
+}
 function parseBattery() {
   try {
     if (process.platform === "darwin") {
@@ -282,6 +338,9 @@ import { readdir, readFile, stat, open } from "fs/promises";
 import { join } from "path";
 import { homedir } from "os";
 var CLAUDE_DIR = join(homedir(), ".claude", "projects");
+function pathToProjectDir(dir) {
+  return dir.replace(/\//g, "-");
+}
 async function loadSession(sessionId) {
   const sessionFile = `${sessionId}.jsonl`;
   let projectDirs;
@@ -304,10 +363,14 @@ var UUID_RE = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i;
 function isValidSessionUUID(name) {
   return UUID_RE.test(name);
 }
-async function listSessions() {
+async function listSessions(workingDirectory) {
   let projectDirs;
   try {
-    projectDirs = await readdir(CLAUDE_DIR);
+    if (workingDirectory) {
+      projectDirs = [pathToProjectDir(workingDirectory)];
+    } else {
+      projectDirs = await readdir(CLAUDE_DIR);
+    }
   } catch {
     return [];
   }
@@ -335,16 +398,12 @@ async function listSessions() {
     candidates.map(async (c) => {
       try {
         const fileStat = await stat(c.filePath);
-        const [titleInfo, turnCount] = await Promise.all([
-          extractTitle(c.filePath),
-          countTurns(c.filePath)
-        ]);
+        const titleInfo = await extractTitle(c.filePath);
         return {
           sessionId: c.sessionId,
           project: c.project,
           title: titleInfo.title,
           lastModified: fileStat.mtimeMs,
-          turnCount,
           ...titleInfo.summary && { summary: titleInfo.summary }
         };
       } catch {
@@ -411,7 +470,9 @@ async function extractFirstUserMessage(fh) {
     try {
       const entry = JSON.parse(line);
       if (entry.isMeta || entry.parentUuid && metaUuids.has(entry.parentUuid)) {
-        if (entry.uuid) metaUuids.add(entry.uuid);
+        if (entry.uuid && entry.message?.role !== "assistant") {
+          metaUuids.add(entry.uuid);
+        }
         continue;
       }
       if (entry.type === "user" && entry.message?.role === "user") {
@@ -421,9 +482,9 @@ async function extractFirstUserMessage(fh) {
         }
         if (Array.isArray(content)) {
           const textBlock = content.find(
-            (b) => b.type === "text" && b.text && !b.text.startsWith("[Request interrupted")
+            (b) => b.type === "text" && "text" in b && b.text && !b.text.startsWith("[Request interrupted")
           );
-          if (textBlock?.text) {
+          if (textBlock && "text" in textBlock && typeof textBlock.text === "string") {
             return textBlock.text.slice(0, 100);
           }
         }
@@ -433,41 +494,9 @@ async function extractFirstUserMessage(fh) {
   }
   return "Untitled session";
 }
-async function countTurns(filePath) {
-  const content = await readFile(filePath, "utf-8");
-  const lines = content.split("\n");
-  let count = 0;
-  const metaUuids = /* @__PURE__ */ new Set();
-  for (const line of lines) {
-    if (!line.trim()) continue;
-    try {
-      const entry = JSON.parse(line);
-      if (entry.isMeta || entry.parentUuid && metaUuids.has(entry.parentUuid)) {
-        if (entry.uuid) metaUuids.add(entry.uuid);
-        continue;
-      }
-      if (entry.type !== "user" || !entry.message || entry.message.role !== "user") continue;
-      const msgContent = entry.message.content;
-      if (typeof msgContent === "string") {
-        const trimmed = msgContent.trim();
-        if (trimmed && !trimmed.startsWith("<") && !trimmed.startsWith("[") && !trimmed.startsWith("/")) {
-          count++;
-        }
-      } else if (Array.isArray(msgContent)) {
-        const textBlock = msgContent.find(
-          (b) => b.type === "text" && b.text
-        );
-        if (textBlock?.text) {
-          const text = textBlock.text.trim();
-          if (text && !text.startsWith("<") && !text.startsWith("[") && !text.startsWith("/")) {
-            count++;
-          }
-        }
-      }
-    } catch {
-    }
-  }
-  return count;
+var ANSI_RE = /\u001b\[\d*m/g;
+function stripAnsi(text) {
+  return text.replace(ANSI_RE, "");
 }
 function parseSessionFile(content) {
   const messages = [];
@@ -477,7 +506,51 @@ function parseSessionFile(content) {
     try {
       const entry = JSON.parse(line);
       if (entry.isMeta || entry.parentUuid && metaUuids.has(entry.parentUuid)) {
-        if (entry.uuid) metaUuids.add(entry.uuid);
+        if (entry.uuid && entry.message?.role !== "assistant") {
+          metaUuids.add(entry.uuid);
+        }
+        if (entry.message?.role === "user" && typeof entry.message.content === "string") {
+          const content2 = entry.message.content;
+          const cmdMatch = content2.match(/<command-name>\/(.+?)<\/command-name>/);
+          if (cmdMatch) {
+            messages.push({
+              role: "user",
+              content: [{ type: "text", text: `/${cmdMatch[1]}` }],
+              timestamp: entry.timestamp,
+              isMeta: entry.isMeta,
+              uuid: entry.uuid,
+              parentUuid: entry.parentUuid,
+              type: entry.type
+            });
+          }
+          const outMatch = content2.match(/<local-command-stdout>([\s\S]*?)<\/local-command-stdout>/);
+          if (outMatch && outMatch[1].trim()) {
+            const text = stripAnsi(outMatch[1].trim());
+            if (text) {
+              messages.push({
+                role: "assistant",
+                content: [{ type: "text", text }],
+                timestamp: entry.timestamp,
+                isMeta: entry.isMeta,
+                uuid: entry.uuid,
+                parentUuid: entry.parentUuid,
+                type: entry.type
+              });
+            }
+          }
+        }
+        continue;
+      }
+      if (entry.type === "system" && entry.subtype === "compact_boundary") {
+        messages.push({
+          role: "assistant",
+          content: [],
+          timestamp: entry.timestamp,
+          type: "system",
+          subtype: entry.subtype,
+          uuid: entry.uuid,
+          parentUuid: entry.parentUuid
+        });
         continue;
       }
       if ((entry.type === "user" || entry.type === "assistant") && entry.message) {
@@ -485,7 +558,10 @@ function parseSessionFile(content) {
         messages.push({
           role: entry.message.role,
           content: typeof msgContent === "string" ? [{ type: "text", text: msgContent }] : msgContent,
-          timestamp: entry.timestamp
+          timestamp: entry.timestamp,
+          uuid: entry.uuid,
+          parentUuid: entry.parentUuid,
+          type: entry.type
         });
       }
     } catch {
@@ -497,12 +573,25 @@ function parseSessionFile(content) {
     const blocks = msg.content;
     if (!Array.isArray(blocks)) continue;
     for (const block of blocks) {
-      if (block.type !== "tool_result" || !Array.isArray(block.content)) continue;
-      const imgBlock = block.content.find(
-        (b) => b?.type === "image" && b?.source?.type === "base64" && b?.source?.data
-      );
-      if (!imgBlock) continue;
-      const dataUri = `data:${imgBlock.source.media_type};base64,${imgBlock.source.data}`;
+      if (block.type !== "tool_result") continue;
+      const content2 = block.content;
+      let dataUri;
+      if (Array.isArray(content2)) {
+        const imgBlock = content2.find(
+          (b) => b?.type === "image" && b?.source?.type === "base64" && b?.source?.data
+        );
+        if (imgBlock) {
+          dataUri = `data:${imgBlock.source.media_type};base64,${imgBlock.source.data}`;
+        }
+      }
+      let resultText;
+      if (typeof content2 === "string") {
+        resultText = content2 || void 0;
+      } else if (Array.isArray(content2)) {
+        const texts = content2.filter((b) => b?.type === "text" && typeof b?.text === "string").map((b) => b.text);
+        resultText = texts.join("\n") || void 0;
+      }
+      if (!dataUri && !resultText) continue;
       for (let j = i - 1; j >= 0; j--) {
         if (messages[j].role !== "assistant") continue;
         const aBlocks = messages[j].content;
@@ -510,7 +599,10 @@ function parseSessionFile(content) {
         const toolUse = aBlocks.find(
           (b) => b.type === "tool_use" && b.id === block.tool_use_id
         );
-        if (toolUse) toolUse.imageUri = dataUri;
+        if (toolUse) {
+          if (dataUri) toolUse.imageUri = dataUri;
+          if (resultText) toolUse.result = resultText;
+        }
         break;
       }
     }
@@ -778,7 +870,7 @@ async function connect(server, options) {
   const activeSessions = /* @__PURE__ */ new Map();
   socket.on("connect", () => {
     logger.info("Connected");
-    const deviceInfo = collectDeviceInfo(deviceId);
+    const deviceInfo = collectDeviceInfo(deviceId, options.name);
     socket.emit("register", deviceInfo, (response) => {
       if (response.success) {
         logger.info({ deviceId }, "Registered");
@@ -826,7 +918,7 @@ async function connect(server, options) {
   });
   socket.on("reconnect", (attemptNumber) => {
     logger.info({ attemptNumber }, "Reconnected");
-    socket.emit("register", collectDeviceInfo(deviceId));
+    socket.emit("register", collectDeviceInfo(deviceId, options.name));
   });
   socket.on("connect_error", (err) => {
     logger.error({ err }, "Connection error");
@@ -870,9 +962,16 @@ function handlePrompt(socket, msg, activeSessions) {
   const handle = runClaude(
     { prompt: prompt2, sessionId, cwd, images, options },
     {
+      onSessionCreated: (info) => {
+        send(socket, "response", { type: "session_created", data: info, requestId: id });
+        log2.info({ sessionId: info.sessionId }, "New session created");
+      },
       onText: (text) => {
         send(socket, "response", { type: "text", text, requestId: id });
       },
+      onSlashCommandOutput: (output) => {
+        send(socket, "response", { type: "command_output", output, requestId: id });
+      },
       onThinking: (thinking) => {
         send(socket, "response", { type: "thinking", thinking, requestId: id });
       },
@@ -882,8 +981,8 @@ function handlePrompt(socket, msg, activeSessions) {
       onToolResult: (toolUseId, content, isError) => {
         send(socket, "response", { type: "tool_result", tool_use_id: toolUseId, content, is_error: isError, requestId: id });
       },
-      onResult: (sid) => {
-        send(socket, "response", { type: "result", session_id: sid, requestId: id });
+      onResult: (sid, result) => {
+        send(socket, "response", { type: "result", session_id: sid, ...result && { result }, requestId: id });
         activeSessions.delete(id);
         log2.info("Session done");
       },
@@ -917,8 +1016,9 @@ function handleCancel(id, activeSessions) {
 }
 async function handleListSessions(socket, msg) {
   const { id } = msg;
+  const workingDirectory = msg.workingDirectory ?? process.cwd();
   logger.info("Listing sessions...");
-  const sessions = await listSessions();
+  const sessions = await listSessions(workingDirectory);
   send(socket, "response", { type: "sessions_list", sessions, requestId: id });
   logger.info({ count: sessions.length }, "Listed sessions");
 }
@@ -1022,7 +1122,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)").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").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.4",
+  "version": "0.1.6",
   "description": "Control Claude Code from your phone",
   "type": "module",
   "bin": {