hello world on encrypted badger.

pkg/storage/cmd.go

@@ -0,0 +1,146 @@
+package storage
+
+import (
+	"github.com/spf13/viper"
+	"os"
+	"strings"
+)
+
+var (
+	isNewKey bool
+	key      Key
+)
+
+func configure() {
+
+	log.I.Ln("initializing storage")
+
+	configureKey()
+	configureDirPath()
+	configureFile()
+}
+
+func configureKey() {
+
+	log.I.Ln("looking for encryption key")
+
+	var err error
+
+	if viper.GetString(storeKeyFlag) != "" {
+
+		log.I.Ln("found key")
+
+		key.Decode(viper.GetString(storeKeyFlag))
+
+		return
+	}
+
+	log.I.Ln("no key found, checking for keyfile")
+
+	if viper.GetString(storeFilePathFlag) != "" {
+
+		var keyBytes []byte
+
+		if keyBytes, err = os.ReadFile(viper.GetString(storeFilePathFlag)); err != nil {
+			startupErrors <- err
+
+			return
+		}
+
+		key.Decode(string(keyBytes))
+
+		return
+	}
+
+	log.I.Ln("no keyfile found, generating a new key")
+
+	isNewKey = true
+
+	if key, err = KeyGen(); err != nil {
+		startupErrors <- err
+		return
+	}
+
+	log.W.Ln("")
+	log.W.Ln("--------------------------------------------------------")
+	log.W.Ln("--")
+	log.W.Ln("-- WARNING: The following key will be used to store")
+	log.W.Ln("-- your database securely, please ensure that you make")
+	log.W.Ln("-- a copy and store it in a secure place before using")
+	log.W.Ln("-- this software in a production environment.")
+	log.W.Ln("--")
+	log.W.Ln("--")
+	log.W.Ln("-- Failure to store this key properly will result in")
+	log.W.Ln("-- no longer being able to decrypt this database.")
+	log.W.Ln("--")
+	log.W.Ln("--")
+	log.W.Ln("-- It is recommended to use the following to generate")
+	log.W.Ln("-- your key:")
+	log.W.Ln("--")
+	log.W.Ln("-- indra seed keygen")
+	log.W.Ln("--")
+	log.W.Ln("--  OR")
+	log.W.Ln("--")
+	log.W.Ln("-- indra seed keygen --keyfile=/path/to/keyfile")
+	log.W.Ln("--")
+	log.W.Ln("--")
+	log.W.Ln("-- YOU HAVE BEEN WARNED!")
+	log.W.Ln("--")
+	log.W.Ln("-------------------------------------------------------")
+	log.W.Ln("-- KEY:", key.Encode(), "--")
+	log.W.Ln("-------------------------------------------------------")
+	log.W.Ln("")
+
+	viper.Set(storeKeyFlag, key.Encode())
+}
+
+func configureDirPath() {
+
+	var err error
+
+	if viper.GetString(storeFilePathFlag) == "" {
+		viper.Set(storeFilePathFlag, viper.GetString("data-dir")+"/"+fileName)
+	}
+
+	err = os.MkdirAll(
+		strings.TrimSuffix(viper.GetString(storeFilePathFlag), "/"+fileName),
+		0755,
+	)
+
+	if err != nil {
+		startupErrors <- err
+		return
+	}
+
+}
+
+func configureFile() {
+
+	log.I.Ln("using storage db path:")
+	log.I.Ln("-", viper.GetString(storeFilePathFlag))
+
+	var err error
+
+	if _, err = os.Stat(viper.GetString(storeFilePathFlag)); err != nil {
+
+		log.I.Ln("none found, creating a new one")
+
+		//file, err := os.OpenFile(viper.GetString(storeFilePathFlag), os.O_WRONLY, 0666)
+		//
+		//if err != nil && os.IsPermission(err) {
+		//	startupErrors <- err
+		//	return
+		//}
+		//
+		//file.Close()
+		//
+		//os.Remove(viper.GetString(storeFilePathFlag))
+
+		return
+	}
+
+	if isNewKey {
+		log.F.Ln("new key generated for an existing database. Check your configuration. Exiting!")
+		os.Exit(1)
+	}
+}