package storage

import (
	"database/sql"
	"log"

	_ "modernc.org/sqlite"
)

func InitDB(filepath string) *sql.DB {
	db, err := sql.Open("sqlite", filepath)
	if err != nil {
		log.Fatal("❌ Failed to open DB:", err)
	}

	createTable := `
	CREATE TABLE IF NOT EXISTS draws (
		id INTEGER PRIMARY KEY AUTOINCREMENT,
		date TEXT,
		numbers TEXT,
		spend REAL,
		return REAL
	);`

	_, err = db.Exec(createTable)
	if err != nil {
		log.Fatal("❌ Failed to create table:", err)
	}

	return db
}