package config

import (
	"errors"

	"gorm.io/gorm"
)

type Repository struct {
	db *gorm.DB
}

func NewRepository(db *gorm.DB) *Repository {
	return &Repository{db: db}
}

func (r *Repository) Get(key string) (*ConfigEntry, error) {
	var e ConfigEntry
	if err := r.db.First(&e, "key = ?", key).Error; err != nil {
		return nil, err
	}
	return &e, nil
}

func (r *Repository) Upsert(key, value string) error {
	// Try update first, then insert if nothing updated.
	res := r.db.Model(&ConfigEntry{}).Where("key = ?", key).Update("value", value)
	if res.Error != nil {
		return res.Error
	}
	if res.RowsAffected > 0 {
		return nil
	}
	return r.db.Create(&ConfigEntry{Key: key, Value: value}).Error
}

func (r *Repository) List() ([]ConfigEntry, error) {
	var entries []ConfigEntry
	if err := r.db.Order("key asc").Find(&entries).Error; err != nil {
		return nil, err
	}
	return entries, nil
}

func (r *Repository) CreateIfMissing(key, value string) error {
	var e ConfigEntry
	err := r.db.First(&e, "key = ?", key).Error
	if err == nil {
		return nil
	}
	if errors.Is(err, gorm.ErrRecordNotFound) {
		return r.db.Create(&ConfigEntry{Key: key, Value: value}).Error
	}
	return err
}