package repo

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

type IWhiteListRepo interface {
	List(pageNo, pageLen int) ([]*entity.WhiteList, error)
	Create(et *entity.WhiteList) error
	Edit(et *entity.WhiteList) error
	Delete(id int) error
}

func NewWhiteListRepo(db *gorm.DB) IWhiteListRepo {
	return newWhiteListRepoImpl(db)
}

type WhiteListRepoImpl struct {
	db *gorm.DB
}

func newWhiteListRepoImpl(db *gorm.DB) *WhiteListRepoImpl {
	return &WhiteListRepoImpl{db: db}
}

func (repo *WhiteListRepoImpl) List(pageNo, pageLen int) ([]*entity.WhiteList, error) {
	list := make([]*model.WhiteList, 0)
	err := repo.db.Find(&list).Error
	if err != nil {
		return nil, errcode.New(errcode.DBError, "find WhiteList error:%v", err)
	}

	entityList := make([]*entity.WhiteList, 0, len(list))
	for _, WhiteList := range list {
		entityList = append(entityList, entity.FromWhiteListPo(WhiteList))
	}

	return entityList, nil
}

func (repo *WhiteListRepoImpl) Create(et *entity.WhiteList) error {
	po := et.ToPo()
	err := repo.db.Create(po).Error
	if err != nil {
		return errcode.New(errcode.DBError, "create obj:%+v error:%v", et, err)
	}
	et.Id = po.ID
	return nil
}

func (repo *WhiteListRepoImpl) Edit(et *entity.WhiteList) error {
	po := et.ToPo()
	err := repo.db.Where("id=?", et.Id).Updates(po).Error
	if err != nil {
		return errcode.New(errcode.DBError, "edit obj:%+v error:%v", et, err)
	}
	return nil
}

func (repo *WhiteListRepoImpl) Delete(id int) error {
	err := repo.db.Where("id=?", id).Unscoped().Delete(&model.WhiteList{}).Error
	if err != nil {
		return errcode.New(errcode.DBError, "delete obj:%+v error:%v", id, err)
	}
	return nil
}