claude-yolo 1.7.0 → 1.9.1

package/bin/claude-yolo.js

@@ -3,6 +3,7 @@
 
 import fs from 'fs';
 import path from 'path';
+import os from 'os';
 import { createRequire } from 'module';
 import { fileURLToPath } from 'url';
 import { execSync } from 'child_process';
@@ -12,9 +13,27 @@ import readline from 'readline';
 const RED = '\x1b[31m';
 const YELLOW = '\x1b[33m';
 const CYAN = '\x1b[36m';
+const GREEN = '\x1b[32m';
 const RESET = '\x1b[0m';
 const BOLD = '\x1b[1m';
 
+// Path to persistent state file
+const stateFile = path.join(os.homedir(), '.claude_yolo_state');
+
+// Function to get current mode from state file
+function getMode() {
+  try {
+    return fs.readFileSync(stateFile, 'utf8').trim();
+  } catch {
+    return 'YOLO'; // Default mode
+  }
+}
+
+// Function to set mode in state file
+function setMode(mode) {
+  fs.writeFileSync(stateFile, mode);
+}
+
 // Debug logging function that only logs if DEBUG env var is set
 const debug = (message) => {
   if (process.env.DEBUG) {
@@ -30,20 +49,21 @@ function askForConsent() {
       output: process.stdout
     });
 
-    console.log(`\n${BOLD}${YELLOW}🔥 CLAUDE-YOLO INSTALLATION CONSENT REQUIRED 🔥${RESET}\n`);
+    console.log(`\n${BOLD}${YELLOW}🔥 CLAUDE-YOLO CONSENT REQUIRED 🔥${RESET}\n`);
     console.log(`${CYAN}----------------------------------------${RESET}`);
     console.log(`${BOLD}What is claude-yolo?${RESET}`);
     console.log(`This package creates a wrapper around the official Claude CLI tool that:`);
     console.log(`  1. ${RED}BYPASSES safety checks${RESET} by automatically adding the --dangerously-skip-permissions flag`);
     console.log(`  2. Automatically updates to the latest Claude CLI version`);
-    console.log(`  3. Adds colorful YOLO-themed loading messages\n`);
+    console.log(`  3. Adds colorful YOLO-themed loading messages`);
+    console.log(`  4. ${GREEN}NOW SUPPORTS SAFE MODE${RESET} with --safe flag\n`);
 
     console.log(`${BOLD}${RED}⚠️ IMPORTANT SECURITY WARNING ⚠️${RESET}`);
     console.log(`The ${BOLD}--dangerously-skip-permissions${RESET} flag was designed for use in containers`);
     console.log(`and bypasses important safety checks. This includes ignoring file access`);
     console.log(`permissions that protect your system and privacy.\n`);
 
-    console.log(`${BOLD}By using claude-yolo:${RESET}`);
+    console.log(`${BOLD}By using claude-yolo in YOLO mode:${RESET}`);
     console.log(`  • You acknowledge these safety checks are being bypassed`);
     console.log(`  • You understand this may allow Claude CLI to access sensitive files`);
     console.log(`  • You accept full responsibility for any security implications\n`);
@@ -94,7 +114,29 @@ async function checkForUpdates() {
     const dependencies = packageJson.dependencies || {};
     const currentVersion = dependencies['@anthropic-ai/claude-code'];
     
-    debug(`Current dependency in package.json: ${currentVersion}`);
+    debug(`Claude version from package.json: ${currentVersion}`);
+    
+    // Get the global Claude version if available
+    let globalVersion;
+    if (globalClaudeDir) {
+      try {
+        const globalPackageJsonPath = path.join(globalClaudeDir, 'package.json');
+        if (fs.existsSync(globalPackageJsonPath)) {
+          const globalPackageJson = JSON.parse(fs.readFileSync(globalPackageJsonPath, 'utf8'));
+          globalVersion = globalPackageJson.version;
+          debug(`Global Claude version: ${globalVersion}`);
+          
+          // If global version is latest, inform user
+          if (globalVersion === latestVersion) {
+            debug(`Global Claude installation is already the latest version`);
+          } else if (globalVersion && latestVersion) {
+            debug(`Global Claude installation (${globalVersion}) differs from latest (${latestVersion})`);
+          }
+        }
+      } catch (err) {
+        debug(`Error getting global Claude version: ${err.message}`);
+      }
+    }
     
     // If using a specific version (not "latest"), and it's out of date, update
     if (currentVersion !== "latest" && currentVersion !== latestVersion) {
@@ -119,22 +161,123 @@ async function checkForUpdates() {
   }
 }
 
-// Path to the Claude CLI file
-const claudeDir = path.join(nodeModulesDir, 'node_modules', '@anthropic-ai', 'claude-code');
-let originalCliPath = path.join(claudeDir, 'cli.mjs');
-if(!fs.existsSync(originalCliPath)) {
-  originalCliPath = path.join(claudeDir, 'cli.js');
-  if(!fs.existsSync(originalCliPath)) {
-    console.error(`Error: ${originalCliPath} not found. Make sure @anthropic-ai/claude-code is installed.`);
-    process.exit(1);
+// Try to find global installation of Claude CLI first
+let globalClaudeDir;
+try {
+  const globalNodeModules = execSync('npm -g root').toString().trim();
+  debug(`Global node_modules: ${globalNodeModules}`);
+  const potentialGlobalDir = path.join(globalNodeModules, '@anthropic-ai', 'claude-code');
+  
+  if (fs.existsSync(potentialGlobalDir)) {
+    globalClaudeDir = potentialGlobalDir;
+    debug(`Found global Claude installation at: ${globalClaudeDir}`);
   }
+} catch (error) {
+  debug(`Error finding global Claude installation: ${error.message}`);
 }
 
-const yoloCliPath = path.join(claudeDir, 'cli-yolo.mjs');
+// Path to the local Claude CLI installation
+const localClaudeDir = path.join(nodeModulesDir, 'node_modules', '@anthropic-ai', 'claude-code');
+
+// For AGENTS mode, always use LOCAL installation to ensure patches work correctly
+// Global installation may have different minified variable names
+const claudeDir = localClaudeDir;
+debug(`Using Claude installation from: ${claudeDir}`);
+debug(`Using ${claudeDir === globalClaudeDir ? 'GLOBAL' : 'LOCAL'} Claude installation`);
+
+// Check for both .js and .mjs versions of the CLI
+let mjs = path.join(claudeDir, 'cli.mjs');
+let js = path.join(claudeDir, 'cli.js');
+let originalCliPath;
+let yoloCliPath;
+
+if (fs.existsSync(js)) {
+  originalCliPath = js;
+  yoloCliPath = path.join(claudeDir, 'cli-yolo.js');
+  debug(`Found Claude CLI at ${originalCliPath} (js version)`);
+} else if (fs.existsSync(mjs)) {
+  originalCliPath = mjs;
+  yoloCliPath = path.join(claudeDir, 'cli-yolo.mjs');
+  debug(`Found Claude CLI at ${originalCliPath} (mjs version)`);
+} else {
+  console.error(`Error: Claude CLI not found in ${claudeDir}. Make sure @anthropic-ai/claude-code is installed.`);
+  process.exit(1);
+}
 const consentFlagPath = path.join(claudeDir, '.claude-yolo-consent');
 
 // Main function to run the application
 async function run() {
+  // Handle mode commands first
+  const args = process.argv.slice(2);
+  if (args[0] === 'mode') {
+    if (args[1] === 'yolo') {
+      console.log(`${YELLOW}🔥 Switching to YOLO mode...${RESET}`);
+      console.log(`${RED}⚠️  WARNING: All safety checks will be DISABLED!${RESET}`);
+      setMode('YOLO');
+      console.log(`${YELLOW}✓ YOLO mode activated${RESET}`);
+      return;
+    } else if (args[1] === 'safe') {
+      console.log(`${CYAN}🛡️  Switching to SAFE mode...${RESET}`);
+      console.log(`${GREEN}✓ Safety checks will be enabled${RESET}`);
+      setMode('SAFE');
+      console.log(`${CYAN}✓ SAFE mode activated${RESET}`);
+      return;
+    } else {
+      const currentMode = getMode();
+      console.log(`Current mode: ${currentMode === 'YOLO' ? YELLOW : CYAN}${currentMode}${RESET}`);
+      return;
+    }
+  }
+
+  // Check for --safe or --no-yolo flags
+  const safeMode = process.argv.includes('--safe') || 
+                   process.argv.includes('--no-yolo') ||
+                   getMode() === 'SAFE';
+  
+  if (safeMode) {
+    // Remove our flags before passing to original CLI
+    process.argv = process.argv.filter(arg => 
+      arg !== '--safe' && arg !== '--no-yolo'
+    );
+    
+    console.log(`${CYAN}[SAFE] Running Claude in SAFE mode${RESET}`);
+    
+    // Update if needed
+    await checkForUpdates();
+    
+    // Ensure original CLI exists
+    if (!fs.existsSync(originalCliPath)) {
+      console.error(`Error: ${originalCliPath} not found. Make sure @anthropic-ai/claude-code is installed.`);
+      process.exit(1);
+    }
+    
+    // Run original CLI without modifications
+    await import(originalCliPath);
+    return; // Exit early
+  }
+
+  // YOLO MODE continues below
+  console.log(`${YELLOW}[YOLO] Running Claude in YOLO mode${RESET}`);
+
+  // Enable bypass permissions mode for YOLO mode (allows auto-accept of plans)
+  if (!process.argv.includes('--dangerously-skip-permissions')) {
+    process.argv.push('--dangerously-skip-permissions');
+    debug("Added --dangerously-skip-permissions flag for YOLO mode");
+  }
+
+  // Temporarily fake non-root for YOLO mode
+  if (process.getuid && process.getuid() === 0) {
+    console.log(`${YELLOW}⚠️  Running as root - applying YOLO bypass...${RESET}`);
+    // Store original getuid
+    const originalGetuid = process.getuid;
+    // Override getuid to return non-root
+    process.getuid = () => 1000; // Fake regular user ID
+    // Restore after a delay to allow CLI to start
+    setTimeout(() => {
+      process.getuid = originalGetuid;
+    }, 100);
+  }
+
   // Check and update Claude package first
   await checkForUpdates();
 
@@ -167,9 +310,10 @@ async function run() {
   // Read the original CLI file content
   let cliContent = fs.readFileSync(originalCliPath, 'utf8');
 
-  // Replace all instances of "punycode" with "punycode/"
-  cliContent = cliContent.replace(/"punycode"/g, '"punycode/"');
-  debug('Replaced all instances of "punycode" with "punycode/"');
+  if (claudeDir === localClaudeDir) {
+    cliContent = cliContent.replace(/"punycode"/g, '"punycode/"');
+    debug('Replaced all instances of "punycode" with "punycode/"');
+  }
 
   // Replace getIsDocker() calls with true
   cliContent = cliContent.replace(/[a-zA-Z0-9_]*\.getIsDocker\(\)/g, 'true');
@@ -179,6 +323,37 @@ async function run() {
   cliContent = cliContent.replace(/[a-zA-Z0-9_]*\.hasInternetAccess\(\)/g, 'false');
   debug("Replaced all instances of *.hasInternetAccess() with false");
 
+  // Replace root check patterns
+  // Pattern 1: process.getuid() === 0
+  cliContent = cliContent.replace(/process\.getuid\(\)\s*===\s*0/g, 'false');
+  debug("Replaced process.getuid() === 0 checks with false");
+
+  // Pattern 2: process.getuid?.() === 0
+  cliContent = cliContent.replace(/process\.getuid\?\.\(\)\s*===\s*0/g, 'false');
+  debug("Replaced process.getuid?.() === 0 checks with false");
+
+  // Pattern 3: getuid() === 0 (with any variable)
+  cliContent = cliContent.replace(/(\w+)\.getuid\(\)\s*===\s*0/g, 'false');
+  debug("Replaced all getuid() === 0 checks with false");
+
+  // Pattern 4: Replace any EUID checks
+  cliContent = cliContent.replace(/process\.geteuid\(\)\s*===\s*0/g, 'false');
+  cliContent = cliContent.replace(/process\.geteuid\?\.\(\)\s*===\s*0/g, 'false');
+  debug("Replaced geteuid() checks with false");
+
+  // Auto-accept plan mode confirmation
+  // Inject a useEffect that automatically triggers "yes-bypass-permissions" when:
+  // - bypass permissions mode is available (G.toolPermissionContext.isBypassPermissionsModeAvailable)
+  // - there's a valid plan (!F means plan is not empty)
+  const planAutoAcceptPatch = `k5.useEffect(()=>{if(G.toolPermissionContext.isBypassPermissionsModeAvailable&&!F){N("yes-bypass-permissions")}},[]);`;
+  const targetString = 'let M=Md(),R=M?oH(M):null';
+  if (cliContent.includes(targetString)) {
+    cliContent = cliContent.replace(targetString, targetString + ';' + planAutoAcceptPatch);
+    debug("Patched plan mode to auto-accept when bypass permissions is available");
+  } else {
+    debug("WARNING: Could not find target string for plan auto-accept patch");
+  }
+
   // Add warning message
   console.log(`${YELLOW}🔥 YOLO MODE ACTIVATED 🔥${RESET}`);
 

package/demo.txt

@@ -0,0 +1 @@
+goodbye world

package/package.json

@@ -1,34 +1,38 @@
 {
   "name": "claude-yolo",
-  "version": "1.7.0",
-  "description": "YOLO wrapper for Claude CLI with danger mode always enabled, auto-updates, and colorful loading messages",
+  "version": "1.9.1",
+  "description": "Claude CLI wrapper with YOLO mode (bypass safety) and SAFE mode support, auto-updates, and colorful loading messages",
   "bin": {
-    "claude-yolo": "./bin/claude-yolo.js"
+    "claude-yolo": "./bin/claude-yolo.js",
+    "cl": "./bin/cl"
   },
   "dependencies": {
-    "@anthropic-ai/claude-code": "0.2.35",
-    "punycode": "latest"
+    "@anthropic-ai/claude-code": "2.0.76",
+    "punycode": "2.3.1"
   },
   "type": "module",
   "scripts": {
+    "postinstall": "chmod +x ./bin/cl"
   },
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/eastlondoner/claude-yolo.git"
+    "url": "git+https://github.com/maxparez/claude-yolo.git"
   },
   "keywords": [
     "claude",
     "cli",
     "anthropic",
     "wrapper",
-    "yolo"
+    "yolo",
+    "safe-mode",
+    "permissions"
   ],
-  "author": "eastlondoner",
+  "author": "eastlondoner (fork by maxparez)",
   "license": "MIT",
   "bugs": {
-    "url": "https://github.com/eastlondoner/claude-yolo/issues"
+    "url": "https://github.com/maxparez/claude-yolo/issues"
   },
-  "homepage": "https://github.com/eastlondoner/claude-yolo#readme",
+  "homepage": "https://github.com/maxparez/claude-yolo#readme",
   "engines": {
     "node": ">=14.16"
   }

package/test.txt

@@ -0,0 +1 @@
+hello world

package/yolo-test.txt

@@ -0,0 +1 @@
+YOLO mode works