package main

import (
	"database/sql"
	"log"

	"golang.org/x/crypto/bcrypt"

	"stevenlr.com/timer/model"
)

func initializeDatabaseV1(db *sql.DB) error {
	tx, err := db.Begin()
	if err != nil {
		return err
	}
	defer tx.Rollback()

	_, err = tx.Exec(`PRAGMA user_version = 1`)
	if err != nil {
		return err
	}

	_, err = tx.Exec(`
		CREATE TABLE Timer (
			Id			BLOB	NOT NULL UNIQUE,
			Name		TEXT	NOT NULL,
			StartTime	TEXT	NOT NULL,
			EndTime		TEXT	NOT NULL,
			Owner		BLOB	NOT NULL,
			Token		TEXT	NOT NULL UNIQUE,
			PRIMARY KEY (Id)
		)`)
	if err != nil {
		return err
	}

	_, err = tx.Exec(`
		CREATE TABLE User (
			Id			BLOB	NOT NULL UNIQUE,
			Name		TEXT	NOT NULL,
			Salt		TEXT	NOT NULL,
			Password	BLOB	NOT NULL,
			PRIMARY KEY (id)
		)`)
	if err != nil {
		return err
	}

	userName := "admin"
	userPassword := "admin"
	salt, err := GenerateRandomString(33)
	if err != nil {
		return err
	}

	password, err := bcrypt.GenerateFromPassword([]byte(salt+userPassword), bcrypt.MinCost)
	if err != nil {
		return err
	}

	_, err = tx.Exec(`INSERT INTO User VALUES ($1, $2, $3, $4)`, model.MakeUUID(), userName, salt, password)
	if err != nil {
		return err
	}

	return tx.Commit()
}

func migrateDatabaseV2(db *sql.DB) error {
	tx, err := db.Begin()
	if err != nil {
		return err
	}
	defer tx.Rollback()

	_, err = tx.Exec(`PRAGMA user_version = 2`)
	if err != nil {
		return err
	}

	_, err = tx.Exec("CREATE INDEX TimerTokenIndex ON Timer(Token)")
	if err != nil {
		return err
	}

	return tx.Commit()
}

func InitializeDatabase(db *sql.DB) error {
	initialVersion := 0
	row := db.QueryRow("PRAGMA user_version")
	row.Scan(&initialVersion)

	if initialVersion < 1 {
		log.Println("Initializing DB V1")
		err := initializeDatabaseV1(db)
		if err != nil {
			return err
		}
	}

	if initialVersion < 2 {
		log.Println("Migrating DB to V2")
		err := migrateDatabaseV2(db)
		if err != nil {
			return err
		}
	}

	return nil
}