Foundry-Scrapping-API/internal/foundry/models/db/setup.go

package db

import (
	"context"
	"database/sql"
	"errors"
	"strconv"
	"time"

	"github.com/jmoiron/sqlx"
	"golang.org/x/sync/errgroup"
)

type Setup struct {
	ID        uint
	CreatedAt time.Time

	IsAdmin         bool
	IsSetup         bool
	CoreUpdate      CoreUpdate
	FeaturedContent FeaturedContent
	Files           Files
	Options         *SetupOptions
	Release         Release
	Languages       []*SetupLanguage
	Modules         []*Module
	News            []*News
	PackageWarnings []*PackageWarning
	Systems         []*System
	Worlds          []*World
}

func (s *Setup) InsertObjects(tx *sqlx.Tx) error {
	relData := InsertId[uint]{id: s.ID, fieldName: "setup_id"}
	group, _ := errgroup.WithContext(context.Background())

	InsertWithCtxParallel(group, tx, &s.CoreUpdate, relData)
	InsertWithCtxParallel(group, tx, &s.FeaturedContent, relData)
	InsertWithCtxParallel(group, tx, &s.Files, relData)
	InsertWithCtxParallel(group, tx, s.Options, relData)
	InsertWithCtxParallel(group, tx, &s.Release, relData)

	InsertSliceParallel(group, tx, s.Languages, relData)
	InsertSliceParallel(group, tx, s.Modules, relData)
	InsertSliceParallel(group, tx, s.News, relData)
	InsertSliceParallel(group, tx, s.PackageWarnings, relData)

	relDataString := InsertId[string]{
		id:        strconv.FormatUint(uint64(s.ID), 10),
		fieldName: "setup_id",
	}
	InsertSliceParallel(group, tx, s.Systems, relDataString)
	InsertSliceParallel(group, tx, s.Worlds, relDataString)

	err := group.Wait()
	if err != nil && !errors.Is(err, sql.ErrNoRows) {
		return err
	}
	return nil
}

func (s *Setup) Insert(db *sqlx.DB) error {
	tx := db.MustBegin()
	defer tx.Rollback()

	const query = `
		INSERT INTO setup (is_admin, is_setup)
		VALUES ($1, $2)
		RETURNING id, created_at`

	args := []any{s.IsAdmin, s.IsSetup}

	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
	defer cancel()

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

	err = s.InsertObjects(tx)
	if err != nil {
		return err
	}

	return tx.Commit()
}

type FeaturedContent struct {
	ID uint

	Title   string
	Caption string
	URL     string
	Image   string
}

func (f *FeaturedContent) Query(data *InsertId[uint]) {
	data.query = `
		INSERT INTO featured_content (setup_id, title, caption, url, image)
		VALUES ($1, $2, $3, $4, $5)
		RETURNING id`
}

func (f *FeaturedContent) Insert(tx *sqlx.Tx, data *InsertId[uint]) error {
	if data.query == "" {
		return ErrNoQuery
	}

	args := []any{data.id, f.Title, f.Caption, f.URL, f.Image}

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

	return nil
}

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

	args := []any{data.id, f.Title, f.Caption, f.URL, f.Image}

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

	return nil
}

type News struct {
	ID uint

	Title   string
	Caption string
	URL     string
	Image   string
}

func (n *News) Query(data *InsertId[uint]) {
	data.query = `
		INSERT INTO news (setup_id, title, caption, url, image)
		VALUES ($1, $2, $3, $4, $5)
		RETURNING id`
}

func (n *News) Insert(tx *sqlx.Tx, data *InsertId[uint]) error {
	if data.query == "" {
		return ErrNoQuery
	}

	args := []any{data.id, n.Title, n.Caption, n.URL, n.Image}

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

	return nil
}

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

	args := []any{data.id, n.Title, n.Caption, n.URL, n.Image}

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

	return nil
}

func HasSetup(db *sqlx.DB) (bool, error) {
	const query = `
		SELECT EXISTS(SELECT 1 FROM setup)`

	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
	defer cancel()

	var exists bool
	err := db.GetContext(ctx, &exists, query)
	if err != nil {
		return false, err
	}
	return exists, nil
}