package repo

import (
	"admin/apps/game/domain/entity"
	"admin/apps/game/model"
	"admin/internal/errcode"
	"admin/internal/model/dto"
	"gorm.io/gorm"
)

type IAccountRepo interface {
	GetWhiteListAll(projectEt *entity.Project) ([]*dto.WhiteListInfo, error)
}

type accountRepoImpl struct {
	db *gorm.DB
}

func NewAccountRepo(db *gorm.DB) IAccountRepo {
	return &accountRepoImpl{db: db}
}

func (impl *accountRepoImpl) GetWhiteListAll(projectEt *entity.Project) ([]*dto.WhiteListInfo, error) {
	list := make([]*model.WhiteList, 0)
	err := impl.db.Where("project_id = ?", projectEt.Po.ID).Find(&list).Error
	if err != nil {
		return nil, errcode.New(errcode.DBError, "list project %v whitelist error:%v", projectEt.Po.ID, err)
	}
	retAccountList := make([]*dto.WhiteListInfo, 0, len(list))
	for _, a := range list {
		retAccountList = append(retAccountList, &dto.WhiteListInfo{
			WType: a.WType,
			Value: a.Value,
		})
	}
	return retAccountList, nil
}