Support search for host name and IP (#7)

README.md

@@ -4,4 +4,18 @@
 
 ```bash
 go install github.com/no-src/hosts/...@latest
+```
+
+## Usage
+
+Print all host items.
+
+```bash
+$ hosts
+```
+
+Search for the specified hostname or IP, and multiple keywords are split with spaces.
+
+```bash
+$ hosts 192.168.100.1 github.com
 ```

cmd/hosts/main.go

@@ -0,0 +1,15 @@
+package main
+
+import (
+	"os"
+
+	"github.com/no-src/hosts"
+)
+
+func main() {
+	var search []string
+	if len(os.Args) > 1 {
+		search = os.Args[1:]
+	}
+	hosts.PrintHosts(search...)
+}

hosts.go

@@ -15,7 +15,7 @@ func isWindows() bool {
 }
 
 // PrintHosts print the hosts info
-func PrintHosts() {
+func PrintHosts(search ...string) {
 	hostsFile := "/etc/hosts"
 	if isWindows() {
 		sysRoot := os.Getenv("SYSTEMROOT")
@@ -67,8 +67,27 @@ func PrintHosts() {
 	log.Log("-------------------------------------------------")
 	hosts = recombine(hosts)
 	for _, item := range hosts {
-		log.Log("%s %s", item.IP, strings.Join(item.HostNameList, " "))
+		var source []string
+		source = append(source, item.IP)
+		source = append(source, item.HostNameList...)
+		if find(source, search...) {
+			log.Log("%s %s", item.IP, strings.Join(item.HostNameList, " "))
+		}
+	}
+}
+
+func find(source []string, search ...string) bool {
+	if len(search) == 0 {
+		return true
+	}
+	for _, searchItem := range search {
+		for _, name := range source {
+			if strings.Contains(strings.ToLower(name), strings.ToLower(searchItem)) {
+				return true
+			}
+		}
 	}
+	return false
 }
 
 // recombine try to recombine hosts,group by ip and distinct,order by ip