#!/usr/bin/env node

const { GITHUB_TOKEN, HEROKU_API_TOKEN } = process.env

if (!GITHUB_TOKEN) {

throw new Error('You must supply a GITHUB_TOKEN environment variable!')

if (!HEROKU_API_TOKEN) {

throw new Error('You must supply a HEROKU_API_TOKEN environment variable!')

const program = require('commander')

const { has } = require('lodash')

const getOctokit = require('./helpers/github')

const parsePrUrl = require('./deployment/parse-pr-url')

const deployToStaging = require('./deployment/deploy-to-staging')

const undeployFromStaging = require('./deployment/undeploy-from-staging')

const STAGING_FLAG = '--staging'

const PRODUCTION_FLAG = '--production'

const ALLOWED_OWNER = 'github'

const ALLOWED_SOURCE_REPOS = ['docs', 'docs-internal']

const EXPECTED_PR_URL_FORMAT = `https://github.com/${ALLOWED_OWNER}/(${ALLOWED_SOURCE_REPOS.join('|')})/pull/123`

.description('Trigger a deployment to Heroku for either staging or production apps')

.option(PRODUCTION_FLAG, 'Deploy the latest internal main branch to Production')

.option(`${STAGING_FLAG} <PR_URL>`, 'Deploy a pull request to Staging')

.option('--rebuild', 'Force a Staging deployment to rebuild the Heroku App from scratch')

.option('--destroy', 'Undeploy a Staging deployment by deleting the Heroku App')

.parse(process.argv)

const opts = program.opts()

const isProduction = opts.production === true

const isStaging = has(opts, 'staging')

const prUrl = opts.staging

const forceRebuild = opts.rebuild === true

const destroy = opts.destroy === true

if (!isProduction && !isStaging) {

return invalidateAndExit(

'commander.missingArgument',

`error: must specify option '${STAGING_FLAG} <PR_URL>' or '${PRODUCTION_FLAG}'`

)

if (isProduction && isStaging) {

return invalidateAndExit(

'commander.conflictingArgument',

`error: must specify option '${STAGING_FLAG} <PR_URL>' or '${PRODUCTION_FLAG}' but not both`

)

if (isProduction && forceRebuild) {

return invalidateAndExit(

'commander.conflictingArgument',

`error: cannot specify option '--rebuild' combined with option '${PRODUCTION_FLAG}'`

)

if (isProduction && destroy) {

return invalidateAndExit(

'commander.conflictingArgument',

`error: cannot specify option '--destroy' combined with option '${PRODUCTION_FLAG}'`

)

const { owner, repo, pullNumber } = parsePrUrl(prUrl)

if (isStaging) {

if (owner !== ALLOWED_OWNER || !ALLOWED_SOURCE_REPOS.includes(repo) || !pullNumber) {

return invalidateAndExit(

'commander.invalidOptionArgument',

`error: option '${STAGING_FLAG}' argument '${prUrl}' is invalid.

)

}

function invalidateAndExit (errorType, message) {

program._displayError(1, errorType, message)

process.exit(1)

async function deploy () {

async function deployProduction () {

async function deployStaging ({ owner, repo, pullNumber, forceRebuild = false, destroy = false }) {