package db

import (
	"database/sql"
	"fmt"
	"os"

	"github.com/jordan-wright/gophish/config"
	_ "github.com/mattn/go-sqlite3"
)

var Conn *sql.DB

// Setup initializes the Conn object
// It also populates the Gophish Config object
func Setup() error {
	//If the file already exists, delete it and recreate it
	_, err := os.Stat(config.Conf.DBPath)
	if err == nil {
		os.Remove(config.Conf.DBPath)
	}
	fmt.Println("Creating db at " + config.Conf.DBPath)
	Conn, err = sql.Open("sqlite3", config.Conf.DBPath)
	if err != nil {
		return err
	}
	//Create the tables needed
	_, err = Conn.Exec(
		`CREATE TABLE Users (id INTEGER PRIMARY KEY AUTOINCREMENT, username TEXT, hash VARCHAR(60), apikey VARCHAR(32));`)
	if err != nil {
		return err
	}
	//Create the default user
	stmt, err := Conn.Prepare(`INSERT INTO Users (username, hash, apikey) VALUES (?, ?, ?);`)
	defer stmt.Close()
	if err != nil {
		return err
	}
	_, err = stmt.Exec("jordan", "$2a$10$d4OtT.RkEOQn.iruVWIQ5u8CeV/85ZYF41y8wKeUwsAPqPNFvTccW", "12345678901234567890123456789012")
	if err != nil {
		return err
	}
	return nil
}