/**
 * @license
 * Copyright 2023 Google Inc.
 * SPDX-License-Identifier: Apache-2.0
 */

/**
 * Downloads test browser binaries to test/.cache/server folder that
 * mirrors the structure of the download server.
 */

import {existsSync, mkdirSync, copyFileSync, rmSync} from 'node:fs';
import {normalize, join, dirname} from 'node:path';

import {BrowserPlatform, install} from '@puppeteer/browsers';

import {downloadPaths} from '../lib/esm/browser-data/browser-data.js';
import * as versions from '../test/build/versions.js';

function getBrowser(str) {
  const regex = /test(.+)BuildId/;
  const match = str.match(regex);

  if (match && match[1]) {
    const lowercased = match[1].toLowerCase();
    if (lowercased === 'chromeheadlessshell') {
      return 'chrome-headless-shell';
    }
    return lowercased;
  } else {
    return null;
  }
}

const cacheDir = normalize(join('.', 'test', '.cache'));

// Needed to test Firefox nightly build download
function mockFirefoxNightly(browser, platform, targetPath) {
  if (browser === 'firefox' && platform === 'linux') {
    const nightlyTarget = join(
      cacheDir,
      'server',
      ...downloadPaths.firefox(
        'linux',
        versions.testFirefoxBuildId.split('_').at(-1),
      ),
    );

    if (existsSync(nightlyTarget)) {
      return;
    }

    copyFileSync(targetPath, nightlyTarget);
  }
}

for (const version of Object.keys(versions)) {
  const browser = getBrowser(version);
  if (!browser) {
    continue;
  }

  const buildId = versions[version];

  for (const platform of Object.values(BrowserPlatform)) {
    if (platform === BrowserPlatform.LINUX_ARM) {
      continue;
    }
    const targetPath = join(
      cacheDir,
      'server',
      ...downloadPaths[browser](platform, buildId),
    );

    if (existsSync(targetPath)) {
      mockFirefoxNightly(browser, platform, targetPath);
      continue;
    }

    const archivePath = await install({
      browser,
      buildId,
      platform,
      cacheDir: join(cacheDir, 'tmp'),
      unpack: false,
    });

    mkdirSync(dirname(targetPath), {
      recursive: true,
    });
    copyFileSync(archivePath, targetPath);

    mockFirefoxNightly(browser, platform, targetPath);
  }
}

rmSync(join(cacheDir, 'tmp'), {
  recursive: true,
  force: true,
  maxRetries: 10,
});
