setted banner and flags

kopolindo · Mar 27, 2018 · c6d6f51 · c6d6f51
1 parent de9bd41
commit c6d6f51
Showing 1 changed file with 53 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -1,10 +1,61 @@
 package main
 
 import (
-	"github.com/kopolindo/cve-scraper/db"
+	"flag"
+	"fmt"
+	"os"
+
+	"github.com/alecthomas/template"
 	_ "github.com/mattn/go-sqlite3"
 )
 
+const banner = `
+                                                                            
+ ####  #    # ######        ####   ####  #####    ##   #####  ###### #####  
+#    # #    # #            #      #    # #    #  #  #  #    # #      #    # 
+#      #    # #####  #####  ####  #      #    # #    # #    # #####  #    # 
+#      #    # #                 # #      #####  ###### #####  #      #####  
+#    #  #  #  #            #    # #    # #   #  #    # #      #      #   #  
+ ####    ##   ######        ####   ####  #    # #    # #      ###### #    #   
+ 
+ |Version: {{.Ver}}
+ |Commit:  {{.Commit}}
+ |Build:   {{.Build}}
+
+`
+
+type Version struct {
+	Ver    string
+	Build  string
+	Commit string
+}
+
+var ActualVersion = Version{"0.1", "Developing", "de9bd41fa1450f6a5bbb667567040615c945c0c6"}
+
+//FLAGS
+var (
+	cve       = flag.String("cve", "", "CVE-ID to look for")
+	sw        = flag.String("sw", "", "Software name to look for")
+	swVersion = flag.String("swv", "", "Software name AND version software")
+	version   = flag.Bool("v", false, "Print version and exit")
+)
+
+func printbanner() {
+	tmpl := template.New("banner")
+	template.Must(tmpl.Parse(banner))
+	_ = tmpl.Execute(os.Stderr, ActualVersion)
+}
+
+func Init() {
+	printbanner()
+	if flag.NFlag() == 0 {
+		fmt.Println("\nUsage flags:")
+		flag.PrintDefaults()
+		return
+	}
+	flag.Parse()
+}
+
 func main() {
-	db.Populate()
+	Init()
 }