Lower & upper IP tool

ipinfo/cmd_tool.go

@@ -12,6 +12,8 @@ import (
 var completionsTool = &complete.Command{
 	Sub: map[string]*complete.Command{
 		"aggregate": completionsToolAggregate,
+		"lower":     completionsToolLower,
+		"upper":     completionsToolUpper,
 	},
 	Flags: map[string]complete.Predictor{
 		"-h":     predict.Nothing,
@@ -27,6 +29,8 @@ func printHelpTool() {
 Commands:
   aggregate    aggregate IPs, IP ranges, and CIDRs.
 
+  lower        get start IP of IP, IP ranges, and CIDRs.
+  upper        get end IP of IP, IP ranges, and CIDRs.
 Options:
   --help, -h
     show help.
@@ -56,6 +60,10 @@ func cmdTool() error {
 	switch {
 	case cmd == "aggregate":
 		err = cmdToolAggregate()
+	case cmd == "lower":
+		err = cmdToolLower()
+	case cmd == "upper":
+		err = cmdToolUpper()
 	default:
 		err = toolHelp()
 	}

ipinfo/cmd_tool_lower.go

@@ -0,0 +1,56 @@
+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 completionsToolLower = &complete.Command{
+	Flags: map[string]complete.Predictor{
+		"-h":      predict.Nothing,
+		"--help":  predict.Nothing,
+		"-q":      predict.Nothing,
+		"--quiet": predict.Nothing,
+	},
+}
+
+func printHelpToolLower() {
+	fmt.Printf(
+		`Usage: %s tool lower [<opts>] <cidr>
+
+Description:
+  Calculates the lower IP address (start address of a network) for the given inputs.
+  Input can be IP, IP range or CIDR.
+
+  If input contains CIDRs, it calculates the lower IP address for each CIDR.
+  If input contains IP ranges, it calculates the lower IP address for each Ip range.
+
+Examples:
+  # Calculate lower IP for IP, IP range and CIDR.
+  $ %[1]s tool lower 192.168.1.0/24
+
+  # Calculate lower IPs for IPs, IP ranges and CIDRs.
+  $ %[1]s tool lower 192.168.1.0/24 10.0.0.0/16
+
+  # Calculate lower IPs from stdin.
+  $ cat /path/to/file.txt | %[1]s tool lower
+
+Options:
+  --help, -h
+    show help.
+  --quiet, -q
+    quiet mode; suppress additional output.
+`, progBase)
+}
+
+func cmdToolLower() (err error) {
+	f := lib.CmdToolLowerFlags{}
+	f.Init()
+	pflag.Parse()
+
+	return lib.CmdToolLower(f, pflag.Args()[2:], printHelpToolLower)
+}

ipinfo/cmd_tool_upper.go

@@ -0,0 +1,55 @@
+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 completionsToolUpper = &complete.Command{
+	Flags: map[string]complete.Predictor{
+		"-h":      predict.Nothing,
+		"--help":  predict.Nothing,
+		"-q":      predict.Nothing,
+		"--quiet": predict.Nothing,
+	},
+}
+
+func printHelpToolUpper() {
+	fmt.Printf(
+		`Usage: %s tool upper [<opts>] <cidr>
+
+Description:
+  Calculates the upper IP address (end address of a network) for the given inputs.
+  Input can be Ip, IP range or CIDR.
+
+  If input contains CIDRs, it calculates the upper IP address for each CIDR.
+  If input contains Ip ranges, it calculates the upper IP address for each IP range.
+Examples:
+  # Calculate upper IP for IP, IP range and CIDR.
+  $ %[1]s tool upper 192.168.1.0/24
+
+  # Calculate upper IPs for IPs, IP ranges and CIDRs.
+  $ %[1]s tool upper 192.168.1.0/24 10.0.0.0/16
+
+  # Calculate upper IPs from stdin.
+  $ cat /path/to/file.txt | %[1]s tool upper
+
+Options:
+  --help, -h
+    Show help.
+  --quiet, -q
+    Quiet mode; suppress additional output.
+`, progBase)
+}
+
+func cmdToolUpper() (err error) {
+	f := lib.CmdToolUpperFlags{}
+	f.Init()
+	pflag.Parse()
+
+	return lib.CmdToolUpper(f, pflag.Args()[2:], printHelpToolUpper)
+}

lib/cmd_tool_lower.go

@@ -0,0 +1,79 @@
+package lib
+
+import (
+	"fmt"
+	"net"
+	"os"
+
+	"github.com/spf13/pflag"
+)
+
+type CmdToolLowerFlags struct {
+	Help  bool
+	Quiet bool
+}
+
+func (f *CmdToolLowerFlags) Init() {
+	pflag.BoolVarP(
+		&f.Help,
+		"help", "h", false,
+		"show help.",
+	)
+	pflag.BoolVarP(
+		&f.Quiet,
+		"quiet", "q", false,
+		"quiet mode; suppress additional output.",
+	)
+}
+
+func CmdToolLower(
+	f CmdToolLowerFlags,
+	args []string,
+	printHelp func(),
+) error {
+	if f.Help {
+		printHelp()
+		return nil
+	}
+
+	stat, _ := os.Stdin.Stat()
+	isStdin := (stat.Mode() & os.ModeCharDevice) == 0
+
+	if len(args) == 0 && !isStdin {
+		printHelp()
+		return nil
+	}
+
+	if isStdin {
+		return scanrdr(os.Stdin, processIPRangeOrCIDRLower)
+	}
+
+	for _, input := range args {
+		if err := processIPRangeOrCIDRLower(input); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func processIPRangeOrCIDRLower(input string) error {
+	if ipRange, err := IPRangeStrFromStr(input); err == nil {
+		// If it's an IP range, print the starting IP in the range.
+		fmt.Printf("%s\n", ipRange.Start)
+		return nil
+	}
+
+	if ip := net.ParseIP(input); ip != nil {
+		// If it's a simple IP address, print the IP itself.
+		fmt.Printf("%s\n", ip)
+		return nil
+	}
+
+	if _, ipnet, err := net.ParseCIDR(input); err == nil {
+		// If it's a CIDR, print the Starting IP address of the CIDR.
+		fmt.Printf("%s\n", ipnet.IP)
+		return nil
+	}
+
+	return fmt.Errorf("invalid input: %s", input)
+}

lib/cmd_tool_upper.go

@@ -0,0 +1,81 @@
+package lib
+
+import (
+	"fmt"
+	"net"
+	"os"
+
+	"github.com/spf13/pflag"
+)
+
+type CmdToolUpperFlags struct {
+	Help  bool
+	Quiet bool
+}
+
+func (f *CmdToolUpperFlags) Init() {
+	pflag.BoolVarP(
+		&f.Help,
+		"help", "h", false,
+		"show help.",
+	)
+	pflag.BoolVarP(
+		&f.Quiet,
+		"quiet", "q", false,
+		"quiet mode; suppress additional output.",
+	)
+}
+
+func CmdToolUpper(
+	f CmdToolUpperFlags,
+	args []string,
+	printHelp func(),
+) error {
+	if f.Help {
+		printHelp()
+		return nil
+	}
+
+	stat, _ := os.Stdin.Stat()
+	isStdin := (stat.Mode() & os.ModeCharDevice) == 0
+
+	if len(args) == 0 && !isStdin {
+		printHelp()
+		return nil
+	}
+
+	if isStdin {
+		return scanrdr(os.Stdin, processIPRangeOrCIDRUpper)
+	}
+
+	for _, input := range args {
+		if err := processIPRangeOrCIDRUpper(input); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func processIPRangeOrCIDRUpper(input string) error {
+	ipRange, err := IPRangeStrFromStr(input)
+	if err == nil {
+		fmt.Println(ipRange.End)
+		return nil
+	}
+
+	if ip := net.ParseIP(input); ip != nil {
+		fmt.Println(input)
+		return nil
+	}
+
+	if _, ipnet, err := net.ParseCIDR(input); err == nil {
+		ipRange, err := IPRangeStrFromCIDR(ipnet.String())
+		if err == nil {
+			fmt.Println(ipRange.End)
+			return nil
+		}
+	}
+
+	fmt.Printf("Error parsing input: %v\n", err)
+	return err
+}

lib/cmd_tool_utils.go

@@ -0,0 +1,36 @@
+package lib
+
+import (
+	"bufio"
+	"io"
+	"strings"
+)
+
+// Function type for the callback used in scanrdr
+type CIDRProcessor func(string) error
+
+// Function to read lines from the io.Reader and process each line as a CIDR using the provided callback function
+func scanrdr(r io.Reader, processCIDR CIDRProcessor) error {
+	buf := bufio.NewReader(r)
+	for {
+		d, err := buf.ReadString('\n')
+		if err == io.EOF {
+			if len(d) == 0 {
+				break
+			}
+		} else if err != nil {
+			return err
+		}
+
+		sepIdx := strings.IndexAny(d, "\n")
+		if sepIdx == -1 {
+			sepIdx = len(d)
+		}
+
+		cidrStr := d[:sepIdx]
+		if err := processCIDR(cidrStr); err != nil {
+			return err
+		}
+	}
+	return nil
+}