package db

import (
	"context"

	"github.com/samber/lo"
	"github.com/uptrace/bun"

	"git.lehouerou.net/laurent/sorarebuddy/model"
)

type CompetitionRepository struct {
	*Repository[model.Competition]
}

func NewCompetitionRepository(db *bun.DB) *CompetitionRepository {
	return &CompetitionRepository{
		Repository: NewRepository[model.Competition](db, []string{"slug"}),
	}
}

func (r *CompetitionRepository) GetAllClubCompetitions(ctx context.Context, zones []int) ([]model.Competition, error) {
	var res []model.Competition
	request := r.db.NewSelect().Model(&res).
		Where("competition_type = ?", "CLUB").
		Relation("Zone")
	if len(zones) > 0 {
		request = request.Where("zone_id IN (?)", bun.In(zones))
	}
	if err := request.Scan(ctx); err != nil {
		return nil, err
	}
	return res, nil
}

func (r *CompetitionRepository) GetCompetitionSlugsNotInDb(
	ctx context.Context,
	competitionSlugs []string,
) ([]string, error) {
	var competitions []model.Competition
	err := r.db.NewSelect().Model(&competitions).Where("slug IN (?)", bun.In(competitionSlugs)).Scan(ctx)
	if err != nil {
		return nil, err
	}
	diff, _ := lo.Difference(
		competitionSlugs,
		lo.Map(competitions, func(c model.Competition, index int) string { return c.Slug }),
	)
	return diff, nil
}