package db

import (
	"context"
	"fmt"

	"github.com/jmoiron/sqlx"
)

type Style struct {
	ID uint

	Src string
}

func (s *Style) Query(data *InsertId[string]) {
	data.query = fmt.Sprintf(`
		INSERT INTO style (%s, src)
		VALUES ($1, $2)
		RETURNING id`, data.fieldName)
}

func (s *Style) Insert(tx *sqlx.Tx, data *InsertId[string]) error {
	if data.query == "" {
		return ErrNoQuery
	}

	args := []any{data.id, s.Src}

	err := tx.QueryRowx(data.query, args...).Scan(&s.ID)
	if err != nil {
		return err
	}

	return nil
}

func (s *Style) InsertCtx(ctx context.Context, tx *sqlx.Tx, data *InsertId[string]) error {
	if data.query == "" {
		return ErrNoQuery
	}

	args := []any{data.id, s.Src}

	err := tx.QueryRowxContext(ctx, data.query, args...).Scan(&s.ID)
	if err != nil {
		return err
	}

	return nil
}