now-cli/bin/now-billing.js

#!/usr/bin/env node

// Native
const { resolve } = require("path");

// Packages
const chalk = require("chalk");
const minimist = require("minimist");
const ms = require("ms");

// Ours
const login = require("../lib/login");
const cfg = require("../lib/cfg");
const { error } = require("../lib/error");
const NowCreditCards = require("../lib/credit-cards");
const indent = require("../lib/indent");
const listInput = require("../lib/utils/input/list");
const success = require("../lib/utils/output/success");
const promptBool = require("../lib/utils/input/prompt-bool");
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 subcommand = argv._[0];

const help = () => {
  console.log(
    `
  ${chalk.bold(`${logo} now billing`)} <ls | add | rm | set-default>

  ${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("–")} Lists all your credit cards:

      ${chalk.cyan("$ now billing ls")}

  ${chalk.gray("–")} Adds a credit card (interactively):

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

  ${chalk.gray("–")} Removes a credit card:

      ${chalk.cyan(`$ now billing rm <id>`)}

      ${chalk.gray("–")} If the id is omitted, you can choose interactively

  ${chalk.gray("–")} Selects your default credit card:

      ${chalk.cyan(`$ now billing set-default <id>`)}

      ${chalk.gray("–")} If the id is omitted, you can choose interactively
  `
  );
};

// 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 || !subcommand) {
  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);
    });
}

// Builds a `choices` object that can be passesd to inquirer.prompt()
function buildInquirerChoices(cards) {
  return cards.cards.map(card => {
    const _default = card.id === cards.defaultCardId
      ? " " + chalk.bold("(default)")
      : "";
    const id = `${chalk.cyan(`ID: ${card.id}`)}${_default}`;
    const number = `${chalk.gray("#### ").repeat(3)}${card.last4}`;
    const str = [
      id,
      indent(card.name, 2),
      indent(`${card.brand} ${number}`, 2)
    ].join("\n");

    return {
      name: str, // Will be displayed by Inquirer
      value: card.id, // Will be used to identify the answer
      short: card.id // Will be displayed after the users answers
    };
  });
}

async function run(token) {
  const start = new Date();
  const creditCards = new NowCreditCards(apiUrl, token, { debug });
  const args = argv._.slice(1);

  switch (subcommand) {
    case "ls":
    case "list": {
      const cards = await creditCards.ls();
      const text = cards.cards
        .map(card => {
          const _default = card.id === cards.defaultCardId
            ? " " + chalk.bold("(default)")
            : "";
          const id = `${chalk.gray("-")} ${chalk.cyan(`ID: ${card.id}`)}${_default}`;
          const number = `${chalk.gray("#### ").repeat(3)}${card.last4}`;
          let address = card.address_line1;

          if (card.address_line2) {
            address += `, ${card.address_line2}.`;
          } else {
            address += ".";
          }

          address += `\n${card.address_city}, `;

          if (card.address_state) {
            address += `${card.address_state}, `;
          }

          // Stripe is returning a two digit code for the country,
          // but we want the full country name
          address += `${card.address_zip}. ${card.address_country}`;

          return [
            id,
            indent(card.name, 2),
            indent(`${card.brand} ${number}`, 2),
            indent(address, 2)
          ].join("\n");
        })
        .join("\n\n");

      const elapsed = ms(new Date() - start);
      console.log(
        `> ${cards.cards.length} card${cards.cards.length === 1 ? "" : "s"} found ${chalk.gray(`[${elapsed}]`)}`
      );
      if (text) {
        console.log(`\n${text}\n`);
      }

      break;
    }

    case "set-default": {
      if (args.length > 1) {
        error("Invalid number of arguments");
        return exit(1);
      }

      const start = new Date();
      const cards = await creditCards.ls();

      if (cards.cards.length === 0) {
        error("You have no credit cards to choose from");
        return exit(0);
      }

      let cardId = args[0];

      if (cardId === undefined) {
        const elapsed = ms(new Date() - start);
        const message = `Selecting a new default payment card from ${cards.cards.length} total ${chalk.gray(`[${elapsed}]`)}`;
        const choices = buildInquirerChoices(cards);

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

      // Check if the provided cardId (in case the user
      // typed `now billing set-default <some-id>`) is valid
      if (cardId) {
        const label = `Are you sure that you to set this card as the default?`;
        const confirmation = await promptBool(label);
        if (!confirmation) {
          console.log("Aborted");
          break;
        }
        const start = new Date();
        await creditCards.setDefault(cardId);

        const card = cards.cards.find(card => card.id === cardId);
        const elapsed = ms(new Date() - start);
        success(
          `${card.brand} ending in ${card.last4} is now the default ${chalk.gray(`[${elapsed}]`)}`
        );
      } else {
        console.log("No changes made");
      }

      break;
    }

    case "rm":
    case "remove": {
      if (args.length > 1) {
        error("Invalid number of arguments");
        return exit(1);
      }

      const start = new Date();
      const cards = await creditCards.ls();

      if (cards.cards.length === 0) {
        error("You have no credit cards to choose from to delete");
        return exit(0);
      }

      let cardId = args[0];

      if (cardId === undefined) {
        const elapsed = ms(new Date() - start);
        const message = `Selecting a card to ${chalk.underline("remove")} from ${cards.cards.length} total ${chalk.gray(`[${elapsed}]`)}`;
        const choices = buildInquirerChoices(cards);

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

      // Shoud check if the provided cardId (in case the user
      // typed `now billing rm <some-id>`) is valid
      if (cardId) {
        const label = `Are you sure that you want to remove this card?`;
        const confirmation = await promptBool(label);
        if (!confirmation) {
          console.log("Aborted");
          break;
        }
        const start = new Date();
        await creditCards.rm(cardId);

        const deletedCard = cards.cards.find(card => card.id === cardId);
        const remainingCards = cards.cards.filter(card => card.id !== cardId);

        let text = `${deletedCard.brand} ending in ${deletedCard.last4} was deleted`;
        //  ${chalk.gray(`[${elapsed}]`)}

        if (cardId === cards.defaultCardId) {
          if (remainingCards.length === 0) {
            // The user deleted the last card in their account
            text += `\n${chalk.yellow("Warning!")} You have no default card`;
          } else {
            // We can't guess the current default card – let's ask the API
            const cards = await creditCards.ls();
            const newDefaultCard = cards.cards.find(
              card => card.id === cards.defaultCardId
            );

            text += `\n${newDefaultCard.brand} ending in ${newDefaultCard.last4} in now default`;
          }
        }

        const elapsed = ms(new Date() - start);
        text += ` ${chalk.gray(`[${elapsed}]`)}`;
        success(text);
      } else {
        console.log("No changes made");
      }

      break;
    }

    case "add": {
      require(resolve(__dirname, "now-billing-add.js"))(creditCards);

      break;
    }

    default:
      error("Please specify a valid subcommand: ls | add | rm | set-default");
      help();
      exit(1);
  }

  creditCards.close();
}