@xaidenlabs/uso 1.1.67 → 1.1.68

package/src/commands/uninstall.js

@@ -1,240 +1,415 @@
-const shell = require('shelljs');
-const { log, spinner } = require('../utils/logger');
-const readline = require('readline');
-const fs = require('fs');
-const path = require('path');
-const os = require('os');
+const shell = require("shelljs");
+const { log, spinner } = require("../utils/logger");
+const readline = require("readline");
+const fs = require("fs");
+const path = require("path");
+const os = require("os");
+const { isStealthMode } = require("../utils/stealth");
+const { runWsl } = require("../utils/wsl-bridge");
 
 const askQuestion = (query) => {
-    const rl = readline.createInterface({
-        input: process.stdin,
-        output: process.stdout,
-    });
-    return new Promise(resolve => rl.question(query, ans => {
-        rl.close();
-        resolve(ans);
-    }));
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+  });
+  return new Promise((resolve) =>
+    rl.question(query, (ans) => {
+      rl.close();
+      resolve(ans);
+    }),
+  );
+};
+
+const getStealthContext = () => {
+  if (os.platform() !== "win32") return { enabled: false, distro: "Ubuntu" };
+  return isStealthMode();
+};
+
+const runInStealth = (command, stealth, silent = true) => {
+  const envSetup =
+    'source $HOME/.cargo/env 2>/dev/null; export PATH="$HOME/.local/share/solana/install/active_release/bin:$HOME/.avm/bin:$PATH"';
+  return runWsl(`${envSetup} && ${command}`, {
+    distro: stealth.distro,
+    execOpts: { silent },
+  });
+};
+
+const commandExists = (command, stealth) => {
+  if (stealth.enabled) {
+    const result = runInStealth(`command -v ${command}`, stealth, true);
+    return result.code === 0;
+  }
+  return !!shell.which(command);
+};
+
+const wslPathExists = (wslPath, stealth) => {
+  const result = runInStealth(`[ -e "${wslPath}" ]`, stealth, true);
+  return result.code === 0;
 };
 
 /**
  * Runs a command and attempts to elevate privileges if it fails with a permission error.
  */
