/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

import { execFile } from 'node:child_process';
import { appendFile, readFile } from 'node:fs/promises';
import { dirname, join } from 'node:path';
import { fileURLToPath, pathToFileURL } from 'node:url';
import { promisify } from 'node:util';
import { parseProductReleaseVersion } from './release-version.mjs';

const repoRoot = dirname(dirname(fileURLToPath(import.meta.url)));
const execFileAsync = promisify(execFile);

function releaseToolchainFromManifest(rootManifest) {
  const appleTeamIdentifier = rootManifest.releaseToolchain?.appleTeamIdentifier;
  const nodeVersion = rootManifest.releaseToolchain?.node;
  const nodeArchiveSha256 = rootManifest.releaseToolchain?.nodeDarwinArm64Sha256;
  const npmMatch = /^npm@(\d+\.\d+\.\d+)$/u.exec(rootManifest.packageManager ?? '');
  if (typeof nodeVersion !== 'string' || !/^\d+\.\d+\.\d+$/u.test(nodeVersion)) {
    throw new Error('package.json must define an exact releaseToolchain.node version');
  }
  if (typeof appleTeamIdentifier !== 'string' || !/^[A-Z0-9]{10}$/u.test(appleTeamIdentifier)) {
    throw new Error('releaseToolchain.appleTeamIdentifier must be an exact Apple Team ID');
  }
  const nodeArchive = `node-v${nodeVersion}-darwin-arm64.tar.xz`;
  if (typeof nodeArchiveSha256 !== 'string' || !/^[0-9a-f]{64}$/u.test(nodeArchiveSha256)) {
    throw new Error('releaseToolchain.nodeDarwinArm64Sha256 must be an exact SHA-256 digest');
  }
  if (!npmMatch) throw new Error('package.json packageManager must pin an exact npm version');
  return {
    appleTeamIdentifier,
    nodeVersion,
    nodeArchive,
    nodeArchiveSha256,
    nodeSourceUrl: `https://nodejs.org/download/release/v${nodeVersion}/${nodeArchive}`,
    npmVersion: npmMatch[1],
  };
}

export function parseAsfSourceReferenceTag(tag) {
  const match = typeof tag === 'string' ? /^v(.+)-incubating-rc([1-9]\d*)$/u.exec(tag) : undefined;
  if (!match) {
    throw new Error('ASF source reference must match v<version>-incubating-rc<positive-integer>');
  }
  try {
    const product = parseProductReleaseVersion(match[1]);
    if (product.prerelease.length > 0) throw new Error('prerelease');
  } catch {
    throw new Error('ASF source reference must match v<version>-incubating-rc<positive-integer>');
  }
  return { rcNumber: match[2], tag, version: match[1] };
}

export function resolveProductManifestIdentity({ rootManifest, desktopManifest, cliManifest }) {
  const { version, prerelease } = parseProductReleaseVersion(rootManifest.version);
  if (prerelease.length > 0) {
    throw new Error('Formal product releases require a stable version');
  }
  for (const [label, manifest] of [
    ['Desktop', desktopManifest],
    ['CLI', cliManifest],
  ]) {
    if (manifest.version !== version) {
      throw new Error(
        `${label} version ${manifest.version ?? 'missing'} does not match root ${version}`,
      );
    }
  }
  if (JSON.stringify(cliManifest.bin) !== JSON.stringify({ maka: './dist/cli.js' })) {
    throw new Error('The only public CLI command must be maka');
  }

  return {
    version,
    runtimeHostSetupPackage: `maka-agent@${version}`,
    publicCommands: ['maka'],
  };
}

