feat: add uv mirror detection and python bootstrap (#24)

Co-authored-by: chatgpt-codex-connector[bot] <199175422+chatgpt-codex-connector[bot]@users.noreply.github.com>

electron/utils/uv-env.ts

@@ -0,0 +1,120 @@
+import { app } from 'electron';
+import { request } from 'https';
+import { logger } from './logger';
+
+const UV_MIRROR_ENV: Record<string, string> = {
+  UV_PYTHON_INSTALL_MIRROR: 'https://registry.npmmirror.com/-/binary/python-build-standalone/',
+  UV_INDEX_URL: 'https://pypi.tuna.tsinghua.edu.cn/simple/',
+};
+
+const GOOGLE_204_HOST = 'www.google.com';
+const GOOGLE_204_PATH = '/generate_204';
+const GOOGLE_204_TIMEOUT_MS = 2000;
+
+let cachedOptimized: boolean | null = null;
+let cachedPromise: Promise<boolean> | null = null;
+let loggedOnce = false;
+
+function getLocaleAndTimezone(): { locale: string; timezone: string } {
+  const locale = app.getLocale?.() || '';
+  const timezone = Intl.DateTimeFormat().resolvedOptions().timeZone || '';
+  return { locale, timezone };
+}
+
+function isRegionOptimized(locale: string, timezone: string): boolean {
+  // Prefer timezone when available to reduce false positives from locale alone.
+  if (timezone) return timezone === 'Asia/Shanghai';
+  return locale === 'zh-CN';
+}
+
+function probeGoogle204(timeoutMs: number): Promise<boolean> {
+  return new Promise((resolve) => {
+    let done = false;
+    const finish = (value: boolean) => {
+      if (done) return;
+      done = true;
+      resolve(value);
+    };
+
+    const req = request(
+      {
+        method: 'GET',
+        hostname: GOOGLE_204_HOST,
+        path: GOOGLE_204_PATH,
+      },
+      (res) => {
+        const status = res.statusCode || 0;
+        res.resume();
+        finish(status >= 200 && status < 300);
+      }
+    );
+
+    req.setTimeout(timeoutMs, () => {
+      req.destroy(new Error('google_204_timeout'));
+    });
+
+    req.on('error', () => finish(false));
+    req.end();
+  });
+}
+
+async function computeOptimization(): Promise<boolean> {
+  const { locale, timezone } = getLocaleAndTimezone();
+
+  if (isRegionOptimized(locale, timezone)) {
+    if (!loggedOnce) {
+      logger.info(`Region optimization enabled via locale/timezone (locale=${locale || 'unknown'}, tz=${timezone || 'unknown'})`);
+      loggedOnce = true;
+    }
+    return true;
+  }
+
+  const reachable = await probeGoogle204(GOOGLE_204_TIMEOUT_MS);
+  const isOptimized = !reachable;
+
+  if (!loggedOnce) {
+    const reason = reachable ? 'google_204_reachable' : 'google_204_unreachable';
+    logger.info(`Network optimization probe: ${reason} (locale=${locale || 'unknown'}, tz=${timezone || 'unknown'})`);
+    loggedOnce = true;
+  }
+
+  return isOptimized;
+}
+
+export async function shouldOptimizeNetwork(): Promise<boolean> {
+  if (cachedOptimized !== null) return cachedOptimized;
+  if (cachedPromise) return cachedPromise;
+
+  if (!app.isReady()) {
+    await app.whenReady();
+  }
+
+  cachedPromise = computeOptimization()
+    .then((result) => {
+      cachedOptimized = result;
+      return result;
+    })
+    .catch((err) => {
+      logger.warn('Network optimization check failed, defaulting to enabled:', err);
+      cachedOptimized = true;
+      return true;
+    })
+    .finally(() => {
+      cachedPromise = null;
+    });
+
+  return cachedPromise;
+}
+
+export async function getUvMirrorEnv(): Promise<Record<string, string>> {
+  const isOptimized = await shouldOptimizeNetwork();
+  return isOptimized ? { ...UV_MIRROR_ENV } : {};
+}
+
+export async function warmupNetworkOptimization(): Promise<void> {
+  try {
+    await shouldOptimizeNetwork();
+  } catch {
+    // Ignore warmup failures
+  }
+}

electron/utils/uv-setup.ts

@@ -2,6 +2,7 @@ import { app } from 'electron';
 import { spawn } from 'child_process';
 import { existsSync } from 'fs';
 import { join } from 'path';
+import { getUvMirrorEnv } from './uv-env';
 
 /**
  * Get the path to the bundled uv binary
@@ -53,6 +54,33 @@ export async function installUv(): Promise<void> {
   console.log('uv is available and ready to use');
 }
 
+/**
+ * Check if a managed Python 3.12 is ready and accessible
+ */
+export async function isPythonReady(): Promise<boolean> {
+  // Use 'uv' if in PATH, otherwise use full bundled path
+  const inPath = await new Promise<boolean>((resolve) => {
+    const cmd = process.platform === 'win32' ? 'where.exe' : 'which';
+    const child = spawn(cmd, ['uv']);
+    child.on('close', (code) => resolve(code === 0));
+    child.on('error', () => resolve(false));
+  });
+
+  const uvBin = inPath ? 'uv' : getBundledUvPath();
+
+  return new Promise<boolean>((resolve) => {
+    try {
+      const child = spawn(uvBin, ['python', 'find', '3.12'], {
+        shell: process.platform === 'win32',
+      });
+      child.on('close', (code) => resolve(code === 0));
+      child.on('error', () => resolve(false));
+    } catch {
+      resolve(false);
+    }
+  });
+}
+
 /**
  * Use bundled uv to install a managed Python version (default 3.12)
  * Automatically picks the best available uv binary
@@ -69,10 +97,15 @@ export async function setupManagedPython(): Promise<void> {
   const uvBin = inPath ? 'uv' : getBundledUvPath();
   
   console.log(`Setting up python with: ${uvBin}`);
+  const uvEnv = await getUvMirrorEnv();
 
   await new Promise<void>((resolve, reject) => {
     const child = spawn(uvBin, ['python', 'install', '3.12'], {
-      shell: process.platform === 'win32'
+      shell: process.platform === 'win32',
+      env: {
+        ...process.env,
+        ...uvEnv,
+      },
     });
 
     child.stdout?.on('data', (data) => {
@@ -96,7 +129,11 @@ export async function setupManagedPython(): Promise<void> {
   try {
     const findPath = await new Promise<string>((resolve) => {
       const child = spawn(uvBin, ['python', 'find', '3.12'], {
-        shell: process.platform === 'win32'
+        shell: process.platform === 'win32',
+        env: {
+          ...process.env,
+          ...uvEnv,
+        },
       });
       let output = '';
       child.stdout?.on('data', (data) => { output += data; });
@@ -111,4 +148,4 @@ export async function setupManagedPython(): Promise<void> {
   } catch (err) {
     console.warn('Could not determine Python path:', err);
   }
-}
+}