package db

import (
	"context"

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

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

type TeamRepository struct {
	*Repository[model.Team]
}

func NewTeamRepository(db *bun.DB) *TeamRepository {
	return &TeamRepository{
		Repository: NewRepository[model.Team](db, []string{"slug"}),
	}
}

func (r *TeamRepository) GetTeamSlugsNotInDb(ctx context.Context, teamSlugs []string) ([]string, error) {
	var teams []model.Team
	err := r.db.NewSelect().
		Model(&teams).
		Where("slug IN (?)", bun.In(teamSlugs)).
		Scan(ctx)
	if err != nil {
		return nil, errors.Wrap(err, "getting players not in db")
	}
	diff, _ := lo.Difference(teamSlugs, lo.Map(teams, func(t model.Team, index int) string { return t.Slug }))
	return diff, nil
}

func (r *TeamRepository) SearchByDisplayName(ctx context.Context, displayName string, limit int) ([]model.Team, error) {
	var teams []model.Team
	err := r.db.NewSelect().
		Model(&teams).
		Relation("Country").
		Relation("DomesticLeague").
		Relation("DomesticLeague.Zone").
		Where("f_unaccent(team.display_name) ILIKE ?", "%"+displayName+"%").
		Limit(limit).
		Scan(ctx)
	return teams, err
}