package spu

import (
	"context"

	globaldedicatedv1 "github.com/tidbcloud/tidb-management-service/api/spec/global/dedicated/v1"
	"github.com/tidbcloud/tidb-management-service/dedicated-global-service/domain/entity"
	"github.com/tidbcloud/tidb-management-service/utilities/bizerr"
)

type QuerySpuFilter struct {
	ComponentTypes []globaldedicatedv1.ComponentType
	Name           string
	Versions       []string
}

//go:generate go run go.uber.org/mock/mockgen@latest -source=$GOFILE -package=mock_$GOPACKAGE -destination=./mock_$GOPACKAGE/mock_$GOFILE
type RepoSpu interface {
	Save(ctx context.Context, spu *entity.Spu) (uint, bizerr.BizError)
	Query(ctx context.Context, filter *QuerySpuFilter) ([]*entity.Spu, bizerr.BizError)
	LoadByKeys(ctx context.Context, keys []string) (map[string]*entity.Spu, bizerr.BizError)
	SaveSpuMappings(ctx context.Context, spuMappings []*entity.SpuMapping) ([]uint, bizerr.BizError)
	Get(ctx context.Context, id uint64) (*entity.Spu, bizerr.BizError)
	Delete(ctx context.Context, id uint64) bizerr.BizError
}