Merge pull request #4 from dogpatch626/feat/cobra

dogpatch626 · web-flow · commit 206d44c8b360 · 2024-11-23T13:19:55.000-05:00
Feat/cobra
diff --git a/.env b/.env
diff --git a/.gitignore b/.gitignore
@@ -1 +1,23 @@
 
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Dependency directories (remove the comment below to include it)
+# vendor/
+
+# Go workspace file
+go.work
+go.work.sum
+
+# env file
+.env
diff --git a/CodexCli b/CodexCli
diff --git a/banner/banner.go b/banner/banner.go
@@ -6,6 +6,6 @@ import (
 
 func Banner() {
 
-	banner := figure.NewColorFigure("Codex", "isometric1", "red", true)
+	banner := figure.NewColorFigure("Codex", "banner3-D", "white", true)
 	banner.Print()
 }
diff --git a/commands/Info.go b/commands/Info.go
@@ -0,0 +1,21 @@
+package Commands
+
+import (
+	"github.com/fatih/color"
+	"github.com/spf13/cobra"
+)
+
+var infoCmd = &cobra.Command{
+	Use:     "info",
+	Aliases: []string{"info"},
+	Short:   "CodexCli is a in development cli",
+	Run: func(cmd *cobra.Command, args []string) {
+		color.Red("CodexCli is a in-development cli made in go as a project to learn what is even going on with this lang.")
+	},
+}
+
+func init() {
+
+	rootCmd.AddCommand(infoCmd)
+
+}
diff --git a/commands/Root.go b/commands/Root.go
@@ -0,0 +1,36 @@
+package Commands
+
+import (
+	"CodexCli/banner"
+	"fmt"
+	"os"
+	"os/exec"
+
+	"github.com/spf13/cobra"
+)
+
+var rootCmd = &cobra.Command{
+	Use:   "Codex",
+	Short: "Codex is a virus total cli",
+	Long:  "Codex is a virus total cli built by dogpatch626 in Go",
+	Run: func(cmd *cobra.Command, args []string) {
+		banner.Banner()
+		ran := exec.Command("./CodexCli", "--help")
+
+		out, err := ran.Output()
+
+		if err != nil {
+			fmt.Println("Could not run: --help", err)
+		}
+
+		fmt.Println("\n", string(out))
+
+	},
+}
+
+func Execute() {
+	if err := rootCmd.Execute(); err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+}
diff --git a/commands/commands.go b/commands/commands.go
diff --git a/commands/scan.go b/commands/scan.go
@@ -0,0 +1,95 @@
+package Commands
+
+import (
+	"encoding/base64"
+	"encoding/json"
+	"fmt"
+	"log"
+	"net/url"
+	"os"
+
+	vt "github.com/VirusTotal/vt-go"
+	"github.com/spf13/cobra"
+)
+
+var vtkey string
+var scanCmd = &cobra.Command{
+	Use:   "scan",
+	Short: "Scan a url using the Virus total api",
+
+	Run: func(cmd *cobra.Command, args []string) {
+
+		if vtkey == "" {
+			fmt.Println("Must pass --vtkey argument.")
+			os.Exit(0)
+		}
+
+		var urlID = base64.RawURLEncoding.EncodeToString([]byte(args[0]))
+
+		client := vt.NewClient(vtkey)
+
+		analyseslink := "https://www.virustotal.com/api/v3/urls/" + urlID
+
+		analysisurl, parseurlerr := url.Parse(analyseslink)
+		fmt.Println(analyseslink)
+		if parseurlerr != nil {
+			log.Fatal(parseurlerr)
+		}
+
+		getId, idErr := client.Get(analysisurl)
+
+		if idErr != nil {
+			log.Fatal(idErr)
+		}
+		var dat map[string]interface{}
+
+		decodeerr := json.Unmarshal(getId.Data, &dat)
+		if decodeerr != nil {
+			log.Panic(decodeerr)
+		}
+		// Iterate through data
+		for key, value := range dat {
+			fmt.Printf("Key: %s\n", key)
+
+			switch v := value.(type) {
+			case string:
+				fmt.Printf("Value: %s\n", v)
+			case float64:
+				fmt.Printf("Value: %f\n", v)
+			case map[string]interface{}:
+				// Recursively handle nested maps
+				for nestedKey, nestedValue := range v {
+
+					switch nv := nestedValue.(type) {
+
+					case map[string]interface{}:
+
+						for nestedKey, nestedValue := range nv {
+							switch fn := nestedValue.(type) {
+
+							case map[string]interface{}:
+								for fkey, nvalue := range fn {
+									fmt.Printf("  ----: %s", fkey)
+									fmt.Printf("%v\n", nvalue)
+								}
+							}
+
+							fmt.Printf("  --------: %s", nestedKey)
+							fmt.Printf(": %v\n", nestedValue)
+
+						}
+						fmt.Printf("  --: %s", nestedKey)
+						fmt.Printf(": %v\n", nestedValue)
+					}
+				}
+			default:
+				fmt.Printf("Unknown type: %T\n", v)
+			}
+		}
+	},
+}
+
+func init() {
+	scanCmd.Flags().StringVarP(&vtkey, "vtkey", "v", "", "Virus Total key for api calls")
+	rootCmd.AddCommand(scanCmd)
+}
diff --git a/go.mod b/go.mod
@@ -3,10 +3,14 @@ module CodexCli
 go 1.23.1
 
 require (
+	github.com/VirusTotal/vt-go v1.0.1 // indirect
 	github.com/common-nighthawk/go-figure v0.0.0-20210622060536-734e95fb86be // indirect
 	github.com/fatih/color v1.17.0 // indirect
-	github.com/joho/godotenv v1.5.1 // indirect
+	github.com/inconshreveable/mousetrap v1.1.0 // indirect
 	github.com/mattn/go-colorable v0.1.13 // indirect
 	github.com/mattn/go-isatty v0.0.20 // indirect
+	github.com/spf13/cobra v1.8.1 // indirect
+	github.com/spf13/pflag v1.0.5 // indirect
+	github.com/thedevsaddam/gojsonq/v2 v2.5.2 // indirect
 	golang.org/x/sys v0.18.0 // indirect
 )
diff --git a/go.sum b/go.sum
@@ -1,15 +1,32 @@
+github.com/VirusTotal/vt-go v1.0.1 h1:rj/qugIY8XNC6ogwOaeJAGCOsb3nmY63+yuDMtHOx0Q=
+github.com/VirusTotal/vt-go v1.0.1/go.mod h1:u1+HeRyl/gQs67eDgVEWNE7+x+zCyXhdtNVrRJR5YPE=
 github.com/common-nighthawk/go-figure v0.0.0-20210622060536-734e95fb86be h1:J5BL2kskAlV9ckgEsNQXscjIaLiOYiZ75d4e94E6dcQ=
 github.com/common-nighthawk/go-figure v0.0.0-20210622060536-734e95fb86be/go.mod h1:mk5IQ+Y0ZeO87b858TlA645sVcEcbiX6YqP98kt+7+w=
+github.com/cpuguy83/go-md2man/v2 v2.0.4/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/fatih/color v1.17.0 h1:GlRw1BRJxkpqUCBKzKOw098ed57fEsKeNjpTe3cSjK4=
 github.com/fatih/color v1.17.0/go.mod h1:YZ7TlrGPkiz6ku9fK3TLD/pl3CpsiFyu8N92HLgmosI=
-github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
-github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
+github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
+github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
 github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
 github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
 github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
 github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
 github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
+github.com/spf13/cobra v1.8.1 h1:e5/vxKd/rZsfSJMUX1agtjeTDf+qv1/JdBF8gg5k9ZM=
+github.com/spf13/cobra v1.8.1/go.mod h1:wHxEcudfqmLYa8iTfL+OuZPbBZkmvliBWKIezN3kD9Y=
+github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
+github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/thedevsaddam/gojsonq/v2 v2.5.2 h1:CoMVaYyKFsVj6TjU6APqAhAvC07hTI6IQen8PHzHYY0=
+github.com/thedevsaddam/gojsonq/v2 v2.5.2/go.mod h1:bv6Xa7kWy82uT0LnXPE2SzGqTj33TAEeR560MdJkiXs=
 golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.18.0 h1:DBdB3niSjOA/O0blCZBqDefyWNYveAYMNF1Wum0DYQ4=
 golang.org/x/sys v0.18.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
diff --git a/main.go b/main.go
@@ -1,50 +1,9 @@
 package main
 
 import (
-	"CodexCli/banner"
-	TextProcessor "CodexCli/textprocessor"
-	"bufio"
-	"fmt"
-	"log"
-	"os"
-	"strings"
-
-	"github.com/joho/godotenv"
+	Commands "CodexCli/commands"
 )
 
 func main() {
-
-	scanner := bufio.NewScanner(os.Stdin)
-	banner.Banner()
-
-	err := godotenv.Load()
-
-	if err != nil {
-		f, CreationError := os.Create(".env")
-		if CreationError != nil {
-			log.Fatal("Error loading and writing .env file")
-		}
-		defer f.Close()
-	}
-
-	var myEnv map[string]string
-	myEnv, ReadError := godotenv.Read()
-
-	if ReadError != nil {
-		log.Fatal("Error reading .env")
-	}
-
-	if len(myEnv) == 0 {
-		fmt.Println("your .env is empty")
-	}
-
-	for scanner.Scan() {
-
-		in := scanner.Text()
-		input := strings.Split(in, " ")
-		// pass args to the processor
-		// arg format :<Command> ...Perams
-		TextProcessor.Processor(input)
-
-	}
+	Commands.Execute()
 }
diff --git a/textprocessor/TextProcessor.go b/textprocessor/TextProcessor.go

Original file line number	Diff line number	Diff line change
`@@ -6,6 +6,6 @@ import (`
`6`	`6`
`7`	`7`	`func Banner() {`
`8`	`8`
`9`		`- banner := figure.NewColorFigure("Codex", "isometric1", "red", true)`
	`9`	`+ banner := figure.NewColorFigure("Codex", "banner3-D", "white", true)`
`10`	`10`	`banner.Print()`
`11`	`11`	`}`