package internal

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

	"golang.org/x/crypto/bcrypt"
)

func Migrate(db *sql.DB, webmaster string, passOne string, passTwo string) {
	_, table_check := db.Query("SELECT * FROM logins;")
	if table_check != nil {
		_, err := db.Exec("CREATE TABLE logins (id SERIAL PRIMARY KEY, name VARCHAR(50) NOT NULL, time TIMESTAMP DEFAULT CURRENT_TIMESTAMP, pass_one TEXT NOT NULL, pass_two TEXT NOT NULL);")
		if err != nil {
			fmt.Fprintf(os.Stderr, "Unable to create logins table: %v\n", err)
			os.Exit(1)
		}

		hashOne, errHashOne := bcrypt.GenerateFromPassword([]byte(passOne), 12)
		hashTwo, errHashTwo := bcrypt.GenerateFromPassword([]byte(passTwo), 12)
		if errHashOne != nil || errHashTwo != nil {
			fmt.Fprintf(os.Stderr, "Failed to hash password")
			os.Exit(1)
		}

		_, err = db.Exec(fmt.Sprintf("INSERT INTO logins (name, pass_one, pass_two) VALUES ('%s', '%s', '%s');", webmaster, hashOne, hashTwo))
		if err != nil {
			fmt.Fprintf(os.Stderr, "Unable to add webmaster to logins table: %v\n", err)
			os.Exit(1)
		}
	}

	_, table_check = db.Query("SELECT * FROM posts;")
	if table_check != nil {
		_, err := db.Exec("CREATE TABLE posts (id SERIAL PRIMARY KEY, user_id INTEGER REFERENCES logins(id), time TIMESTAMP DEFAULT CURRENT_TIMESTAMP, brief TEXT NOT NULL, content TEXT NOT NULL);")
		if err != nil {
			fmt.Fprintf(os.Stderr, "Unable to create posts table: %v\n", err)
			os.Exit(1)
		}
	}

	_, table_check = db.Query("SELECT * FROM feeds;")
	if table_check != nil {
		_, err := db.Exec("CREATE TABLE feeds (id SERIAL PRIMARY KEY, name VARCHAR(50) NOT NULL, user_id INTEGER REFERENCES logins(id), url VARCHAR(255) NOT NULL);")
		if err != nil {
			fmt.Fprintf(os.Stderr, "Unable to create feeds table: %v\n", err)
			os.Exit(1)
		}
	}

	_, table_check = db.Query("SELECT * FROM cookies;")
	if table_check != nil {
		_, err := db.Exec("CREATE TABLE cookies (id SERIAL PRIMARY KEY, content VARCHAR(255) NOT NULL, user_id INTEGER REFERENCES logins(id), expiration TIMESTAMP);")
		if err != nil {
			fmt.Fprintf(os.Stderr, "Unable to create cookies table: %v\n", err)
		}
	}
}