package dao

import (
	"gorm.io/gorm"
	"hospital-admin-api/api/model"
	"hospital-admin-api/api/requests"
	"hospital-admin-api/global"
)

type PharmacyDao struct{}

// GetPharmacyById 获取药房数据-药房id
func (r *PharmacyDao) GetPharmacyById(pharmacyId int64) (m *model.Pharmacy, err error) {
	err = global.Db.Where("pharmacy_id = ? AND status != ?", pharmacyId, 2).First(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetPharmacyByCode 获取药房数据-药房代码
func (r *PharmacyDao) GetPharmacyByCode(code string) (m *model.Pharmacy, err error) {
	err = global.Db.Where("pharmacy_code = ? AND status != ?", code, 2).First(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// AddPharmacy 新增药房
func (r *PharmacyDao) AddPharmacy(tx *gorm.DB, m *model.Pharmacy) (*model.Pharmacy, error) {
	if err := tx.Create(m).Error; err != nil {
		return nil, err
	}
	return m, nil
}

// EditPharmacyById 修改药房-药房id
func (r *PharmacyDao) EditPharmacyById(tx *gorm.DB, pharmacyId int64, data interface{}) error {
	err := tx.Model(&model.Pharmacy{}).Where("pharmacy_id = ?", pharmacyId).Updates(data).Error
	if err != nil {
		return err
	}
	return nil
}

// DeletePharmacyById 软删除药房-药房id
func (r *PharmacyDao) DeletePharmacyById(tx *gorm.DB, pharmacyId int64) error {
	err := tx.Model(&model.Pharmacy{}).Where("pharmacy_id = ?", pharmacyId).Update("status", 2).Error
	if err != nil {
		return err
	}
	return nil
}

// GetPharmacyList 获取药房列表
func (r *PharmacyDao) GetPharmacyList(maps interface{}) (m []*model.Pharmacy, err error) {
	err = global.Db.Where(maps).Where("status != ?", 2).Order("created_at desc").Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetPharmacyPageSearch 获取药房列表-分页
func (r *PharmacyDao) GetPharmacyPageSearch(req requests.GetPharmacyPage, page, pageSize int) (m []*model.Pharmacy, total int64, err error) {
	var totalRecords int64

	query := global.Db.Model(&model.Pharmacy{}).Where("status != ?", 2)

	if req.PharmacyName != "" {
		query = query.Where("pharmacy_name LIKE ?", "%"+req.PharmacyName+"%")
	}

	if req.PharmacyCode != "" {
		query = query.Where("pharmacy_code LIKE ?", "%"+req.PharmacyCode+"%")
	}

	if req.Status != nil {
		query = query.Where("status = ?", *req.Status)
	}

	if req.IsPickup != nil {
		query = query.Where("is_pickup = ?", *req.IsPickup)
	}

	if req.ProvinceId != 0 {
		query = query.Where("province_id = ?", req.ProvinceId)
	}

	if req.CityId != nil {
		query = query.Where("city_id = ?", req.CityId)
	}

	if req.CountyId != nil {
		query = query.Where("county_id = ?", req.CountyId)
	}

	query = query.Order("created_at desc")

	if err := query.Count(&totalRecords).Error; err != nil {
		return nil, 0, err
	}

	err = query.Scopes(model.Paginate(page, pageSize)).Find(&m).Error
	if err != nil {
		return nil, 0, err
	}

	return m, totalRecords, nil
}