droid-patch 0.1.2 → 0.2.0

package/dist/cli.js

@@ -1,12 +1,665 @@
 #!/usr/bin/env node
-import { createAlias, listAliases, patchDroid, removeAlias } from "./alias-DcCF7R2B.js";
+import { createAlias, createAliasForWrapper, listAliases, patchDroid, removeAlias } from "./alias-C9LRaTwF.js";
 import bin from "tiny-bin";
 import { styleText } from "node:util";
 import { existsSync, readFileSync } from "node:fs";
 import { dirname, join } from "node:path";
 import { homedir } from "node:os";
 import { fileURLToPath } from "node:url";
+import { chmod, mkdir, writeFile } from "node:fs/promises";
 
+//#region src/websearch-patch.ts
+/**
+* Generate search proxy server code (runs in background)
+* Since BUN_CONFIG_PRELOAD doesn't work with compiled binaries,
+* use a local proxy server to intercept search requests instead
+*
+* Features:
+* - Auto-shutdown on idle (default 5 minutes without requests)
+* - Timeout configurable via DROID_PROXY_IDLE_TIMEOUT env var (seconds)
+* - Set to 0 to disable timeout
+*/
+function generateSearchProxyServer() {
+	return `#!/usr/bin/env node
+// Droid WebSearch Proxy Server
+// Auto-generated by droid-patch --websearch
+
+const http = require('http');
+const https = require('https');
+const { execSync } = require('child_process');
+const fs = require('fs');
+
+const DEBUG = process.env.DROID_SEARCH_DEBUG === '1';
+const PORT = parseInt(process.env.SEARCH_PROXY_PORT || '23119');
+
+// Idle timeout in seconds, default 5 minutes, set to 0 to disable
+const IDLE_TIMEOUT = parseInt(process.env.DROID_PROXY_IDLE_TIMEOUT || '300');
+let lastActivityTime = Date.now();
+let idleCheckTimer = null;
+
+function log(...args) {
+  if (DEBUG) console.error('[websearch]', ...args);
+}
+
+// Update activity time
+function updateActivity() {
+  lastActivityTime = Date.now();
+}
+
+// Check if any droid process is running (droid instances using the proxy)
+function isDroidRunning() {
+  try {
+    const { execSync } = require('child_process');
+    // Use ps to check if droid.patched binary is running
+    // Exclude scripts and grep itself, only match actual droid binary processes
+    const result = execSync(
+      'ps aux | grep -E "[d]roid\\\\.patched" | grep -v grep | wc -l',
+      { encoding: 'utf-8', timeout: 1000 }
+    ).trim();
+    return parseInt(result) > 0;
+  } catch {
+    return false;
+  }
+}
+
+// Check idle time and possibly exit
+function checkIdleAndExit() {
+  if (IDLE_TIMEOUT <= 0) return; // Timeout disabled
+
+  // If droid process is running, refresh activity time (like heartbeat)
+  if (isDroidRunning()) {
+    log('Droid process detected, keeping proxy alive');
+    updateActivity();
+    return;
+  }
+
+  const idleMs = Date.now() - lastActivityTime;
+  const timeoutMs = IDLE_TIMEOUT * 1000;
+
+  if (idleMs >= timeoutMs) {
+    log(\`Idle for \${Math.round(idleMs / 1000)}s and no droid running, shutting down...\`);
+    cleanup();
+    process.exit(0);
+  }
+}
+
+// Cleanup resources
+function cleanup() {
+  if (idleCheckTimer) {
+    clearInterval(idleCheckTimer);
+    idleCheckTimer = null;
+  }
+  // Delete PID file
+  try {
+    fs.unlinkSync('/tmp/droid-search-proxy.pid');
+  } catch {}
+}
+
+// === Search Implementation ===
+
+// Smithery Exa MCP - highest priority, requires SMITHERY_API_KEY and SMITHERY_PROFILE
+async function searchSmitheryExa(query, numResults) {
+  const apiKey = process.env.SMITHERY_API_KEY;
+  const profile = process.env.SMITHERY_PROFILE;
+  if (!apiKey || !profile) return null;
+
+  try {
+    // Construct URL with authentication
+    const serverUrl = \`https://server.smithery.ai/exa/mcp?api_key=\${encodeURIComponent(apiKey)}&profile=\${encodeURIComponent(profile)}\`;
+    log('Smithery Exa request');
+
+    // Use MCP protocol to call the search tool via HTTP POST
+    const requestBody = JSON.stringify({
+      jsonrpc: '2.0',
+      id: 1,
+      method: 'tools/call',
+      params: {
+        name: 'web_search_exa',
+        arguments: {
+          query: query,
+          numResults: numResults
+        }
+      }
+    });
+
+    const curlCmd = \`curl -s -X POST "\${serverUrl}" -H "Content-Type: application/json" -d '\${requestBody.replace(/'/g, "'\\\\\\\\''")}'\`;
+    const jsonStr = execSync(curlCmd, { encoding: 'utf-8', timeout: 30000 });
+    const response = JSON.parse(jsonStr);
+
+    // Parse MCP response
+    if (response.result && response.result.content) {
+      // MCP returns content as array of text blocks
+      const textContent = response.result.content.find(c => c.type === 'text');
+      if (textContent && textContent.text) {
+        try {
+          const searchResults = JSON.parse(textContent.text);
+          if (Array.isArray(searchResults) && searchResults.length > 0) {
+            return searchResults.slice(0, numResults).map(item => ({
+              title: item.title || '',
+              url: item.url || '',
+              content: item.text || item.snippet || item.highlights?.join(' ') || '',
+              publishedDate: item.publishedDate || null,
+              author: item.author || null,
+              score: item.score || null
+            }));
+          }
+        } catch (parseErr) {
+          log('Smithery response parsing failed');
+        }
+      }
+    }
+
+    if (response.error) {
+      log('Smithery Exa error:', response.error.message || response.error);
+      return null;
+    }
+  } catch (e) {
+    log('Smithery Exa failed:', e.message);
+    return null;
+  }
+  return null;
+}
+
+async function searchGooglePSE(query, numResults) {
+  const apiKey = process.env.GOOGLE_PSE_API_KEY;
+  const cx = process.env.GOOGLE_PSE_CX;
+  if (!apiKey || !cx) return null;
+
+  try {
+    const url = \`https://www.googleapis.com/customsearch/v1?key=\${apiKey}&cx=\${cx}&q=\${encodeURIComponent(query)}&num=\${Math.min(numResults, 10)}\`;
+    log('Google PSE request:', url.replace(apiKey, '***'));
+
+    const curlCmd = \`curl -s "\${url}"\`;
+    const jsonStr = execSync(curlCmd, { encoding: 'utf-8', timeout: 15000 });
+    const data = JSON.parse(jsonStr);
+
+    if (data.error) {
+      log('Google PSE error:', data.error.message);
+      return null;
+    }
+    return (data.items || []).map(item => ({
+      title: item.title,
+      url: item.link,
+      content: item.snippet || '',
+      publishedDate: null,
+      author: null,
+      score: null
+    }));
+  } catch (e) {
+    log('Google PSE failed:', e.message);
+    return null;
+  }
+}
+
+// SearXNG - self-hosted meta search engine
+async function searchSearXNG(query, numResults) {
+  const searxngUrl = process.env.SEARXNG_URL;
+  if (!searxngUrl) return null;
+
+  try {
+    const url = \`\${searxngUrl}/search?q=\${encodeURIComponent(query)}&format=json&engines=google,bing,duckduckgo\`;
+    log('SearXNG request:', url);
+
+    const curlCmd = \`curl -s "\${url}" -H "Accept: application/json"\`;
+    const jsonStr = execSync(curlCmd, { encoding: 'utf-8', timeout: 15000 });
+    const data = JSON.parse(jsonStr);
+
+    if (data.results && data.results.length > 0) {
+      return data.results.slice(0, numResults).map(item => ({
+        title: item.title,
+        url: item.url,
+        content: item.content || '',
+        publishedDate: null,
+        author: null,
+        score: null
+      }));
+    }
+  } catch (e) {
+    log('SearXNG failed:', e.message);
+  }
+  return null;
+}
+
+// Serper API - free tier available (2500 queries/month)
+async function searchSerper(query, numResults) {
+  const apiKey = process.env.SERPER_API_KEY;
+  if (!apiKey) return null;
+
+  try {
+    const curlCmd = \`curl -s "https://google.serper.dev/search" -H "X-API-KEY: \${apiKey}" -H "Content-Type: application/json" -d '{"q":"\${query.replace(/"/g, '\\\\"')}","num":\${numResults}}'\`;
+    log('Serper request');
+
+    const jsonStr = execSync(curlCmd, { encoding: 'utf-8', timeout: 15000 });
+    const data = JSON.parse(jsonStr);
+
+    if (data.organic && data.organic.length > 0) {
+      return data.organic.slice(0, numResults).map(item => ({
+        title: item.title,
+        url: item.link,
+        content: item.snippet || '',
+        publishedDate: null,
+        author: null,
+        score: null
+      }));
+    }
+  } catch (e) {
+    log('Serper failed:', e.message);
+  }
+  return null;
+}
+
+// Brave Search API - free tier available
+async function searchBrave(query, numResults) {
+  const apiKey = process.env.BRAVE_API_KEY;
+  if (!apiKey) return null;
+
+  try {
+    const url = \`https://api.search.brave.com/res/v1/web/search?q=\${encodeURIComponent(query)}&count=\${numResults}\`;
+    const curlCmd = \`curl -s "\${url}" -H "Accept: application/json" -H "X-Subscription-Token: \${apiKey}"\`;
+    log('Brave request');
+
+    const jsonStr = execSync(curlCmd, { encoding: 'utf-8', timeout: 15000 });
+    const data = JSON.parse(jsonStr);
+
+    if (data.web && data.web.results && data.web.results.length > 0) {
+      return data.web.results.slice(0, numResults).map(item => ({
+        title: item.title,
+        url: item.url,
+        content: item.description || '',
+        publishedDate: null,
+        author: null,
+        score: null
+      }));
+    }
+  } catch (e) {
+    log('Brave failed:', e.message);
+  }
+  return null;
+}
+
+// DuckDuckGo - limited reliability due to bot detection
+async function searchDuckDuckGo(query, numResults) {
+  // DuckDuckGo Instant Answer API (limited results but more reliable)
+  try {
+    const apiUrl = \`https://api.duckduckgo.com/?q=\${encodeURIComponent(query)}&format=json&no_html=1&skip_disambig=1\`;
+    const curlCmd = \`curl -s "\${apiUrl}" -H "User-Agent: Mozilla/5.0"\`;
+    const jsonStr = execSync(curlCmd, { encoding: 'utf-8', timeout: 15000 });
+    const data = JSON.parse(jsonStr);
+
+    const results = [];
+
+    if (data.Abstract && data.AbstractURL) {
+      results.push({
+        title: data.Heading || query,
+        url: data.AbstractURL,
+        content: data.Abstract,
+        publishedDate: null,
+        author: null,
+        score: null
+      });
+    }
+
+    for (const topic of (data.RelatedTopics || [])) {
+      if (results.length >= numResults) break;
+      if (topic.Text && topic.FirstURL) {
+        results.push({
+          title: topic.Text.substring(0, 100),
+          url: topic.FirstURL,
+          content: topic.Text,
+          publishedDate: null,
+          author: null,
+          score: null
+        });
+      }
+      if (topic.Topics) {
+        for (const st of topic.Topics) {
+          if (results.length >= numResults) break;
+          if (st.Text && st.FirstURL) {
+            results.push({
+              title: st.Text.substring(0, 100),
+              url: st.FirstURL,
+              content: st.Text,
+              publishedDate: null,
+              author: null,
+              score: null
+            });
+          }
+        }
+      }
+    }
+
+    if (results.length > 0) {
+      log('DDG API:', results.length, 'results');
+      return results;
+    }
+  } catch (e) {
+    log('DDG API failed:', e.message);
+  }
+
+  return [];
+}
+
+function parseDDGLiteHTML(html, maxResults) {
+  const results = [];
+  const linkRegex = /<a[^>]+rel="nofollow"[^>]+href="([^"]+)"[^>]*>([^<]+)<\\/a>/gi;
+  const snippetRegex = /<td[^>]*class="result-snippet"[^>]*>([^<]*)<\\/td>/gi;
+
+  const links = [];
+  let match;
+
+  while ((match = linkRegex.exec(html)) !== null && links.length < maxResults) {
+    let url = match[1];
+    if (url.includes('duckduckgo.com') && !url.includes('uddg=')) continue;
+    if (url.includes('uddg=')) {
+      const uddgMatch = url.match(/uddg=([^&]+)/);
+      if (uddgMatch) url = decodeURIComponent(uddgMatch[1]);
+    }
+    links.push({
+      url: url,
+      title: decodeHTMLEntities(match[2].trim())
+    });
+  }
+
+  const snippets = [];
+  while ((match = snippetRegex.exec(html)) !== null && snippets.length < maxResults) {
+    snippets.push(decodeHTMLEntities(match[1].trim()));
+  }
+
+  for (let i = 0; i < links.length && results.length < maxResults; i++) {
+    results.push({
+      title: links[i].title,
+      url: links[i].url,
+      content: snippets[i] || '',
+      publishedDate: null,
+      author: null,
+      score: null
+    });
+  }
+
+  return results;
+}
+
+function decodeHTMLEntities(str) {
+  return str
+    .replace(/&amp;/g, '&')
+    .replace(/&lt;/g, '<')
+    .replace(/&gt;/g, '>')
+    .replace(/&quot;/g, '"')
+    .replace(/&#39;/g, "'")
+    .replace(/&nbsp;/g, ' ');
+}
+
+async function search(query, numResults = 10) {
+  // Priority order:
+  // 1. Smithery Exa MCP (best quality if configured)
+  // 2. Google PSE (most reliable if configured)
+  // 3. Serper (free tier: 2500/month)
+  // 4. Brave Search (free tier available)
+  // 5. SearXNG (self-hosted)
+  // 6. DuckDuckGo (limited due to bot detection)
+
+  // 1. Smithery Exa MCP (highest priority)
+  const smitheryResults = await searchSmitheryExa(query, numResults);
+  if (smitheryResults && smitheryResults.length > 0) {
+    log('Using Smithery Exa');
+    return { results: smitheryResults, source: 'smithery-exa' };
+  }
+
+  // 2. Google PSE
+  const googleResults = await searchGooglePSE(query, numResults);
+  if (googleResults && googleResults.length > 0) {
+    log('Using Google PSE');
+    return { results: googleResults, source: 'google-pse' };
+  }
+
+  // 3. Serper
+  const serperResults = await searchSerper(query, numResults);
+  if (serperResults && serperResults.length > 0) {
+    log('Using Serper');
+    return { results: serperResults, source: 'serper' };
+  }
+
+  // 4. Brave Search
+  const braveResults = await searchBrave(query, numResults);
+  if (braveResults && braveResults.length > 0) {
+    log('Using Brave Search');
+    return { results: braveResults, source: 'brave' };
+  }
+
+  // 5. SearXNG
+  const searxngResults = await searchSearXNG(query, numResults);
+  if (searxngResults && searxngResults.length > 0) {
+    log('Using SearXNG');
+    return { results: searxngResults, source: 'searxng' };
+  }
+
+  // 6. DuckDuckGo (last resort, limited results)
+  log('Using DuckDuckGo (fallback)');
+  const ddgResults = await searchDuckDuckGo(query, numResults);
+  return { results: ddgResults, source: 'duckduckgo' };
+}
+
+// === HTTP Proxy Server ===
+
+const FACTORY_API = 'https://api.factory.ai';
+
+const server = http.createServer(async (req, res) => {
+  const url = new URL(req.url, \`http://\${req.headers.host}\`);
+
+  // Health check - don't update activity time to avoid self-ping preventing timeout
+  if (url.pathname === '/health') {
+    const idleSeconds = Math.round((Date.now() - lastActivityTime) / 1000);
+    const droidRunning = isDroidRunning();
+    res.writeHead(200, { 'Content-Type': 'application/json' });
+    res.end(JSON.stringify({
+      status: 'ok',
+      port: PORT,
+      idleTimeout: IDLE_TIMEOUT,
+      idleSeconds: idleSeconds,
+      droidRunning: droidRunning,
+      // If droid is running, won't shutdown; otherwise calculate based on idle time
+      willShutdownIn: IDLE_TIMEOUT > 0 && !droidRunning ? Math.max(0, IDLE_TIMEOUT - idleSeconds) : null
+    }));
+    return;
+  }
+
+  // Update activity time (only non-health-check requests refresh it)
+  updateActivity();
+
+  // Search endpoint - intercept
+  if (url.pathname === '/api/tools/exa/search' && req.method === 'POST') {
+    let body = '';
+    req.on('data', c => body += c);
+    req.on('end', async () => {
+      try {
+        const { query, numResults } = JSON.parse(body);
+        log('Search query:', query);
+        const { results, source } = await search(query, numResults || 10);
+        log('Results:', results.length, 'from', source);
+        res.writeHead(200, { 'Content-Type': 'application/json' });
+        res.end(JSON.stringify({ results }));
+      } catch (e) {
+        log('Search error:', e.message);
+        res.writeHead(500, { 'Content-Type': 'application/json' });
+        res.end(JSON.stringify({ error: String(e), results: [] }));
+      }
+    });
+    return;
+  }
+
+  // Proxy all other requests to Factory API
+  log('Proxy:', req.method, url.pathname);
+
+  const proxyUrl = new URL(FACTORY_API + url.pathname + url.search);
+  const proxyReq = https.request(proxyUrl, {
+    method: req.method,
+    headers: { ...req.headers, host: proxyUrl.host }
+  }, proxyRes => {
+    res.writeHead(proxyRes.statusCode, proxyRes.headers);
+    proxyRes.pipe(res);
+  });
+
+  proxyReq.on('error', e => {
+    log('Proxy error:', e.message);
+    res.writeHead(502, { 'Content-Type': 'application/json' });
+    res.end(JSON.stringify({ error: 'Proxy failed: ' + e.message }));
+  });
+
+  if (req.method !== 'GET' && req.method !== 'HEAD') {
+    req.pipe(proxyReq);
+  } else {
+    proxyReq.end();
+  }
+});
+
+// If port is 0, system will automatically assign an available port
+server.listen(PORT, '127.0.0.1', () => {
+  const actualPort = server.address().port;
+  const hasGoogle = process.env.GOOGLE_PSE_API_KEY && process.env.GOOGLE_PSE_CX;
+
+  // Write port file for parent process to read
+  const portFile = process.env.SEARCH_PROXY_PORT_FILE;
+  if (portFile) {
+    fs.writeFileSync(portFile, String(actualPort));
+  }
+
+  const hasSmithery = process.env.SMITHERY_API_KEY && process.env.SMITHERY_PROFILE;
+  log('Search proxy started on http://127.0.0.1:' + actualPort);
+  log('Smithery Exa:', hasSmithery ? 'configured (priority 1)' : 'not set');
+  log('Google PSE:', hasGoogle ? 'configured' : 'not set');
+  log('Serper:', process.env.SERPER_API_KEY ? 'configured' : 'not set');
+  log('Brave:', process.env.BRAVE_API_KEY ? 'configured' : 'not set');
+  log('SearXNG:', process.env.SEARXNG_URL || 'not set');
+
+  // Start idle check timer
+  // Check interval = min(timeout/2, 30s) to ensure timely timeout detection
+  if (IDLE_TIMEOUT > 0) {
+    const checkInterval = Math.min(IDLE_TIMEOUT * 500, 30000); // milliseconds
+    log(\`Idle timeout: \${IDLE_TIMEOUT}s (will auto-shutdown when idle)\`);
+    idleCheckTimer = setInterval(checkIdleAndExit, checkInterval);
+  } else {
+    log('Idle timeout: disabled (will run forever)');
+  }
+});
+
+process.on('SIGTERM', () => { cleanup(); server.close(); process.exit(0); });
+process.on('SIGINT', () => { cleanup(); server.close(); process.exit(0); });
+`;
+}
+/**
+* Generate unified Wrapper script
+* Uses shared proxy server mode:
+* - All droid instances share the same proxy process
+* - Proxy starts automatically if not running
+* - Proxy runs as background daemon, doesn't exit with droid
+*/
+function generateUnifiedWrapper(droidPath, proxyScriptPath) {
+	return `#!/bin/bash
+# Droid with WebSearch
+# Auto-generated by droid-patch --websearch
+
+PROXY_SCRIPT="${proxyScriptPath}"
+DROID_BIN="${droidPath}"
+PORT=23119
+PID_FILE="/tmp/droid-search-proxy.pid"
+LOG_FILE="/tmp/droid-search-proxy.log"
+
+# Check if proxy is running
+is_proxy_running() {
+  if [ -f "$PID_FILE" ]; then
+    local pid
+    pid=$(cat "$PID_FILE")
+    if kill -0 "$pid" 2>/dev/null; then
+      # Process exists, check if port responds
+      if curl -s "http://127.0.0.1:$PORT/health" > /dev/null 2>&1; then
+        return 0
+      fi
+    fi
+    # PID file exists but process doesn't exist or port not responding, cleanup
+    rm -f "$PID_FILE"
+  fi
+  return 1
+}
+
+# Start shared proxy
+start_shared_proxy() {
+  # First check if port is occupied by another program
+  if lsof -i:"$PORT" > /dev/null 2>&1; then
+    # Port is occupied, check if it's our proxy
+    if curl -s "http://127.0.0.1:$PORT/health" > /dev/null 2>&1; then
+      [ -n "$DROID_SEARCH_DEBUG" ] && echo "[websearch] Proxy already running on port $PORT" >&2
+      return 0
+    else
+      echo "[websearch] Port $PORT is occupied by another process" >&2
+      return 1
+    fi
+  fi
+
+  [ -n "$DROID_SEARCH_DEBUG" ] && echo "[websearch] Starting shared proxy on port $PORT..." >&2
+
+  # Start proxy as background daemon
+  SEARCH_PROXY_PORT="$PORT" nohup node "$PROXY_SCRIPT" >> "$LOG_FILE" 2>&1 &
+  echo $! > "$PID_FILE"
+
+  # Wait for proxy to start
+  for i in {1..30}; do
+    if curl -s "http://127.0.0.1:$PORT/health" > /dev/null 2>&1; then
+      [ -n "$DROID_SEARCH_DEBUG" ] && echo "[websearch] Proxy ready on port $PORT (PID: $(cat $PID_FILE))" >&2
+      return 0
+    fi
+    sleep 0.1
+  done
+
+  echo "[websearch] Failed to start proxy" >&2
+  rm -f "$PID_FILE"
+  return 1
+}
+
+# Ensure proxy is running
+ensure_proxy() {
+  if is_proxy_running; then
+    [ -n "$DROID_SEARCH_DEBUG" ] && echo "[websearch] Using existing proxy on port $PORT" >&2
+    return 0
+  fi
+  start_shared_proxy
+}
+
+# Start/reuse proxy
+if ! ensure_proxy; then
+  echo "[websearch] Running without search proxy" >&2
+  exec "$DROID_BIN" "$@"
+fi
+
+# Run droid, set API to point to local proxy
+export FACTORY_API_BASE_URL_OVERRIDE="http://127.0.0.1:$PORT"
+exec "$DROID_BIN" "$@"
+`;
+}
+/**
+* Create unified WebSearch files
+*
+* Approach: Proxy server mode
+* - wrapper script starts local proxy server
+* - proxy server intercepts search requests, passes through other requests
+* - uses FACTORY_API_BASE_URL_OVERRIDE env var to point to proxy
+* - alias works directly, no extra steps needed
+*/
+async function createWebSearchUnifiedFiles(outputDir, droidPath, aliasName) {
+	if (!existsSync(outputDir)) await mkdir(outputDir, { recursive: true });
+	const proxyScriptPath = join(outputDir, `${aliasName}-proxy.js`);
+	const wrapperScriptPath = join(outputDir, aliasName);
+	await writeFile(proxyScriptPath, generateSearchProxyServer());
+	console.log(`[*] Created proxy script: ${proxyScriptPath}`);
+	await writeFile(wrapperScriptPath, generateUnifiedWrapper(droidPath, proxyScriptPath));
+	await chmod(wrapperScriptPath, 493);
+	console.log(`[*] Created wrapper: ${wrapperScriptPath}`);
+	return {
+		wrapperScript: wrapperScriptPath,
+		preloadScript: proxyScriptPath
+	};
+}
+
+//#endregion
 //#region src/cli.ts
 const __dirname = dirname(fileURLToPath(import.meta.url));
 function getVersion() {
@@ -29,26 +682,70 @@ function findDefaultDroidPath() {
 	for (const p of paths) if (existsSync(p)) return p;
 	return join(home, ".droid/bin/droid");
 }
-bin("droid-patch", "CLI tool to patch droid binary with various modifications").package("droid-patch", version).option("--is-custom", "Patch isCustom:!0 to isCustom:!1 (enable context compression for custom models)").option("--skip-login", "Inject a fake FACTORY_API_KEY to bypass login requirement (no real key needed)").option("--dry-run", "Verify patches without actually modifying the binary").option("-p, --path <path>", "Path to the droid binary").option("-o, --output <dir>", "Output directory for patched binary").option("--no-backup", "Do not create backup of original binary").option("-v, --verbose", "Enable verbose output").argument("[alias]", "Alias name for the patched binary").action(async (options, args) => {
+bin("droid-patch", "CLI tool to patch droid binary with various modifications").package("droid-patch", version).option("--is-custom", "Patch isCustom:!0 to isCustom:!1 (enable context compression for custom models)").option("--skip-login", "Inject a fake FACTORY_API_KEY to bypass login requirement (no real key needed)").option("--api-base <url>", "Replace Factory API base URL (https://api.factory.ai) with custom URL").option("--websearch", "Enable local WebSearch via fetch hook (Google PSE + DuckDuckGo fallback)").option("--dry-run", "Verify patches without actually modifying the binary").option("-p, --path <path>", "Path to the droid binary").option("-o, --output <dir>", "Output directory for patched binary").option("--no-backup", "Do not create backup of original binary").option("-v, --verbose", "Enable verbose output").argument("[alias]", "Alias name for the patched binary").action(async (options, args) => {
 	const alias = args?.[0];
 	const isCustom = options["is-custom"];
 	const skipLogin = options["skip-login"];
+	const apiBase = options["api-base"];
+	const webSearch = options["websearch"];
 	const dryRun = options["dry-run"];
 	const path = options.path || findDefaultDroidPath();
 	const outputDir = options.output;
 	const backup = options.backup !== false;
 	const verbose = options.verbose;
 	const outputPath = outputDir && alias ? join(outputDir, alias) : void 0;
-	if (!isCustom && !skipLogin) {
+	if (webSearch && !isCustom && !skipLogin && !apiBase) {
+		if (!alias) {
+			console.log(styleText("red", "Error: Alias name required for --websearch"));
+			console.log(styleText("gray", "Usage: npx droid-patch --websearch <alias>"));
+			process.exit(1);
+		}
+		console.log(styleText("cyan", "═".repeat(60)));
+		console.log(styleText(["cyan", "bold"], "  Droid WebSearch Setup"));
+		console.log(styleText("cyan", "═".repeat(60)));
+		console.log();
+		const websearchDir = join(homedir(), ".droid-patch", "websearch");
+		const { wrapperScript } = await createWebSearchUnifiedFiles(websearchDir, path, alias);
+		await createAliasForWrapper(wrapperScript, alias, verbose);
+		console.log();
+		console.log(styleText("green", "═".repeat(60)));
+		console.log(styleText(["green", "bold"], "  WebSearch Ready!"));
+		console.log(styleText("green", "═".repeat(60)));
+		console.log();
+		console.log("Run directly:");
+		console.log(styleText("yellow", `  ${alias}`));
+		console.log();
+		console.log(styleText("cyan", "Auto-shutdown:"));
+		console.log(styleText("gray", "  Proxy auto-shuts down after 5 min idle (no manual cleanup needed)"));
+		console.log(styleText("gray", "  To disable: export DROID_PROXY_IDLE_TIMEOUT=0"));
+		console.log();
+		console.log("Search providers (in priority order):");
+		console.log(styleText("yellow", "  1. Smithery Exa (best quality):"));
+		console.log(styleText("gray", "     export SMITHERY_API_KEY=your_api_key"));
+		console.log(styleText("gray", "     export SMITHERY_PROFILE=your_profile"));
+		console.log(styleText("gray", "  2. Google PSE:"));
+		console.log(styleText("gray", "     export GOOGLE_PSE_API_KEY=your_api_key"));
+		console.log(styleText("gray", "     export GOOGLE_PSE_CX=your_search_engine_id"));
+		console.log(styleText("gray", "  3-6. Serper, Brave, SearXNG, DuckDuckGo (fallbacks)"));
+		console.log();
+		console.log("Debug mode:");
+		console.log(styleText("gray", "  export DROID_SEARCH_DEBUG=1"));
+		return;
+	}
+	if (!isCustom && !skipLogin && !apiBase && !webSearch) {
 		console.log(styleText("yellow", "No patch flags specified. Available patches:"));
 		console.log(styleText("gray", "  --is-custom    Patch isCustom for custom models"));
 		console.log(styleText("gray", "  --skip-login   Bypass login by injecting a fake API key"));
+		console.log(styleText("gray", "  --api-base     Replace Factory API URL with custom server"));
+		console.log(styleText("gray", "  --websearch    Enable local WebSearch (Google PSE + DuckDuckGo)"));
 		console.log();
 		console.log("Usage examples:");
 		console.log(styleText("cyan", "  npx droid-patch --is-custom droid-custom"));
 		console.log(styleText("cyan", "  npx droid-patch --skip-login droid-nologin"));
 		console.log(styleText("cyan", "  npx droid-patch --is-custom --skip-login droid-patched"));
 		console.log(styleText("cyan", "  npx droid-patch --skip-login -o . my-droid"));
+		console.log(styleText("cyan", "  npx droid-patch --api-base http://localhost:3000 droid-local"));
+		console.log(styleText("cyan", "  npx droid-patch --websearch droid-search"));
 		process.exit(1);
 	}
 	if (!alias && !dryRun) {
@@ -73,6 +770,29 @@ bin("droid-patch", "CLI tool to patch droid binary with various modifications").
 		pattern: Buffer.from("process.env.FACTORY_API_KEY"),
 		replacement: Buffer.from("\"fk-droid-patch-skip-00000\"")
 	});
+	if (apiBase) {
+		const originalUrl = "https://api.factory.ai";
+		const originalLength = originalUrl.length;
+		let normalizedUrl = apiBase.replace(/\/+$/, "");
+		if (normalizedUrl.length > originalLength) {
+			console.log(styleText("red", `Error: API base URL must be ${originalLength} characters or less`));
+			console.log(styleText("gray", `  Your URL: "${normalizedUrl}" (${normalizedUrl.length} chars)`));
+			console.log(styleText("gray", `  Maximum:  ${originalLength} characters`));
+			console.log();
+			console.log(styleText("yellow", "Tip: Use a shorter URL or set up a local redirect."));
+			console.log(styleText("gray", "  Examples:"));
+			console.log(styleText("gray", "    http://127.0.0.1:3000 (19 chars)"));
+			console.log(styleText("gray", "    http://localhost:80  (19 chars)"));
+			process.exit(1);
+		}
+		const paddedUrl = normalizedUrl.padEnd(originalLength, " ");
+		patches.push({
+			name: "apiBase",
+			description: `Replace Factory API URL with "${normalizedUrl}"`,
+			pattern: Buffer.from(originalUrl),
+			replacement: Buffer.from(paddedUrl)
+		});
+	}
 	try {
 		const result = await patchDroid({
 			inputPath: path,
@@ -103,7 +823,15 @@ bin("droid-patch", "CLI tool to patch droid binary with various modifications").
 		}
 		if (result.success && result.outputPath && alias) {
 			console.log();
-			await createAlias(result.outputPath, alias, verbose);
+			if (webSearch) {
+				const websearchDir = join(homedir(), ".droid-patch", "websearch");
+				const { wrapperScript } = await createWebSearchUnifiedFiles(websearchDir, result.outputPath, alias);
+				await createAliasForWrapper(wrapperScript, alias, verbose);
+				console.log();
+				console.log("Optional: Set Google PSE for better results:");
+				console.log(styleText("gray", "  export GOOGLE_PSE_API_KEY=your_api_key"));
+				console.log(styleText("gray", "  export GOOGLE_PSE_CX=your_search_engine_id"));
+			} else await createAlias(result.outputPath, alias, verbose);
 		}
 		if (result.success) {
 			console.log();
@@ -122,9 +850,9 @@ bin("droid-patch", "CLI tool to patch droid binary with various modifications").
 }).command("remove", "Remove a droid-patch alias or patched binary file").argument("<alias-or-path>", "Alias name or file path to remove").action(async (_options, args) => {
 	const target = args[0];
 	if (target.includes("/") || existsSync(target)) {
-		const { unlink } = await import("node:fs/promises");
+		const { unlink: unlink$1 } = await import("node:fs/promises");
 		try {
-			await unlink(target);
+			await unlink$1(target);
 			console.log(styleText("green", `[*] Removed: ${target}`));
 		} catch (error) {
 			console.error(styleText("red", `Error: ${error.message}`));
@@ -133,6 +861,86 @@ bin("droid-patch", "CLI tool to patch droid binary with various modifications").
 	} else await removeAlias(target);
 }).command("version", "Print droid-patch version").action(() => {
 	console.log(`droid-patch v${version}`);
+}).command("proxy-status", "Check websearch proxy status").action(async () => {
+	const pidFile = "/tmp/droid-search-proxy.pid";
+	const logFile = "/tmp/droid-search-proxy.log";
+	const port = 23119;
+	console.log(styleText("cyan", "═".repeat(60)));
+	console.log(styleText(["cyan", "bold"], "  WebSearch Proxy Status"));
+	console.log(styleText("cyan", "═".repeat(60)));
+	console.log();
+	try {
+		const response = await fetch(`http://127.0.0.1:${port}/health`);
+		if (response.ok) {
+			const data = await response.json();
+			console.log(styleText("green", `  Status: Running ✓`));
+			console.log(styleText("white", `  Port: ${port}`));
+			if (existsSync(pidFile)) {
+				const { readFileSync: readFileSync$1 } = await import("node:fs");
+				const pid = readFileSync$1(pidFile, "utf-8").trim();
+				console.log(styleText("white", `  PID: ${pid}`));
+			}
+			if (data.droidRunning !== void 0) console.log(styleText("white", `  Droid running: ${data.droidRunning ? "yes (proxy will stay alive)" : "no"}`));
+			if (data.idleTimeout !== void 0) if (data.idleTimeout > 0) {
+				const idleMins = Math.floor((data.idleSeconds || 0) / 60);
+				const idleSecs = (data.idleSeconds || 0) % 60;
+				if (data.droidRunning) console.log(styleText("white", `  Idle: ${idleMins}m ${idleSecs}s (won't shutdown while droid runs)`));
+				else if (data.willShutdownIn !== null) {
+					const shutdownMins = Math.floor((data.willShutdownIn || 0) / 60);
+					const shutdownSecs = (data.willShutdownIn || 0) % 60;
+					console.log(styleText("white", `  Idle: ${idleMins}m ${idleSecs}s`));
+					console.log(styleText("white", `  Auto-shutdown in: ${shutdownMins}m ${shutdownSecs}s`));
+				}
+			} else console.log(styleText("white", `  Auto-shutdown: disabled`));
+			console.log(styleText("white", `  Log: ${logFile}`));
+			console.log();
+			console.log(styleText("gray", "To stop the proxy manually:"));
+			console.log(styleText("cyan", "  npx droid-patch proxy-stop"));
+			console.log();
+			console.log(styleText("gray", "To disable auto-shutdown:"));
+			console.log(styleText("cyan", "  export DROID_PROXY_IDLE_TIMEOUT=0"));
+		}
+	} catch {
+		console.log(styleText("yellow", `  Status: Not running`));
+		console.log();
+		console.log(styleText("gray", "The proxy will start automatically when you run droid-full."));
+		console.log(styleText("gray", "It will auto-shutdown after 5 minutes of idle (configurable)."));
+	}
+	console.log();
+}).command("proxy-stop", "Stop the websearch proxy").action(async () => {
+	const pidFile = "/tmp/droid-search-proxy.pid";
+	if (!existsSync(pidFile)) {
+		console.log(styleText("yellow", "Proxy is not running (no PID file)"));
+		return;
+	}
+	try {
+		const { readFileSync: readFileSync$1, unlinkSync: unlinkSync$1 } = await import("node:fs");
+		const pid = readFileSync$1(pidFile, "utf-8").trim();
+		process.kill(parseInt(pid), "SIGTERM");
+		unlinkSync$1(pidFile);
+		console.log(styleText("green", `[*] Proxy stopped (PID: ${pid})`));
+	} catch (error) {
+		console.log(styleText("yellow", `[!] Could not stop proxy: ${error.message}`));
+		try {
+			const { unlinkSync: unlinkSync$1 } = await import("node:fs");
+			unlinkSync$1(pidFile);
+			console.log(styleText("gray", "Cleaned up stale PID file"));
+		} catch {}
+	}
+}).command("proxy-log", "Show websearch proxy logs").action(async () => {
+	const logFile = "/tmp/droid-search-proxy.log";
+	if (!existsSync(logFile)) {
+		console.log(styleText("yellow", "No log file found"));
+		return;
+	}
+	const { readFileSync: readFileSync$1 } = await import("node:fs");
+	const log = readFileSync$1(logFile, "utf-8");
+	const lines = log.split("\n").slice(-50);
+	console.log(styleText("cyan", "═".repeat(60)));
+	console.log(styleText(["cyan", "bold"], "  WebSearch Proxy Logs (last 50 lines)"));
+	console.log(styleText("cyan", "═".repeat(60)));
+	console.log();
+	console.log(lines.join("\n"));
 }).run().catch((err) => {
 	console.error(err);
 	process.exit(1);