package encounter

import (
	"errors"
	"fmt"
	uir "simrs-vx/internal/use-case/main-use-case/internal-reference"
	"strconv"

	ea "simrs-vx/internal/domain/main-entities/ambulatory"
	ec "simrs-vx/internal/domain/main-entities/chemo"
	edc "simrs-vx/internal/domain/main-entities/death-cause"
	ee "simrs-vx/internal/domain/main-entities/emergency"
	eem "simrs-vx/internal/domain/main-entities/employee"
	e "simrs-vx/internal/domain/main-entities/encounter"
	ei "simrs-vx/internal/domain/main-entities/inpatient"

	ua "simrs-vx/internal/use-case/main-use-case/ambulatory"
	uc "simrs-vx/internal/use-case/main-use-case/chemo"
	udc "simrs-vx/internal/use-case/main-use-case/death-cause"
	ue "simrs-vx/internal/use-case/main-use-case/emergency"
	uem "simrs-vx/internal/use-case/main-use-case/employee"
	ui "simrs-vx/internal/use-case/main-use-case/inpatient"

	erc "simrs-vx/internal/domain/references/common"
	ere "simrs-vx/internal/domain/references/encounter"

	pl "simrs-vx/pkg/logger"
	pu "simrs-vx/pkg/use-case-helper"

	dg "github.com/karincake/apem/db-gorm-pg"
	d "github.com/karincake/dodol"

	"gorm.io/gorm"
)

const source = "encounter"

func Create(input e.CreateDto) (*d.Data, error) {
	data := e.Encounter{}

	event := pl.Event{
		Feature: "Create",
		Source:  source,
	}

	// Start log
	pl.SetLogInfo(&event, input, "started", "create")

	err := dg.I.Transaction(func(tx *gorm.DB) error {
		mwRunner := newMiddlewareRunner(&event, tx)
		mwRunner.setMwType(pu.MWTPre)
		// Run pre-middleware
		if err := mwRunner.RunCreateMiddleware(createPreMw, &input, &data); err != nil {
			return err
		}

		if emp, err := uem.ReadDetailData(eem.ReadDetailDto{User_Id: &input.AuthInfo.User_Id}, &event, tx); err != nil {
			return err
		} else {
			input.Adm_Employee_Id = &emp.Id
		}

		if resData, err := CreateData(input, &event, tx); err != nil {
			return err
		} else {
			data = *resData
		}

		switch input.Class_Code {
		case ere.ECAmbulatory:
			subCode, err := ua.CheckClassCode(input.SubClass_Code)
			if err != nil {
				return err
			}
			ambCreate := ea.CreateDto{
				Encounter_Id: &data.Id,
				Class_Code:   subCode,
			}
			_, err = ua.CreateData(ambCreate, &event, tx)
			if err != nil {
				return err
			}

			if subCode == ere.ACCCac || subCode == ere.ACCCad {
				chemoCreate := ec.CreateDto{
					Encounter_Id: &data.Id,
					Status_Code:  erc.DVCNew,
					SrcUnit_Id:   input.Unit_Id,
				}

				_, err = uc.CreateData(chemoCreate, &event, tx)
				if err != nil {
					return err
				}
			}
		case ere.ECEmergency:
			subCode, err := ue.CheckClassCode(input.SubClass_Code)
			if err != nil {
				return err
			}
			emerCreate := ee.CreateDto{
				Encounter_Id: &data.Id,
				Class_Code:   subCode,
			}
			_, err = ue.CreateData(emerCreate, &event, tx)
			if err != nil {
				return err
			}
		case ere.ECInpatient:
			subCode, err := ui.CheckClassCode(input.SubClass_Code)
			if err != nil {
				return err
			}
			inpCreate := ei.CreateDto{
				Encounter_Id: &data.Id,
				Class_Code:   subCode,
				Infra_Id:     input.Infra_Id,
			}
			_, err = ui.CreateData(inpCreate, &event, tx)
			if err != nil {
				return err
			}
		default:
			return errors.New("invalid encounter class code")
		}

		mwRunner.setMwType(pu.MWTPost)
		// Run post-middleware
		if err := mwRunner.RunCreateMiddleware(createPostMw, &input, &data); err != nil {
			return err
		}

		pl.SetLogInfo(&event, nil, "complete")

		return nil
	})

	if err != nil {
		return nil, err
	}

	return &d.Data{
		Meta: d.II{
			"source":    source,
			"structure": "single-data",
			"status":    "created",
		},
		Data: data.ToResponse(),
	}, nil
}

