Bulk ASN

ipinfo/cmd_asn.go

@@ -1,113 +1,65 @@
 package main
 
 import (
-	"errors"
 	"fmt"
-	"net/http"
-
-	"github.com/fatih/color"
 	"github.com/ipinfo/cli/lib/complete"
 	"github.com/ipinfo/cli/lib/complete/predict"
-	"github.com/ipinfo/go/v2/ipinfo"
-	"github.com/spf13/pflag"
+	"os"
 )
 
 var completionsASN = &complete.Command{
+	Sub: map[string]*complete.Command{
+		"bulk": completionsASNBulk,
+	},
 	Flags: map[string]complete.Predictor{
-		"-t":        predict.Nothing,
-		"--token":   predict.Nothing,
-		"--nocache": predict.Nothing,
-		"-h":        predict.Nothing,
-		"--help":    predict.Nothing,
-		"-f":        predict.Set(asnFields),
-		"--field":   predict.Set(asnFields),
-		"--nocolor": predict.Nothing,
-		"-p":        predict.Nothing,
-		"--pretty":  predict.Nothing,
-		"-j":        predict.Nothing,
-		"--json":    predict.Nothing,
-		"-c":        predict.Nothing,
-		"--csv":     predict.Nothing,
+		"-h":     predict.Nothing,
+		"--help": predict.Nothing,
 	},
 }
 
