Initial Commit

internal/storage/storage.go

@@ -0,0 +1,121 @@
+package storage
+
+import (
+	"fmt"
+	"path/filepath"
+	"os"
+
+	"git.skdevstudios.com/SK-Development-Studios/go-cal-tui/calendar"
+	"git.skdevstudios.com/SK-Development-Studios/go-cal-tui/internal/logger"
+
+	"github.com/dgraph-io/badger/v4"
+	"google.golang.org/protobuf/proto"
+)
+
+type EventInput struct {
+	Title       string
+	Description string
+	Year        int32
+	Month       int32
+	Day         int32
+	StartHour   int32
+	EndHour     int32
+	Color       string
+}
+
+var db *badger.DB
+
+func InitDB(path string) error {
+	if err := os.MkdirAll(path, 0755); err != nil {
+		logStr := fmt.Errorf("Error creating data directory: %w", err)	
+		logger.Log.Error(logStr)
+	}
+	var err error
+	db, err = badger.Open(badger.DefaultOptions(filepath.Join(path,"events_db")).WithLogger(nil))
+	if err != nil {
+		logStr := fmt.Errorf("Failed to open database: %w", err)
+		logger.Log.Error(logStr)
+	}
+	return nil
+}
+
+func CloseDB() {
+	if db != nil {
+		db.Close()
+	}
+}
+
+func SaveEvent(input EventInput) error {
+	if db == nil {
+		logger.Log.Error("Badger DB is not initialized")
+	}
+
+	event := &calendar.Event{
+		Title:       input.Title,
+		Description: input.Description,
+		Year:        input.Year,
+		Month:       input.Month,
+		Day:         input.Day,
+		StartHour:   input.StartHour,
+		EndHour:     input.EndHour,
+		Color:       input.Color,
+	}
+
+	data,err := proto.Marshal(event)
+	if err != nil {
+		return err
+	}
+	var key []byte
+	key = fmt.Appendf(key, "events/%04d-%02d-%02d/%02d", event.Year, event.Month, event.Day, event.StartHour)
+
+	return db.Update(func(txn *badger.Txn) error {
+		return txn.Set(key, data)
+	})
+}
+
+func GetEvent(year,month,day,hour int32) (*calendar.Event,error) {
+	var evt calendar.Event
+	var key []byte
+	key = fmt.Appendf(key, "events/%04d-%02d-%02d/%02d", year, month, day, hour)
+
+	err := db.View(func(txn *badger.Txn) error {
+		item,err := txn.Get(key)
+		if err != nil {
+			return err
+		}
+		return item.Value(func(val []byte) error {
+			return proto.Unmarshal(val, &evt)
+		})
+	})
+	return &evt,err
+}
+
+func GetEventsForDay(year,month,day int32) ([]*calendar.Event,error) {
+	var prefix []byte
+
+	prefix = fmt.Appendf(prefix, "events/%04d-%02d-%02d/", year,month,day)
+	
+	var events []*calendar.Event
+
+	err := db.View(func(txn *badger.Txn) error {
+		it := txn.NewIterator(badger.DefaultIteratorOptions)
+		defer it.Close()
+
+		for it.Seek(prefix); it.ValidForPrefix(prefix); it.Next(){
+			item := it.Item()
+			err := item.Value(func(val []byte) error {
+				var e calendar.Event
+				if err := proto.Unmarshal(val, &e); err != nil{
+					return err
+				}
+				events = append(events,&e)
+				return nil
+			})
+			if err != nil {
+				return err
+			}
+		}
+		return nil
+	})
+	return events,err
+}