package dao

import (
	"errors"
	"strconv"
	"strings"

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

type DoctorPharmacyDao struct{}

// GetDoctorPharmacyById 获取绑定关系-主键id
func (r *DoctorPharmacyDao) GetDoctorPharmacyById(doctorPharmacyId int64) (m *model.DoctorPharmacy, err error) {
	err = global.Db.Where("doctor_pharmacy_id = ?", doctorPharmacyId).First(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetDoctorPharmacyListByDoctorId 获取医生绑定的药房列表（带预加载药房详情）
func (r *DoctorPharmacyDao) GetDoctorPharmacyListByDoctorId(doctorId int64) (m []*model.DoctorPharmacy, err error) {
	err = global.Db.Preload("Pharmacy").Where("doctor_id = ? AND status = 1", doctorId).Order("is_default desc, created_at desc").Find(&m).Error
	if err != nil {
		return nil, err
	}

	// GORM 在外键为 0 时会自动跳过 Preload，因此针对未能预加载的记录进行兜底补全
	var zeroPharmacy *model.Pharmacy
	for _, dp := range m {
		if dp.Pharmacy == nil {
			if dp.PharmacyId == 0 {
				if zeroPharmacy == nil {
					var ph model.Pharmacy
					if err := global.Db.Where("pharmacy_id = 0").First(&ph).Error; err == nil {
						zeroPharmacy = &ph
					}
				}
				dp.Pharmacy = zeroPharmacy
			} else {
				var ph model.Pharmacy
				if err := global.Db.Where("pharmacy_id = ?", dp.PharmacyId).First(&ph).Error; err == nil {
					dp.Pharmacy = &ph
				}
			}
		}
	}

	return m, nil
}

// GetDoctorPharmacyByDoctorAndPharmacy 获取某医生与某药房的绑定关系
func (r *DoctorPharmacyDao) GetDoctorPharmacyByDoctorAndPharmacy(doctorId, pharmacyId int64) (m *model.DoctorPharmacy, err error) {
	err = global.Db.Where("doctor_id = ? AND pharmacy_id = ?", doctorId, pharmacyId).First(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// AddDoctorPharmacy 新增绑定
func (r *DoctorPharmacyDao) AddDoctorPharmacy(tx *gorm.DB, m *model.DoctorPharmacy) (*model.DoctorPharmacy, error) {
	if err := tx.Create(m).Error; err != nil {
		return nil, err
	}
	return m, nil
}

// DeleteDoctorPharmacy 删除绑定记录
func (r *DoctorPharmacyDao) DeleteDoctorPharmacy(tx *gorm.DB, maps interface{}) error {
	err := tx.Where(maps).Delete(&model.DoctorPharmacy{}).Error
	if err != nil {
		return err
	}
	return nil
}

// DeleteDoctorPharmacyById 根据绑定关系主键id删除
func (r *DoctorPharmacyDao) DeleteDoctorPharmacyById(tx *gorm.DB, doctorPharmacyId int64) error {
	err := tx.Where("doctor_pharmacy_id = ?", doctorPharmacyId).Delete(&model.DoctorPharmacy{}).Error
	if err != nil {
		return err
	}
	return nil
}

// DeleteDoctorPharmacyByDoctorAndPharmacy 解绑指定医生和指定药房
func (r *DoctorPharmacyDao) DeleteDoctorPharmacyByDoctorAndPharmacy(tx *gorm.DB, doctorId, pharmacyId int64) error {
	err := tx.Where("doctor_id = ? AND pharmacy_id = ?", doctorId, pharmacyId).Delete(&model.DoctorPharmacy{}).Error
	if err != nil {
		return err
	}
	return nil
}

// ResetDoctorDefaultPharmacy 重置医生的默认药房（设为非默认）
func (r *DoctorPharmacyDao) ResetDoctorDefaultPharmacy(tx *gorm.DB, doctorId int64) error {
	err := tx.Model(&model.DoctorPharmacy{}).Where("doctor_id = ?", doctorId).Update("is_default", 0).Error
	if err != nil {
		return err
	}
	return nil
}

// SetDefaultPharmacy 设指定药房为默认
func (r *DoctorPharmacyDao) SetDefaultPharmacy(tx *gorm.DB, doctorId, pharmacyId int64) error {
	err := tx.Model(&model.DoctorPharmacy{}).Where("doctor_id = ? AND pharmacy_id = ?", doctorId, pharmacyId).Update("is_default", 1).Error
	if err != nil {
		return err
	}
	return nil
}

// SetDefaultPharmacyById 根据主键设为默认
func (r *DoctorPharmacyDao) SetDefaultPharmacyById(tx *gorm.DB, doctorPharmacyId int64) error {
	err := tx.Model(&model.DoctorPharmacy{}).Where("doctor_pharmacy_id = ?", doctorPharmacyId).Update("is_default", 1).Error
	if err != nil {
		return err
	}
	return nil
}

// GetPharmacyDoctorPageSearch 获取药房绑定的医生列表-分页
func (r *DoctorPharmacyDao) GetPharmacyDoctorPageSearch(pharmacyId int64, req requests.GetPharmacyDoctorPage, page, pageSize int) (m []*model.DoctorPharmacy, total int64, err error) {
	var totalRecords int64

	query := global.Db.Model(&model.DoctorPharmacy{}).
		Where("gdxz_doctor_pharmacy.pharmacy_id = ? AND gdxz_doctor_pharmacy.status = 1", pharmacyId)

	// 如果有医生姓名或手机号筛选，做表连接
	needJoinDoctor := req.DoctorName != "" || req.Mobile != ""
	if needJoinDoctor {
		query = query.Joins("JOIN gdxz_user_doctor ON gdxz_user_doctor.doctor_id = gdxz_doctor_pharmacy.doctor_id")
		if req.DoctorName != "" {
			query = query.Where("gdxz_user_doctor.user_name LIKE ?", "%"+req.DoctorName+"%")
		}
		if req.Mobile != "" {
			query = query.Joins("JOIN gdxz_user ON gdxz_user.user_id = gdxz_user_doctor.user_id").
				Where("gdxz_user.mobile LIKE ?", "%"+req.Mobile+"%")
		}
	}

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

	err = query.Order("gdxz_doctor_pharmacy.is_default desc, gdxz_doctor_pharmacy.created_at desc").
		Preload("UserDoctor.User").
		Preload("UserDoctor.Hospital").
		Scopes(model.Paginate(page, pageSize)).
		Find(&m).Error
	if err != nil {
		return nil, 0, err
	}

	return m, totalRecords, nil
}

// GetPharmacyDoctorListByPharmacyId 获取药房绑定的所有医生列表
func (r *DoctorPharmacyDao) GetPharmacyDoctorListByPharmacyId(pharmacyId int64) (m []*model.DoctorPharmacy, err error) {
	err = global.Db.Preload("UserDoctor.User").
		Preload("UserDoctor.Hospital").
		Where("pharmacy_id = ? AND status = 1", pharmacyId).
		Order("is_default desc, created_at desc").
		Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetDefaultPharmacyDoctorListByPharmacyId 获取将该药房设为默认药房的医生列表
func (r *DoctorPharmacyDao) GetDefaultPharmacyDoctorListByPharmacyId(pharmacyId int64) (m []*model.DoctorPharmacy, err error) {
	err = global.Db.Preload("UserDoctor").
		Preload("UserDoctor.User").
		Preload("UserDoctor.Hospital").
		Where("pharmacy_id = ? AND status = 1 AND is_default = 1", pharmacyId).
		Order("created_at desc").
		Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetDoctorPharmacyCountsByPharmacyId 获取药房绑定的医生数量（有效绑定数、设为默认药房数）
func (r *DoctorPharmacyDao) GetDoctorPharmacyCountsByPharmacyId(pharmacyId int64) (totalCount int64, defaultCount int64, err error) {
	err = global.Db.Model(&model.DoctorPharmacy{}).
		Where("pharmacy_id = ? AND status = 1", pharmacyId).
		Count(&totalCount).Error
	if err != nil {
		return 0, 0, err
	}

	err = global.Db.Model(&model.DoctorPharmacy{}).
		Where("pharmacy_id = ? AND status = 1 AND is_default = 1", pharmacyId).
		Count(&defaultCount).Error
	if err != nil {
		return 0, 0, err
	}

	return totalCount, defaultCount, nil
}

// GetOtherPharmaciesByDoctorIds 批量获取一组医生绑定的其他有效药房
func (r *DoctorPharmacyDao) GetOtherPharmaciesByDoctorIds(doctorIds []int64, excludePharmacyId int64) (m []*model.DoctorPharmacy, err error) {
	if len(doctorIds) == 0 {
		return nil, nil
	}

	err = global.Db.Preload("Pharmacy").
		Where("doctor_id IN (?) AND status = 1 AND pharmacy_id != ?", doctorIds, excludePharmacyId).
		Order("is_default desc, created_at desc").
		Find(&m).Error
	if err != nil {
		return nil, err
	}

	// GORM 在外键为 0 时会自动跳过 Preload，因此针对未能预加载的记录进行兜底补全
	var zeroPharmacy *model.Pharmacy
	for _, dp := range m {
		if dp.Pharmacy == nil {
			if dp.PharmacyId == 0 {
				if zeroPharmacy == nil {
					var ph model.Pharmacy
					if err := global.Db.Where("pharmacy_id = 0").First(&ph).Error; err == nil {
						zeroPharmacy = &ph
					}
				}
				dp.Pharmacy = zeroPharmacy
			} else {
				var ph model.Pharmacy
				if err := global.Db.Where("pharmacy_id = ?", dp.PharmacyId).First(&ph).Error; err == nil {
					dp.Pharmacy = &ph
				}
			}
		}
	}

	return m, nil
}

// GetPharmacyDoctorExportListSearch 获取药房关联医生列表-导出
func (r *DoctorPharmacyDao) GetPharmacyDoctorExportListSearch(req requests.PharmacyDoctorExportList) (m []*model.DoctorPharmacy, err error) {
	query := global.Db.Model(&model.DoctorPharmacy{}).
		Where("gdxz_doctor_pharmacy.status = 1")

	// 药房id过滤
	if req.PharmacyId != "" {
		pharmacyId, _ := strconv.ParseInt(req.PharmacyId, 10, 64)
		if pharmacyId >= 0 {
			query = query.Where("gdxz_doctor_pharmacy.pharmacy_id = ?", pharmacyId)
		}
	}

	// 1:当前搜索数据
	if req.Type == 1 {
		needJoinDoctor := req.DoctorName != "" || req.Mobile != ""
		if needJoinDoctor {
			query = query.Joins("JOIN gdxz_user_doctor ON gdxz_user_doctor.doctor_id = gdxz_doctor_pharmacy.doctor_id")
			if req.DoctorName != "" {
				query = query.Where("gdxz_user_doctor.user_name LIKE ?", "%"+req.DoctorName+"%")
			}
			if req.Mobile != "" {
				query = query.Joins("JOIN gdxz_user ON gdxz_user.user_id = gdxz_user_doctor.user_id").
					Where("gdxz_user.mobile LIKE ?", "%"+req.Mobile+"%")
			}
		}
	}

	// 2:当前选中数据
	if req.Type == 2 {
		if req.Id == "" {
			return nil, errors.New("未提供需导出数据编号")
		}
		id := strings.Split(req.Id, ",")
		query = query.Where("gdxz_doctor_pharmacy.doctor_pharmacy_id IN (?) OR gdxz_doctor_pharmacy.doctor_id IN (?)", id, id)
	}

	// 3:全部数据（无额外搜索条件）

	err = query.Order("gdxz_doctor_pharmacy.is_default desc, gdxz_doctor_pharmacy.created_at desc").
		Preload("Pharmacy").
		Preload("UserDoctor.User").
		Preload("UserDoctor.Hospital").
		Find(&m).Error
	if err != nil {
		return nil, err
	}

	// GORM 在外键为 0 时会自动跳过 Preload，因此针对未能预加载的记录进行兜底补全
	var zeroPharmacy *model.Pharmacy
	for _, dp := range m {
		if dp.Pharmacy == nil {
			if dp.PharmacyId == 0 {
				if zeroPharmacy == nil {
					var ph model.Pharmacy
					if err := global.Db.Where("pharmacy_id = 0").First(&ph).Error; err == nil {
						zeroPharmacy = &ph
					}
				}
				dp.Pharmacy = zeroPharmacy
			} else {
				var ph model.Pharmacy
				if err := global.Db.Where("pharmacy_id = ?", dp.PharmacyId).First(&ph).Error; err == nil {
					dp.Pharmacy = &ph
				}
			}
		}
	}

	return m, nil
}