import path from 'node:path';

import { Sema } from 'async-sema';
import { getHostname } from 'tldts-experimental';
import { task } from './trace';
import { $fetch } from './lib/make-fetch-happen';
import { SHARED_DESCRIPTION } from './lib/constants';
import { readFileIntoProcessedArray } from './lib/fetch-text-by-line';

import { DomainsetOutput } from './lib/create-file';
import { OUTPUT_SURGE_DIR } from './constants/dir';

const KEYWORDS = [
  'Hong Kong',
  'Taiwan',
  'China Telecom',
  'China Mobile',
  'China Unicom',
  'Japan',
  'Tokyo',
  'Singapore',
  'Korea',
  'Seoul',
  'Canada',
  'Toronto',
  'Montreal',
  'Los Ang',
  'San Jos',
  'Seattle',
  'New York',
  'Dallas',
  'Miami',
  'Berlin',
  'Frankfurt',
  'London',
  'Paris',
  'Amsterdam',
  'Moscow',
  'Australia',
  'Sydney',
  'Brazil',
  'Turkey'
];

const PREDEFINE_DOMAINS = [
  // speedtest.net
  '.speedtest.net',
  '.speedtestcustom.com',
  '.ooklaserver.net',
  '.speed.misaka.one',
  '.speedtest.rt.ru',
  '.speedtest.aptg.com.tw',
  '.speedtest.gslnetworks.com',
  '.speedtest.jsinfo.net',
  '.speedtest.i3d.net',
  '.speedtestkorea.com',
  '.speedtest.telus.com',
  '.speedtest.telstra.net',
  '.speedtest.clouvider.net',
  '.speedtest.idv.tw',
  '.speedtest.frontier.com',
  '.speedtest.orange.fr',
  '.speedtest.centurylink.net',
  '.srvr.bell.ca',
  '.speedtest.contabo.net',
  'speedtest.hk.chinamobile.com',
  'speedtestbb.hk.chinamobile.com',
  '.hizinitestet.com',
  '.linknetspeedtest.net.br',
  'speedtest.rit.edu',
  'speedtest.ropa.de',
  'speedtest.sits.su',
  'speedtest.tigo.cr',
  'speedtest.upp.com',
  '.speedtest.pni.tw',
  '.speed.pfm.gg',
  '.speedtest.faelix.net',
  '.speedtest.labixe.net',
  '.speedtest.warian.net',
  '.speedtest.starhub.com',
  '.speedtest.gibir.net.tr',
  '.speedtest.ozarksgo.net',
  '.speedtest.exetel.com.au',
  '.speedtest.sbcglobal.net',
  '.speedtest.leaptel.com.au',
  '.speedtest.windstream.net',
  '.speedtest.vodafone.com.au',
  '.speedtest.rascom.ru',
  '.speedtest.dchost.com',
  '.speedtest.highnet.com',
  '.speedtest.seattle.wa.limewave.net',
  '.speedtest.optitel.com.au',
  '.speednet.net.tr',
  '.speedtest.angolacables.co.ao',
  '.ookla-speedtest.fsr.com',
  '.speedtest.comnet.com.tr',
  '.speedtest.gslnetworks.com.au',
  '.test.gslnetworks.com.au',
  '.speedtest.gslnetworks.com',
  '.speedtestunonet.com.br',
  '.speedtest.alagas.net',
  'speedtest.surfshark.com',
  '.speedtest.aarnet.net.au',
  '.ookla.rcp.net',
  '.ookla-speedtests.e2ro.com',
  '.speedtest.com.sg',
  '.ookla.ddnsgeek.com',
  '.speedtest.pni.tw',
  '.speedtest.cmcnetworks.net',
  '.speedtestwnet.com.br',
  // Cloudflare
  '.speed.cloudflare.com',
  // Wi-Fi Man
  '.wifiman.com',
  '.wifiman.me',
  '.wifiman.ubncloud.com',
  '.wifiman-stability-test.ubncloud.com',
  // Fast.com
  '.fast.com',
  // MacPaw
  'speedtest.macpaw.com',
  // speedtestmaster
  '.netspeedtestmaster.com',
  // Google Search Result of "speedtest", powered by this
  '.measurement-lab.org',
  '.measurementlab.net',
  // Google Fiber legacy speedtest site (new fiber speedtest use speedtestcustom.com)
  '.speed.googlefiber.net',
  // librespeed
  '.backend.librespeed.org',
  // Apple,
  'mensura.cdn-apple.com', // From netQuality command
  // OpenSpeedtest
  'open.cachefly.net' // This is also used for openspeedtest server download

];

const s = new Sema(2);

const latestTopUserAgentsPromise = $fetch('https://cdn.jsdelivr.net/npm/top-user-agents@latest/src/desktop.json')
  .then(res => res.json())
  .then((userAgents: string[]) => userAgents.filter(ua => ua.startsWith('Mozilla/5.0 ')));

const querySpeedtestApi = async (keyword: string): Promise<Array<string | null>> => {
  const topUserAgents = await latestTopUserAgentsPromise;

  const url = `https://www.speedtest.net/api/js/servers?engine=js&search=${keyword}&limit=100`;

  try {
    const randomUserAgent = topUserAgents[Math.floor(Math.random() * topUserAgents.length)];

    return await s.acquire().then(() => $fetch(url, {
      headers: {
        dnt: '1',
        Referer: 'https://www.speedtest.net/',
        accept: 'application/json, text/plain, */*',
        'User-Agent': randomUserAgent,
        'Accept-Language': 'en-US,en;q=0.9',
        ...(randomUserAgent.includes('Chrome')
          ? {
            'Sec-Ch-Ua-Mobile': '?0',
            'Sec-Fetch-Dest': 'empty',
            'Sec-Fetch-Mode': 'cors',
            'Sec-Fetch-Site': 'same-origin',
            'Sec-Gpc': '1'
          }
          : {})
      },
      timeout: 1000 * 60
    })).then(r => r.json() as any).then((data: Array<{ url: string, host: string }>) => data.reduce<string[]>(
      (prev, cur) => {
        const line = cur.host || cur.url;
        const hn = getHostname(line, { detectIp: false, validateHostname: true });
        if (hn) {
          prev.push(hn);
        }
        return prev;
      }, []
    )).finally(() => s.release());
  } catch (e) {
    console.error(e);
    return [];
  }
};

export const buildSpeedtestDomainSet = task(require.main === module, __filename)(async (span) => {
  const output = new DomainsetOutput(span, 'speedtest')
    .withTitle('Sukka\'s Ruleset - Speedtest Domains')
    .withDescription([
      ...SHARED_DESCRIPTION,
      '',
      'This file contains common speedtest endpoints.'
    ])
    .addFromDomainset(PREDEFINE_DOMAINS)
    .addFromDomainset(await readFileIntoProcessedArray(path.resolve(OUTPUT_SURGE_DIR, 'domainset/speedtest.conf')));

  await Promise.all(KEYWORDS.map((keyword) => span.traceChildAsync(
    `fetch speedtest endpoints: ${keyword}`,
    () => querySpeedtestApi(keyword)
  ).then(hostnameGroup => hostnameGroup.forEach(hostname => {
    if (hostname) {
      output.addDomain(hostname);
    }
  }))));

  return output.write();
});