-const runOrElevate = (command, description) => {
-    // We run without silent:true initially to let the user see output, 
-    // but detecting the error code is what matters.
-    // actually, to detect the specific string "os error 1314", we need to capture output.
-    // So we run silently first? Or we just run and if it fails, we assume it *might* be elevation if on Windows?
-    // Let's run synchronously and capture output.
-
-    const result = shell.exec(command, { silent: true });
+const runOrElevate = (
+  command,
+  description,
+  stealth = { enabled: false, distro: "Ubuntu" },
+) => {
+  if (stealth.enabled) {
+    const result = runInStealth(command, stealth, true);
 
     if (result.code === 0) {
-        console.log(result.stdout);
-        return true;
+      if (result.stdout) console.log(result.stdout);
+      return true;
     }
 
-    // Print the error output to the user
-    console.log(result.stdout);
-    console.error(result.stderr);
-
-    const output = result.stderr + result.stdout;
-
-    // Check for common permission errors
-    // "os error 1314" is specific to Windows symlink privilege
-    if ((output.includes("os error 1314") || output.includes("EPERM") || output.includes("permission denied")) && os.platform() === 'win32') {
-        log.warn(`⚠️  Permission denied during: ${description}`);
-        log.info("🛡️  Triggering Run as Administrator (UAC) to retry...");
+    if (result.stdout) console.log(result.stdout);
+    if (result.stderr) console.error(result.stderr);
+    log.error(`❌ Command failed in WSL: ${description}`);
+    return false;
+  }
 
-        // Construct PowerShell command to run cmd /c <command> as admin
-        // We need to be careful with quoting.
-        const escapedCommand = command.replace(/'/g, "''"); // Basic PowerShell escaping for single quotes
-        const elevateCmd = `powershell -Command "Start-Process -FilePath 'cmd.exe' -ArgumentList '/c ${escapedCommand}' -Verb RunAs -Wait"`;
+  // We run without silent:true initially to let the user see output,
+  // but detecting the error code is what matters.
+  // actually, to detect the specific string "os error 1314", we need to capture output.
+  // So we run silently first? Or we just run and if it fails, we assume it *might* be elevation if on Windows?
+  // Let's run synchronously and capture output.
 
-        const elevatedRun = shell.exec(elevateCmd);
+  const result = shell.exec(command, { silent: true });
 
-        if (elevatedRun.code === 0) {
-            log.success(`✅ ${description} completed (Elevated).`);
-            return true;
-        } else {
-            log.error(`❌ Elevated execution failed for: ${description}`);
-            return false;
-        }
+  if (result.code === 0) {
+    console.log(result.stdout);
+    return true;
+  }
+
+  // Print the error output to the user
+  console.log(result.stdout);
+  console.error(result.stderr);
+
+  const output = result.stderr + result.stdout;
+
+  // Check for common permission errors
+  // "os error 1314" is specific to Windows symlink privilege
+  if (
+    (output.includes("os error 1314") ||
+      output.includes("EPERM") ||
+      output.includes("permission denied")) &&
+    os.platform() === "win32"
+  ) {
+    log.warn(`⚠️  Permission denied during: ${description}`);
+    log.info("🛡️  Triggering Run as Administrator (UAC) to retry...");
+
+    // Construct PowerShell command to run cmd /c <command> as admin
+    // We need to be careful with quoting.
+    const escapedCommand = command.replace(/'/g, "''"); // Basic PowerShell escaping for single quotes
+    const elevateCmd = `powershell -Command "Start-Process -FilePath 'cmd.exe' -ArgumentList '/c ${escapedCommand}' -Verb RunAs -Wait"`;
+
+    const elevatedRun = shell.exec(elevateCmd);
+
+    if (elevatedRun.code === 0) {
+      log.success(`✅ ${description} completed (Elevated).`);
+      return true;
+    } else {
+      log.error(`❌ Elevated execution failed for: ${description}`);
+      return false;
     }
+  }
 
-    log.error(`❌ Command failed: ${description}`);
-    return false;
+  log.error(`❌ Command failed: ${description}`);
+  return false;
 };
 
 const uninstall = async (component) => {
-    log.header("🗑️  USO Uninstallation & Cleanup");
-
-    if (component) {
-        component = component.toLowerCase();
-        log.info(`🎯 Targeted uninstallation: ${component}`);
-
-        if (component === 'anchor') {
-            const anchorInstalled = shell.which('anchor');
-            if (anchorInstalled) {
-                log.info("Removing Anchor...");
-                // Try avm uninstall first if available
-                if (shell.which('avm')) {
-                    runOrElevate('avm uninstall latest', 'Uninstall Anchor (AVM)');
-                }
-                runOrElevate('cargo uninstall anchor-cli', 'Uninstall anchor-cli');
-                runOrElevate('cargo uninstall avm', 'Uninstall avm');
-                log.success("Anchor removal steps completed.");
-            } else {
-                log.success("✅ Anchor is not installed.");
-            }
-            return;
+  const stealth = getStealthContext();
+  log.header("🗑️  USO Uninstallation & Cleanup");
+  if (stealth.enabled) {
+    log.info(
+      `🐧 Stealth Mode detected. Uninstall targets WSL distro: ${stealth.distro}`,
+    );
+  }
+
+  if (component) {
+    component = component.toLowerCase();
+    log.info(`🎯 Targeted uninstallation: ${component}`);
+
+    if (component === "anchor") {
+      const anchorInstalled = commandExists("anchor", stealth);
+      if (anchorInstalled) {
+        log.info("Removing Anchor...");
+        // Try avm uninstall first if available
+        if (commandExists("avm", stealth)) {
+          runOrElevate(
+            "avm uninstall latest",
+            "Uninstall Anchor (AVM)",
+            stealth,
+          );
         }
+        runOrElevate(
+          "cargo uninstall anchor-cli",
+          "Uninstall anchor-cli",
+          stealth,
+        );
+        runOrElevate("cargo uninstall avm", "Uninstall avm", stealth);
+        log.success("Anchor removal steps completed.");
+      } else {
+        log.success("✅ Anchor is not installed.");
+      }
+      return;
+    }
 
-        if (component === 'solana') {
-            // Check PATH first
-            let solanaInstalled = shell.which('solana');
-            const localShareSolana = path.join(os.homedir(), '.local', 'share', 'solana');
-
-            // If not found in PATH, check default location
-            if (!solanaInstalled && fs.existsSync(localShareSolana)) {
-                solanaInstalled = true;
-            }
-
-            if (solanaInstalled) {
-                log.info("Removing Solana CLI...");
-
-                if (fs.existsSync(localShareSolana)) {
-                    try {
-                        fs.rmSync(localShareSolana, { recursive: true, force: true });
-                        log.success(`Removed ${localShareSolana}`);
-                    } catch (err) {
-                        log.warn(`Failed to remove ${localShareSolana} directly: ${err.message}`);
-                        log.info("Trying to remove via elevated command...");
-                        runOrElevate(`rmdir /s /q "${localShareSolana}"`, `Remove folder ${localShareSolana}`);
-                    }
-                } else {
-                    log.warn(`Could not find Solana folder at ${localShareSolana}. It might be removed already.`);
-                }
-            } else {
-                log.success("✅ Solana CLI is not installed.");
-            }
-            return;
+    if (component === "solana") {
+      // Check PATH first
+      let solanaInstalled = commandExists("solana", stealth);
+      const localShareSolana = path.join(
+        os.homedir(),
+        ".local",
+        "share",
+        "solana",
+      );
+      const wslSolanaPath = "$HOME/.local/share/solana";
+
+      // If not found in PATH, check default location
+      if (!solanaInstalled) {
+        if (stealth.enabled) {
+          solanaInstalled = wslPathExists(wslSolanaPath, stealth);
+        } else if (fs.existsSync(localShareSolana)) {
+          solanaInstalled = true;
         }
-
-        if (component === 'rust') {
-            const rustInstalled = shell.which('rustc');
-            if (rustInstalled) {
-                log.info("Running rustup self uninstall...");
-                runOrElevate('rustup self uninstall -y', 'Uninstall Rust');
-            } else {
-                log.success("✅ Rust is not installed.");
+      }
+
+      if (solanaInstalled) {
+        log.info("Removing Solana CLI...");
+
+        if (stealth.enabled) {
+          runOrElevate(
+            `rm -rf "${wslSolanaPath}"`,
+            `Remove folder ${wslSolanaPath}`,
+            stealth,
+          );
+          log.success(`Removed ${wslSolanaPath}`);
+        } else {
+          if (fs.existsSync(localShareSolana)) {
+            try {
+              fs.rmSync(localShareSolana, { recursive: true, force: true });
+              log.success(`Removed ${localShareSolana}`);
+            } catch (err) {
+              log.warn(
+                `Failed to remove ${localShareSolana} directly: ${err.message}`,
+              );
+              log.info("Trying to remove via elevated command...");
+              runOrElevate(
+                `rmdir /s /q "${localShareSolana}"`,
+                `Remove folder ${localShareSolana}`,
+                stealth,
+              );
             }
-            return;
+          } else {
+            log.warn(
+              `Could not find Solana folder at ${localShareSolana}. It might be removed already.`,
+            );
+          }
         }
-
-        log.error(`❌ Unknown component: ${component}. Available: rust, solana, anchor`);
-        return;
+      } else {
+        log.success("✅ Solana CLI is not installed.");
+      }
+      return;
     }
 
-    // --- FULL INTERACTIVE UNINSTALL ---
-
-    log.warn("This process allows you to remove components installed by uso.");
-    log.warn("Please be careful, especially with wallet removal!");
-
-    const proceed = await askQuestion("👉 Do you want to proceed with uninstallation? (y/N): ");
-    if (proceed.toLowerCase() !== 'y') {
-        log.info("Operation cancelled.");
-        return;
+    if (component === "rust") {
+      const rustInstalled = commandExists("rustc", stealth);
+      if (rustInstalled) {
+        log.info("Running rustup self uninstall...");
+        runOrElevate("rustup self uninstall -y", "Uninstall Rust", stealth);
+      } else {
+        log.success("✅ Rust is not installed.");
+      }
+      return;
     }
 
-    // 1. Uninstall Anchor
-    const anchorInstalled = shell.which('anchor');
-    if (anchorInstalled) {
-        const removeAnchor = await askQuestion("\n⚓ Remove Anchor Framework? (y/N): ");
-        if (removeAnchor.toLowerCase() === 'y') {
-            log.info("Removing Anchor...");
-            // Try avm uninstall first if available
-            if (shell.which('avm')) {
-                runOrElevate('avm uninstall latest', 'Uninstall Anchor (AVM)');
-            }
-            runOrElevate('cargo uninstall anchor-cli', 'Uninstall anchor-cli');
-            runOrElevate('cargo uninstall avm', 'Uninstall avm');
-            log.success("Anchor removal steps completed.");
+    log.error(
+      `❌ Unknown component: ${component}. Available: rust, solana, anchor`,
+    );
+    return;
+  }
+
+  // --- FULL INTERACTIVE UNINSTALL ---
+
+  log.warn("This process allows you to remove components installed by uso.");
+  log.warn("Please be careful, especially with wallet removal!");
+
+  const proceed = await askQuestion(
+    "👉 Do you want to proceed with uninstallation? (y/N): ",
+  );
+  if (proceed.toLowerCase() !== "y") {
+    log.info("Operation cancelled.");
+    return;
+  }
+
+  // 1. Uninstall Anchor
+  const anchorInstalled = commandExists("anchor", stealth);
+  if (anchorInstalled) {
+    const removeAnchor = await askQuestion(
+      "\n⚓ Remove Anchor Framework? (y/N): ",
+    );
+    if (removeAnchor.toLowerCase() === "y") {
+      log.info("Removing Anchor...");
+      // Try avm uninstall first if available
+      if (commandExists("avm", stealth)) {
+        runOrElevate("avm uninstall latest", "Uninstall Anchor (AVM)", stealth);
+      }
+      runOrElevate(
+        "cargo uninstall anchor-cli",
+        "Uninstall anchor-cli",
+        stealth,
+      );
+      runOrElevate("cargo uninstall avm", "Uninstall avm", stealth);
+      log.success("Anchor removal steps completed.");
+    }
+  }
+
+  // 2. Uninstall Solana
+  let solanaInstalled = commandExists("solana", stealth);
+  const localShareSolana = path.join(os.homedir(), ".local", "share", "solana");
+  const wslSolanaPath = "$HOME/.local/share/solana";
+
+  // If not found in PATH, check default location (like doctor does)
+  if (!solanaInstalled) {
+    if (stealth.enabled) {
+      solanaInstalled = wslPathExists(wslSolanaPath, stealth);
+    } else if (fs.existsSync(localShareSolana)) {
+      solanaInstalled = true;
+    }
+  }
+
+  if (solanaInstalled) {
+    const removeSolana = await askQuestion("\n☀️  Remove Solana CLI? (y/N): ");
+    if (removeSolana.toLowerCase() === "y") {
+      log.info("Removing Solana CLI...");
+
+      // Default locations
+      // const localShareSolana = path.join(os.homedir(), '.local', 'share', 'solana'); // Already defined
+
+      if (stealth.enabled) {
+        runOrElevate(
+          `rm -rf "${wslSolanaPath}"`,
+          `Remove folder ${wslSolanaPath}`,
+          stealth,
+        );
+        log.success(`Removed ${wslSolanaPath}`);
+      } else {
+        if (fs.existsSync(localShareSolana)) {
+          try {
+            fs.rmSync(localShareSolana, { recursive: true, force: true });
+            log.success(`Removed ${localShareSolana}`);
+          } catch (err) {
+            log.warn(
+              `Failed to remove ${localShareSolana} directly: ${err.message}`,
+            );
+            log.info("Trying to remove via elevated command...");
+            runOrElevate(
+              `rmdir /s /q "${localShareSolana}"`,
+              `Remove folder ${localShareSolana}`,
+              stealth,
+            );
+          }
+        } else {
+          log.warn(
+            `Could not find Solana at ${localShareSolana}. It might be already removed.`,
+          );
         }
+      }
     }
-
-    // 2. Uninstall Solana
-    let solanaInstalled = shell.which('solana');
-    const localShareSolana = path.join(os.homedir(), '.local', 'share', 'solana');
-
-    // If not found in PATH, check default location (like doctor does)
-    if (!solanaInstalled && fs.existsSync(localShareSolana)) {
-        solanaInstalled = true;
+  }
+
+  // 3. Uninstall Rust
+  const rustInstalled = commandExists("rustc", stealth);
+  if (rustInstalled) {
+    const removeRust = await askQuestion("\n🦀 Remove Rust? (y/N): ");
+    if (removeRust.toLowerCase() === "y") {
+      log.info("Running rustup self uninstall...");
+      runOrElevate("rustup self uninstall -y", "Uninstall Rust", stealth);
     }
-
-    if (solanaInstalled) {
-        const removeSolana = await askQuestion("\n☀️  Remove Solana CLI? (y/N): ");
-        if (removeSolana.toLowerCase() === 'y') {
-            log.info("Removing Solana CLI...");
-
-            // Default locations
-            // const localShareSolana = path.join(os.homedir(), '.local', 'share', 'solana'); // Already defined
-
-            if (fs.existsSync(localShareSolana)) {
-                try {
-                    fs.rmSync(localShareSolana, { recursive: true, force: true });
-                    log.success(`Removed ${localShareSolana}`);
-                } catch (err) {
-                    log.warn(`Failed to remove ${localShareSolana} directly: ${err.message}`);
-                    log.info("Trying to remove via elevated command...");
-                    runOrElevate(`rmdir /s /q "${localShareSolana}"`, `Remove folder ${localShareSolana}`);
-                }
-            } else {
-                log.warn(`Could not find Solana at ${localShareSolana}. It might be already removed.`);
+  }
+
+  // 4. WALLET REMOVAL (DANGER)
+  const walletPath = path.join(os.homedir(), ".config", "solana", "id.json");
+  const wslWalletPath = "$HOME/.config/solana/id.json";
+  const hasNativeWallet = fs.existsSync(walletPath);
+  const hasWslWallet = stealth.enabled
+    ? wslPathExists(wslWalletPath, stealth)
+    : false;
+
+  if (hasNativeWallet || hasWslWallet) {
+    log.error("\n⚠️  DANGER ZONE ⚠️");
+    if (hasNativeWallet) log.warn(`Found a Solana wallet at: ${walletPath}`);
+    if (hasWslWallet)
+      log.warn(`Found a Solana wallet in WSL at: ${wslWalletPath}`);
+    log.warn(
+      "If you delete this without a backup, your funds will be LOST FOREVER.",
+    );
+
+    const removeWallet = await askQuestion(
+      "💥 Do you REALLY want to delete this wallet? (type 'DELETE' to confirm): ",
+    );
+    if (removeWallet === "DELETE") {
+      if (hasNativeWallet) {
+        try {
+          fs.unlinkSync(walletPath);
+          log.success("Native wallet deleted.");
+
+          // Clean up parent config dir if empty
+          const configDir = path.dirname(walletPath);
+          try {
+            if (fs.readdirSync(configDir).length === 0) {
+              fs.rmSync(configDir, { recursive: true, force: true });
             }
+          } catch (e) {}
+        } catch (err) {
+          log.error(`Failed to delete native wallet: ${err.message}`);
         }
-    }
-
-    // 3. Uninstall Rust
-    const rustInstalled = shell.which('rustc');
-    if (rustInstalled) {
-        const removeRust = await askQuestion("\n🦀 Remove Rust? (y/N): ");
-        if (removeRust.toLowerCase() === 'y') {
-            log.info("Running rustup self uninstall...");
-            runOrElevate('rustup self uninstall -y', 'Uninstall Rust');
+      }
+
+      if (hasWslWallet) {
+        const deleted = runOrElevate(
+          `rm -f "${wslWalletPath}"`,
+          "Delete WSL wallet",
+          stealth,
+        );
+        if (deleted) {
+          // Best effort cleanup of config dir if empty.
+          runOrElevate(
+            'rmdir "$HOME/.config/solana" 2>/dev/null || true',
+            "Cleanup WSL wallet config directory",
+            stealth,
+          );
+          log.success("WSL wallet deleted.");
         }
+      }
+    } else {
+      log.info("Skipping wallet deletion.");
     }
-
-    // 4. WALLET REMOVAL (DANGER)
-    const walletPath = path.join(os.homedir(), '.config', 'solana', 'id.json');
-    if (fs.existsSync(walletPath)) {
-        log.error("\n⚠️  DANGER ZONE ⚠️");
-        log.warn(`Found a Solana wallet at: ${walletPath}`);
-        log.warn("If you delete this without a backup, your funds will be LOST FOREVER.");
-
-        const removeWallet = await askQuestion("💥 Do you REALLY want to delete this wallet? (type 'DELETE' to confirm): ");
-        if (removeWallet === 'DELETE') {
-            try {
-                fs.unlinkSync(walletPath);
-                log.success("Wallet deleted.");
-
-                // Clean up parent config dir if empty
-                const configDir = path.dirname(walletPath);
-                try {
-                    if (fs.readdirSync(configDir).length === 0) {
-                        fs.rmSync(configDir, { recursive: true, force: true });
-                    }
-                } catch (e) { }
-            } catch (err) {
-                log.error(`Failed to delete wallet: ${err.message}`);
-            }
-        } else {
-            log.info("Skipping wallet deletion.");
+  }
+
+  if (stealth.enabled) {
+    const configPath = path.join(os.homedir(), ".uso-config.json");
+    if (fs.existsSync(configPath)) {
+      const disableStealth = await askQuestion(
+        "\n🐧 Disable Stealth WSL mode for USO? (y/N): ",
+      );
+      if (disableStealth.toLowerCase() === "y") {
+        try {
+          fs.rmSync(configPath, { force: true });
+          log.success("Stealth mode config removed.");
+        } catch (err) {
+          log.warn(`Failed to remove stealth config: ${err.message}`);
         }
+      }
     }
+  }
 
-    log.header("\n✅ Cleanup complete.");
-    log.info("To remove the 'uso' tool itself, run:");
-    log.info("   npm uninstall -g uso");
+  log.header("\n✅ Cleanup complete.");
+  log.info("To remove the 'uso' tool itself, run:");
+  log.info("   npm uninstall -g uso");
 };
 
 module.exports = { uninstall };