File size: 2,716 Bytes
1f21206 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 | import { SettingsService } from './settingsService.js'
export type NetworkProxyMode = 'system' | 'manual'
export type NetworkSettings = {
aiRequestTimeoutMs: number
proxy: {
mode: NetworkProxyMode
url: string
}
}
export const DEFAULT_AI_REQUEST_TIMEOUT_MS = 120_000
export const MIN_AI_REQUEST_TIMEOUT_MS = 5_000
export const MAX_AI_REQUEST_TIMEOUT_MS = 600_000
const DEFAULT_NETWORK_SETTINGS: NetworkSettings = {
aiRequestTimeoutMs: DEFAULT_AI_REQUEST_TIMEOUT_MS,
proxy: {
mode: 'system',
url: '',
},
}
function isNetworkProxyMode(value: unknown): value is NetworkProxyMode {
return value === 'system' || value === 'manual'
}
function clampTimeoutMs(value: number): number {
return Math.min(Math.max(value, MIN_AI_REQUEST_TIMEOUT_MS), MAX_AI_REQUEST_TIMEOUT_MS)
}
function parseTimeoutMs(value: unknown): number {
if (typeof value !== 'number' || !Number.isFinite(value)) {
return DEFAULT_NETWORK_SETTINGS.aiRequestTimeoutMs
}
return clampTimeoutMs(Math.round(value))
}
function parseProxy(value: unknown): NetworkSettings['proxy'] {
if (!value || typeof value !== 'object') {
return DEFAULT_NETWORK_SETTINGS.proxy
}
const record = value as Record<string, unknown>
return {
mode: isNetworkProxyMode(record.mode) ? record.mode : DEFAULT_NETWORK_SETTINGS.proxy.mode,
url: typeof record.url === 'string' ? record.url.trim() : '',
}
}
export function normalizeNetworkSettings(settings: unknown): NetworkSettings {
if (!settings || typeof settings !== 'object') {
return DEFAULT_NETWORK_SETTINGS
}
const record = settings as Record<string, unknown>
const rawNetwork = record.network
const network = rawNetwork && typeof rawNetwork === 'object'
? rawNetwork as Record<string, unknown>
: {}
return {
aiRequestTimeoutMs: parseTimeoutMs(network.aiRequestTimeoutMs),
proxy: parseProxy(network.proxy),
}
}
export function getManualNetworkProxyUrl(settings: NetworkSettings): string | undefined {
if (settings.proxy.mode !== 'manual') return undefined
const url = settings.proxy.url.trim()
return url || undefined
}
export function buildNetworkEnvironment(settings: NetworkSettings): Record<string, string> {
const env: Record<string, string> = {
API_TIMEOUT_MS: String(settings.aiRequestTimeoutMs),
}
const proxyUrl = getManualNetworkProxyUrl(settings)
if (proxyUrl) {
env.HTTP_PROXY = proxyUrl
env.HTTPS_PROXY = proxyUrl
env.http_proxy = proxyUrl
env.https_proxy = proxyUrl
}
return env
}
export async function loadNetworkSettings(): Promise<NetworkSettings> {
const settings = await new SettingsService().getUserSettings()
return normalizeNetworkSettings(settings)
}
|