export function resolveProductReleaseIdentity({
  rootManifest,
  desktopManifest,
  cliManifest,
  sha,
  sourceReferenceTag,
}) {
  const manifestIdentity = resolveProductManifestIdentity({
    rootManifest,
    desktopManifest,
    cliManifest,
  });
  const { version } = manifestIdentity;
  if (typeof sha !== 'string' || !/^[0-9a-f]{40}$/u.test(sha)) {
    throw new Error('Product releases require an exact 40-character source commit SHA');
  }
  if (sourceReferenceTag !== undefined) {
    const sourceReference = parseAsfSourceReferenceTag(sourceReferenceTag);
    if (sourceReference.version !== version) {
      throw new Error(
        `ASF source reference version ${sourceReference.version} does not match product ${version}`,
      );
    }
  }

  const toolchain = releaseToolchainFromManifest(rootManifest);
  const dmg = `Maka-${version}-mac-arm64.dmg`;
  const macZip = `Maka-${version}-mac-arm64.zip`;
  const exe = `Maka-${version}-win-x64.exe`;
  const windowsZip = `Maka-${version}-win-x64.zip`;
  const cliArchive = `Maka-${version}-cli-mac-arm64.zip`;
  const artifacts = {
    'desktop-macos': [dmg, `${dmg}.sha256`, macZip, `${macZip}.blockmap`, 'latest-mac.yml'],
    'desktop-windows': [
      exe,
      `${exe}.blockmap`,
      `${exe}.sha256`,
      windowsZip,
      `${windowsZip}.sha256`,
      'latest.yml',
    ],
    'cli-macos-arm64': [cliArchive, `${cliArchive}.sha256`],
  };

  return {
    ...toolchain,
    ...manifestIdentity,
    tag: `v${version}`,
    sourceCommit: sha,
    sourceReferenceTag,
    dmg,
    exe,
    cliArchive,
    artifacts,
  };
}

async function readProductManifests(root = repoRoot) {
  const [rootManifest, desktopManifest, cliManifest] = await Promise.all([
    readFile(join(root, 'package.json'), 'utf8').then(JSON.parse),
    readFile(join(root, 'apps/desktop/package.json'), 'utf8').then(JSON.parse),
    readFile(join(root, 'packages/cli/package.json'), 'utf8').then(JSON.parse),
  ]);
  return { rootManifest, desktopManifest, cliManifest };
}

export async function readProductManifestIdentity({ root = repoRoot } = {}) {
  return resolveProductManifestIdentity(await readProductManifests(root));
}

export async function readProductReleaseIdentity({
  sha,
  sourceReferenceTag = process.env.SOURCE_REFERENCE_TAG,
  root = process.env.PRODUCT_MANIFEST_ROOT ?? repoRoot,
} = {}) {
  const { rootManifest, desktopManifest, cliManifest } = await readProductManifests(root);
  const sourceCommit =
    sha ??
    process.env.GITHUB_SHA ??
    (await execFileAsync('git', ['rev-parse', 'HEAD'], { cwd: repoRoot })).stdout.trim();
  return resolveProductReleaseIdentity({
    rootManifest,
    desktopManifest,
    cliManifest,
    sha: sourceCommit,
    sourceReferenceTag,
  });
}

function githubOutputEntries(identity) {
  return {
    version: identity.version,
    tag: identity.tag,
    source_commit: identity.sourceCommit,
    source_reference_tag: identity.sourceReferenceTag,
    dmg: identity.dmg,
    exe: identity.exe,
    cli_archive: identity.cliArchive,
    node_version: identity.nodeVersion,
    node_archive: identity.nodeArchive,
    node_archive_sha256: identity.nodeArchiveSha256,
    node_source_url: identity.nodeSourceUrl,
    npm_version: identity.npmVersion,
  };
}

if (process.argv[1] && import.meta.url === pathToFileURL(process.argv[1]).href) {
  const identity = await readProductReleaseIdentity();
  const expectedVersion = process.env.EXPECTED_PRODUCT_VERSION;
  if (expectedVersion !== undefined && identity.version !== expectedVersion) {
    throw new Error(
      `Checked product version ${identity.version} does not match requested release ${expectedVersion}`,
    );
  }
  if (process.env.GITHUB_OUTPUT) {
    const output = Object.entries(githubOutputEntries(identity))
      .filter(([, value]) => value !== undefined)
      .map(([name, value]) => `${name}=${value}`)
      .join('\n');
    await appendFile(process.env.GITHUB_OUTPUT, `${output}\n`, 'utf8');
  }
  console.log(`Product release ${identity.tag} from ${identity.sourceCommit}`);
}