Refactor: CLI commands

old usage: geoip -l
new usage: geoip list

old usage: geoip -c config.json
new usage: geoip convert -c config.json

1 files changed, 9 insertions, 24 deletions

diff --git a/main.go b/main.go
index 8dd8f45b..efc6a46a 100644
--- a/main.go
+++ b/main.go
@@ -1,36 +1,21 @@
 package main
 
 import (
-	"flag"
 	"log"
 
-	"github.com/Loyalsoldier/geoip/lib"
+	"github.com/spf13/cobra"
 )
 
-var (
-	list       = flag.Bool("l", false, "List all available input and output formats")
-	configFile = flag.String("c", "config.json", "URI of the JSON format config file, support both local file path and remote HTTP(S) URL")
-)
+var rootCmd = &cobra.Command{
+	Use:   "geoip",
+	Short: "geoip is a convenient tool to merge, convert and lookup IP & CIDR from various formats of geoip data.",
+	CompletionOptions: cobra.CompletionOptions{
+		HiddenDefaultCmd: true,
+	},
+}
 
 func main() {
-	flag.Parse()
-
-	if *list {
-		lib.ListInputConverter()
-		lib.ListOutputConverter()
-		return
-	}
-
-	instance, err := lib.NewInstance()
-	if err != nil {
-		log.Fatal(err)
-	}
-
-	if err := instance.Init(*configFile); err != nil {
-		log.Fatal(err)
-	}
-
-	if err := instance.Run(); err != nil {
+	if err := rootCmd.Execute(); err != nil {
 		log.Fatal(err)
 	}
 }