@zhigang1992/happy-cli 0.12.1 → 0.12.4

package/dist/{index-DsHtmQqP.mjs → index-BERBU6rR.mjs}

@@ -1,11 +1,15 @@
 import chalk from 'chalk';
 import os$1, { homedir } from 'node:os';
 import { randomUUID, randomBytes, createHmac } from 'node:crypto';
-import { l as logger, p as projectPath, j as backoff, k as delay, R as RawJSONLinesSchema, m as AsyncLock, c as configuration, n as readDaemonState, o as clearDaemonState, i as packageJson, r as readSettings, q as readCredentials, g as encodeBase64, u as updateSettings, s as encodeBase64Url, d as decodeBase64, w as writeCredentialsLegacy, t as writeCredentialsDataKey, v as acquireDaemonLock, x as writeDaemonState, A as ApiClient, y as releaseDaemonLock, z as authChallenge, B as clearCredentials, C as clearMachineId, D as getLatestDaemonLog } from './types-CGco5Y-r.mjs';
+import { l as logger, p as projectPath, j as backoff, k as delay, R as RawJSONLinesSchema, m as AsyncLock, c as configuration, n as readDaemonState, o as clearDaemonState, i as packageJson, r as readSettings, q as readCredentials, g as encodeBase64, u as updateSettings, s as encodeBase64Url, d as decodeBase64, w as writeCredentialsLegacy, t as writeCredentialsDataKey, v as acquireDaemonLock, x as writeDaemonState, A as ApiClient, y as releaseDaemonLock, z as authChallenge, B as clearCredentials, C as clearMachineId, D as getLatestDaemonLog } from './types-Cw6y7GyQ.mjs';
 import { spawn, execSync, execFileSync } from 'node:child_process';
 import { resolve, join } from 'node:path';
 import { createInterface } from 'node:readline';
 import { existsSync, readFileSync, mkdirSync, watch, readdirSync, statSync, rmSync } from 'node:fs';
+import { exec, spawn as spawn$1, execSync as execSync$1 } from 'child_process';
+import { promisify } from 'util';
+import { existsSync as existsSync$1, readFileSync as readFileSync$1, writeFileSync, chmodSync, unlinkSync } from 'fs';
+import { join as join$1, dirname, basename } from 'path';
 import { readFile } from 'node:fs/promises';
 import fs, { watch as watch$1, access } from 'fs/promises';
 import { useStdout, useInput, Box, Text, render } from 'ink';
@@ -17,9 +21,6 @@ import 'socket.io-client';
 import tweetnacl from 'tweetnacl';
 import 'expo-server-sdk';
 import { createHash, randomBytes as randomBytes$1 } from 'crypto';
-import { spawn as spawn$1, execSync as execSync$1, exec } from 'child_process';
-import { readFileSync as readFileSync$1, existsSync as existsSync$1, writeFileSync, chmodSync, unlinkSync } from 'fs';
-import { join as join$1 } from 'path';
 import psList from 'ps-list';
 import spawn$2 from 'cross-spawn';
 import os from 'os';
@@ -34,7 +35,6 @@ import { createServer } from 'node:http';
 import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
 import * as hex from '@stablelib/hex';
 import { createServer as createServer$1 } from 'http';
-import { promisify } from 'util';
 
 class Session {
   path;
@@ -218,6 +218,99 @@ const systemPrompt = (() => {
   }
 })();
 
