add flags

Author: charliegerard

src/commands/threat-feed.ts

@@ -22,16 +22,47 @@ export const threatFeed: CliSubcommand = {
         throw new AuthError("User must be authenticated to run this command. To log in, run the command `socket login` and enter your API key.")
       }
       const spinner = ora(`Looking up the threat feed \n`).start()
-      await fetchThreatFeed(spinner, apiKey)
+      await fetchThreatFeed(input, spinner, apiKey)
     }
   }
 }
 
+const threatFeedFlags = {
+  perPage: {
+    type: 'number',
+    shortFlag: 'pp',
+    default: 30,
+    description: 'Number of items per page'
+  },
+  page: {
+    type: 'string',
+    shortFlag: 'p',
+    default: '1',
+    description: 'Page token'
+  },
+  direction: {
+    type: 'string',
+    shortFlag: 'd',
+    default: 'desc',
+    description: 'Order asc or desc by the createdAt attribute.'
+  },
+  filter: {
+    type: 'string',
+    shortFlag: 'f',
+    default: 'mal',
+    description: 'Filter what type of threats to return'
+  }
+}
+
 // Internal functions
 
 type CommandContext = {
   outputJson: boolean
   outputMarkdown: boolean
+  per_page: number
+  page: string
+  direction: string
+  filter: string
 }
 
 function setupCommand(
@@ -41,6 +72,7 @@ function setupCommand(
   importMeta: ImportMeta
 ): CommandContext | undefined {
   const flags: { [key: string]: any } = {
+    ...threatFeedFlags,
     ...outputFlags
   }
 
@@ -66,21 +98,36 @@ function setupCommand(
   const {
     json: outputJson,
     markdown: outputMarkdown,
+    perPage: per_page,
+    page,
+    direction,
+    filter
   } = cli.flags
 
   return <CommandContext>{
     outputJson,
-    outputMarkdown
+    outputMarkdown,
+    per_page,
+    page,
+    direction,
+    filter
   }
 }
 
 async function fetchThreatFeed(
+  { per_page, page, direction, filter }: CommandContext,
   spinner: Ora,
   apiKey: string
 ): Promise<void> {
-  const response = await queryAPI(`threat-feed`, apiKey)
+  const formattedQueryParams = formatQueryParams({ per_page, page, direction, filter }).join('&')
+  
+  const response = await queryAPI(`threat-feed?${formattedQueryParams}`, apiKey)
   const data = await response.json();
 
   spinner.stop()
   console.log(data)
 }
+
+const formatQueryParams = (params: any) => {
+  return Object.entries(params).map(entry => `${entry[0]}=${entry[1]}`)
+}