new-server/internal/mysql/presetrepo.go

package mysql

import (
	"context"
	"git.aiterp.net/lucifer/new-server/internal/color"
	"git.aiterp.net/lucifer/new-server/models"
	"github.com/jmoiron/sqlx"
)

type presetRecord struct {
	ID         int     `db:"id"`
	Name       string  `db:"name"`
	Hue        float64 `db:"hue"`
	Saturation float64 `db:"saturation"`
	Kelvin     int     `db:"kelvin"`
	Value      string  `db:"value"`
}

type ColorPresetRepo struct {
	DBX *sqlx.DB
}

func (c *ColorPresetRepo) Find(ctx context.Context, id int) (models.ColorPreset, error) {
	var record presetRecord
	err := c.DBX.GetContext(ctx, &record, "SELECT * FROM color_preset WHERE id = ?", id)
	if err != nil {
		return models.ColorPreset{}, dbErr(err)
	}

	return c.fromRecords(record)[0], nil
}

func (c *ColorPresetRepo) FetchAll(ctx context.Context) ([]models.ColorPreset, error) {
	records := make([]presetRecord, 0, 16)
	err := c.DBX.SelectContext(ctx, &records, "SELECT * FROM color_preset")
	if err != nil {
		return nil, dbErr(err)
	}

	return c.fromRecords(records...), nil
}

func (c *ColorPresetRepo) Save(ctx context.Context, preset *models.ColorPreset) error {
	if preset.ID > 0 {
		_, err := c.DBX.ExecContext(
			ctx,
			"UPDATE color_preset SET name = ?, value = ? WHERE id = ?",
			preset.Name, preset.Value.String(), preset.ID,
		)

		if err != nil {
			return dbErr(err)
		}
	} else {
		rs, err := c.DBX.ExecContext(
			ctx,
			"INSERT INTO color_preset (name, value, hue, saturation, kelvin) VALUES (?, ?, 0, 0, 0)",
			preset.Name, preset.Value.String(),
		)

		if err != nil {
			return dbErr(err)
		}

		id, err := rs.LastInsertId()
		if err != nil {
			return dbErr(err)
		}

		preset.ID = int(id)
	}

	return nil
}

func (c *ColorPresetRepo) Delete(ctx context.Context, preset *models.ColorPreset) error {
	_, err := c.DBX.ExecContext(ctx, "DELETE FROM color_preset WHERE id = ?", preset.ID)
	if err != nil {
		return dbErr(err)
	}

	preset.ID = 0
	return nil
}

func (c *ColorPresetRepo) fromRecords(records ...presetRecord) []models.ColorPreset {
	newList := make([]models.ColorPreset, len(records), len(records))
	for i, record := range records {
		color, _ := color.Parse(record.Value)

		newList[i] = models.ColorPreset{
			ID:    record.ID,
			Name:  record.Name,
			Value: color,
		}
	}

	return newList
}