package main

import (
	"github.com/google/uuid"
	"go.mills.io/bitcask/v2"
	"log"
)

type Database struct {
	connection *bitcask.Bitcask
	Handle     *bitcask.Collection
}

func NewDatabase(path string) *Database {
	db, err := bitcask.Open(path)
	if err != nil {
		log.Fatal(err)
	}
	return &Database{
		connection: db,
		Handle:     db.Collection("noodles"),
	}
}

func (db *Database) MakeID() string {
	id, err := uuid.NewUUID()
	if err != nil {
		log.Print(err)
		return ""
	}
	return id.String()
}

func (db *Database) GetAll() []Noodle {
	var data []Noodle
	err := db.Handle.List(&data)
	if err != nil {
		log.Print(err)
		return nil
	}
	return data
}

func (db *Database) GetAllGeneric() []interface{} {
	var data []interface{}
	err := db.Handle.List(&data)
	if err != nil {
		log.Print(err)
		return nil
	}
	return data
}

func (db *Database) Get(id string) Noodle {
	var noodle Noodle
	log.Printf("Looking up noodle key='%s'", id)
	log.Printf("key='%s' exists=%b", id, db.Handle.Has(id))
	err := db.Handle.Get(id, &noodle)
	if err != nil {
		log.Print(err)
		return Noodle{}
	}
	return noodle
}

func (db *Database) Add(noodle Noodle) error {
	err := db.Handle.Add(noodle.Id, noodle)
	if err != nil {
		log.Print(err)
		return err
	}
	return nil
}

func (db *Database) Delete(id string) error {
	err := db.Handle.Delete(id)
	if err != nil {
		log.Print(err)
		return err
	}
	return nil
}

func (db *Database) Close() error {
	return db.connection.Close()
}