Rewire CLI args and flags once more

Author: huangsam

cmd/namigo/flag.go

@@ -7,10 +7,10 @@ import (
 	"github.com/urfave/cli/v2"
 )
 
-// checkCountFlag checks for valid count flag.
-func checkCountFlag(ctx *cli.Context, i int) error {
+// checkSizeFlag checks for valid size flag.
+func checkSizeFlag(ctx *cli.Context, i int) error {
 	if i <= 0 {
-		return fmt.Errorf("count %d is invalid", i)
+		return fmt.Errorf("size %d is invalid", i)
 	}
 	return nil
 }

cmd/namigo/main.go

@@ -39,14 +39,14 @@ func main() {
 								Usage: "Target interests",
 							},
 							&cli.IntFlag{
-								Name:   "count",
-								Usage:  "Maximum count of names",
+								Name:   "size",
+								Usage:  "Max size for generated names",
 								Value:  10,
-								Action: checkCountFlag,
+								Action: checkSizeFlag,
 							},
 							&cli.IntFlag{
 								Name:   "length",
-								Usage:  "Maximum length for each name",
+								Usage:  "Max length for each generated name",
 								Value:  20,
 								Action: checkLengthFlag,
 							},
@@ -60,10 +60,10 @@ func main() {
 				Usage: "Search for terms across entities",
 				Flags: []cli.Flag{
 					&cli.IntFlag{
-						Name:   "count",
-						Usage:  "Maximum count of results",
+						Name:   "size",
+						Usage:  "Max size for search results",
 						Value:  10,
-						Action: checkCountFlag,
+						Action: checkSizeFlag,
 					},
 					&cli.StringFlag{
 						Name:   "format",

cmd/namigo/sub/generate.go

@@ -26,10 +26,10 @@ func GeneratePromptAction(c *cli.Context) error {
 	if err != nil {
 		return err
 	}
-	maxResults := c.Int("count")
+	maxSize := c.Int("size")
 	maxLength := c.Int("length")
 
-	prompt, err := generate.GeneratePrompt(purpose, theme, demographics, interests, maxResults, maxLength)
+	prompt, err := generate.GeneratePrompt(purpose, theme, demographics, interests, maxSize, maxLength)
 	if err != nil {
 		return err
 	}

cmd/namigo/sub/search.go

@@ -20,15 +20,15 @@ func SearchPackageAction(c *cli.Context) error {
 	if len(searchTerm) == 0 {
 		return ErrMissingSearchTerm
 	}
-	maxResults := c.Int("count")
+	maxSize := c.Int("size")
 	outputFormat := model.GetOutputFormat(c.String("format"))
 
 	ptf := search.NewPortfolio()
 
 	ptf.Run(func(ptf *search.Portfolio) {
 		defer ptf.Done()
 		fmt.Printf("🔍 Search for %s results\n", ptf.Fmt.Golang.Label())
-		if searchResults, err := golang.SearchByScrape(searchTerm, maxResults); err == nil {
+		if searchResults, err := golang.SearchByScrape(searchTerm, maxSize); err == nil {
 			ptf.Res.Golang = searchResults
 		} else {
 			ptf.Err.Golang = err
@@ -38,7 +38,7 @@ func SearchPackageAction(c *cli.Context) error {
 	ptf.Run(func(ptf *search.Portfolio) {
 		defer ptf.Done()
 		fmt.Printf("🔍 Search for %s results\n", ptf.Fmt.NPM.Label())
-		if searchResults, err := npm.SearchByAPI(searchTerm, maxResults); err == nil {
+		if searchResults, err := npm.SearchByAPI(searchTerm, maxSize); err == nil {
 			ptf.Res.NPM = searchResults
 		} else {
 			ptf.Err.NPM = err
@@ -48,7 +48,7 @@ func SearchPackageAction(c *cli.Context) error {
 	ptf.Run(func(ptf *search.Portfolio) {
 		defer ptf.Done()
 		fmt.Printf("🔍 Search for %s results\n", ptf.Fmt.PyPI.Label())
-		if searchResults, err := pypi.SearchByAPI(searchTerm, maxResults); err == nil {
+		if searchResults, err := pypi.SearchByAPI(searchTerm, maxSize); err == nil {
 			ptf.Res.PyPI = searchResults
 		} else {
 			ptf.Err.PyPI = err
@@ -75,13 +75,13 @@ func SearchDNSAction(c *cli.Context) error {
 	if len(searchTerm) == 0 {
 		return ErrMissingSearchTerm
 	}
-	maxResults := c.Int("count")
+	maxSize := c.Int("size")
 	outputFormat := model.GetOutputFormat(c.String("format"))
 
 	ptf := search.NewPortfolio()
 
 	fmt.Printf("🔍 Search for %s results\n", ptf.Fmt.DNS.Label())
-	if searchResults, err := dns.SearchByProbe(searchTerm, maxResults); err == nil {
+	if searchResults, err := dns.SearchByProbe(searchTerm, maxSize); err == nil {
 		ptf.Res.DNS = searchResults
 	} else {
 		ptf.Err.DNS = err
@@ -104,13 +104,13 @@ func SearchEmailAction(c *cli.Context) error {
 	if len(searchTerm) == 0 {
 		return ErrMissingSearchTerm
 	}
-	maxResults := c.Int("count")
+	maxSize := c.Int("size")
 	outputFormat := model.GetOutputFormat(c.String("format"))
 
 	ptf := search.NewPortfolio()
 
 	fmt.Printf("🔍 Search for %s results\n", ptf.Fmt.Email.Label())
-	if searchResults, err := email.SearchByProbe(searchTerm, maxResults); err == nil {
+	if searchResults, err := email.SearchByProbe(searchTerm, maxSize); err == nil {
 		ptf.Res.Email = searchResults
 	} else {
 		ptf.Err.Email = err

pkg/generate/generate.go

@@ -7,13 +7,13 @@ import (
 )
 
 // GeneratePrompt generates the prompt for AI chatbots.
-func GeneratePrompt(purpose, theme, demographics, interests string, count, length int) (string, error) {
+func GeneratePrompt(purpose, theme, demographics, interests string, size, length int) (string, error) {
 	data := promptData{
 		Purpose:      purpose,
 		Demographics: demographics,
 		Interests:    interests,
 		Theme:        theme,
-		MaxCount:     count,
+		MaxSize:      size,
 		MaxLength:    length,
 	}
 	tmpl, err := template.New(promptName).Parse(promptTemplate)

pkg/generate/template.go

@@ -9,7 +9,7 @@ type promptData struct {
 	Demographics string
 	Interests    string
 	Theme        string
-	MaxCount     int
+	MaxSize      int
 	MaxLength    int
 }
 
@@ -21,7 +21,7 @@ const promptTemplate = `Given the following parameters:
 - Target Demographics: {{.Demographics}}
 - Target Interests: {{.Interests}}
 
-Generate up to {{.MaxCount}} names for a side business / project. All names
+Generate up to {{.MaxSize}} names for a side business / project. All names
 should have at most {{.MaxLength}} characters.
 
 For each of the names generated, please provide the following:

pkg/search/dns/search.go

@@ -8,7 +8,7 @@ import (
 )
 
 // SearchByProbe searches for DNS records via nameserver lookups.
-func SearchByProbe(name string, count int) ([]model.DNSRecord, error) {
+func SearchByProbe(name string, size int) ([]model.DNSRecord, error) {
 	domains := []string{"com", "org", "net", "io", "tech", "ai", "me", "shop"}
 	domainChan := make(chan string)
 
@@ -26,7 +26,7 @@ func SearchByProbe(name string, count int) ([]model.DNSRecord, error) {
 
 	for i := 0; i < 4; i++ {
 		wg.Add(1)
-		go netWorker(domainChan, &wg, &mu, &result, &errors, count)
+		go netWorker(domainChan, &wg, &mu, &result, &errors, size)
 	}
 
 	wg.Wait()

pkg/search/email/search.go

@@ -23,10 +23,10 @@ var (
 // To learn more:
 //
 // https://docs.abstractapi.com/email-validation
-func SearchByProbe(name string, count int) ([]model.EmailRecord, error) {
+func SearchByProbe(name string, size int) ([]model.EmailRecord, error) {
 	result := []model.EmailRecord{}
 	for _, domain := range domains {
-		if len(result) >= count {
+		if len(result) >= size {
 			break
 		}
 		email := fmt.Sprintf("%s@%s", name, domain)

pkg/search/golang/search.go

@@ -9,13 +9,13 @@ import (
 )
 
 // SearchByScrape searches for Go packages by scraping pkg.go.dev.
-func SearchByScrape(name string, count int) ([]model.GoPackage, error) {
+func SearchByScrape(name string, size int) ([]model.GoPackage, error) {
 	client := &http.Client{Timeout: 5 * time.Second}
 	pipeline := util.NewDocumentPipeline(client, ScrapeList(name))
 	doc, err := pipeline.Execute()
 	if err != nil {
 		return []model.GoPackage{}, err
 	}
 
-	return docWorker(doc, count, name), nil
+	return docWorker(doc, size, name), nil
 }

pkg/search/npm/search.go

@@ -11,21 +11,21 @@ import (
 )
 
 // SearchByScrape searches for NPM packages by scraping www.npmjs.com.
-func SearchByScrape(name string, count int) ([]model.NPMPackage, error) {
+func SearchByScrape(name string, size int) ([]model.NPMPackage, error) {
 	client := &http.Client{Timeout: 5 * time.Second}
 	pipeline := util.NewDocumentPipeline(client, ScrapeList(name))
 	doc, err := pipeline.Execute()
 	if err != nil {
 		return []model.NPMPackage{}, err
 	}
 
-	return docWorker(doc, count), nil
+	return docWorker(doc, size), nil
 }
 
 // SearchByAPI searches for NPM packages by querying registry.npmjs.com.
-func SearchByAPI(name string, count int) ([]model.NPMPackage, error) {
+func SearchByAPI(name string, size int) ([]model.NPMPackage, error) {
 	client := &http.Client{Timeout: 5 * time.Second}
-	bl, err := util.RESTAPIQuery(client, APIList(name, count))
+	bl, err := util.RESTAPIQuery(client, APIList(name, size))
 	if err != nil {
 		return []model.NPMPackage{}, err
 	}