func ReadList(input e.ReadListDto) (*d.Data, error) {
	var data *e.Encounter
	var dataList []e.Encounter
	var metaList *e.MetaDto
	var err error

	event := pl.Event{
		Feature: "ReadList",
		Source:  source,
	}

	// Start log
	pl.SetLogInfo(&event, input, "started", "readList")

	err = dg.I.Transaction(func(tx *gorm.DB) error {
		mwRunner := newMiddlewareRunner(&event, tx)
		mwRunner.setMwType(pu.MWTPre)
		// Run pre-middleware
		if err := mwRunner.RunReadListMiddleware(readListPreMw, &input, data); err != nil {
			return err
		}

		if dataList, metaList, err = ReadListData(input, &event, tx); err != nil {
			return err
		}

		mwRunner.setMwType(pu.MWTPost)
		// Run post-middleware
		if err := mwRunner.RunReadListMiddleware(readListPostMw, &input, data); err != nil {
			return err
		}

		return nil
	})

	if err != nil {
		return nil, err
	}

	return &d.Data{
		Meta: d.IS{
			"source":              source,
			"structure":           "list-data",
			"status":              "fetched",
			"page_number":         strconv.Itoa(metaList.PageNumber),
			"page_size":           strconv.Itoa(metaList.PageSize),
			"record_totalCount":   strconv.Itoa(metaList.Count),
			"record_currentCount": strconv.Itoa(len(dataList)),
		},
		Data: e.ToResponseList(dataList),
	}, nil
}

func ReadDetail(input e.ReadDetailDto) (*d.Data, error) {
	var data *e.Encounter
	var err error

	event := pl.Event{
		Feature: "ReadDetail",
		Source:  source,
	}

	// Start log
	pl.SetLogInfo(&event, input, "started", "readDetail")

	err = dg.I.Transaction(func(tx *gorm.DB) error {
		mwRunner := newMiddlewareRunner(&event, tx)
		mwRunner.setMwType(pu.MWTPre)
		// Run pre-middleware
		if err := mwRunner.RunReadDetailMiddleware(readDetailPreMw, &input, data); err != nil {
			return err
		}

		if data, err = ReadDetailData(input, &event, tx); err != nil {
			return err
		}

		mwRunner.setMwType(pu.MWTPost)
		// Run post-middleware
		if err := mwRunner.RunReadDetailMiddleware(readDetailPostMw, &input, data); err != nil {
			return err
		}

		return nil
	})

	if err != nil {
		return nil, err
	}

	return &d.Data{
		Meta: d.IS{
			"source":    source,
			"structure": "single-data",
			"status":    "fetched",
		},
		Data: data.ToResponse(),
	}, nil
}

