package database

import (
	"database/sql"
	"fmt"

	_ "modernc.org/sqlite"
)

var db *sql.DB

// InitDB opens the sqlite file at path and ensures the schema exists.
// Call this once at startup before AddBookmark/RetrieveBookmarks are used.
func InitDB(path string) error {
	var err error
	db, err = sql.Open("sqlite", path)
	if err != nil {
		return fmt.Errorf("open db: %w", err)
	}

	if err := db.Ping(); err != nil {
		return fmt.Errorf("ping db: %w", err)
	}

	schema := `
	CREATE TABLE IF NOT EXISTS bookmarks (
		id       INTEGER PRIMARY KEY AUTOINCREMENT,
		title    TEXT NOT NULL,
		url      TEXT NOT NULL,
		category TEXT NOT NULL
	);`

	if _, err := db.Exec(schema); err != nil {
		return fmt.Errorf("create schema: %w", err)
	}

	return nil
}

type Bookmark struct {
	ID       uint64
	Title    string
	URL      string
	Category string // kebab-case, e.g. "market-data-and-filling-dbs"
}

func AddBookmark(b Bookmark) (id uint64, err error) {
	res, err := db.Exec(
		`INSERT INTO bookmarks (title, url, category) VALUES (?, ?, ?)`,
		b.Title, b.URL, b.Category,
	)
	if err != nil {
		return 0, fmt.Errorf("insert bookmark: %w", err)
	}

	lastID, err := res.LastInsertId()
	if err != nil {
		return 0, fmt.Errorf("get inserted id: %w", err)
	}

	return uint64(lastID), nil
}

func RetrieveBookmarks() (marks []Bookmark, err error) {
	rows, err := db.Query(`SELECT id, title, url, category FROM bookmarks ORDER BY id DESC`)
	if err != nil {
		return nil, fmt.Errorf("query bookmarks: %w", err)
	}
	defer rows.Close()

	for rows.Next() {
		var b Bookmark
		if err := rows.Scan(&b.ID, &b.Title, &b.URL, &b.Category); err != nil {
			return nil, fmt.Errorf("scan bookmark: %w", err)
		}
		marks = append(marks, b)
	}

	if err := rows.Err(); err != nil {
		return nil, fmt.Errorf("rows error: %w", err)
	}

	return marks, nil
}