// @ts-check

const { execSync } = require('node:child_process');

/**
 * This function is invoked by the publish.yml GitHub Action workflow and contains all of the dynamic logic needed
 * for the various workflow trigger types. This avoids the need for the logic to be stored in fragile inline
 * shell commands.
 *
 * @typedef {'--dry-run' | ''} DryRunFlag
 *
 * @typedef {{
 *   version: string;
 *   dry_run_flag: DryRunFlag;
 *   success_comment: string;
 *   publish_branch: string;
 *   ref: string;
 *   repo: string;
 *   pr_number: string;
 *   pr_author: string;
 * }} PublishResolveData
 *
 * Partial from https://github.com/actions/toolkit/blob/c6b487124a61d7dc6c7bd6ea0208368af3513a6e/packages/github/src/context.ts
 * @typedef {{
 *   actor: string;
 *   runId: number;
 *   repo: { owner: string; repo: string };
 * }} GitHubContext
 *
 * @param {{
 *  github: import('octokit/dist-types').Octokit;
 *  context: GitHubContext;
 *  core: import('@actions/core');
 * }} param
 */
module.exports = async ({ github, context, core }) => {
  const data = await getPublishResolveData({ github, context });

  // Ensure that certain outputs are always set
  if (!data.version) {
    throw new Error('The "version" to release could not be determined');
  }
  if (!data.publish_branch) {
    throw new Error('The "publish_branch" could not be determined');
  }

  // Set the outputs to be consumed in later steps
  core.setOutput('version', data.version);
  core.setOutput('dry_run_flag', data.dry_run_flag);
  core.setOutput('success_comment', JSON.stringify(data.success_comment)); // Escape the multi-line string
  core.setOutput('publish_branch', data.publish_branch);
  core.setOutput('ref', data.ref);
  core.setOutput('repo', data.repo);
  core.setOutput('pr_number', data.pr_number);
  core.setOutput('pr_author', data.pr_author);
};

/**
 * @param {{
 *  github: import('octokit/dist-types').Octokit;
 *  context: GitHubContext;
 * }} param
 *
 * @returns {Promise<PublishResolveData>}
 */
async function getPublishResolveData({ github, context }) {
  // We use empty strings as default values so that we can let the `actions/checkout` action apply its default resolution
  /**
   * "The short ref name of the branch or tag that triggered the workflow run. This value matches the branch or tag name shown
   * on GitHub. For example, feature-branch-1."
   *
   * Source: https://docs.github.com/en/actions/learn-github-actions/variables#default-environment-variables
   */
  const refName = process.env.GITHUB_REF_NAME;
  if (!refName) {
    throw new Error('The github ref name could not be determined');
  }

  const DEFAULT_PUBLISH_BRANCH = `publish/${refName}`;
  const DEFAULT_REPO = '';
  const DEFAULT_REF = '';

  /** @type {DryRunFlag} */
  const DRY_RUN_DISABLED = '';
  /** @type {DryRunFlag} */
  const DRY_RUN_ENABLED = '--dry-run';

  switch (process.env.GITHUB_EVENT_NAME) {
    case 'schedule': {
      const data = {
        version: 'canary',
        dry_run_flag: DRY_RUN_DISABLED,
        success_comment: '',
        publish_branch: DEFAULT_PUBLISH_BRANCH,
        repo: DEFAULT_REPO,
        ref: DEFAULT_REF,
        pr_number: '',
        pr_author: '',
      };
      console.log('"schedule" trigger detected', { data });
      return data;
    }

    case 'release': {
      const data = {
        version: refName,
        dry_run_flag: DRY_RUN_DISABLED,
        success_comment: '',
        publish_branch: DEFAULT_PUBLISH_BRANCH,
        repo: DEFAULT_REPO,
        ref: DEFAULT_REF,
        pr_number: '',
        pr_author: '',
      };
      console.log('"release" trigger detected', { data });
      return data;
    }

    case 'workflow_dispatch': {
      const prNumber = context.payload.inputs?.pr;

      if (!prNumber) {
        const data = {
          version: '0.0.0-dry-run.0',
          dry_run_flag: DRY_RUN_ENABLED,
          success_comment: '',
          publish_branch: DEFAULT_PUBLISH_BRANCH,
          // In this case the default checkout logic should use the branch/tag selected when triggering the workflow
          repo: DEFAULT_REPO,
          ref: DEFAULT_REF,
          pr_number: '',
          pr_author: '',
        };
        console.log(
          '"workflow_dispatch" trigger detected, no PR number provided',
          { data }
        );
        return data;
      }

      const pr = await github.rest.pulls.get({
        owner: context.repo.owner,
        repo: context.repo.repo,
        pull_number: Number(prNumber),
      });
      if (!pr?.data?.head?.repo) {
        throw new Error(
          `The PR data for PR number ${prNumber} is missing the head branch information`
        );
      }

      const fullSHA = pr.data.head.sha;
      const shortSHA = fullSHA.slice(0, 7);

      // Use the base version from nx@next so PR releases align with the current next/beta release line
      const currentNextVersion = execSync('npm view nx@next version', {
        windowsHide: false,
      })
        .toString()
        .trim();
      const nextVersionMatch = currentNextVersion.match(/^(\d+\.\d+\.\d+)/);
      if (!nextVersionMatch) {
        throw new Error(
          `Unable to parse the current next version from the npm registry: "${currentNextVersion}"`
        );
      }
      const prBaseVersion = nextVersionMatch[1];
      const version = `${prBaseVersion}-pr.${prNumber}.${shortSHA}`;
      const repo = pr.data.head.repo.full_name;
      const ref = pr.data.head.ref;
      const pr_author = pr.data.user?.login || 'unknown';

      const data = {
        version,
        dry_run_flag: DRY_RUN_DISABLED,
        success_comment: getSuccessCommentForPR({
          context,
          version,
          repo,
          ref,
          pr_short_sha: shortSHA,
          pr_full_sha: fullSHA,
        }),
        publish_branch: `publish-pr-${prNumber}`,
        // Override the default git checkout for the build job
        repo,
        ref,
        pr_number: prNumber,
        pr_author,
      };
      console.log('"workflow_dispatch" trigger detected, PR number provided', {
        data,
      });
      return data;
    }

    default:
      throw new Error(
        `The publish.yml workflow was triggered by an unexpected event: "${process.env.GITHUB_EVENT_NAME}"`
      );
  }
}

function getSuccessCommentForPR({
  context,
  version,
  repo,
  ref,
  pr_short_sha,
  pr_full_sha,
}) {
  return `## 🐳 We have a release for that!

  This PR has a release associated with it. You can try it out using this command:

  \`\`\`bash
  npx create-nx-workspace@${version} my-workspace
  \`\`\`

  Or just copy this version and use it in your own command:
  \`\`\`bash
  ${version}
  \`\`\`

  | Release details | 📑 |
  | ------------- | ------------- |
  | **Published version** | [${version}](https://www.npmjs.com/package/nx/v/${version}) |
  | **Triggered by** | @${context.actor} |
  | **Branch** | [${ref}](https://github.com/${repo}/tree/${ref}) |
  | **Commit** | [${pr_short_sha}](https://github.com/${repo}/commit/${pr_full_sha}) |
  | **Workflow run** | [${context.runId}](https://github.com/nrwl/nx/actions/runs/${context.runId}) |

  To request a new release for this pull request, mention someone from the Nx team or the \`@nrwl/nx-pipelines-reviewers\`.
`;
}