-func printHelpASN(asn string) {
+func printHelpASN() {
 	fmt.Printf(
-		`Usage: %s %s [<opts>]
+		`Usage: %s asn <cmd> [<opts>] 
+
+Commands:
+  bulk        lookup ASNs in bulk
 
 Options:
   General:
-    --token <tok>, -t <tok>
-      use <tok> as API token.
-    --nocache
-      do not use the cache.
     --help, -h
       show help.
-
-  Outputs:
-    --field <field>, -f <field>
-      lookup only specific fields in the output.
-      field names correspond to JSON keys, e.g. 'registry' or 'allocated'.
-      multiple field names must be separated by commas.
-    --nocolor
-      disable colored output.
-
-  Formats:
-    --json, -j
-      output JSON format. (default)
-    --yaml, -y
-      output YAML format.
-`, progBase, asn)
+`, progBase)
 }
 
-func cmdASN(asn string) error {
-	var fTok string
-	var fField []string
-	var fJSON bool
-	var fYAML bool
-
-	pflag.StringVarP(&fTok, "token", "t", "", "the token to use.")
-	pflag.BoolVar(&fNoCache, "nocache", false, "disable the cache.")
-	pflag.BoolVarP(&fHelp, "help", "h", false, "show help.")
-	pflag.StringSliceVarP(&fField, "field", "f", nil, "specific field to lookup.")
-	pflag.BoolVarP(&fJSON, "json", "j", true, "output JSON format. (default)")
-	pflag.BoolVarP(&fYAML, "yaml", "y", false, "output YAML format.")
-	pflag.BoolVar(&fNoColor, "nocolor", false, "disable color output.")
-	pflag.Parse()
-
-	if fNoColor {
-		color.NoColor = true
-	}
-
-	if fHelp {
-		printHelpASN(asn)
+func cmdASNDefault() error {
+	// check whether the standard input (stdin) is being piped
+	// or redirected from another source or whether it's being read from the terminal (interactive mode).
+	stat, _ := os.Stdin.Stat()
+	if (stat.Mode() & os.ModeCharDevice) != 0 {
+		printHelpASN()
 		return nil
 	}
+	return cmdASNBulk(true)
+}
 
-	ii = prepareIpinfoClient(fTok)
-
-	// require token for ASN API.
-	if ii.Token == "" {
-		return errors.New("ASN lookups require a token; login via `ipinfo init`.")
+// cmdASN is the handler for the "asn" command.
+func cmdASN() error {
+	var err error
+	cmd := ""
+	if len(os.Args) > 2 {
+		cmd = os.Args[2]
 	}
 
-	data, err := ii.GetASNDetails(asn)
-	if err != nil {
-		iiErr, ok := err.(*ipinfo.ErrorResponse)
-		if ok && (iiErr.Response.StatusCode == http.StatusUnauthorized) {
-			return errors.New("Token does not have access to ASN API")
-		}
-		return err
+	switch {
+	case cmd == "bulk":
+		err = cmdASNBulk(false)
+	default:
+		err = cmdASNDefault()
 	}
 
-	if len(fField) > 0 {
-		d := make(ipinfo.BatchASNDetails, 1)
-		d[data.ASN] = data
-		return outputFieldBatchASNDetails(d, fField, false, false)
-	}
-	if fYAML {
-		return outputYAML(data)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "err: %v\n", err)
 	}
 
-	return outputJSON(data)
+	return nil
 }

ipinfo/cmd_asn_bulk.go

@@ -0,0 +1,95 @@
+package main
+
+import (
+	"fmt"
+	"github.com/ipinfo/cli/lib"
+	"github.com/ipinfo/cli/lib/complete"
+	"github.com/ipinfo/cli/lib/complete/predict"
+	"github.com/spf13/pflag"
+)
+
+var completionsASNBulk = &complete.Command{
+	Flags: map[string]complete.Predictor{
+		"-t":        predict.Nothing,
+		"--token":   predict.Nothing,
+		"--nocache": predict.Nothing,
+		"-h":        predict.Nothing,
+		"--help":    predict.Nothing,
+		"-f":        predict.Set(asnFields),
+		"--field":   predict.Set(asnFields),
+		"-j":        predict.Nothing,
+		"--json":    predict.Nothing,
+	},
+}
+
+// printHelpASNBulk prints the help message for the asn bulk command.
+func printHelpASNBulk() {
+	fmt.Printf(
+		`Usage: %s asn bulk [<opts>] <ASNs | filepath>
+
+Description:
+  Accepts ASNs and file paths.
+
+Examples:
+  # Lookup all ASNs in multiple files.
+  $ %[1]s asn bulk /path/to/asnlist1.txt /path/to/asnlist2.txt
+
+  # Lookup multiple ASNs.
+  $ %[1]s asn bulk AS123 AS456 AS789
+
+  # Lookup ASNs from multiple sources simultaneously.
+  $ %[1]s asn bulk AS123 AS456 AS789 asns.txt
+
+Options:
+  General:
+    --token <tok>, -t <tok>
+      use <tok> as API token.
+    --nocache
+      do not use the cache.
+    --help, -h
+      show help.
+
+  Outputs:
+    --field <field>, -f <field>
+      lookup only specific fields in the output.
+      field names correspond to JSON keys, e.g. 'name' or 'registry'.
+      multiple field names must be separated by commas.
+
+  Formats:
+    --json, -j
+      output JSON format. (default)
+    --yaml, -y
+      output YAML format.
+`, progBase)
+}
+
+// cmdASNBulk is the asn bulk command.
+func cmdASNBulk(piped bool) error {
+	f := lib.CmdASNBulkFlags{}
+	f.Init()
+	pflag.Parse()
+
+	ii = prepareIpinfoClient(f.Token)
+	var args []string
+	if !piped {
+		args = pflag.Args()[2:]
+	}
+
+	data, err := lib.CmdASNBulk(f, ii, args, printHelpASNBulk)
+	if err != nil {
+		return err
+	}
+	if (data) == nil {
+		return nil
+	}
+
+	if len(f.Field) > 0 {
+		return outputFieldBatchASNDetails(data, f.Field, false, false)
+	}
+
+	if f.Yaml {
+		return outputYAML(data)
+	}
+
+	return outputJSON(data)
+}

ipinfo/cmd_asn_single.go

@@ -0,0 +1,111 @@
+package main
+
+import (
+	"errors"
+	"fmt"
+	"net/http"
+
+	"github.com/fatih/color"
+	"github.com/ipinfo/cli/lib/complete"
+	"github.com/ipinfo/cli/lib/complete/predict"
+	"github.com/ipinfo/go/v2/ipinfo"
+	"github.com/spf13/pflag"
+)
+
+var completionsASNSingle = &complete.Command{
+	Flags: map[string]complete.Predictor{
+		"-t":        predict.Nothing,
+		"--token":   predict.Nothing,
+		"--nocache": predict.Nothing,
+		"-h":        predict.Nothing,
+		"--help":    predict.Nothing,
+		"-f":        predict.Set(asnFields),
+		"--field":   predict.Set(asnFields),
+		"--nocolor": predict.Nothing,
+		"-p":        predict.Nothing,
+		"--pretty":  predict.Nothing,
+		"-j":        predict.Nothing,
+		"--json":    predict.Nothing,
+	},
+}
+
+func printHelpASNSingle(asn string) {
+	fmt.Printf(
+		`Usage: %s %s [<opts>]
+
+Options:
+  General:
+    --token <tok>, -t <tok>
+      use <tok> as API token.
+    --nocache
+      do not use the cache.
+    --help, -h
+      show help.
+
+  Outputs:
+    --field <field>, -f <field>
+      lookup only specific fields in the output.
+      field names correspond to JSON keys, e.g. 'registry' or 'allocated'.
+      multiple field names must be separated by commas.
+    --nocolor
+      disable colored output.
+
+  Formats:
+    --json, -j
+      output JSON format. (default)
+    --yaml, -y
+      output YAML format.
+`, progBase, asn)
+}
+
+func cmdASNSingle(asn string) error {
+	var fTok string
+	var fField []string
+	var fJSON bool
+	var fYAML bool
+
+	pflag.StringVarP(&fTok, "token", "t", "", "the token to use.")
+	pflag.BoolVar(&fNoCache, "nocache", false, "disable the cache.")
+	pflag.BoolVarP(&fHelp, "help", "h", false, "show help.")
+	pflag.StringSliceVarP(&fField, "field", "f", nil, "specific field to lookup.")
+	pflag.BoolVarP(&fJSON, "json", "j", true, "output JSON format. (default)")
+	pflag.BoolVarP(&fYAML, "yaml", "y", false, "output YAML format.")
+	pflag.BoolVar(&fNoColor, "nocolor", false, "disable color output.")
+	pflag.Parse()
+
+	if fNoColor {
+		color.NoColor = true
+	}
+
+	if fHelp {
+		printHelpASNSingle(asn)
+		return nil
+	}
+
+	ii = prepareIpinfoClient(fTok)
+
+	// require token for ASN API.
+	if ii.Token == "" {
+		return errors.New("ASN lookups require a token; login via `ipinfo init`.")
+	}
+
+	data, err := ii.GetASNDetails(asn)
+	if err != nil {
+		iiErr, ok := err.(*ipinfo.ErrorResponse)
+		if ok && (iiErr.Response.StatusCode == http.StatusUnauthorized) {
+			return errors.New("Token does not have access to ASN API")
+		}
+		return err
+	}
+
+	if len(fField) > 0 {
+		d := make(ipinfo.BatchASNDetails, 1)
+		d[data.ASN] = data
+		return outputFieldBatchASNDetails(d, fField, false, false)
+	}
+	if fYAML {
+		return outputYAML(data)
+	}
+
+	return outputJSON(data)
+}

ipinfo/cmd_default.go

@@ -21,6 +21,7 @@ Commands:
   <asn>       look up details for an ASN, e.g. AS123 or as123.
   myip        get details for your IP.
   bulk        get details for multiple IPs in bulk.
+  asn         tools related to ASNs.
   summarize   get summarized data for a group of IPs.
   map         open a URL to a map showing the locations of a group of IPs.
   prips       print IP list from CIDR or range.

ipinfo/completions.go

@@ -12,6 +12,7 @@ var completions = &complete.Command{
 	Sub: map[string]*complete.Command{
 		"myip":       completionsMyIP,
 		"bulk":       completionsBulk,
+		"asn":        completionsASN,
 		"summarize":  completionsSummarize,
 		"map":        completionsMap,
 		"prips":      completionsPrips,
@@ -49,7 +50,7 @@ func handleCompletions() {
 		if lib.StrIsIPStr(cmdSecondArg) {
 			completions.Sub[cmdSecondArg] = completionsIP
 		} else if lib.StrIsASNStr(cmdSecondArg) {
-			completions.Sub[cmdSecondArg] = completionsASN
+			completions.Sub[cmdSecondArg] = completionsASNSingle
 		}
 	}
 

ipinfo/main.go

@@ -38,13 +38,15 @@ func main() {
 		err = cmdIP(cmd)
 	case lib.StrIsASNStr(cmd):
 		asn := strings.ToUpper(cmd)
-		err = cmdASN(asn)
+		err = cmdASNSingle(asn)
 	case len(cmd) >= 3 && strings.IndexByte(cmd, '.') != -1:
 		err = cmdDomain(cmd)
 	case cmd == "myip":
 		err = cmdMyIP()
 	case cmd == "bulk":
 		err = cmdBulk()
+	case cmd == "asn":
+		err = cmdASN()
 	case cmd == "summarize" || cmd == "sum":
 		err = cmdSum()
 	case cmd == "map":