added inquirer

Author: Ajay Narain Mathur

.nvmrc

@@ -0,0 +1 @@
+16

lib/utils/builds.ts

@@ -4,6 +4,25 @@ import * as times from './times';
 
 const defaultBuildDir = path.join(__dirname, '../..');
 
+export async function getAllDownloadedSlugs(cwd = defaultBuildDir) {
+  let allDownloadedSlugs = [];
+  const pathToCache = path.join(cwd, '.data');
+
+  const allHosts = await fs.readDir(pathToCache);
+
+  await Promise.all(allHosts.map(async host => {
+    const allUsers = await fs.readDir(path.join(pathToCache, host));
+    await Promise.all(allUsers.map(async user => {
+      const allRepos = await fs.readDir(path.join(pathToCache, host, user));
+      allRepos.forEach(repo => {
+        allDownloadedSlugs.push(`${host}:${user}/${repo}`);
+      });
+    }))
+  }));
+
+  return allDownloadedSlugs;
+}
+
 export async function getBuildDir(cwd = defaultBuildDir, host, user, repo) {
   const buildsDir = path.join(cwd, '.data', host, user, repo, 'builds');
   await fs.mkdirp(buildsDir);
@@ -86,6 +105,7 @@ export async function getLastDownloadedBuildNumber(buildsDir) {
 }
 
 module.exports = {
+  getAllDownloadedSlugs,
   getBuildDir,
   getHistory,
   getLastDownloadedBuildNumber,

lib/utils/sanitize.ts

@@ -1,4 +1,8 @@
 import inquirer from 'inquirer';
+import { SUPPORTED_COMMANDS } from '../../index';
+import { InvalidInputError } from '../error';
+import { getAllDownloadedSlugs } from './builds';
+import flatten from 'lodash.flatten';
 
 const getHost = {
   name: 'host',
@@ -32,12 +36,32 @@ const getRepo = {
   message: 'Enter the repo name',
 };
 
+const getSlug = (choices) => ({
+  name: 'slug',
+  type: 'list',
+  message: 'Select the repo:',
+  choices
+})
+
 export async function sanitizeInput(input): Promise<[Array<string>, string]> {
   if (/(.*):(.*)\/(.*)/.test(input[0])) {
     return [input[0].match(/(.*):(.*)\/(.*)/).slice(1), input[1]];
   }
 
-  const { host, user, repo } = await inquirer.prompt([getHost, getUser, getRepo]);
+  if (input[0] === 'download') {
+    const { host, user, repo } = await inquirer.prompt([getHost, getUser, getRepo]);
+
+    return [[host, user, repo], input[0]];
+  }
+
+  if (SUPPORTED_COMMANDS.indexOf(input[0]) !== -1) {
+    const slugs = await getAllDownloadedSlugs();
+    const { slug } = await inquirer.prompt([getSlug(slugs)]);
+
+    if (/(.*):(.*)\/(.*)/.test(slug)) {
+      return [slug.match(/(.*):(.*)\/(.*)/).slice(1), input[0]];
+    }
+  }
 
-  return [[host, user, repo], input[0]];
+  throw new InvalidInputError(`Invalid input. Please run build-stats --help to documention for the tool.`, input);
 }

package.json

@@ -39,8 +39,10 @@
     "got": "^8.0.1",
     "inquirer": "^8.2.0",
     "left-pad": "^1.2.0",
+    "lodash.flatten": "^4.4.0",
     "lodash.groupby": "^4.6.0",
     "lodash.pick": "^4.4.0",
+    "lodash.without": "^4.4.0",
     "meow": "^v9.0.0",
     "mkdirp": "^0.5.1",
     "ora": "^5.4.1",

yarn.lock

@@ -3108,6 +3108,11 @@ lodash.debounce@^4.0.8:
   resolved "https://registry.yarnpkg.com/lodash.debounce/-/lodash.debounce-4.0.8.tgz#82d79bff30a67c4005ffd5e2515300ad9ca4d7af"
   integrity sha1-gteb/zCmfEAF/9XiUVMArZyk168=
 
+lodash.flatten@^4.4.0:
+  version "4.4.0"
+  resolved "https://registry.yarnpkg.com/lodash.flatten/-/lodash.flatten-4.4.0.tgz#f31c22225a9632d2bbf8e4addbef240aa765a61f"
+  integrity sha1-8xwiIlqWMtK7+OSt2+8kCqdlph8=
+
 lodash.groupby@^4.6.0:
   version "4.6.0"
   resolved "https://registry.yarnpkg.com/lodash.groupby/-/lodash.groupby-4.6.0.tgz#0b08a1dcf68397c397855c3239783832df7403d1"
@@ -3123,6 +3128,11 @@ lodash.pick@^4.4.0:
   resolved "https://registry.yarnpkg.com/lodash.pick/-/lodash.pick-4.4.0.tgz#52f05610fff9ded422611441ed1fc123a03001b3"
   integrity sha1-UvBWEP/53tQiYRRB7R/BI6AwAbM=
 
+lodash.without@^4.4.0:
+  version "4.4.0"
+  resolved "https://registry.yarnpkg.com/lodash.without/-/lodash.without-4.4.0.tgz#3cd4574a00b67bae373a94b748772640507b7aac"
+  integrity sha1-PNRXSgC2e643OpS3SHcmQFB7eqw=
+
 lodash@^4.17.21, lodash@^4.7.0:
   version "4.17.21"
   resolved "https://registry.yarnpkg.com/lodash/-/lodash-4.17.21.tgz#679591c564c3bffaae8454cf0b3df370c3d6911c"