improve list command

Author: charliegerard

lib/commands/index.js

@@ -10,3 +10,4 @@ export * from './report/index.js'
 export * from './wrapper/index.js'
 export * from './scan/index.js'
 export * from './audit-log/index.js'
+export * from './repos/index.js'

lib/commands/repos/list.js

@@ -1,17 +1,17 @@
-// @ts-nocheck
 /* eslint-disable no-console */
 
-// import chalk from 'chalk'
+import chalk from 'chalk'
+import chalkTable from 'chalk-table'
 import meow from 'meow'
 import ora from 'ora'
 
 import { outputFlags } from '../../flags/index.js'
-// import { handleApiCall, handleUnsuccessfulApiResponse } from '../../utils/api-helpers.js'
-import { InputError } from '../../utils/errors.js'
+import { handleApiCall, handleUnsuccessfulApiResponse } from '../../utils/api-helpers.js'
+import { prepareFlags } from '../../utils/flags.js'
 import { printFlagList } from '../../utils/formatting.js'
-// import { getDefaultKey, setupSdk } from '../../utils/sdk.js'
+import { getDefaultKey, setupSdk } from '../../utils/sdk.js'
 
-/** @type {import('../../utils/meow-with-subcommands').CliSubcommand} */
+/** @type {import('../../utils/meow-with-subcommands.js').CliSubcommand} */
 export const list = {
   description: 'List repositories in an organization',
   async run (argv, importMeta, { parentName }) {
@@ -26,13 +26,43 @@ export const list = {
   }
 }
 
+const listRepoFlags = prepareFlags({
+  sort: {
+    type: 'string',
+    shortFlag: 's',
+    default: 'created_at',
+    description: 'Sorting option',
+  },
+  direction: {
+    type: 'string',
+    default: 'desc',
+    description: 'Direction option',
+  },
+  perPage: {
+    type: 'number',
+    shortFlag: 'pp',
+    default: 30,
+    description: 'Number of results per page'
+  },
+  page: {
+    type: 'number',
+    shortFlag: 'p',
+    default: 1,
+    description: 'Page number'
+  },
+})
+
 // Internal functions
 
 /**
  * @typedef CommandContext
  * @property {boolean} outputJson
  * @property {boolean} outputMarkdown
  * @property {string} orgSlug
+ * @property {string} sort
+ * @property {string} direction
+ * @property {number} per_page
+ * @property {number} page
  */
 
 /**
@@ -44,7 +74,8 @@ export const list = {
  */
 function setupCommand (name, description, argv, importMeta) {
   const flags = {
-    ...outputFlags
+    ...outputFlags,
+    ...listRepoFlags
   }
 
   const cli = meow(`
@@ -65,28 +96,35 @@ function setupCommand (name, description, argv, importMeta) {
 
   const {
     json: outputJson,
-    markdown: outputMarkdown
+    markdown: outputMarkdown,
+    perPage,
+    sort,
+    direction,
+    page
   } = cli.flags
 
   if (!cli.input[0]) {
-    throw new InputError(`Please specify an organization slug. \n
-Example:
-socket scan list FakeOrg
-`)
+    console.error(`${chalk.bgRed('Input error')}: Please provide an organization slug \n`)
+    cli.showHelp()
+    return
   }
 
-  const orgSlug = cli.input[0] || ''
+  const [orgSlug = ''] = cli.input
 
   return {
     outputJson,
     outputMarkdown,
-    orgSlug
+    orgSlug,
+    sort,
+    direction,
+    page,
+    per_page: perPage
   }
 }
 
 /**
  * @typedef RepositoryData
- * @property {import('@socketsecurity/sdk').SocketSdkReturnType<'getOrgFullScanList'>["data"]} data
+ * @property {import('@socketsecurity/sdk').SocketSdkReturnType<'getOrgRepoList'>["data"]} data
  */
 
 /**
@@ -96,10 +134,36 @@ socket scan list FakeOrg
  * @returns {Promise<void|RepositoryData>}
  */
 async function listOrgRepos (orgSlug, input, spinner) {
-  // const socketSdk = await setupSdk(getDefaultKey())
-  console.log(input)
+  const socketSdk = await setupSdk(getDefaultKey())
+  const result = await handleApiCall(socketSdk.getOrgRepoList(orgSlug, input), 'looking up package')
+
+  if (!result.success) {
+    return handleUnsuccessfulApiResponse('getOrgRepoList', result, spinner)
+  }
+
+  spinner.stop()
 
-//   return {
-//     // data: result.data
-//   }
+  const options = {
+    columns: [
+      { field: 'id', name: chalk.magenta('ID') },
+      { field: 'name', name: chalk.magenta('Name') },
+      { field: 'visibility', name: chalk.magenta('Visibility') },
+      { field: 'default_branch', name: chalk.magenta('Default branch') },
+      { field: 'archived', name: chalk.magenta('Archived') }
+    ]
+  }
+
+  const formattedResults = result.data.results.map(d => {
+    return {
+      ...d
+    }
+  })
+
+  const table = chalkTable(options, formattedResults)
+
+  console.log(table, '\n')
+
+  return {
+    data: result.data
+  }
 }