package repo

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

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

func NewBanRepo(db *gorm.DB) IBanRepo {
	return newBanRepoImpl(db)
}

type BanRepoImpl struct {
	db *gorm.DB
}

func newBanRepoImpl(db *gorm.DB) *BanRepoImpl {
	return &BanRepoImpl{db: db}
}

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

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

	return entityList, nil
}

func (repo *BanRepoImpl) Create(et *entity.Ban) 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 *BanRepoImpl) Edit(et *entity.Ban) 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 *BanRepoImpl) Delete(id int) error {
	err := repo.db.Where("id=?", id).Unscoped().Delete(&model.Ban{}).Error
	if err != nil {
		return errcode.New(errcode.DBError, "delete obj:%+v error:%v", id, err)
	}
	return nil
}