+const execAsync$1 = promisify(exec);
+let direnvAvailable = null;
+async function isDirenvAvailable() {
+  if (direnvAvailable !== null) {
+    return direnvAvailable;
+  }
+  try {
+    await execAsync$1("direnv version", { timeout: 5e3 });
+    direnvAvailable = true;
+    logger.debug("[direnv] direnv is available");
+  } catch {
+    direnvAvailable = false;
+    logger.debug("[direnv] direnv is not available");
+  }
+  return direnvAvailable;
+}
+function findEnvrcDirectory(startDir) {
+  let dir = startDir;
+  const root = "/";
+  while (dir !== root) {
+    const envrcPath = join$1(dir, ".envrc");
+    if (existsSync$1(envrcPath)) {
+      logger.debug(`[direnv] Found .envrc at: ${envrcPath}`);
+      return dir;
+    }
+    const parent = dirname(dir);
+    if (parent === dir) {
+      break;
+    }
+    dir = parent;
+  }
+  logger.debug(`[direnv] No .envrc found in directory tree for: ${startDir}`);
+  return null;
+}
+async function loadDirenvEnvironment(cwd) {
+  if (!await isDirenvAvailable()) {
+    return {};
+  }
+  const envrcDir = findEnvrcDirectory(cwd);
+  if (!envrcDir) {
+    return {};
+  }
+  try {
+    logger.debug(`[direnv] Loading environment for: ${cwd}`);
+    const { stdout, stderr } = await execAsync$1("direnv export json", {
+      cwd,
+      timeout: 1e4,
+      // 10 second timeout
+      env: {
+        ...process.env,
+        // Ensure direnv doesn't prompt for allowance interactively
+        DIRENV_LOG_FORMAT: ""
+      }
+    });
+    if (stderr) {
+      logger.debug(`[direnv] stderr: ${stderr}`);
+    }
+    const trimmed = stdout.trim();
+    if (!trimmed) {
+      logger.debug("[direnv] No environment changes from direnv");
+      return {};
+    }
+    const direnvVars = JSON.parse(trimmed);
+    const varCount = Object.keys(direnvVars).length;
+    logger.debug(`[direnv] Loaded ${varCount} environment variables`);
+    return direnvVars;
+  } catch (error) {
+    if (error instanceof Error) {
+      logger.debug(`[direnv] Failed to load environment: ${error.message}`);
+      if (error.message.includes("is blocked")) {
+        logger.debug('[direnv] Hint: Run "direnv allow" in the project directory to allow the .envrc');
+      }
+    }
+    return {};
+  }
+}
+async function loadAndMergeEnvironment(cwd, existingEnv = process.env, overrideEnv = {}) {
+  const direnvVars = await loadDirenvEnvironment(cwd);
+  const merged = {};
+  for (const [key, value] of Object.entries(existingEnv)) {
+    if (value !== void 0) {
+      merged[key] = value;
+    }
+  }
+  for (const [key, value] of Object.entries(direnvVars)) {
+    merged[key] = value;
+  }
+  for (const [key, value] of Object.entries(overrideEnv)) {
+    merged[key] = value;
+  }
+  return merged;
+}
+
 const claudeCliPath = resolve(join(projectPath(), "scripts", "claude_local_launcher.cjs"));
 async function claudeLocal(opts) {
   const projectDir = getProjectPath(opts.path);
@@ -259,6 +352,15 @@ async function claudeLocal(opts) {
     }
   };
   try {
+    const direnvVars = await loadDirenvEnvironment(opts.path);
+    if (Object.keys(direnvVars).length > 0) {
+      logger.debug(`[ClaudeLocal] Loaded ${Object.keys(direnvVars).length} direnv environment variables`);
+    }
+    const env = {
+      ...process.env,
+      ...direnvVars,
+      ...opts.claudeEnvVars
+    };
     process.stdin.pause();
     await new Promise((r, reject) => {
       const args = [];
@@ -278,10 +380,6 @@ async function claudeLocal(opts) {
       if (!claudeCliPath || !existsSync(claudeCliPath)) {
         throw new Error("Claude local launcher not found. Please ensure HAPPY_PROJECT_ROOT is set correctly for development.");
       }
-      const env = {
-        ...process.env,
-        ...opts.claudeEnvVars
-      };
       const child = spawn("node", [claudeCliPath, ...args], {
         stdio: ["inherit", "inherit", "inherit", "pipe"],
         signal: opts.abort,
@@ -1522,6 +1620,13 @@ async function claudeRemote(opts) {
       }
     }
   }
+  const direnvVars = await loadDirenvEnvironment(opts.path);
+  if (Object.keys(direnvVars).length > 0) {
+    logger.debug(`[claudeRemote] Loaded ${Object.keys(direnvVars).length} direnv environment variables`);
+  }
+  Object.entries(direnvVars).forEach(([key, value]) => {
+    process.env[key] = value;
+  });
   if (opts.claudeEnvVars) {
     Object.entries(opts.claudeEnvVars).forEach(([key, value]) => {
       process.env[key] = value;
@@ -1628,12 +1733,32 @@ Echo message: ${echoMessage}` : "");
       }
     }
   };
+  async function buildMessageContent(text, imageRefs) {
+    if (!imageRefs || imageRefs.length === 0 || !opts.resolveImageRefs) {
+      return text;
+    }
+    logger.debug(`[claudeRemote] Resolving ${imageRefs.length} image references`);
+    const resolvedImages = await opts.resolveImageRefs(imageRefs);
+    logger.debug(`[claudeRemote] Resolved ${resolvedImages.length} images`);
+    if (resolvedImages.length === 0) {
+      return text;
+    }
+    const content = [];
+    for (const img of resolvedImages) {
+      content.push(img);
+    }
+    if (text) {
+      content.push({ type: "text", text });
+    }
+    return content;
+  }
   let messages = new PushableAsyncIterable();
+  const initialContent = await buildMessageContent(initial.message, initial.mode.imageRefs);
   messages.push({
     type: "user",
     message: {
       role: "user",
-      content: initial.message
+      content: initialContent
     }
   });
   const response = query({
@@ -1674,7 +1799,8 @@ Echo message: ${echoMessage}` : "");
           return;
         }
         mode = next.mode;
-        messages.push({ type: "user", message: { role: "user", content: next.message } });
+        const nextContent = await buildMessageContent(next.message, next.mode.imageRefs);
+        messages.push({ type: "user", message: { role: "user", content: nextContent } });
       }
       if (message.type === "user") {
         const msg = message;
@@ -2780,6 +2906,18 @@ async function claudeRemoteLauncher(session) {
           isAborted: (toolCallId) => {
             return permissionHandler.isAborted(toolCallId);
           },
+          resolveImageRefs: async (imageRefs) => {
+            const resolved = [];
+            for (const imageRef of imageRefs) {
+              const image = await session.client.resolveImageRef(imageRef);
+              if (image) {
+                resolved.push(image);
+              } else {
+                logger.debug(`[remote]: Failed to resolve image ref: ${imageRef.blobId}`);
+              }
+            }
+            return resolved;
+          },
           nextMessage: async () => {
             if (pending) {
               let p = pending;
@@ -2823,9 +2961,10 @@ async function claudeRemoteLauncher(session) {
           onReady: () => {
             if (!pending && session.queue.size() === 0) {
               session.client.sendSessionEvent({ type: "ready" });
+              const folderName = basename(session.path);
               session.api.push().sendToAllDevices(
-                "It's ready!",
-                `Claude is waiting for your command`,
+                `Ready`,
+                folderName,
                 { sessionId: session.client.sessionId }
               );
             }
@@ -2884,6 +3023,16 @@ async function claudeRemoteLauncher(session) {
 
 async function loop(opts) {
   const logPath = logger.logFilePath;
+  const sessionEnv = await loadAndMergeEnvironment(
+    opts.path,
+    process.env,
+    opts.claudeEnvVars ?? {}
+  );
+  logger.debug(`[loop] Loaded session environment with ${Object.keys(sessionEnv).length} variables`);
+  opts.session.rpcHandlerManager.setSessionContext({
+    path: opts.path,
+    env: sessionEnv
+  });
   let session = new Session({
     api: opts.api,
     client: opts.session,
@@ -4807,6 +4956,27 @@ function registerKillSessionHandler(rpcHandlerManager, killThisHappy) {
   });
 }
 
+function extractTextFromContent(content) {
+  if (!Array.isArray(content)) {
+    if (content.type === "text") {
+      return content.text;
+    }
+    return "";
+  }
+  const textParts = [];
+  for (const block of content) {
+    if (block.type === "text") {
+      textParts.push(block.text);
+    }
+  }
+  return textParts.join("\n");
+}
+function extractImageRefs(content) {
+  if (!Array.isArray(content)) {
+    return [];
+  }
+  return content.filter((block) => block.type === "image_ref");
+}
 async function runClaude(credentials, options = {}) {
   const workingDirectory = process.cwd();
   const sessionTag = randomUUID();
@@ -4973,7 +5143,12 @@ async function runClaude(credentials, options = {}) {
     } else {
       logger.debug(`[loop] User message received with no disallowed tools override, using current: ${currentDisallowedTools ? currentDisallowedTools.join(", ") : "none"}`);
     }
-    const specialCommand = parseSpecialCommand(message.content.text);
+    const textContent = extractTextFromContent(message.content);
+    const imageRefs = extractImageRefs(message.content);
+    if (imageRefs.length > 0) {
+      logger.debug(`[loop] User message contains ${imageRefs.length} image(s)`);
+    }
+    const specialCommand = parseSpecialCommand(textContent);
     if (specialCommand.type === "compact") {
       logger.debug("[start] Detected /compact command");
       const enhancedMode2 = {
@@ -4983,9 +5158,10 @@ async function runClaude(credentials, options = {}) {
         customSystemPrompt: messageCustomSystemPrompt,
         appendSystemPrompt: messageAppendSystemPrompt,
         allowedTools: messageAllowedTools,
-        disallowedTools: messageDisallowedTools
+        disallowedTools: messageDisallowedTools,
+        imageRefs: imageRefs.length > 0 ? imageRefs : void 0
       };
-      messageQueue.pushIsolateAndClear(specialCommand.originalMessage || message.content.text, enhancedMode2);
+      messageQueue.pushIsolateAndClear(specialCommand.originalMessage || textContent, enhancedMode2);
       logger.debugLargeJson("[start] /compact command pushed to queue:", message);
       return;
     }
@@ -4998,9 +5174,10 @@ async function runClaude(credentials, options = {}) {
         customSystemPrompt: messageCustomSystemPrompt,
         appendSystemPrompt: messageAppendSystemPrompt,
         allowedTools: messageAllowedTools,
-        disallowedTools: messageDisallowedTools
+        disallowedTools: messageDisallowedTools,
+        imageRefs: imageRefs.length > 0 ? imageRefs : void 0
       };
-      messageQueue.pushIsolateAndClear(specialCommand.originalMessage || message.content.text, enhancedMode2);
+      messageQueue.pushIsolateAndClear(specialCommand.originalMessage || textContent, enhancedMode2);
       logger.debugLargeJson("[start] /compact command pushed to queue:", message);
       return;
     }
@@ -5011,9 +5188,10 @@ async function runClaude(credentials, options = {}) {
       customSystemPrompt: messageCustomSystemPrompt,
       appendSystemPrompt: messageAppendSystemPrompt,
       allowedTools: messageAllowedTools,
-      disallowedTools: messageDisallowedTools
+      disallowedTools: messageDisallowedTools,
+      imageRefs: imageRefs.length > 0 ? imageRefs : void 0
     };
-    messageQueue.push(message.content.text, enhancedMode);
+    messageQueue.push(textContent, enhancedMode);
     logger.debugLargeJson("User message pushed to queue:", message);
   });
   const cleanup = async () => {
@@ -6192,7 +6370,7 @@ async function handleConnectVendor(vendor, displayName) {
     return;
   } else if (subcommand === "codex") {
     try {
-      const { runCodex } = await import('./runCodex-CLGYMNs2.mjs');
+      const { runCodex } = await import('./runCodex-CYkmZphO.mjs');
       let startedBy = void 0;
       for (let i = 1; i < args.length; i++) {
         if (args[i] === "--started-by") {
@@ -6237,7 +6415,7 @@ async function handleConnectVendor(vendor, displayName) {
   } else if (subcommand === "list") {
     try {
       const { credentials } = await authAndSetupMachineIfNeeded();
-      const { listSessions } = await import('./list-hET5tyMc.mjs');
+      const { listSessions } = await import('./list-BvtUKVTq.mjs');
       let sessionId;
       let titleFilter;
       let recentMsgs;
@@ -6339,7 +6517,7 @@ Examples:
         process.exit(1);
       }
       const { credentials } = await authAndSetupMachineIfNeeded();
-      const { promptSession } = await import('./prompt-Dz7G8yGx.mjs');
+      const { promptSession } = await import('./prompt-BbMNN7fl.mjs');
       await promptSession(credentials, sessionId, promptText, timeoutMinutes ?? void 0);
     } catch (error) {
       console.error(chalk.red("Error:"), error instanceof Error ? error.message : "Unknown error");