func Update(input e.UpdateDto) (*d.Data, error) {
	rdDto := e.ReadDetailDto{Id: uint16(input.Id)}
	var data *e.Encounter
	var err error

	event := pl.Event{
		Feature: "Update",
		Source:  source,
	}

	// Start log
	pl.SetLogInfo(&event, input, "started", "update")

	err = dg.I.Transaction(func(tx *gorm.DB) error {
		pl.SetLogInfo(&event, rdDto, "started", "DBReadDetail")
		if data, err = ReadDetailData(rdDto, &event, tx); err != nil {
			return err
		}

		if data.IsDone() {
			event.Status = "failed"
			event.ErrInfo = pl.ErrorInfo{
				Code:   "data-state-mismatch",
				Detail: "encounter is done",
				Raw:    errors.New("encounter is done"),
			}
			return pl.SetLogError(&event, input)
		}

		mwRunner := newMiddlewareRunner(&event, tx)
		mwRunner.setMwType(pu.MWTPre)
		// Run pre-middleware
		if err := mwRunner.RunUpdateMiddleware(readDetailPreMw, &rdDto, data); err != nil {
			return err
		}

		if err := UpdateData(input, data, &event, tx); err != nil {
			return err
		}

		pl.SetLogInfo(&event, nil, "complete")

		mwRunner.setMwType(pu.MWTPost)
		// Run post-middleware
		if err := mwRunner.RunUpdateMiddleware(readDetailPostMw, &rdDto, data); err != nil {
			return err
		}

		return nil
	})

	if err != nil {
		return nil, err
	}

	return &d.Data{
		Meta: d.IS{
			"source":    source,
			"structure": "single-data",
			"status":    "updated",
		},
		Data: data.ToResponse(),
	}, nil

}

func Delete(input e.DeleteDto) (*d.Data, error) {
	rdDto := e.ReadDetailDto{Id: uint16(input.Id)}
	var data *e.Encounter
	var err error

	event := pl.Event{
		Feature: "Delete",
		Source:  source,
	}

	// Start log
	pl.SetLogInfo(&event, input, "started", "delete")

	err = dg.I.Transaction(func(tx *gorm.DB) error {
		pl.SetLogInfo(&event, rdDto, "started", "DBReadDetail")
		if data, err = ReadDetailData(rdDto, &event, tx); err != nil {
			return err
		}

		mwRunner := newMiddlewareRunner(&event, tx)
		mwRunner.setMwType(pu.MWTPre)
		// Run pre-middleware
		if err := mwRunner.RunDeleteMiddleware(readDetailPreMw, &rdDto, data); err != nil {
			return err
		}

		if err := DeleteData(data, &event, tx); err != nil {
			return err
		}

		mwRunner.setMwType(pu.MWTPost)
		// Run post-middleware
		if err := mwRunner.RunDeleteMiddleware(readDetailPostMw, &rdDto, data); err != nil {
			return err
		}

		return nil
	})

	if err != nil {
		return nil, err
	}

	return &d.Data{
		Meta: d.IS{
			"source":    source,
			"structure": "single-data",
			"status":    "deleted",
		},
		Data: data.ToResponse(),
	}, nil

}

