nquest/pkg/service/game.go

package service

import (
	"context"

	"github.com/google/uuid"
	"gitrepo.ru/neonxp/nquest/pkg/models"
	"gorm.io/gorm"
	"gorm.io/gorm/clause"
)

type Game struct {
	DB *gorm.DB
}

// NewGame returns new Game.
func NewGame(db *gorm.DB) *Game {
	return &Game{
		DB: db,
	}
}

func (gs *Game) GetByID(ctx context.Context, id uuid.UUID) (*models.Game, error) {
	g := &models.Game{}

	return g, gs.DB.
		WithContext(ctx).
		Preload("Tasks", func(db *gorm.DB) *gorm.DB {
			return db.Order("tasks.task_order ASC")
		}).
		Preload("Tasks.Codes").
		First(g, id).
		Error
}

func (gs *Game) List(ctx context.Context) ([]*models.Game, error) {
	games := make([]*models.Game, 0)

	return games, gs.DB.
		WithContext(ctx).
		Order("created_at DESC").
		Preload("Tasks").
		Preload("Authors").
		Find(&games, "visible = true").
		Error
}

func (gs *Game) GetTaskID(ctx context.Context, id uuid.UUID) (*models.Task, error) {
	t := &models.Task{}

	return t, gs.DB.WithContext(ctx).First(t, id).Error
}

func (gs *Game) ListByAuthor(ctx context.Context, author *models.User) ([]*models.Game, error) {
	games := make([]*models.Game, 0)
	model := gs.DB.
		WithContext(ctx).
		Model(&models.Game{})
	if author.Role == models.RoleCreator {
		model.Preload("Authors", gs.DB.Where("id = ?", author.ID))
	}

	return games,
		model.
			Order("created_at DESC").
			Find(&games).
			Error
}

func (gs *Game) UpsertGame(ctx context.Context, game *models.Game) (*models.Game, error) {

	ids := []uuid.UUID{}
	for _, t := range game.Tasks {
		ids = append(ids, t.ID)
	}
	gs.DB.Delete([]models.Task{}, `game_id = ? and id not in (?)`, game.ID, ids)

	err := gs.DB.
		Session(&gorm.Session{FullSaveAssociations: true}).
		Clauses(clause.OnConflict{
			Columns: []clause.Column{{Name: "id"}},
			DoUpdates: clause.AssignmentColumns([]string{
				"title", "description",
			}),
		}).
		Create(game).
		Error
	if err != nil {
		return nil, err
	}

	return game, gs.DB.Save(game).Error
}