Add BoltDB integration for data storage and retrieval

Author: Z3R0zz

go.mod

@@ -13,6 +13,7 @@ require (
 	github.com/spf13/pflag v1.0.6 // indirect
 	github.com/valyala/bytebufferpool v1.0.0 // indirect
 	github.com/valyala/fasthttp v1.59.0 // indirect
+	go.etcd.io/bbolt v1.4.0 // indirect
 	golang.org/x/sys v0.30.0 // indirect
 	gopkg.in/natefinch/lumberjack.v2 v2.2.1 // indirect
 )

go.sum

@@ -26,6 +26,8 @@ github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyC
 github.com/valyala/fasthttp v1.59.0 h1:Qu0qYHfXvPk1mSLNqcFtEk6DpxgA26hy6bmydotDpRI=
 github.com/valyala/fasthttp v1.59.0/go.mod h1:GTxNb9Bc6r2a9D0TWNSPwDz78UxnTGBViY3xZNEqyYU=
 github.com/xyproto/randomstring v1.0.5/go.mod h1:rgmS5DeNXLivK7YprL0pY+lTuhNQW3iGxZ18UQApw/E=
+go.etcd.io/bbolt v1.4.0 h1:TU77id3TnN/zKr7CO/uk+fBCwF2jGcMuw2B/FMAzYIk=
+go.etcd.io/bbolt v1.4.0/go.mod h1:AsD+OCi/qPN1giOX1aiLAha3o1U8rAz65bvN4j0sRuk=
 golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8 h1:0A+M6Uqn+Eje4kHMK80dtF3JCXC4ykBgQG4Fe06QRhQ=
 golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.30.0 h1:QjkSwP/36a20jFYWkSue1YwXzLmsV5Gfq7Eiy72C1uc=

src/database/database.go

@@ -0,0 +1,68 @@
+package database
+
+import (
+	"encoding/json"
+	"errors"
+	"os"
+	"path/filepath"
+	"time"
+
+	bolt "go.etcd.io/bbolt"
+)
+
+var dbPath = filepath.Join(os.Getenv("HOME"), ".config/zeroctl/zero.db")
+var DB *bolt.DB
+
+func InitBoltDB() error {
+	if err := os.MkdirAll(filepath.Dir(dbPath), 0755); err != nil {
+		return err
+	}
+
+	var err error
+	DB, err = bolt.Open(dbPath, 0600, &bolt.Options{Timeout: 1 * time.Second})
+	if err != nil {
+		return err
+	}
+
+	return DB.Update(func(tx *bolt.Tx) error {
+		_, err := tx.CreateBucketIfNotExists([]byte("cache"))
+		return err
+	})
+}
+
+func CloseBoltDB() {
+	if DB != nil {
+		DB.Close()
+	}
+}
+
+func StoreJsonData(key string, value interface{}) error {
+	data, err := json.Marshal(value)
+	if err != nil {
+		return err
+	}
+
+	return DB.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("cache"))
+		if b == nil {
+			return errors.New("cache bucket does not exist")
+		}
+		return b.Put([]byte(key), data)
+	})
+}
+
+func GetJsonData(key string, dest interface{}) error {
+	return DB.View(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("cache"))
+		if b == nil {
+			return errors.New("cache bucket does not exist")
+		}
+
+		data := b.Get([]byte(key))
+		if data == nil {
+			return errors.New("key not found")
+		}
+
+		return json.Unmarshal(data, dest)
+	})
+}

src/main.go

@@ -5,6 +5,7 @@ import (
 	"os"
 	"zeroctl/src/config"
 	"zeroctl/src/daemon"
+	"zeroctl/src/database"
 	"zeroctl/src/types"
 
 	"github.com/spf13/cobra"
@@ -16,6 +17,12 @@ func main() {
 		os.Exit(1)
 	}
 
+	if err := database.InitBoltDB(); err != nil {
+		fmt.Println("Error:", err)
+		os.Exit(1)
+	}
+	defer database.CloseBoltDB()
+
 	rootCmd := &cobra.Command{
 		Use:   "zeroctl",
 		Short: "zeroctl is a custom CLI tool",