// Copyright (C) 2021 Marius Schellenberger

package db

import (
	"io"
	"path/filepath"

	"git.giftfish.de/ston1th/docstore/pkg/core"
	"git.giftfish.de/ston1th/docstore/pkg/index"
	"git.giftfish.de/ston1th/docstore/pkg/store"
	"github.com/go-logr/logr"
)

const (
	storeFile  = "store.db"
	bcryptCost = 13
)

const blevePath = "bleve"

type DB struct {
	log        logr.Logger
	rm         countMutex
	IndexMutex countMutex
	store      store.Store
	Index      *index.Index
}

func New(log logr.Logger, cfg *core.Config) (db *DB, err error) {
	db, err = NewPlain(log, cfg)
	if err != nil {
		return
	}
	db.Index, err = index.NewIndex(filepath.Join(cfg.RunDir, blevePath))
	if err != nil {
		return
	}
	err = db.UnlockUser()
	if err != nil && err != store.ErrKeyNotFound {
		return
	}
	err = db.store.Get(versionKey, nil)
	if err == store.ErrKeyNotFound {
		err = db.addDefaultTags()
		if err != nil {
			return
		}
	}
	err = db.runMigrations()
	return
}

func NewPlain(log logr.Logger, cfg *core.Config) (db *DB, err error) {
	db = &DB{log: log}
	dbFile := filepath.Join(cfg.RunDir, storeFile)
	db.store, err = store.NewBoltStore(dbFile, nil)
	return
}

func (db *DB) Dump(w io.Writer) error {
	return db.store.Dump(w)
}

func (db *DB) Restore(r io.Reader) error {
	return db.store.Restore(r)
}

func (db *DB) Close() (err error) {
	err = db.store.Close()
	if err != nil {
		return
	}
	if db.Index != nil {
		return db.Index.Close()
	}
	return nil
}