finish world initialization, set up init on startup and on shutdown, move all models to main directory

internal/foundry/models/db/folder.go

@@ -0,0 +1,151 @@
+package db
+
+import (
+	"context"
+	"database/sql"
+	"errors"
+	"fmt"
+	"strconv"
+
+	"github.com/jmoiron/sqlx"
+	"golang.org/x/sync/errgroup"
+)
+
+type Folder struct {
+	ID uint
+
+	Name    string
+	Sorting string
+	Color   string
+	Packs   []string
+	Folders []*Folder
+}
+
+func (f *Folder) Query(data *InsertId[string]) {
+	data.query = fmt.Sprintf(`
+		INSERT INTO folder (%s, name, sorting, color)
+		VALUES ($1, $2, $3, $4)
+		RETURNING id`, data.fieldName)
+}
+
+func (f *Folder) InsertObjects(tx *sqlx.Tx) error {
+	relId := InsertId[string]{
+		id:        strconv.FormatUint(uint64(f.ID), 10),
+		fieldName: "folder_id",
+		tableName: "folder_packs",
+	}
+	group, _ := errgroup.WithContext(context.Background())
+
+	InsertSimpleSliceParallel(group, tx, f.Packs, &relId)
+	InsertSliceParallel(group, tx, f.Folders, relId)
+
+	err := group.Wait()
+	if err != nil && !errors.Is(err, sql.ErrNoRows) {
+		return err
+	}
+	return nil
+}
+
+func (f *Folder) Insert(tx *sqlx.Tx, data *InsertId[string]) error {
+	if data.query == "" {
+		return ErrNoQuery
+	}
+
+	args := []any{data.id, f.Name, f.Sorting, f.Color}
+
+	err := tx.QueryRowx(data.query, args...).Scan(&f.ID)
+	if err != nil {
+		return err
+	}
+
+	return f.InsertObjects(tx)
+}
+
+func (f *Folder) InsertCtx(ctx context.Context, tx *sqlx.Tx, data *InsertId[string]) error {
+	if data.query == "" {
+		return ErrNoQuery
+	}
+
+	args := []any{data.id, f.Name, f.Sorting, f.Color}
+
+	err := tx.QueryRowxContext(ctx, data.query, args...).Scan(&f.ID)
+	if err != nil {
+		return err
+	}
+
+	return f.InsertObjects(tx)
+}
+
+type WorldFolder struct {
+	ID string
+
+	Name        string
+	Type        string
+	Folder      string
+	Sorting     string
+	Description string
+	Color       string
+	Sort        int
+	Stats       Stats
+}
+
+func (w *WorldFolder) Query(data *InsertId[uint]) {
+	data.query = `
+		INSERT INTO world_folder (game_id, id, name, type, folder, sorting, description, color, sort)
+		VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)
+		ON CONFLICT(id) DO UPDATE SET
+			game_id = EXCLUDED.game_id,
+			updated_at = datetime('now')
+		RETURNING (created_at == updated_at) AS is_inserted`
+}
+
+func (w *WorldFolder) InsertObjects(tx *sqlx.Tx) error {
+	group, ctx := errgroup.WithContext(context.Background())
+
+	relId := InsertId[string]{id: w.ID, fieldName: "world_folder_id"}
+	InsertWithCtxParallel(group, ctx, tx, w.Stats, relId)
+
+	err := group.Wait()
+	if err != nil && !errors.Is(err, sql.ErrNoRows) {
+		return err
+	}
+	return nil
+}
+
+func (w *WorldFolder) Insert(tx *sqlx.Tx, data *InsertId[uint]) error {
+	if data.query == "" {
+		return ErrNoQuery
+	}
+
+	args := []any{data.id, w.ID, w.Name, w.Type, w.Folder, w.Sorting, w.Description, w.Color, w.Sort}
+
+	var isInserted bool
+	err := tx.QueryRowx(data.query, args...).Scan(&isInserted)
+	if err != nil {
+		return err
+	}
+
+	if isInserted {
+		return w.InsertObjects(tx)
+	}
+	return nil
+}
+
+func (w *WorldFolder) InsertCtx(ctx context.Context, tx *sqlx.Tx, data *InsertId[uint]) error {
+	if data.query == "" {
+		return ErrNoQuery
+	}
+
+	args := []any{data.id, w.ID, w.Name, w.Type, w.Folder, w.Sorting, w.Description, w.Color, w.Sort}
+
+	var isInserted bool
+	err := tx.QueryRowxContext(ctx, data.query, args...).Scan(&isInserted)
+	if err != nil {
+		return err
+	}
+
+	if isInserted {
+		return w.InsertObjects(tx)
+	}
+	return nil
+}