package database

import (
	"path/filepath"

	"certmanager/pkg/logger"

	"github.com/jmoiron/sqlx"
	_ "github.com/mattn/go-sqlite3"
)

const schema = `
    DROP TABLE IF EXISTS issuer;
    CREATE TABLE IF NOT EXISTS issuer (
	id      INT  NOT NULL,
	name    TEXT NOT NULL,
	cert    TEXT NOT NULL,
	key     TEXT NOT NULL,
	revoked BOOL
    );
    CREATE INDEX IF NOT EXISTS issuer_index
        ON issuer(id, name);

    DROP TABLE IF EXISTS service;
    CREATE TABLE IF NOT EXISTS service (
	id          INT  NOT NULL,
        issuer_id   INT  NOT NULL,
	name        TEXT NOT NULL,
	cert        TEXT NOT NULL,
	key         TEXT NOT NULL,
	revoked     BOOL
    );
    CREATE INDEX IF NOT EXISTS service_index
        ON issuer(id, name);
`

type Database struct {
	datapath string
	db       *sqlx.DB
	log      *logger.Logger
}

func NewDatabase(datapath string) (*Database, error) {
	var err error
	db := &Database{
		datapath: datapath,
	}
	db.log = logger.NewLogger("database")
	return db, err

}

func (db *Database) InitDatabase() error {
	var err error
	dbPath := filepath.Join(db.datapath, "certmanager.db")
	db.log.Infof("Initialize database %s", dbPath)

	db.db, err = sqlx.Open("sqlite3", dbPath)
	if err != nil {
		return err
	}

	err = db.db.Ping()
	if err != nil {
		return err
	}
	_, err = db.db.Exec(schema)
	if err != nil {
		return err
	}
	return err
}