package encounter

import (
	ecore "simrs-vx/internal/domain/base-entities/core"
	ea "simrs-vx/internal/domain/main-entities/appointment"
	ed "simrs-vx/internal/domain/main-entities/doctor"
	ep "simrs-vx/internal/domain/main-entities/patient"
	es "simrs-vx/internal/domain/main-entities/specialist"
	ess "simrs-vx/internal/domain/main-entities/subspecialist"
	eu "simrs-vx/internal/domain/main-entities/unit"
	erc "simrs-vx/internal/domain/references/common"
	ere "simrs-vx/internal/domain/references/encounter"
	"time"
)

type CreateDto struct {
	Patient_Id            *uint                  `json:"patient_id"`
	RegisteredAt          *time.Time             `json:"registeredAt"`
	Class_Code            ere.EncounterClassCode `json:"class_code" validate:"maxLength=10"`
	SubClass_Code         *string                `json:"subClass_code" validate:"maxLength=10"` // for sub
	Infra_Id              *uint16                `json:"infra_id"`                              // for inpatient
	Unit_Id               *uint                  `json:"unit_id"`
	Specialist_Id         *uint16                `json:"specialist_id"`
	Subspecialist_Id      *uint16                `json:"subspecialist_id"`
	VisitDate             time.Time              `json:"visitDate"`
	Appointment_Doctor_Id *uint                  `json:"appointment_doctor_id"`
	Responsible_Doctor_Id *uint                  `json:"responsible_doctor_id"`
	RefSource_Name        *string                `json:"refSource_name" validate:"maxLength=100"`
	Appointment_Id        *uint                  `json:"appointment_id"`
}

type ReadListDto struct {
	FilterDto
	Includes string   `json:"includes"`
	Preloads []string `json:"-"`
}

type FilterDto struct {
	Patient_Id                *uint                   `json:"patient_id"`
	Patient                   *ep.Patient             `json:"patient,omitempty"`
	RegisteredAt              *time.Time              `json:"registeredAt"`
	Class_Code                ere.EncounterClassCode  `json:"class_code" validate:"maxLength=10"`
	Unit_Id                   *uint                   `json:"unit_id"`
	Specialist_Id             *uint16                 `json:"specialist_id"`
	Subspecialist_Id          *uint16                 `json:"subspecialist_id"`
	VisitDate                 time.Time               `json:"visitDate"`
	Appoinment_Doctor_Id      *uint                   `json:"appointment_doctor_id"`
	Responsible_Doctor_Id     *uint                   `json:"responsible_doctor_id"`
	DischargeMethod_Code      ere.DischargeMethodCode `json:"dischargeMethod_code" validate:"maxLength=10"`
	RefSource_Name            *string                 `json:"refSource_name" validate:"maxLength=100"`
	Appointment_Id            *uint                   `json:"appointment_id"`
	EarlyEducation            *string                 `json:"earlyEducation"`
	MedicalDischargeEducation *string                 `json:"medicalDischargeEducation"`
	AdmDischargeEducation     *string                 `json:"admDischargeEducation"`
	DischargeReason           *string                 `json:"dischargeReason"`

	Page         int `json:"page"`
	PageSize     int `json:"page_size"`
	NoPagination int `json:"no_pagination"`
}

type ReadDetailDto struct {
	Id uint16 `json:"id"`
}

type UpdateDto struct {
	Id uint16 `json:"id"`
	CreateDto
}

type DeleteDto struct {
	Id uint16 `json:"id"`
}

type MetaDto struct {
	PageNumber int `json:"page_number"`
	PageSize   int `json:"page_size"`
	Count      int `json:"count"`
}
type DischargeDto struct {
	Id                        uint                     `json:"id"`
	DischargeMethod_Code      *ere.DischargeMethodCode `json:"dischargeMethod_code" validate:"maxLength=10"`
	EarlyEducation            *string                  `json:"earlyEducation"`
	MedicalDischargeEducation *string                  `json:"medicalDischargeEducation"`
	AdmDischargeEducation     *string                  `json:"admDischargeEducation"`
	DischargeReason           *string                  `json:"dischargeReason"`
}

type ResponseDto struct {
	ecore.Main
	Patient_Id                *uint                    `json:"patient_id"`
	Patient                   *ep.Patient              `json:"patient,omitempty"`
	RegisteredAt              *time.Time               `json:"registeredAt"`
	Class_Code                ere.EncounterClassCode   `json:"class_code"`
	Unit_Id                   *uint                    `json:"unit_id"`
	Specialist_Id             *uint16                  `json:"specialist_id"`
	Specialist                *es.Specialist           `json:"specialist,omitempty"`
	Subspecialist_Id          *uint16                  `json:"subspecialist_id"`
	Subspecialist             *ess.Subspecialist       `json:"subspecialist,omitempty"`
	Unit                      *eu.Unit                 `json:"unit,omitempty"`
	VisitDate                 time.Time                `json:"visitDate"`
	Appointment_Doctor_Id     *uint                    `json:"assignment_doctor_id"`
	Appointment_Doctor        *ed.Doctor               `json:"assignment_doctor,omitempty"`
	Responsible_Doctor_Id     *uint                    `json:"responsible_doctor_id"`
	Responsible_Doctor        *ed.Doctor               `json:"responsible_doctor,omitempty"`
	DischargeMethod_Code      *ere.DischargeMethodCode `json:"dischargeMethod_code"`
	RefSource_Name            *string                  `json:"refSource_name"`
	Appointment_Id            *uint                    `json:"appointment_id"`
	Appointment               *ea.Appointment          `json:"appointment,omitempty"`
	EarlyEducation            *string                  `json:"earlyEducation"`
	MedicalDischargeEducation *string                  `json:"medicalDischargeEducation"`
	AdmDischargeEducation     *string                  `json:"admDischargeEducation"`
	DischargeReason           *string                  `json:"dischargeReason"`
	Status_Code               erc.DataStatusCode       `json:"status_code"`
}

func (d Encounter) ToResponse() ResponseDto {
	resp := ResponseDto{
		Patient_Id:                d.Patient_Id,
		Patient:                   d.Patient,
		RegisteredAt:              d.RegisteredAt,
		Class_Code:                d.Class_Code,
		Unit_Id:                   d.Unit_Id,
		Unit:                      d.Unit,
		Specialist_Id:             d.Specialist_Id,
		Specialist:                d.Specialist,
		Subspecialist_Id:          d.Subspecialist_Id,
		Subspecialist:             d.Subspecialist,
		VisitDate:                 d.VisitDate,
		Appointment_Doctor_Id:     d.Appointment_Doctor_Id,
		Appointment_Doctor:        d.Appointment_Doctor,
		Responsible_Doctor_Id:     d.Responsible_Doctor_Id,
		Responsible_Doctor:        d.Responsible_Doctor,
		DischargeMethod_Code:      d.DischargeMethod_Code,
		RefSource_Name:            d.RefSource_Name,
		Appointment_Id:            d.Appointment_Id,
		Appointment:               d.Appointment,
		EarlyEducation:            d.EarlyEducation,
		MedicalDischargeEducation: d.MedicalDischargeEducation,
		AdmDischargeEducation:     d.AdmDischargeEducation,
		DischargeReason:           d.DischargeReason,
		Status_Code:               d.Status_Code,
	}
	resp.Main = d.Main
	return resp
}

func ToResponseList(data []Encounter) []ResponseDto {
	resp := make([]ResponseDto, len(data))
	for i, u := range data {
		resp[i] = u.ToResponse()
	}
	return resp
}