package dao

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

type OrderPrescriptionDao struct {
}

// GetById 获取处方-处方id
func (r *OrderPrescriptionDao) GetById(orderPrescriptionId int64) (m *model.OrderPrescription, err error) {
	err = global.Db.First(&m, orderPrescriptionId).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// Delete 删除处方
func (r *OrderPrescriptionDao) Delete(tx *gorm.DB, maps interface{}) error {
	err := tx.Where(maps).Delete(&model.BasicBank{}).Error
	if err != nil {
		return err
	}
	return nil
}

// Edit 修改处方
func (r *OrderPrescriptionDao) Edit(tx *gorm.DB, maps interface{}, data interface{}) error {
	err := tx.Model(&model.BasicBank{}).Where(maps).Updates(data).Error
	if err != nil {
		return err
	}
	return nil
}

// EditById 修改处方-处方id
func (r *OrderPrescriptionDao) EditById(tx *gorm.DB, orderPrescriptionId int64, data interface{}) error {
	err := tx.Model(&model.BasicBank{}).Where("order_prescription_id = ?", orderPrescriptionId).Updates(data).Error
	if err != nil {
		return err
	}
	return nil
}

// GetList 获取处方列表
func (r *OrderPrescriptionDao) GetList(maps interface{}) (m []*model.OrderPrescription, err error) {
	err = global.Db.Where(maps).Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// Add 新增处方
func (r *OrderPrescriptionDao) Add(tx *gorm.DB, model *model.OrderPrescription) (*model.OrderPrescription, error) {
	if err := tx.Create(model).Error; err != nil {
		return nil, err
	}
	return model, nil
}

// AddByMap 新增处方-map
func (r *OrderPrescriptionDao) AddByMap(tx *gorm.DB, data map[string]interface{}) (*model.OrderPrescription, error) {
	orderPrescription := &model.OrderPrescription{}
	if err := tx.Model(&model.OrderPrescription{}).Create(data).Error; err != nil {
		return nil, err
	}
	return orderPrescription, nil
}

// GetOrderPrescriptionPageSearch 获取处方列表-分页
func (r *OrderPrescriptionDao) GetOrderPrescriptionPageSearch(req requests.GetOrderPrescriptionPage, page, pageSize int) (m []*model.OrderPrescription, total int64, err error) {
	var totalRecords int64

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

	// 患者表
	query = query.Preload("UserPatient", func(db *gorm.DB) *gorm.DB {
		return db.Omit("open_id", "union_id", "wx_session_key")
	})

	// 用户表
	query = query.Preload("UserPatient.User", func(db *gorm.DB) *gorm.DB {
		return db.Omit("user_password", "salt")
	})

	// 药师表
	query = query.Preload("UserPharmacist", func(db *gorm.DB) *gorm.DB {
		return db.Omit("open_id", "union_id", "wx_session_key")
	})

	// 处方关联疾病表
	query = query.Preload("OrderPrescriptionIcd")

	// 患者家庭成员表
	query = query.Preload("PatientFamily")

	// 处方编号
	if req.PrescriptionCode != "" {
		query = query.Where("prescription_code = ?", req.PrescriptionCode)
	}

	// 医生名称
	if req.DoctorName != "" {
		query = query.Where("doctor_name LIKE ?", "%"+req.DoctorName+"%")
	}

	// 患者姓名-就诊人
	if req.PatientName != "" {
		query = query.Where("patient_name LIKE ?", "%"+req.PatientName+"%")
	}

	// 手机号-医生/患者/就诊人
	if req.Mobile != "" {
		// 就诊人
		patientFamilySubQuery := global.Db.Model(&model.PatientFamily{}).
			Select("family_id").
			Where("mobile = ?", req.Mobile)

		// 患者
		patientUserSubQuery := global.Db.Model(&model.User{}).
			Select("user_id").
			Where("mobile = ?", req.Mobile)

		patientSubQuery := global.Db.Model(&model.UserPatient{}).
			Select("patient_id").
			Where(gorm.Expr("user_id IN (?)", patientUserSubQuery))

		// 医生
		doctorUserSubQuery := global.Db.Model(&model.User{}).
			Select("user_id").
			Where("mobile = ?", req.Mobile)

		doctorSubQuery := global.Db.Model(&model.UserDoctor{}).
			Select("doctor_id").
			Where(gorm.Expr("user_id IN (?)", doctorUserSubQuery))

		query = query.Where("patient_id IN (?)", patientSubQuery).Or("doctor_id IN (?)", doctorSubQuery).Or("family_id IN (?)", patientFamilySubQuery)
	}

	// 处方状态
	if req.PrescriptionStatus != nil {
		query = query.Where("prescription_status = ?", req.PrescriptionStatus)
	}

	// 药师审核状态
	if req.PharmacistAuditStatus != nil {
		query = query.Where("pharmacist_audit_status = ?", req.PharmacistAuditStatus)
	}

	// 问诊订单编号
	if req.InquiryNo != "" {
		subQuery := global.Db.Model(&model.OrderInquiry{}).
			Select("order_inquiry_id").
			Where("inquiry_no = ?", req.InquiryNo)
		query = query.Where("order_inquiry_id IN (?)", subQuery)
	}

	// 药品订单编号
	if req.OrderProductNo != "" {
		subQuery := global.Db.Model(&model.OrderInquiry{}).
			Select("order_prescription_id").
			Where("order_product_no = ?", req.OrderProductNo)
		query = query.Where("order_prescription_id IN (?)", subQuery)
	}

	// 药师审核时间
	if req.PharmacistVerifyTime != "" {
		pharmacistVerifyTime := strings.Split(req.PharmacistVerifyTime, "&")
		if len(pharmacistVerifyTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", pharmacistVerifyTime[0])
			endTime, _ := time.Parse("2006-01-02", pharmacistVerifyTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("pharmacist_verify_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 医生开具处方时间
	if req.DoctorCreatedTime != "" {
		doctorCreatedTime := strings.Split(req.DoctorCreatedTime, "&")
		if len(doctorCreatedTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", doctorCreatedTime[0])
			endTime, _ := time.Parse("2006-01-02", doctorCreatedTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("doctor_created_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 处方过期时间
	if req.ExpiredTime != "" {
		expiredTime := strings.Split(req.ExpiredTime, "&")
		if len(expiredTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", expiredTime[0])
			endTime, _ := time.Parse("2006-01-02", expiredTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("expired_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 排序
	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
}

// GetOrderPrescriptionExportListSearch 获取处方列表-导出
func (r *OrderPrescriptionDao) GetOrderPrescriptionExportListSearch(req requests.OrderPrescriptionExportList) (m []*model.OrderPrescription, err error) {
	// 构建查询条件
	query := global.Db.Model(&model.OrderPrescription{})

	// 患者表
	query = query.Preload("UserPatient", func(db *gorm.DB) *gorm.DB {
		return db.Omit("open_id", "union_id", "wx_session_key")
	})

	// 用户表
	query = query.Preload("UserPatient.User", func(db *gorm.DB) *gorm.DB {
		return db.Omit("user_password", "salt")
	})

	// 医生表
	query = query.Preload("UserDoctor", func(db *gorm.DB) *gorm.DB {
		return db.Omit("open_id", "union_id", "wx_session_key")
	})

	// 药师表
	query = query.Preload("UserPharmacist", func(db *gorm.DB) *gorm.DB {
		return db.Omit("open_id", "union_id", "wx_session_key")
	})

	// 处方关联疾病表
	query = query.Preload("OrderPrescriptionIcd")

	// 患者家庭成员表
	query = query.Preload("PatientFamily")

	// 当前搜索数据
	if req.Type == 1 {
		// 处方编号
		if req.PrescriptionCode != "" {
			query = query.Where("prescription_code = ?", req.PrescriptionCode)
		}

		// 医生名称
		if req.DoctorName != "" {
			query = query.Where("doctor_name LIKE ?", "%"+req.DoctorName+"%")
		}

		// 患者姓名-就诊人
		if req.PatientName != "" {
			query = query.Where("patient_name LIKE ?", "%"+req.PatientName+"%")
		}

		// 手机号-医生/患者/就诊人
		if req.Mobile != "" {
			// 就诊人
			patientFamilySubQuery := global.Db.Model(&model.PatientFamily{}).
				Select("family_id").
				Where("mobile = ?", req.Mobile)

			// 患者
			patientUserSubQuery := global.Db.Model(&model.User{}).
				Select("user_id").
				Where("mobile = ?", req.Mobile)

			patientSubQuery := global.Db.Model(&model.UserPatient{}).
				Select("patient_id").
				Where(gorm.Expr("user_id IN (?)", patientUserSubQuery))

			// 医生
			doctorUserSubQuery := global.Db.Model(&model.User{}).
				Select("user_id").
				Where("mobile = ?", req.Mobile)

			doctorSubQuery := global.Db.Model(&model.UserDoctor{}).
				Select("doctor_id").
				Where(gorm.Expr("user_id IN (?)", doctorUserSubQuery))

			query = query.Where("patient_id IN (?)", patientSubQuery).Or("doctor_id IN (?)", doctorSubQuery).Or("family_id IN (?)", patientFamilySubQuery)
		}

		// 处方状态
		if req.PrescriptionStatus != nil {
			query = query.Where("prescription_status = ?", req.PrescriptionStatus)
		}

		// 药师审核状态
		if req.PharmacistAuditStatus != nil {
			query = query.Where("pharmacist_audit_status = ?", req.PharmacistAuditStatus)
		}

		// 问诊订单编号
		if req.InquiryNo != "" {
			subQuery := global.Db.Model(&model.OrderInquiry{}).
				Select("order_inquiry_id").
				Where("inquiry_no = ?", req.InquiryNo)
			query = query.Where("order_inquiry_id IN (?)", subQuery)
		}

		// 药品订单编号
		if req.OrderProductNo != "" {
			subQuery := global.Db.Model(&model.OrderInquiry{}).
				Select("order_prescription_id").
				Where("order_product_no = ?", req.OrderProductNo)
			query = query.Where("order_prescription_id IN (?)", subQuery)
		}

		// 药师审核时间
		if req.PharmacistVerifyTime != "" {
			pharmacistVerifyTime := strings.Split(req.PharmacistVerifyTime, "&")
			if len(pharmacistVerifyTime) == 2 {
				startTime, _ := time.Parse("2006-01-02", pharmacistVerifyTime[0])
				endTime, _ := time.Parse("2006-01-02", pharmacistVerifyTime[1])

				if startTime == endTime {
					endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
				}

				query = query.Where("pharmacist_verify_time BETWEEN ? AND ?", startTime, endTime)
			}
		}

		// 医生开具处方时间
		if req.DoctorCreatedTime != "" {
			doctorCreatedTime := strings.Split(req.DoctorCreatedTime, "&")
			if len(doctorCreatedTime) == 2 {
				startTime, _ := time.Parse("2006-01-02", doctorCreatedTime[0])
				endTime, _ := time.Parse("2006-01-02", doctorCreatedTime[1])

				if startTime == endTime {
					endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
				}

				query = query.Where("doctor_created_time BETWEEN ? AND ?", startTime, endTime)
			}
		}

		// 处方过期时间
		if req.ExpiredTime != "" {
			expiredTime := strings.Split(req.ExpiredTime, "&")
			if len(expiredTime) == 2 {
				startTime, _ := time.Parse("2006-01-02", expiredTime[0])
				endTime, _ := time.Parse("2006-01-02", expiredTime[1])

				if startTime == endTime {
					endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
				}

				query = query.Where("expired_time BETWEEN ? AND ?", startTime, endTime)
			}
		}
	}

	// 当前选中数据
	if req.Type == 2 {
		if req.Id == "" {
			return nil, errors.New("未提供需导出数据编号")
		}

		id := strings.Split(req.Id, ",")
		query = query.Where("order_prescription_id IN (?)", id)
	}

	// 排序
	query = query.Order("created_at desc")

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

	return m, nil
}