58 lines
1.4 KiB
TypeScript
Vendored
58 lines
1.4 KiB
TypeScript
Vendored
import dns from 'dns';
|
|
import http from 'http';
|
|
import { promisify } from 'util';
|
|
|
|
import logger from '@/options/logger';
|
|
|
|
const resolve = promisify(dns.resolve);
|
|
|
|
const ping = async (host: string) => {
|
|
const lookup = promisify(dns.lookup);
|
|
const ip = await lookup(host);
|
|
const start = new Date();
|
|
|
|
// Prevent timeouts from affecting user experience.
|
|
const requestPromise = new Promise<number>((resolve, reject) => {
|
|
const req = http.get(`http://${ip.address}`, (res) => {
|
|
const delay = new Date().getTime() - start.getTime();
|
|
res.resume();
|
|
resolve(delay);
|
|
});
|
|
|
|
req.on('error', (err) => {
|
|
reject(err);
|
|
});
|
|
});
|
|
|
|
const timeoutPromise = new Promise<number>((_, reject) => {
|
|
setTimeout(() => {
|
|
reject(new Error('Request timed out after 3 seconds'));
|
|
}, 1000);
|
|
});
|
|
|
|
return Promise.race([requestPromise, timeoutPromise]);
|
|
};
|
|
|
|
async function isChinaDomain(domain: string): Promise<boolean> {
|
|
try {
|
|
const [ip] = await resolve(domain);
|
|
return await isChinaIP(ip, domain);
|
|
} catch (error) {
|
|
logger.debug(`${domain} can't be parse!`);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
async function isChinaIP(ip: string, domain: string): Promise<boolean> {
|
|
try {
|
|
const delay = await ping(ip);
|
|
logger.debug(`${domain} latency is ${delay} ms`);
|
|
return delay > 1000;
|
|
} catch (error) {
|
|
logger.debug(`ping ${domain} failed!`);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
export { isChinaDomain, isChinaIP };
|