package service

import (
	"database/sql"
	"fmt"
	"log"

	_ "github.com/mattn/go-sqlite3"
)

type MacAddress struct {
	Id         string
	Prefix     string
	VendorName string
	Private    bool
	BlockType  string
	LastUpdate string
}

type SQLiteHelper struct {
	connectionString string
	db               *sql.DB
}

func NewSQLiteHelper(dbFilePath string) *SQLiteHelper {
	connectionString := fmt.Sprintf("file:%s?cache=shared&mode=rwc", dbFilePath)
	db, err := sql.Open("sqlite3", connectionString)
	if err != nil {
		log.Fatal(err)
	}

	return &SQLiteHelper{
		connectionString: connectionString,
		db:               db,
	}
}

func (helper *SQLiteHelper) Close() {
	helper.db.Close()
}

func (helper *SQLiteHelper) ReloadDatabase() bool {
	helper.db.Close()
	db, err := sql.Open("sqlite3", helper.connectionString)
	if err != nil {
		log.Fatal(err)
		return false
	}
	helper.db = db
	return true
}

func (helper *SQLiteHelper) QueryMacAddressByPrefix(prefix string) (*MacAddress, error) {
	const tableName = "MacAddresses"
	query := fmt.Sprintf("SELECT * FROM %s WHERE Prefix = ?", tableName)
	row := helper.db.QueryRow(query, prefix)

	var macAddress MacAddress
	err := row.Scan(&macAddress.Id, &macAddress.Prefix, &macAddress.VendorName, &macAddress.Private, &macAddress.BlockType, &macAddress.LastUpdate)
	if err != nil {
		if err == sql.ErrNoRows {
			return nil, nil
		}
		return nil, err
	}

	return &macAddress, nil
}

func (helper *SQLiteHelper) querySingleColumn(tableName, columnName string) (string, error) {
	query := fmt.Sprintf("SELECT %s FROM %s LIMIT 1", columnName, tableName)
	row := helper.db.QueryRow(query)

	var result string
	err := row.Scan(&result)
	if err != nil {
		return "", err
	}

	return result, nil
}

func (helper *SQLiteHelper) QueryVersion() (string, error) {
	return helper.querySingleColumn("Config", "Version")
}