func CheckOut(input e.DischargeDto) (*d.Data, error) {
	rdDto := e.ReadDetailDto{Id: uint16(input.Id)}
	var data *e.Encounter
	var err error

	event := pl.Event{
		Feature: "CheckOut",
		Source:  source,
	}

	// Start log
	pl.SetLogInfo(&event, input, "started", "checkOut")

	err = dg.I.Transaction(func(tx *gorm.DB) error {
		pl.SetLogInfo(&event, rdDto, "started", "DBReadDetail")
		if data, err = ReadDetailData(rdDto, &event, tx); err != nil {
			return err
		}

		if data.IsDone() {
			event.Status = "failed"
			event.ErrInfo = pl.ErrorInfo{
				Code:   "data-state-mismatch",
				Detail: "encounter is done",
				Raw:    errors.New("encounter is done"),
			}
			return pl.SetLogError(&event, input)
		}

		if err := checkSoapiByDocExists(data.Id, &event, tx); err != nil {
			return err
		}

		if err := updateDischargeData(input, data, &event, tx); err != nil {
			return err
		}

		if err := checkNewOrdersExist(data.Id, &event, tx); err != nil {
			return err
		}

		if err := createMedication(data.Id, &event, tx); err != nil {
			if !pu.IsDataNotFoundError(err) {
				return err
			}
		}

		switch *input.Discharge_Method_Code {
		case ere.DMCDeath:
			// insert data death-cause
			if _, err := udc.CreateData(edc.CreateDto{
				Encounter_Id: &input.Id,
				Value:        input.DeathCause,
			}, &event, tx); err != nil {
				return err
			}

		case ere.DMCConsulPoly, ere.DMCConsulExecutive:
			unitIDs := make(map[uint16]struct{})
			doctorIDs := make(map[uint]struct{})

			for _, ref := range *input.InternalReferences {
				if ref.Unit_Id != nil {
					unitIDs[*ref.Unit_Id] = struct{}{}
				}
				if ref.Doctor_Id != nil {
					doctorIDs[*ref.Doctor_Id] = struct{}{}
				}
			}

			// validate unitIds
			if len(unitIDs) > 0 {
				var ids []uint16
				for id := range unitIDs {
					ids = append(ids, id)
				}

				units, err := getUnits(ids, &event, tx)
				if err != nil {
					return fmt.Errorf("failed to fetch units: %w", err)
				}
				if len(units) != len(ids) {
					return fmt.Errorf("unit_id not found")
				}
			}

			// validate doctorIds
			if len(doctorIDs) > 0 {
				var ids []uint
				for id := range doctorIDs {
					ids = append(ids, id)
				}

				doctors, err := getDoctors(ids, &event, tx)
				if err != nil {
					return fmt.Errorf("failed to fetch doctors: %w", err)
				}
				if len(doctors) != len(ids) {
					return fmt.Errorf("doctor_id not found")
				}
			}

			if err := uir.CreateBulkData(*input.InternalReferences, input.Id, &event, tx); err != nil {
				return err
			}
		}

		pl.SetLogInfo(&event, nil, "complete")

		return nil
	})

	if err != nil {
		return nil, err
	}

	return &d.Data{
		Meta: d.IS{
			"source":    source,
			"structure": "single-data",
			"status":    "checkOut",
		},
		Data: data.ToResponse(),
	}, nil
}

func UpdateStatusCode(input e.UpdateStatusDto) (*d.Data, error) {
	rdDto := e.ReadDetailDto{Id: input.Id}
	var data *e.Encounter
	var err error

	event := pl.Event{
		Feature: "Update Status Code",
		Source:  source,
	}

	// Start log
	pl.SetLogInfo(&event, input, "started", "update")

	err = dg.I.Transaction(func(tx *gorm.DB) error {
		pl.SetLogInfo(&event, rdDto, "started", "DBReadDetail")
		if data, err = ReadDetailData(rdDto, &event, tx); err != nil {
			return err
		}

		if data.IsDone() {
			event.Status = "failed"
			event.ErrInfo = pl.ErrorInfo{
				Code:   "data-state-mismatch",
				Detail: "encounter is done",
				Raw:    errors.New("encounter is done"),
			}
			return pl.SetLogError(&event, input)
		}

		mwRunner := newMiddlewareRunner(&event, tx)
		mwRunner.setMwType(pu.MWTPre)
		// Run pre-middleware
		if err := mwRunner.RunUpdateMiddleware(readDetailPreMw, &rdDto, data); err != nil {
			return err
		}

		if err := UpdateStatusData(input, data, &event, tx); err != nil {
			return err
		}

		pl.SetLogInfo(&event, nil, "complete")

		mwRunner.setMwType(pu.MWTPost)
		// Run post-middleware
		if err := mwRunner.RunUpdateMiddleware(readDetailPostMw, &rdDto, data); err != nil {
			return err
		}

		return nil
	})

	if err != nil {
		return nil, err
	}

	return &d.Data{
		Meta: d.IS{
			"source":    source,
			"structure": "single-data",
			"status":    "updated",
		},
		Data: data.ToResponse(),
	}, nil

}