Skip to content

Commit

Permalink
fix(load env from file)
Browse files Browse the repository at this point in the history
  • Loading branch information
refaldyrk committed Aug 10, 2023
1 parent 0deb9c2 commit 4232e08
Show file tree
Hide file tree
Showing 3 changed files with 30 additions and 30 deletions.
4 changes: 2 additions & 2 deletions go.mod
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,6 @@ module github.com/refaldyrk/hydra-env
go 1.20

require (
github.com/google/uuid v1.3.0 // indirect
github.com/mervick/aes-everywhere/go/aes256 v0.0.0-20220903070135-f13ed3789ae1 // indirect
github.com/google/uuid v1.3.0
github.com/mervick/aes-everywhere/go/aes256 v0.0.0-20220903070135-f13ed3789ae1
)
5 changes: 3 additions & 2 deletions key/key.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,8 @@ package key

import (
"errors"
"fmt"

"log"

"github.com/google/uuid"
"github.com/refaldyrk/hydra-env/helper"
Expand All @@ -21,7 +22,6 @@ func DefaultKey() *Key {
func (k *Key) NewKey() {
k.key = uuid.NewString()
k.hc.Password = k.key
fmt.Println("[HYDRA] Key: ", k.hc.Password)
}

type KeyStruct struct {
Expand Down Expand Up @@ -73,6 +73,7 @@ func (k *Key) PrintKey() error {
var keyData KeyStruct
if keyVal, ok := fileData["key"].(string); ok {
keyData.Key = keyVal
log.Println("[HYDRA] New Key: ", keyVal)
} else {
return errors.New("key not found or invalid")
}
Expand Down
51 changes: 25 additions & 26 deletions main.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,8 @@ import (
"os"
"strings"

"log"

"github.com/refaldyrk/hydra-env/env"
"github.com/refaldyrk/hydra-env/key"
)
Expand All @@ -29,24 +31,22 @@ func main() {
envis.SetHelperConfig(*envFlag, key.GetHC())
err := envis.CreateKeyFile(map[string]interface{}{})
if err != nil {
panic(err)
log.Fatal("[HYDRA] Failed To Create Key File Env", err.Error())
}
} else {
fmt.Println("[INFO] Env No Set")
}

if *genKeyFlag {
key.NewKey()

err := key.CreateKeyFile()
if err != nil {
panic(err)
log.Fatal("[HYDRA] Error Create Key File: ", err.Error())
return
}

err = key.PrintKey()
if err != nil {
panic(err)
log.Fatal("[HYDRA] Error Print Key", err)
return
}

Expand All @@ -55,13 +55,13 @@ func main() {

if *addKeyFlag != "" {
if *envFlag == "" {
fmt.Println("Error: --env flag is required for -add-key")
log.Fatal("[HYDRA] Require flag --env")
return
}

parts := strings.Split(*addKeyFlag, "|")
if len(parts) != 2 {
fmt.Println("Invalid format for add-key")
log.Fatal("[HYDRA] Invalid Command For Add Key: ", addKeyFlag)
return
}

Expand All @@ -70,74 +70,73 @@ func main() {

err := envis.AddKeyToFile(keys, value)
if err != nil {
panic(err)
log.Fatal("[HYDRA] Error When Add Key: ", err)
return
}

fmt.Printf("Key '%s' added successfully.\n", keys)
log.Println("[HYDRA] Success Add Key: ", keys)
}

if *getKeyFlag != "" {
if *envFlag == "" {
fmt.Println("Error: --env flag is required for -get-key")
log.Fatal("[HYDRA] Require flag --env")
return
}

value, err := envis.GetKey(*getKeyFlag)
if err != nil {
panic(err)
log.Fatal("[HYDRA] ERROR: ", err)
}

fmt.Printf("Value of key '%s': %s\n", *getKeyFlag, value)
log.Printf("[HYDRA] Value of key '%s': %s\n", *getKeyFlag, value)
}

if *listKeysFlag {
if *envFlag == "" {
fmt.Println("Error: --env flag is required for -list-keys")
log.Fatal("[HYDRA] Require flag --env")
return
}

data, err := envis.GetExistingData()
if err != nil {
panic(err)
log.Fatal("[HYDRA] ERROR: ", err)
}

fmt.Println("List of keys:")
log.Println("[HYDRA] List of keys:")
for key := range data {
fmt.Println(key)
}
log.Println("[HYDRA] Key: ", len(data))
}

if *delKeyFlag != "" {
if *envFlag == "" {
fmt.Println("Error: --env flag is required for -add-key")
log.Fatal("[HYDRA] Require flag --env")
return
}

err := envis.DeleteKey(*delKeyFlag)
if err != nil {
panic(err)
log.Fatal("[HYDRA] ERROR: ", err)
}

fmt.Printf("Key '%s' deleted successfully.\n", *delKeyFlag)
log.Printf("Key '%s' deleted successfully.\n", *delKeyFlag)
}

if *loadEnvFlag != "" {
files, err := os.Open(*loadEnvFlag)
if err != nil {
panic(err)
log.Fatal("[HYDRA] ERROR: ", err)
}

defer files.Close()
data, err := ioutil.ReadAll(files)
if err != nil {
panic(err)
log.Fatal("[HYDRA] ERROR: ", err)
}
dataString := string(data)
arrNewLine := strings.Split(dataString, "\n")
for _, v := range arrNewLine {
if v == "" {
fmt.Println(v)
}
if v == "" || v == "\n" {
continue
}
Expand All @@ -148,13 +147,13 @@ func main() {
}
err := envis.AddKeyToFile(arrEnv[0], arrEnv[1])
if err != nil {
fmt.Println(err)
log.Fatal("[HYDRA] ERROR: ", err)
continue
}
fmt.Println("Add Key: ", arrEnv[0])
log.Println("[HYDRA] Add Key: ", arrEnv[0])
}
}
fmt.Println("[Hydra] Everything Is Load...")
log.Println("[HYDRA] Success Load Env ", loadEnvFlag)
}

}

0 comments on commit 4232e08

Please sign in to comment.