index.js

#!/usr/bin/env node

const chalk = require('chalk');
const cli = require('commander');
const Configstore = require('configstore');
const fs = require('fs');
const promptly = require('promptly');
const pkg = require('./package.json');

const {
  printOrderByRestaurantsGraph,
  printOrdersOfLastMonthsGraph,
  printOrdersByDayGraph,
  calculateTotalMoneySpent,
} = require('./orders.js');
const scrapeZomatoOrders = require('./zomato');

const SECTION_BREAK = '\n\n';

const readOrdersFromFile = file => JSON.parse(fs.readFileSync(file));
const writeOrdersToFile = (orders, file) => fs.writeFileSync(file, JSON.stringify(orders));

const conf = new Configstore(pkg.name);

const getCredentialsFromPrompt = async () => {
  console.log('Please enter your Zomato credentials');

  emailID = await promptly.prompt('Username / Email: ', { trim: true });
  password = await promptly.prompt('Password: ', { trim: true, silent: true, replace: '*' });

  conf.set('emailID', emailID);
  conf.set('password', password);

  return { emailID, password };
};

const getUserCredentials = async () => {
  const emailID = conf.get('emailID');
  const password = conf.get('password');

  if (emailID && password) {
    return { emailID, password };
  }

  return await getCredentialsFromPrompt();
};

async function main() {
  let orders;
  if (cli.input) {
    orders = readOrdersFromFile(cli.input);
  } else {
    const { emailID, password } = await getUserCredentials();
    orders = await scrapeZomatoOrders({ emailID, password });

    if (cli.save) {
      writeOrdersToFile(orders, cli.save);
    }
  }

  const isOrderDelivered = ({ status }) => status === 'Delivered';
  const deliveredOrders = orders.filter(isOrderDelivered);
  console.log(chalk.bold(`\nYour total delivered orders are: ${deliveredOrders.length}`));
  console.log(chalk.bold(`Total money spent: ${calculateTotalMoneySpent(orders)}`))

  console.log(SECTION_BREAK);

  console.log(chalk.bold("Top 10 Restaurants from where you've ordered:"));
  printOrderByRestaurantsGraph(deliveredOrders, 10);

  console.log(SECTION_BREAK);

  console.log(chalk.bold('Distribution of your spendings over the last 12 Months:'));
  printOrdersOfLastMonthsGraph(deliveredOrders, 12);

  console.log(SECTION_BREAK);

  console.log(chalk.bold('Weekday wise distribution of your spendings:'));
  printOrdersByDayGraph(deliveredOrders);
}

(async () => {
  cli
    .version(pkg.version)
    .option('-i, --input <file>', 'Read orders from file')
    .option('-s, --save <file>', 'Save the extracted orders to file')
    .option('-c, --configure', 'Update your user credentials')
    .parse(process.argv);

  if (cli.configure) {
    await getCredentialsFromPrompt();
    console.log('Your credentials have been updated');
    return;
  }

  main();
})();