import process from 'node:process';
import os from 'node:os';
import fs from 'node:fs';

import { downloadPreviousBuild } from './download-previous-build';
import { buildCommon } from './build-common';
import { buildRejectIPList } from './build-reject-ip-list';
import { buildAppleCdn } from './build-apple-cdn';
import { buildCdnDownloadConf } from './build-cdn-download-conf';
import { buildRejectDomainSet } from './build-reject-domainset';
import { buildTelegramCIDR } from './build-telegram-cidr';
import { buildChnCidr } from './build-chn-cidr';
import { buildSpeedtestDomainSet } from './build-speedtest-domainset';
import { buildDomesticRuleset } from './build-domestic-direct-lan-ruleset-dns-mapping-module';
import { buildGlobalRuleset } from './build-global-server-dns-mapping';
import { buildStreamService } from './build-stream-service';

import { buildRedirectModule } from './build-sgmodule-redirect';
import { buildAlwaysRealIPModule } from './build-sgmodule-always-realip';

import { buildMicrosoftCdn } from './build-microsoft-cdn';

import { buildPublic } from './build-public';
import { downloadMockAssets } from './download-mock-assets';

import { buildCloudMounterRules } from './build-cloudmounter-rules';

import { printStats, printTraceResult, whyIsNodeRunning } from './trace';
import type { TraceResult } from './trace';
import { buildDeprecateFiles } from './build-deprecate-files';
import path from 'node:path';
import { ROOT_DIR } from './constants/dir';
import { isCI } from 'ci-info';

process.on('uncaughtException', (error) => {
  console.error('Uncaught exception:', error);
  process.exit(1);
});
process.on('unhandledRejection', (reason) => {
  console.error('Unhandled rejection:', reason);
  process.exit(1);
});

const buildFinishedLock = path.join(ROOT_DIR, '.BUILD_FINISHED');

(async () => {
  console.log('Version:', process.version);

  console.log(`OS: ${os.type()} ${os.release()} ${os.arch()}`);
  console.log(`Node.js: ${process.versions.node}`);
  console.log(`V8: ${process.versions.v8}`);

  const cpus = os.cpus()
    .reduce<Record<string, number>>((o, cpu) => {
      o[cpu.model] = (o[cpu.model] || 0) + 1;
      return o;
    }, {});

  console.log(`CPU: ${
    Object.keys(cpus)
      .map((key) => `${key} x ${cpus[key]}`)
      .join('\n')
  }`);
  if ('availableParallelism' in os) {
    console.log(`Available parallelism: ${os.availableParallelism()}`);
  }

  console.log(`Memory: ${os.totalmem() / (1024 * 1024)} MiB`);

  if (fs.existsSync(buildFinishedLock)) {
    fs.unlinkSync(buildFinishedLock);
  }

  try {
    // only enable why-is-node-running in GitHub Actions debug mode
    if (isCI && process.env.RUNNER_DEBUG === '1') {
      await import('why-is-node-running');
    }

    const downloadPreviousBuildPromise = downloadPreviousBuild();

    await Promise.all([
      downloadPreviousBuildPromise,
      downloadPreviousBuildPromise.then(() => buildCommon()),
      downloadPreviousBuildPromise.then(() => buildRejectIPList()),
      downloadPreviousBuildPromise.then(() => buildAppleCdn()),
      downloadPreviousBuildPromise.then(() => buildCdnDownloadConf()),
      downloadPreviousBuildPromise.then(() => buildRejectDomainSet()),
      downloadPreviousBuildPromise.then(() => buildTelegramCIDR()),
      downloadPreviousBuildPromise.then(() => buildChnCidr()),
      downloadPreviousBuildPromise.then(() => buildSpeedtestDomainSet()),
      downloadPreviousBuildPromise.then(() => buildDomesticRuleset()),
      downloadPreviousBuildPromise.then(() => buildGlobalRuleset()),
      downloadPreviousBuildPromise.then(() => buildRedirectModule()),
      downloadPreviousBuildPromise.then(() => buildAlwaysRealIPModule()),
      downloadPreviousBuildPromise.then(() => buildStreamService()),
      downloadPreviousBuildPromise.then(() => buildMicrosoftCdn()),
      downloadPreviousBuildPromise.then(() => buildCloudMounterRules()),
      downloadMockAssets()
    ]);

    await buildDeprecateFiles();
    await buildPublic();

    // write a file to demonstrate that the build is finished
    fs.writeFileSync(buildFinishedLock, 'BUILD_FINISHED\n');

    const traces: TraceResult[] = [];
    [
      downloadPreviousBuild,
      downloadMockAssets,
      buildCommon,
      buildRejectIPList,
      buildAppleCdn,
      buildCdnDownloadConf,
      buildRejectDomainSet,
      buildTelegramCIDR,
      buildChnCidr,
      buildSpeedtestDomainSet,
      buildDomesticRuleset,
      buildGlobalRuleset,
      buildRedirectModule,
      buildAlwaysRealIPModule,
      buildStreamService,
      buildMicrosoftCdn,
      buildCloudMounterRules,
      buildPublic,
      buildDeprecateFiles
    ].forEach((fn) => {
      const trace = fn.getInternalTraceResult();
      printTraceResult(trace);
      traces.push(trace);
    });
    printStats(traces);

    // Finish the build to avoid leaking timer/fetch ref
    await whyIsNodeRunning();
    process.exit(0);
  } catch (e) {
    console.error('Something went wrong!');
    console.trace(e);
    process.exit(1);
  }
})();