package runtimeflags

import (
	"context"
	"fmt"
	"time"

	"github.com/jmoiron/sqlx"
)

type SQLiteStore struct {
	writer *sqlx.DB
	reader *sqlx.DB
}

func NewSQLiteStore(writer, reader *sqlx.DB) (*SQLiteStore, error) {
	store := &SQLiteStore{writer: writer, reader: reader}
	if err := store.initSchema(); err != nil {
		return nil, err
	}
	return store, nil
}

func (s *SQLiteStore) initSchema() error {
	_, err := s.writer.Exec(`CREATE TABLE IF NOT EXISTS runtime_flag_overrides (
		key TEXT PRIMARY KEY,
		value INTEGER NOT NULL CHECK (value IN (0, 1)),
		created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
		updated_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
	)`)
	if err != nil {
		return fmt.Errorf("create runtime_flag_overrides: %w", err)
	}
	return nil
}

func (s *SQLiteStore) ListOverrides(ctx context.Context) (map[string]bool, error) {
	rows, err := s.reader.QueryxContext(ctx, `SELECT key, value FROM runtime_flag_overrides`)
	if err != nil {
		return nil, fmt.Errorf("list runtime flag overrides: %w", err)
	}
	defer func() { _ = rows.Close() }()

	out := map[string]bool{}
	for rows.Next() {
		var key string
		var value int
		if err := rows.Scan(&key, &value); err != nil {
			return nil, fmt.Errorf("scan runtime flag override: %w", err)
		}
		out[key] = value != 0
	}
	if err := rows.Err(); err != nil {
		return nil, fmt.Errorf("iterate runtime flag overrides: %w", err)
	}
	return out, nil
}

func (s *SQLiteStore) SetOverride(ctx context.Context, key string, value bool) error {
	now := time.Now().UTC()
	intValue := 0
	if value {
		intValue = 1
	}
	_, err := s.writer.ExecContext(ctx, `INSERT INTO runtime_flag_overrides (key, value, created_at, updated_at)
		VALUES (?, ?, ?, ?)
		ON CONFLICT(key) DO UPDATE SET value = excluded.value, updated_at = excluded.updated_at`,
		key, intValue, now, now)
	if err != nil {
		return fmt.Errorf("set runtime flag override %q: %w", key, err)
	}
	return nil
}

func (s *SQLiteStore) DeleteOverride(ctx context.Context, key string) error {
	if _, err := s.writer.ExecContext(ctx, `DELETE FROM runtime_flag_overrides WHERE key = ?`, key); err != nil {
		return fmt.Errorf("delete runtime flag override %q: %w", key, err)
	}
	return nil
}