122 lines
3.6 KiB
TypeScript
122 lines
3.6 KiB
TypeScript
import axios from 'axios';
|
||
|
||
/**
|
||
* 选线：本地缓存线路 → 内置兜底线路 → DNS 解析，同安卓 DetectLineManager。
|
||
*
|
||
* 内置线路全挂时，去解析 VITE_DNS_DOMAIN 的 TXT 记录拿线路（运维提供，多条用 _ 分隔），
|
||
* 解析出来的线路存本地，下次启动优先用它们。
|
||
*/
|
||
|
||
/** DNS 解析出来的线路缓存 key（同安卓 StoreKeys.DETECT_LINE） */
|
||
const LINE_CACHE_KEY = 'detect_line';
|
||
|
||
/** 单次 ping / DNS 请求超时 */
|
||
const TIMEOUT = 1000 * 10;
|
||
|
||
/** DoH 服务商，按顺序试：阿里（安卓用的也是这家）→ Cloudflare 兜底 */
|
||
const DNS_PROVIDERS = ['https://dns.alidns.com/resolve', 'https://cloudflare-dns.com/dns-query'];
|
||
|
||
/** 内置兜底线路（.env 的 VITE_BASE_URL） */
|
||
const builtinLines = (): string[] => {
|
||
try {
|
||
const lines = JSON.parse(process.env.VITE_BASE_URL || '[]');
|
||
return Array.isArray(lines) ? lines.filter(Boolean) : [];
|
||
} catch {
|
||
return [];
|
||
}
|
||
};
|
||
|
||
/** 上次 DNS 解析出来的线路 */
|
||
const cachedLines = (): string[] => {
|
||
try {
|
||
const lines = JSON.parse(localStorage.getItem(LINE_CACHE_KEY) || '[]');
|
||
return Array.isArray(lines) ? lines.filter(Boolean) : [];
|
||
} catch {
|
||
return [];
|
||
}
|
||
};
|
||
|
||
const saveCachedLines = (lines: string[]): void => {
|
||
try {
|
||
localStorage.setItem(LINE_CACHE_KEY, JSON.stringify(lines));
|
||
} catch {
|
||
// 无痕模式等写不进去，不影响本次选线
|
||
}
|
||
};
|
||
|
||
/** ping 单条线路，不抛错：通了返回 true */
|
||
const pingLine = async (line: string): Promise<boolean> => {
|
||
try {
|
||
const { data } = await axios.get(line + '/api/app/ping/check', { timeout: TIMEOUT });
|
||
return data?.code == 200;
|
||
} catch {
|
||
return false;
|
||
}
|
||
};
|
||
|
||
/** 并发 ping 一批线路：谁先成功用谁；全失败返回 '' */
|
||
const pingBatch = (lines: string[]): Promise<string> =>
|
||
new Promise((resolve) => {
|
||
if (!lines.length) return resolve('');
|
||
let failCount = 0;
|
||
let settled = false;
|
||
lines.forEach((line) => {
|
||
void pingLine(line).then((ok) => {
|
||
if (settled) return;
|
||
if (ok) {
|
||
settled = true;
|
||
resolve(line);
|
||
return;
|
||
}
|
||
if (++failCount >= lines.length) {
|
||
settled = true;
|
||
resolve('');
|
||
}
|
||
});
|
||
});
|
||
});
|
||
|
||
/**
|
||
* 解析 DNS TXT 记录拿线路。
|
||
* TXT 值形如 "https://a.com_https://b.com"，带引号，按 _ 拆成多条
|
||
*/
|
||
const dnsSolve = async (): Promise<string[]> => {
|
||
const domain = process.env.VITE_DNS_DOMAIN;
|
||
if (!domain) return [];
|
||
for (const provider of DNS_PROVIDERS) {
|
||
try {
|
||
const { data } = await axios.get(provider, {
|
||
params: { name: domain, type: 'TXT' },
|
||
headers: { accept: 'application/dns-json' },
|
||
timeout: TIMEOUT,
|
||
});
|
||
const lines: string[] = (data?.Answer || [])
|
||
.flatMap((record: any) => String(record?.data || '').split('_'))
|
||
.map((line: string) => line.replace(/"/g, '').trim())
|
||
.filter(Boolean);
|
||
if (lines.length) return lines;
|
||
} catch {
|
||
// 这家解析不出来就换下一家
|
||
}
|
||
}
|
||
return [];
|
||
};
|
||
|
||
/**
|
||
* 完整选线：缓存线路（没有就用内置线路）全挂时回退 DNS 解析。
|
||
* 返回通了的线路，全都不通返回 ''
|
||
*/
|
||
export const detectLine = async (): Promise<string> => {
|
||
const cached = cachedLines();
|
||
const line = await pingBatch(cached.length ? cached : builtinLines());
|
||
if (line) return line;
|
||
|
||
// 本地线路全部失败，回退 DNS 解析
|
||
const dnsLines = await dnsSolve();
|
||
if (!dnsLines.length) return '';
|
||
saveCachedLines(dnsLines);
|
||
return await pingBatch(dnsLines);
|
||
};
|
||
|
||
export default detectLine;
|