package storage

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

func hasDB(db *badger.DB, key []byte) bool {
	return db.View(func(txn *badger.Txn) error {
		_, err := txn.Get(key)
		return err
	}) == nil
}

func putDB(db *badger.DB, key, value []byte) (err error) {
	return db.Update(func(txn *badger.Txn) error {
		return txn.Set(key, value)
	})
}

func getDB(db *badger.DB, key []byte) (value []byte, err error) {
	err = db.View(func(txn *badger.Txn) error {
		item, err := txn.Get(key)
		if err != nil {
			return err
		}
		value, err = item.ValueCopy(value)
		if err != nil {
			return err
		}
		return err
	})
	return
}

func getTxn(txn *badger.Txn, key []byte) (value []byte, err error) {
	item, err := txn.Get(key)
	if err != nil {
		return
	}
	return item.ValueCopy(value)
}