mirror of
https://github.com/SukkaW/Surge.git
synced 2026-04-30 09:56:59 +08:00
Refactor: use jest-worker
This commit is contained in:
@@ -7,10 +7,6 @@ import { bigint2ip } from 'fast-cidr-tools';
|
||||
|
||||
import { base64ToUint8Array, concatUint8Arrays } from 'foxts/uint8array-utils';
|
||||
|
||||
import Worktank from 'worktank';
|
||||
import { wait } from 'foxts/wait';
|
||||
import { once } from 'foxts/once';
|
||||
|
||||
const mtptoto_public_rsa = `-----BEGIN RSA PUBLIC KEY-----
|
||||
MIIBCgKCAQEAyr+18Rex2ohtVy8sroGP
|
||||
BwXD3DOoKCSpjDqYoXgCqB7ioln4eDCFfOBUlfXUEvM/fnKCpF46VkAftlb4VuPD
|
||||
@@ -112,160 +108,3 @@ export function getTelegramBackupIPFromBase64(base64: string) {
|
||||
}
|
||||
}));
|
||||
}
|
||||
|
||||
const pool = new Worktank({
|
||||
pool: {
|
||||
name: 'get-telegram-backup-ips',
|
||||
size: 1 // The number of workers to keep in the pool, if more workers are needed they will be spawned up to this limit
|
||||
},
|
||||
worker: {
|
||||
autoAbort: 10000,
|
||||
autoTerminate: 30000, // The interval of milliseconds at which to check if the pool can be automatically terminated, to free up resources, workers will be spawned up again if needed
|
||||
autoInstantiate: true,
|
||||
methods: {
|
||||
// eslint-disable-next-line object-shorthand -- workertank
|
||||
getTelegramBackupIPs: async function (__filename: string): Promise<{ timestamp: number, ipcidr: string[], ipcidr6: string[] }> {
|
||||
// TODO: createRequire is a temporary workaround for https://github.com/nodejs/node/issues/51956
|
||||
const { default: module } = await import('node:module');
|
||||
const __require = module.createRequire(__filename);
|
||||
|
||||
const picocolors = __require('picocolors') as typeof import('picocolors');
|
||||
const { '~fetch': fetch } = __require('./fetch-retry') as typeof import('./fetch-retry');
|
||||
|
||||
const dns = __require('node:dns/promises') as typeof import('node:dns/promises');
|
||||
|
||||
const { createReadlineInterfaceFromResponse } = __require('./fetch-text-by-line') as typeof import('./fetch-text-by-line');
|
||||
const { getTelegramBackupIPFromBase64 } = __require('./get-telegram-backup-ip') as typeof import('./get-telegram-backup-ip');
|
||||
const { fastIpVersion } = __require('foxts/fast-ip-version') as typeof import('foxts/fast-ip-version');
|
||||
const { fastStringArrayJoin } = __require('foxts/fast-string-array-join') as typeof import('foxts/fast-string-array-join');
|
||||
|
||||
const resp = await fetch('https://core.telegram.org/resources/cidr.txt');
|
||||
const lastModified = resp.headers.get('last-modified');
|
||||
const date = lastModified ? new Date(lastModified) : new Date();
|
||||
|
||||
const ipcidr: string[] = [
|
||||
// Unused secret Telegram backup CIDR, announced by AS62041
|
||||
'95.161.64.0/20'
|
||||
];
|
||||
const ipcidr6: string[] = [];
|
||||
|
||||
for await (const cidr of createReadlineInterfaceFromResponse(resp, true)) {
|
||||
const v = fastIpVersion(cidr);
|
||||
if (v === 4) {
|
||||
ipcidr.push(cidr);
|
||||
} else if (v === 6) {
|
||||
ipcidr6.push(cidr);
|
||||
}
|
||||
}
|
||||
|
||||
const backupIPs = new Set<string>();
|
||||
|
||||
// https://github.com/tdlib/td/blob/master/td/telegram/ConfigManager.cpp
|
||||
|
||||
const resolvers = ['8.8.8.8', '1.0.0.1'].map((ip) => {
|
||||
const resolver = new dns.Resolver();
|
||||
resolver.setServers([ip]);
|
||||
return Object.assign(resolver, { server: ip });
|
||||
});
|
||||
|
||||
// Backup IP Source 1 (DNS)
|
||||
await Promise.all(resolvers.flatMap((resolver) => [
|
||||
'apv3.stel.com', // prod
|
||||
'tapv3.stel.com' // test
|
||||
].map(async (domain) => {
|
||||
try {
|
||||
// tapv3.stel.com was for testing server
|
||||
const resp = await resolver.resolveTxt(domain);
|
||||
const strings = resp.map(r => fastStringArrayJoin(r, '')); // flatten
|
||||
if (strings.length !== 2) {
|
||||
throw new TypeError(`Unexpected TXT record count: ${strings.length}`);
|
||||
}
|
||||
|
||||
const str = strings[0].length > strings[1].length
|
||||
? strings[0] + strings[1]
|
||||
: strings[1] + strings[0];
|
||||
|
||||
const ips = getTelegramBackupIPFromBase64(str);
|
||||
ips.forEach(i => backupIPs.add(i.ip));
|
||||
|
||||
console.log('[telegram backup ip]', picocolors.green('DNS TXT'), { domain, ips, server: resolver.server });
|
||||
} catch (e) {
|
||||
console.error('[telegram backup ip]', picocolors.red('DNS TXT error'), { domain }, e);
|
||||
}
|
||||
})));
|
||||
|
||||
// Backup IP Source 2: Firebase Realtime Database (test server not supported)
|
||||
try {
|
||||
const text = await (await fetch('https://reserve-5a846.firebaseio.com/ipconfigv3.json')).json();
|
||||
if (typeof text === 'string' && text.length === 344) {
|
||||
const ips = getTelegramBackupIPFromBase64(text);
|
||||
ips.forEach(i => backupIPs.add(i.ip));
|
||||
|
||||
console.log('[telegram backup ip]', picocolors.green('Firebase Realtime DB'), { ips });
|
||||
}
|
||||
} catch (e) {
|
||||
console.error('[telegram backup ip]', picocolors.red('Firebase Realtime DB error'), e);
|
||||
// ignore all errors
|
||||
}
|
||||
|
||||
// Backup IP Source 3: Firebase Value Store (test server not supported)
|
||||
try {
|
||||
const json = await (await fetch('https://firestore.googleapis.com/v1/projects/reserve-5a846/databases/(default)/documents/ipconfig/v3', {
|
||||
headers: {
|
||||
Accept: '*/*',
|
||||
Origin: undefined // Without this line, Google API will return "Bad request: Origin doesn't match Host for XD3.". Probably have something to do with sqlite cache store
|
||||
}
|
||||
})).json();
|
||||
|
||||
// const json = await resp.json();
|
||||
if (
|
||||
json && typeof json === 'object'
|
||||
&& 'fields' in json && typeof json.fields === 'object' && json.fields
|
||||
&& 'data' in json.fields && typeof json.fields.data === 'object' && json.fields.data
|
||||
&& 'stringValue' in json.fields.data && typeof json.fields.data.stringValue === 'string' && json.fields.data.stringValue.length === 344
|
||||
) {
|
||||
const ips = getTelegramBackupIPFromBase64(json.fields.data.stringValue);
|
||||
ips.forEach(i => backupIPs.add(i.ip));
|
||||
|
||||
console.log('[telegram backup ip]', picocolors.green('Firebase Value Store'), { ips });
|
||||
} else {
|
||||
console.error('[telegram backup ip]', picocolors.red('Firebase Value Store data format invalid'), { json });
|
||||
}
|
||||
} catch (e) {
|
||||
console.error('[telegram backup ip]', picocolors.red('Firebase Value Store error'), e);
|
||||
}
|
||||
|
||||
// Backup IP Source 4: Google App Engine
|
||||
await Promise.all([
|
||||
'https://dns-telegram.appspot.com',
|
||||
'https://dns-telegram.appspot.com/test'
|
||||
].map(async (url) => {
|
||||
try {
|
||||
const text = await (await fetch(url)).text();
|
||||
if (text.length === 344) {
|
||||
const ips = getTelegramBackupIPFromBase64(text);
|
||||
ips.forEach(i => backupIPs.add(i.ip));
|
||||
|
||||
console.log('[telegram backup ip]', picocolors.green('Google App Engine'), { url, ips });
|
||||
}
|
||||
} catch (e) {
|
||||
console.error('[telegram backup ip]', picocolors.red('Google App Engine error'), { url }, e);
|
||||
}
|
||||
}));
|
||||
|
||||
// tcdnb.azureedge.net no longer works
|
||||
|
||||
console.log('[telegram backup ip]', `Found ${backupIPs.size} backup IPs:`, backupIPs);
|
||||
|
||||
ipcidr.push(...Array.from(backupIPs).map(i => i + '/32'));
|
||||
|
||||
return { timestamp: date.getTime(), ipcidr, ipcidr6 };
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
export const getTelegramCIDRPromise = once(() => wait(0).then(() => pool.exec(
|
||||
'getTelegramBackupIPs',
|
||||
[__filename]
|
||||
)).finally(() => pool.terminate()), false);
|
||||
|
||||
Reference in New Issue
Block a user