package dataloaders

import (
	"context"
	"time"

	"github.com/tribalwarshelp/api/player"
	"github.com/tribalwarshelp/api/tribe"
	"github.com/tribalwarshelp/api/version"
	"github.com/tribalwarshelp/api/village"
	"github.com/tribalwarshelp/shared/models"
)

const (
	wait = 4 * time.Millisecond
)

type DataLoaders struct {
	VersionByCode *VersionLoader
}

type Config struct {
	PlayerRepo  player.Repository
	TribeRepo   tribe.Repository
	VillageRepo village.Repository
	VersionRepo version.Repository
}

func NewDataLoaders(cfg Config) *DataLoaders {
	return &DataLoaders{
		VersionByCode: &VersionLoader{
			wait:     wait,
			maxBatch: 0,
			fetch: func(keys []string) ([]*models.Version, []error) {
				codes := []models.VersionCode{}
				for _, code := range keys {
					codes = append(codes, models.VersionCode(code))
				}
				versions, _, err := cfg.VersionRepo.Fetch(context.Background(), version.FetchConfig{
					Filter: &models.VersionFilter{
						Code: codes,
					},
					Select: true,
				})
				if err != nil {
					return nil, []error{err}
				}

				versionByCode := make(map[models.VersionCode]*models.Version)
				for _, version := range versions {
					versionByCode[version.Code] = version
				}

				inOrder := make([]*models.Version, len(keys))
				for i, code := range codes {
					inOrder[i] = versionByCode[code]
				}

				return inOrder, nil
			},
		},
	}
}