yarr/src/storage/sqlite/settings.go

package sqlite

import (
	"database/sql"
	"encoding/json"
	"log"

	"github.com/nkanaev/yarr/src/storage/model"
)

func (s *SQLiteStorage) GetSettings() model.Settings {
	result := model.SettingsDefault()
	rows, err := s.db.Query(`select key, val from settings;`)
	if err != nil {
		log.Print(err)
		return result
	}
	defer rows.Close()

	for rows.Next() {
		var key string
		var val []byte
		rows.Scan(&key, &val)

		switch key {
		case "filter":
			json.Unmarshal(val, &result.Filter)
		case "feed":
			json.Unmarshal(val, &result.Feed)
		case "feed_list_width":
			json.Unmarshal(val, &result.FeedListWidth)
		case "item_list_width":
			json.Unmarshal(val, &result.ItemListWidth)
		case "sort_newest_first":
			json.Unmarshal(val, &result.SortNewestFirst)
		case "theme_name":
			json.Unmarshal(val, &result.ThemeName)
		case "theme_font":
			json.Unmarshal(val, &result.ThemeFont)
		case "theme_size":
			json.Unmarshal(val, &result.ThemeSize)
		case "refresh_rate":
			json.Unmarshal(val, &result.RefreshRate)
		case "language":
			json.Unmarshal(val, &result.Language)
		}
	}
	return result
}

func (s *SQLiteStorage) UpdateSettings(params model.UpdateSettingsParams) bool {
	tx, err := s.db.Begin()
	if err != nil {
		log.Print(err)
		return false
	}
	defer tx.Rollback()

	update := func(key string, val any) error {
		valEncoded, err := json.Marshal(val)
		if err != nil {
			return err
		}
		_, err = tx.Exec(`
			insert into settings (key, val) values (:key, :val)
			on conflict (key) do update set val=:val`,
			sql.Named("key", key),
			sql.Named("val", valEncoded),
		)
		return err
	}

	var errs []error
	if params.Filter != nil {
		errs = append(errs, update("filter", *params.Filter))
	}
	if params.Feed != nil {
		errs = append(errs, update("feed", *params.Feed))
	}
	if params.FeedListWidth != nil {
		errs = append(errs, update("feed_list_width", *params.FeedListWidth))
	}
	if params.ItemListWidth != nil {
		errs = append(errs, update("item_list_width", *params.ItemListWidth))
	}
	if params.SortNewestFirst != nil {
		errs = append(errs, update("sort_newest_first", *params.SortNewestFirst))
	}
	if params.ThemeName != nil {
		errs = append(errs, update("theme_name", *params.ThemeName))
	}
	if params.ThemeFont != nil {
		errs = append(errs, update("theme_font", *params.ThemeFont))
	}
	if params.ThemeSize != nil {
		errs = append(errs, update("theme_size", *params.ThemeSize))
	}
	if params.RefreshRate != nil {
		errs = append(errs, update("refresh_rate", *params.RefreshRate))
	}
	if params.Language != nil {
		errs = append(errs, update("language", *params.Language))
	}

	for _, err := range errs {
		if err != nil {
			log.Print(err)
			return false
		}
	}

	if err := tx.Commit(); err != nil {
		log.Print(err)
		return false
	}
	return true
}