SCR #0 - fix create default dirs

curkan · Jun 28, 2024 · b620214 · b620214
1 parent dcfb6e1
commit b620214
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/internal/crypto/vault.go b/internal/crypto/vault.go
@@ -9,6 +9,8 @@ import (
 	"os"
 	"path/filepath"
 	"time"
+
+	"github.com/sirupsen/logrus"
 )
 
 type vault struct {
@@ -83,6 +85,22 @@ func backupVault() bool {
 
 }
 
+func CreateDirs() {
+	homeDir := os.Getenv("HOME")
+	filePath := filepath.Join(homeDir, ".local", "share", "go2fa")
+	err := os.MkdirAll(filepath.Join(filePath, "stores"), os.ModePerm)
+
+	if err != nil {
+		logrus.Fatal(err)
+	}
+
+	err = os.MkdirAll(filepath.Join(filePath, "backups"), os.ModePerm)
+
+	if err != nil {
+		logrus.Fatal(err)
+	}
+}
+
 func GetEmptyVault() vault {
 	return vault{}
 }

diff --git a/internal/vault/create.go b/internal/vault/create.go
@@ -3,6 +3,7 @@ package vault
 import "go2fa/internal/crypto"
 
 func Create() bool {
+	crypto.CreateDirs()
 	vault := crypto.GetEmptyVault()
 	crypto.GeneratePublicPrivateKeys()
 	crypto.SetEmptyVault(vault)