now-cli/bin/now-upgrade.js

#!/usr/bin/env node

// Packages
const chalk = require("chalk");
const minimist = require("minimist");
const ms = require("ms");
const stripAnsi = require("strip-ansi");

// Ours
const login = require("../lib/login");
const cfg = require("../lib/cfg");
const NowPlans = require("../lib/plans");
const indent = require("../lib/indent");
const listInput = require("../lib/utils/input/list");
const code = require("../lib/utils/output/code");
const error = require("../lib/utils/output/error");
const success = require("../lib/utils/output/success");
const cmd = require("../lib/utils/output/cmd");
const logo = require("../lib/utils/output/logo");

const argv = minimist(process.argv.slice(2), {
  string: ["config", "token"],
  boolean: ["help", "debug"],
  alias: {
    help: "h",
    config: "c",
    debug: "d",
    token: "t"
  }
});

const help = () => {
  console.log(
    `
  ${chalk.bold(`${logo} now upgrade`)} [plan]

  ${chalk.dim("Options:")}

    -h, --help              Output usage information
    -c ${chalk.bold.underline("FILE")}, --config=${chalk.bold.underline("FILE")}  Config file
    -d, --debug             Debug mode [off]
    -t ${chalk.bold.underline("TOKEN")}, --token=${chalk.bold.underline("TOKEN")} Login token

  ${chalk.dim("Examples:")}

  ${chalk.gray("–")} List available plans and pick one interactively

      ${chalk.cyan("$ now upgrade")}

      ${chalk.yellow("NOTE:")} ${chalk.gray("Make sure you have a payment method, or add one:")}

      ${chalk.cyan(`$ now billing add`)}

  ${chalk.gray("–")} Pick a specific plan (premium):

      ${chalk.cyan(`$ now upgrade premium`)}
  `
  );
};

// options
const debug = argv.debug;
const apiUrl = argv.url || "https://api.zeit.co";

if (argv.config) {
  cfg.setConfigFile(argv.config);
}

const exit = code => {
  // we give stdout some time to flush out
  // because there's a node bug where
  // stdout writes are asynchronous
  // https://github.com/nodejs/node/issues/6456
  setTimeout(() => process.exit(code || 0), 100);
};

if (argv.help) {
  help();
  exit(0);
} else {
  const config = cfg.read();

  Promise.resolve(argv.token || config.token || login(apiUrl))
    .then(async token => {
      try {
        await run(token);
      } catch (err) {
        if (err.userError) {
          error(err.message);
        } else {
          error(`Unknown error: ${err.stack}`);
        }
        exit(1);
      }
    })
    .catch(e => {
      error(`Authentication error – ${e.message}`);
      exit(1);
    });
}

function buildInquirerChoices(current, until) {
  if (until) {
    until = until.split(" ");
    until = " for " + chalk.bold(until[0]) + " more " + until[1];
  } else {
    until = "";
  }
  const ossTitle = current === "oss"
    ? `oss FREE ${" ".repeat(28)} (current)`
    : "oss FREE";
  const premiumTitle = current === "premium"
    ? `premium $15/mo ${" ".repeat(24 - stripAnsi(until).length)} (current${until})`
    : "premium $15/mo";
  return [
    {
      name: [
        ossTitle,
        indent("✓ All code is public and open-source", 2),
        indent("✓ 20 deploys per month | 1GB monthly bandwidth", 2),
        indent("✓ 1GB FREE storage | 1MB size limit per file", 2)
      ].join("\n"),
      value: "oss",
      short: "oss FREE"
    },
    {
      name: [
        premiumTitle,
        indent("✓ All code is private and secure", 2),
        indent("✓ 1000 deploys per month | 50GB monthly bandwidth", 2),
        indent("✓ 100GB storage | No filesize limit", 2)
      ].join("\n"),
      value: "premium",
      short: "premium $15/mo"
    }
  ];
}

async function run(token) {
  const args = argv._;
  if (args.length > 1) {
    error("Invalid number of arguments");
    return exit(1);
  }

  const start = new Date();
  const plans = new NowPlans(apiUrl, token, { debug });

  let planId = args[0];

  if (![undefined, "oss", "premium"].includes(planId)) {
    error(`Invalid plan name – should be ${code("oss")} or ${code("premium")}`);
    return exit(1);
  }

  const currentPlan = await plans.getCurrent();

  if (planId === undefined) {
    const elapsed = ms(new Date() - start);

    let message = `To manage this from the web UI, head to https://zeit.co/account\n`;
    message += `> Selecting a plan for your account ${chalk.gray(`[${elapsed}]`)}`;
    const choices = buildInquirerChoices(currentPlan.id, currentPlan.until);

    planId = await listInput({
      message,
      choices,
      separator: true,
      abort: "end"
    });
  }

  if (
    planId === undefined ||
    (planId === currentPlan.id && currentPlan.until === undefined)
  ) {
    return console.log("No changes made");
  }

  let newPlan;

  try {
    newPlan = await plans.set(planId);
  } catch (err) {
    let errorBody;
    if (err.res && err.res.status === 400) {
      errorBody = err.res.json();
    } else {
      const message = "A network error has occurred. Please retry.";
      errorBody = { message };
    }

    const _err = (await errorBody).error;
    const { code, message } = _err;

    if (code === "customer_not_found" || code === "source_not_found") {
      error(
        `You have no payment methods available. Run ${cmd("now billing add")} to add one`
      );
    } else {
      error(`An unknow error occured. Please try again later ${message}`);
    }
    plans.close();
    return;
  }

  if (currentPlan.until && newPlan.id === "premium") {
    success(
      `The cancelation has been undone. You're back on the ${chalk.bold("Premium plan")}`
    );
  } else if (newPlan.until) {
    success(
      `Your plan will be switched to OSS in ${chalk.bold(newPlan.until)}. Your card will not be charged again`
    );
  } else {
    success(`You're now on the ${chalk.bold("Premium plan")}`);
  }

  plans.close();
}