✨ Added init command
This commit is contained in:
parent
552c98c684
commit
ec02a3c9d4
44
internal/shodan/init.go
Normal file
44
internal/shodan/init.go
Normal file
@ -0,0 +1,44 @@
|
||||
package shodan
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
"git.tcp.direct/perp/shogo/internal/utils"
|
||||
)
|
||||
|
||||
// Initalize Shodan
|
||||
func (s *Shodan) Init() {
|
||||
// Get home directory
|
||||
home, err := os.UserHomeDir()
|
||||
if err != nil {
|
||||
s.Results <- fmt.Sprintf("%s: %s", utils.Red("Error"), err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
// Check if directory exists
|
||||
_, err = os.Stat(home + "/.config/shodan/api_key")
|
||||
if err != nil {
|
||||
// Create directory
|
||||
err := os.MkdirAll(home+"/.config/shodan", 0770)
|
||||
if err != nil {
|
||||
s.Results <- fmt.Sprintf("%s: %s", utils.Red("Error"), err.Error())
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// Write key to file
|
||||
err = os.WriteFile(home+"/.config/shodan/api_key", []byte(s.Flags.Args[0]), 0644)
|
||||
if err != nil {
|
||||
s.Results <- fmt.Sprintf("%s: %s", utils.Red("Error"), err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
// Store lines
|
||||
var lines string
|
||||
|
||||
// Add lines
|
||||
lines += utils.Title.Sprintf("Shogo initalized\n")
|
||||
lines += fmt.Sprintf("%s: %s", utils.Green("Key"), utils.Blue(s.Flags.Args[0]))
|
||||
s.Results <- lines
|
||||
}
|
Loading…
Reference in New Issue
Block a user