WIP add total number of domains in prompt for filtered domains

Need to update types and update logic for when this condition will be true. Currently defaulted to true verify size of filtered domains

packages/cli/src/commands/domains/index.ts

@@ -73,18 +73,65 @@ www.example.com  CNAME            www.example.herokudns.com
     return tableConfig
   }
 
+  splitKeyValuePair(input: string) {
+    let [key, value] = input.split(/=(.+)/)
+
+    key = key.trim()
+    value = value ? value.trim() : ''
+
+    return {key, value}
+  }
+
+  filterInfo = (filterKeyValue: string, domains: Array<Heroku.Domain>) => {
+    // parse --filter key-value pair
+    const {key: filterName, value} = this.splitKeyValuePair(filterKeyValue)
+
+    // filter table headers by value from --filter
+    // return size from filter
+
+    if (filterName === 'Domain Name') {
+      const filteredDomains = domains.filter(domain => domain.hostname!.includes(value))
+      return filteredDomains.length
+    }
+
+    if (filterName === 'DNS Record Type') {
+      // eslint-disable-next-line array-callback-return
+      const filteredDomains = domains.filter(domain => {
+        const kind = isApexDomain(domain.hostname!) ? 'ALIAS or ANAME' : 'CNAME'
+        kind.includes(value)
+      })
+      return filteredDomains.length
+    }
+
+    if (filterName === 'DNS Target') {
+      const filteredDomains = domains.filter(domain => domain.cname!.includes(value))
+      console.log('filteredDomains', filteredDomains.length)
+      return filteredDomains.length
+    }
+
+    if (filterName === 'SNI Endpoint') {
+      const filteredDomains = domains.filter(domain => domain.sni_endpoint!.includes(value))
+      return filteredDomains.length
+    }
+  }
+
   async run() {
     const {flags} = await this.parse(DomainsIndex)
     // const {body: domains, headers: headerInfo} = await this.heroku.get<Array<Heroku.Domain>>(`/apps/${flags.app}/domains`)
     const {body: domains, headers: headerInfo, statusCode: code} = await paginateRequest(this.heroku, `/apps/${flags.app}/domains`, 1000)
     const herokuDomain = domains.find(domain => domain.kind === 'heroku')
     const customDomains = domains.filter(domain => domain.kind === 'custom')
+    let filteredDomainsSize: any = 0
     let displayTotalDomains = false
 
     // console.log('headerInfo', domainsNew)
     // console.log('statusCode', code)
     // console.log('headerInfo', headerInfo)
 
+    if (flags.filter) {
+      filteredDomainsSize = this.filterInfo(flags.filter, domains)
+    }
+
     if (flags.json) {
       ux.styledJSON(domains)
     } else {
@@ -94,9 +141,10 @@ www.example.com  CNAME            www.example.herokudns.com
         // console.log('# of custom domains', customDomains.length)
         // console.log('# of total domains', domains.length)
         // ux.log()
+        console.log('filteredDomainsSize.length', filteredDomainsSize.length)
         if (customDomains.length > 100 && !flags.csv) {
           ux.warn(`This app has over 100 domains. Your terminal may not be configured to display the total amount of domains. We recommend outputting this information to a csv file: ${color.cyan('heroku domains -a example-app --csv > example-file.txt')}`)
-          displayTotalDomains = await confirm({default: false, message: `Display all ${customDomains.length} domains?`})
+          displayTotalDomains = await confirm({default: false, message: `Display all ${flags.filter ? filteredDomainsSize : customDomains.length} domains?`})
 
           if (!displayTotalDomains) {
             return