import { exec, spawn } from "child-process-promise";
import chalk from "chalk";
import path from "path";
import fs from "fs";
import inquirer from "inquirer";

export default async function deploy(args: any, branch: string) {
  const _env = args.deployEnv;
  if (!_env) {
    console.error("--deploy-env must be set.");
    process.exit(1);
  }

  // Validate terraform directory
  let terraformDirIsValid = false;
  try {
    terraformDirIsValid = fs.statSync(args.terraformDir).isDirectory();
  } finally {
    if (!terraformDirIsValid) {
      console.error("--terraform-dir must be a directory.");
      process.exit(1);
    }
  }

  console.log("\n\n");

  // Force env to lowercase so it works with S3 bucket names and other stuff
  const env = _env.toLowerCase();

  process.env.DEPLOYMENT_ENVIRONMENT = env;

  /** If branch does not match deploy environment name, as for confirmation
   *  before continuing */
  if (branch !== env) {
    const { confirm } = await inquirer.prompt([
      {
        type: "confirm",
        name: "confirm",
        message: `You are attempting to deploy from git branch ${chalk.yellow.bold(
          branch !== "" ? branch : "<no branch>"
        )} to environment ${chalk.blue.bold(
          env
        )}. Are you sure you want to continue?`
      }
    ]);

    if (!confirm) {
      console.error(chalk.red("Deployment cancelled."));
      process.exit(1);
    }
  }

  console.log(
    `Building and deploying infrastructure for ${chalk.blue.bold(
      env
    )} environment...`
  );

  if (args.envFile) {
    console.log(
      `> Loading environment variables from ${chalk.bold.blue(args.envFile)}`
    );
    require("dotenv").config({ path: args.envFile });
  } else {
    try {
      console.log(
        `> Loading environment variables from ${chalk.bold.blue(`${env}.env`)}`
      );
      require("dotenv").config({ path: `${env}.env` });
    } catch (e) {
      console.warn("No .env file found for this environment.");
    }
  }

  // Run pre-apply scripts
  if (args.preApply) {
    console.log(`> Running pre-apply script`);
    try {
      await spawn(`sh ${args.preApply}`, [], {
        shell: true,
        stdio: "inherit"
      });
    } catch (e) {
      console.error(chalk.red("Pre-apply script failed!"));
      process.exit(1);
    }
  }

  // Initialize terraform and select the correct workspace
  await exec(
    `terraform init
        terraform workspace new ${env}
        terraform workspace select ${env}`,
    { cwd: args.terraformDir }
  );

  // Try to find a .tfvars file that matches the current env
  const defaultVarFile = "dev.tfvars";
  let varFile = path.join(args.terraformDir, `${env}.tfvars`);
  try {
    varFile = fs.statSync(varFile).isFile() ? `${env}.tfvars` : defaultVarFile;
  } catch (e) {
    varFile = defaultVarFile;
  }

  const mode = args.autoApprove ? "auto-approve" : "interactive";
  console.log(
    `> Running ${chalk.yellow(
      "terraform apply"
    )} using variables from ${chalk.blue.bold(varFile)} in ${chalk.green.bold(
      mode
    )} mode`
  );

  // If we're not running in auto-approve mode, just run apply
  try {
    if (!args.autoApprove) {
      await spawn(`terraform apply -var-file="${varFile}"`, [], {
        shell: true,
        cwd: args.terraformDir,
        stdio: "inherit"
      });
    }
    // Otherwise, we need to run plan and then apply
    else {
      await spawn(
        `terraform plan -input=false -out=tfplan -var-file="${varFile}" &&
          terraform apply -input=false -auto-approve tfplan`,
        [],
        { shell: true, cwd: args.terraformDir, stdio: "inherit" }
      );
    }
  } catch (e) {
    console.error(chalk.red("Terraform deploy failed!"));
    process.exit(1);
  }

  // Run post-apply scripts
  if (args.postApply) {
    console.log(`> Running post-apply script`);
    try {
      await spawn(`sh ${args.postApply}`, [], {
        shell: true,
        stdio: "inherit"
      });

      console.log(`\n\n✨ ${chalk.green.bold("Deployment was sucessful!")} ✨`);
    } catch (e) {
      console.error(chalk.red("Post-apply script failed!"));
      process.exit(1);
    }
  }
}