package dao

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

type HospitalDao struct {
}

// GetHospitalById 获取医院数据-医院id
func (r *HospitalDao) GetHospitalById(hospitalId int64) (m *model.Hospital, err error) {
	err = global.Db.First(&m, hospitalId).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// AddHospital 新增医院
func (r *HospitalDao) AddHospital(tx *gorm.DB, model *model.Hospital) (*model.Hospital, error) {
	if err := tx.Create(model).Error; err != nil {
		return nil, err
	}
	return model, nil
}

// GetHospitalList 获取医院列表
func (r *HospitalDao) GetHospitalList(maps interface{}) (m []*model.Hospital, err error) {
	err = global.Db.Where(maps).Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// DeleteHospitalById 删除医院-医院id
func (r *HospitalDao) DeleteHospitalById(tx *gorm.DB, hospitalId int64) error {
	if err := tx.Delete(&model.Hospital{}, hospitalId).Error; err != nil {
		return err
	}
	return nil
}

// EditHospitalById 修改医院-医院id
func (r *HospitalDao) EditHospitalById(tx *gorm.DB, hospitalId int64, data interface{}) error {
	err := tx.Model(&model.Hospital{}).Where("hospital_id = ?", hospitalId).Updates(data).Error
	if err != nil {
		return err
	}
	return nil
}

// GetHospitalLimitByMaps 获取医院列表-限制条数
func (r *HospitalDao) GetHospitalLimitByMaps(hospitalRequest requests.GetHospitalList) (m []*model.Hospital, err error) {
	result := global.Db
	if hospitalRequest.HospitalName != "" {
		result = result.Where("hospital_name like ?", "%"+hospitalRequest.HospitalName+"%")
	}

	if hospitalRequest.HospitalLevelName != "" {
		result = result.Where("hospital_level_name like ?", "%"+hospitalRequest.HospitalLevelName+"%")
	}

	if hospitalRequest.ProvinceId != 0 {
		result = result.Where("province_id = ?", hospitalRequest.ProvinceId)
	}

	if hospitalRequest.Province != "" {
		result = result.Where("province like ?", "%"+hospitalRequest.Province+"%")
	}

	if hospitalRequest.CityId != 0 {
		result = result.Where("city_id = ?", hospitalRequest.CityId)
	}

	if hospitalRequest.City != "" {
		result = result.Where("city like ?", "%"+hospitalRequest.City+"%")
	}

	if hospitalRequest.CountyId != 0 {
		result = result.Where("county_id = ?", hospitalRequest.CountyId)
	}

	if hospitalRequest.County != "" {
		result = result.Where("county like ?", "%"+hospitalRequest.County+"%")
	}

	err = result.Limit(10).Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetHospitalPageSearch 获取医院列表-分页
func (r *HospitalDao) GetHospitalPageSearch(req requests.GetHospitalPage, page, pageSize int) (m []*model.Hospital, total int64, err error) {
	var totalRecords int64

	// 构建查询条件
	query := global.Db.Model(&model.Hospital{})

	if req.HospitalName != "" {
		query = query.Where("hospital_name like ?", "%"+req.HospitalName+"%")
	}

	if req.HospitalLevelName != "" {
		query = query.Where("hospital_level_name like ?", "%"+req.HospitalLevelName+"%")
	}

	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
}