package db

import (
	"errors"
	badger "github.com/dgraph-io/badger/v3"
	"log"
	"os"
)

var (
	db *badger.DB
)

// Init initializes the database.
func Init() error {
	dataDir := os.Getenv("MIGO_DATA_DIR")

	if dataDir == "" {
		return errors.New("MIGO_DATA_DIR not configured")
	}

	var err error
	db, err = badger.Open(badger.DefaultOptions(dataDir))

	if err != nil {
		return err
	}

	return nil
}

// Close closes the database connection.
func Close() {
	if err := db.Close(); err != nil {
		log.Printf("Error closing database connection: %v", err)
	}
}