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>
This commit is contained in:
120
electron/utils/uv-env.ts
Normal file
120
electron/utils/uv-env.ts
Normal file
@@ -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
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user