package server

import (
	"io/ioutil"
	"os"

	"github.com/otoolep/rqlite/log"
)

// DbStateMachine contains the DB path.
type DbStateMachine struct {
	dbpath string
}

// NewDbStateMachine returns a StateMachine for capturing and restoring
// the state of an sqlite database.
func NewDbStateMachine(path string) *DbStateMachine {
	d := &DbStateMachine{
		dbpath: path,
	}
	log.Tracef("New DB state machine created with path: %s", path)
	return d
}

// Save captures the state of the database. The caller must ensure that
// no transaction is taking place during this call.
//
// http://sqlite.org/howtocorrupt.html states it is safe to do this
// as long as no transaction is in progress.
func (d *DbStateMachine) Save() ([]byte, error) {
	log.Tracef("Capturing database state from path: %s", d.dbpath)
	b, err := ioutil.ReadFile(d.dbpath)
	if err != nil {
		log.Errorf("Failed to save state: %s", err.Error())
		return nil, err
	}
	log.Tracef("Database state successfully saved to %s", d.dbpath)
	return b, nil
}

// Recovery restores the state of the database using the given data.
func (d *DbStateMachine) Recovery(b []byte) error {
	log.Tracef("Restoring database state to path: %s", d.dbpath)
	err := ioutil.WriteFile(d.dbpath, b, os.ModePerm)
	if err != nil {
		log.Errorf("Failed to recover state: %s", err.Error())
		return err
	}
	log.Tracef("Database restored successfully to %s", d.dbpath)
	return nil
}