const path = require('path');
const request = require('request-promise-native');

const repositoryRootPath = path.resolve(__dirname, '..', '..');
const appMetadata = require(path.join(repositoryRootPath, 'package.json'));

const yargs = require('yargs');
const argv = yargs
  .usage('Usage: $0 [options]')
  .help('help')
  .describe('nightly', 'Indicates that a nightly version should be produced')
  .wrap(yargs.terminalWidth()).argv;

function getAppName(version) {
  const match = version.match(/\d+\.\d+\.\d+(-([a-z]+)(\d+|-\w{4,})?)?$/);
  if (!match) {
    throw new Error(`Found incorrectly formatted Atom version ${version}`);
  } else if (match[2]) {
    return `atom-${match[2]}`;
  }

  return 'atom';
}

async function getReleaseVersion() {
  let releaseVersion = process.env.ATOM_RELEASE_VERSION || appMetadata.version;
  if (argv.nightly) {
    const releases = await request({
      url: 'https://api.github.com/repos/atom/atom-nightly-releases/releases',
      headers: {
        Accept: 'application/vnd.github.v3+json',
        'User-Agent': 'Atom Release Build'
      },
      json: true
    });

    let releaseNumber = 0;
    const baseVersion = appMetadata.version.split('-')[0];
    if (releases && releases.length > 0) {
      const latestRelease = releases.find(r => !r.draft);
      const versionMatch = latestRelease.tag_name.match(
        /^v?(\d+\.\d+\.\d+)-nightly(\d+)$/
      );

      if (versionMatch && versionMatch[1] === baseVersion) {
        releaseNumber = parseInt(versionMatch[2]) + 1;
      }
    }

    releaseVersion = `${baseVersion}-nightly${releaseNumber}`;
  }

  // Set our ReleaseVersion build variable and update VSTS' build number to
  // include the version.  Writing these strings to stdout causes VSTS to set
  // the associated variables.
  console.log(
    `##vso[task.setvariable variable=ReleaseVersion;isOutput=true]${releaseVersion}`
  );
  if (!process.env.SYSTEM_PULLREQUEST_PULLREQUESTNUMBER) {
    // Only set the build number on non-PR builds as it causes build errors when
    // non-admins send PRs to the repo
    console.log(
      `##vso[build.updatebuildnumber]${releaseVersion}+${
        process.env.BUILD_BUILDID
      }`
    );
  }

  // Write out some variables that indicate whether artifacts should be uploaded
  const buildBranch = process.env.BUILD_SOURCEBRANCHNAME;
  const isReleaseBranch =
    process.env.IS_RELEASE_BRANCH ||
    argv.nightly ||
    buildBranch.match(/\d\.\d+-releases/) !== null;
  const isSignedZipBranch =
    !isReleaseBranch &&
    (process.env.IS_SIGNED_ZIP_BRANCH ||
      buildBranch.startsWith('electron-') ||
      (buildBranch === 'master' &&
        !process.env.SYSTEM_PULLREQUEST_PULLREQUESTNUMBER));

  console.log(
    `##vso[task.setvariable variable=AppName;isOutput=true]${getAppName(
      releaseVersion
    )}`
  );
  console.log(
    `##vso[task.setvariable variable=IsReleaseBranch;isOutput=true]${isReleaseBranch}`
  );
  console.log(
    `##vso[task.setvariable variable=IsSignedZipBranch;isOutput=true]${isSignedZipBranch}`
  );
}

getReleaseVersion();
