package db

import (
	"database/sql"
	"errors"
	"log"
	"os"
	"strings"

	"go-sjles-pta-vote/server/config"

	_ "github.com/glebarez/go-sqlite"
)

var build_db_query string = `
CREATE TABLE IF NOT EXISTS polls (
    id INTEGER PRIMARY KEY,
    question TEXT NOT NULL,
    member_yes_votes UNSIGNED INT NOT NULL DEFAULT 0,
    member_no_votes UNSIGNED INT NOT NULL DEFAULT 0,
    non_member_yes_votes UNSIGNED INT NOT NULL DEFAULT 0,
    non_member_no_votes UNSIGNED INT NOT NULL DEFAULT 0,
    created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
    updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
    expires_at DATETIME
);

CREATE TABLE IF NOT EXISTS voters (
    poll_id UNSIGNED INT NOT NULL,
    voter_email TEXT NOT NULL,
    FOREIGN KEY (poll_id) REFERENCES polls(id),
    PRIMARY KEY (poll_id, voter_email)
);

CREATE TABLE IF NOT EXISTS members (
    email TEXT NOT NULL,
    member_name TEXT,
    school_year UNSIGNED INT NOT NULL,
    PRIMARY KEY (email, school_year)
);
`

var db *sql.DB

func Connect() (*sql.DB, error) {
	log.Printf("Connecting to database")

	db_config := config.GetConfig()

	log.Printf("Database path: %s", db_config.DBPath)
	db, err := sql.Open("sqlite", db_config.DBPath)
	if err != nil {
		log.Printf("Error opening database: %v", err)
		return nil, err
	}

	_, err = db.Exec(build_db_query)
	if err != nil {
		log.Printf("Error updating schema: %v", err)
		_ = db.Close()
		return nil, err
	}

	return db, nil
}

func Close() {
	if db != nil {
		err := db.Close()
		if err != nil {
			log.Printf("Error closing database: %v", err)
		}
	}
}