import React from 'react';
import { Encounter } from 'fhir/r4';
import { FhirCodeableConceptDisplay } from '../../utils/fhirCodeableConcept';
import { FhirCodeDisplay } from '../../utils/fhirCode';
import { FhirPeriodDisplay } from '../../utils/fhirPeriod';
import { FhirReferenceDisplay } from '../../utils/fhirReference';
import { renderField } from '../../utils/renderField';
import { Dialog, DialogContent, DialogHeader, DialogTitle } from '../ui/dialog';
import { Card, CardHeader, CardContent } from '../ui/card';
import { FieldHistoryProvider, FieldHistoryScope } from '../FieldHistory/FieldHistoryProvider';
import { ShowFieldHistoryToggle } from '../FieldHistory/ShowFieldHistoryToggle';
import { PrintRecord } from '../ui/print-record';
import { RecordEditedMarker } from '../FieldHistory/RecordEditedMarker';
import { FieldGroup, Section } from '../FieldHistory/FieldLayout';
import { EncounterVisitNote } from './EncounterVisitNote';
import type { VisitNoteSource } from './useEncounterVisitNote';

interface FhirEncounterDisplayProps {
    encounter: Encounter;
    isOpen: boolean;
    onClose: () => void;
    title?: string;
    /** Omitted where the encounter is a past version or a bare reference: no note is shown. */
    visitNote?: VisitNoteSource;
}

export const FhirEncounterDisplay: React.FC<FhirEncounterDisplayProps> = ({
                                                                              encounter,
                                                                              isOpen,
                                                                              onClose,
                                                                              title = "Encounter Details",
                                                                              visitNote,
                                                                          }) => {
    const renderParticipants = () => {
        if (!encounter.participant || encounter.participant.length === 0) {
            return (
                <p className="text-sm italic text-muted-foreground">
                    No participants recorded
                </p>
            );
        }

        return (
            <div className="flex flex-col gap-4">
                {encounter.participant.map((participant, index) => (
                    <Card key={index}>
                        <CardHeader>
                            <h5 className="text-sm font-semibold">Participant {index + 1}</h5>
                        </CardHeader>
                        <CardContent>
                            <div className="bg-surface">
                                <FieldGroup>
                                    {renderField('Role of participant in encounter', <FhirCodeableConceptDisplay code={participant.type} multiple={true} />)}
                                    {renderField('List of participants involved in the encounter', <FhirReferenceDisplay references={participant.individual} />)}
                                    {renderField('Period of time during the encounter that the participant participated', <FhirPeriodDisplay period={participant.period} />)}
                                </FieldGroup>
                            </div>
                        </CardContent>
                    </Card>
                ))}
            </div>
        );
    };

    const renderDiagnoses = () => {
        if (!encounter.diagnosis || encounter.diagnosis.length === 0) {
            return (
                <p className="text-sm italic text-muted-foreground">
                    No diagnoses recorded
                </p>
            );
        }

        return (
            <div className="flex flex-col gap-4">
                {encounter.diagnosis.map((diagnosis, index) => (
                    <Card key={index}>
                        <CardHeader>
                            <h5 className="text-sm font-semibold">Diagnosis {index + 1}</h5>
                        </CardHeader>
                        <CardContent>
                            <div className="bg-surface">
                                <FieldGroup>
                                    {renderField('The diagnosis or procedure relevant to the encounter', <FhirReferenceDisplay references={diagnosis.condition} />)}
                                    {renderField('Role that this diagnosis has within the encounter', <FhirCodeableConceptDisplay code={diagnosis.use} />)}
                                    {renderField('Ranking of the diagnosis (for each role type)', diagnosis.rank?.toString())}
                                </FieldGroup>
                            </div>
                        </CardContent>
                    </Card>
                ))}
            </div>
        );
    };

    const renderLocations = () => {
        if (!encounter.location || encounter.location.length === 0) {
            return (
                <p className="text-sm italic text-muted-foreground">
                    No locations recorded
                </p>
            );
        }

        return (
            <div className="flex flex-col gap-4">
                {encounter.location.map((location, index) => (
                    <Card key={index}>
                        <CardHeader>
                            <h5 className="text-sm font-semibold">Location {index + 1}</h5>
                        </CardHeader>
                        <CardContent>
                            <div className="bg-surface">
                                <FieldGroup>
                                    {renderField('Location the encounter takes place', <FhirReferenceDisplay references={location.location} />)}
                                    {renderField('Status', <FhirCodeDisplay code={location.status} />)}
                                    {renderField('The physical type of the location', <FhirCodeableConceptDisplay code={location.physicalType} />)}
                                    {renderField('Time period during which the patient was present at the location', <FhirPeriodDisplay period={location.period} />)}
                                </FieldGroup>
                            </div>
                        </CardContent>
                    </Card>
                ))}
            </div>
        );
    };

    const renderStatusHistory = () => {
        if (!encounter.statusHistory || encounter.statusHistory.length === 0) {
            return (
                <p className="text-sm italic text-muted-foreground">
                    No status history recorded
                </p>
            );
        }

        return (
            <div className="flex flex-col gap-4">
                {encounter.statusHistory.map((statusHistory, index) => (
                    <Card key={index}>
                        <CardHeader>
                            <h5 className="text-sm font-semibold">Status History {index + 1}</h5>
                        </CardHeader>
                        <CardContent>
                            <div className="bg-surface">
                                <FieldGroup>
                                    {renderField('Status', <FhirCodeDisplay code={statusHistory.status} />)}
                                    {renderField('The time that the episode was in the specified status', <FhirPeriodDisplay period={statusHistory.period} />)}
                                </FieldGroup>
                            </div>
                        </CardContent>
                    </Card>
                ))}
            </div>
        );
    };

    const renderClassHistory = () => {
        if (!encounter.classHistory || encounter.classHistory.length === 0) {
            return (
                <p className="text-sm italic text-muted-foreground">
                    No class history recorded
                </p>
            );
        }

        return (
            <div className="flex flex-col gap-4">
                {encounter.classHistory.map((classHistory, index) => (
                    <Card key={index}>
                        <CardHeader>
                            <h5 className="text-sm font-semibold">Class History {index + 1}</h5>
                        </CardHeader>
                        <CardContent>
                            <div className="bg-surface">
                                <FieldGroup>
                                    {renderField('Class', <FhirCodeableConceptDisplay code={{ coding: [classHistory.class]}} />)}
                                    {renderField('The time that the episode was in the specified class', <FhirPeriodDisplay period={classHistory.period} />)}
                                </FieldGroup>
                            </div>
                        </CardContent>
                    </Card>
                ))}
            </div>
        );
    };

    if (!isOpen) return null;

    return (
        <Dialog open={isOpen} onOpenChange={(next) => { if (!next) onClose(); }}>
            <DialogContent className="max-h-[90vh] w-full max-w-3xl overflow-y-auto" data-print-region="true">
                <FieldHistoryProvider>
                    <FieldHistoryScope resourceType="Encounter" id={encounter.id}>
                        <DialogHeader className="flex-row items-center justify-between gap-4 text-left sm:text-left">
                            <div className="flex items-center gap-2">
                                <DialogTitle>{title}</DialogTitle>
                                <RecordEditedMarker resource={encounter} />
                            </div>
                            <ShowFieldHistoryToggle />
            <PrintRecord/>
                        </DialogHeader>

                        <FieldGroup>
                            {renderField('Status', <FhirCodeDisplay code={encounter.status} />, 'status')}
                            {renderField('Classification of patient encounter', <FhirCodeableConceptDisplay code={encounter.class} />)}
                            {renderField('Specific type of encounter', <FhirCodeableConceptDisplay code={encounter.type} multiple={true} />)}
                            {renderField('Specific type of service', <FhirCodeableConceptDisplay code={encounter.serviceType} />, 'serviceType.coding[0].display')}
                            {renderField('Priority', <FhirCodeableConceptDisplay code={encounter.priority} />, 'priority.coding[0].display')}
                            {renderField('The start and end time of the encounter', <FhirPeriodDisplay period={encounter.period} />)}
                            {renderField('The appointment that scheduled this encounter', <FhirReferenceDisplay references={encounter.appointment} />)}
                            {renderField('Another Encounter this encounter is part of', <FhirReferenceDisplay references={encounter.partOf} />)}
                            {renderField('Episode(s) of care that this encounter should be recorded against', <FhirReferenceDisplay references={encounter.episodeOfCare} />)}
                        </FieldGroup>

                        {visitNote && <EncounterVisitNote encounter={encounter} source={visitNote} />}

                        <Section title="Reason">
                            <FieldGroup>
                                {renderField('Reason the encounter takes place', <FhirCodeableConceptDisplay code={encounter.reasonCode} multiple={true} />)}
                                {renderField('Reason the encounter takes place (reference)', <FhirReferenceDisplay references={encounter.reasonReference} />)}
                            </FieldGroup>
                        </Section>

                        <Section title="Participants">
                            {renderParticipants()}
                        </Section>

                        <Section title="Diagnosis">
                            {renderDiagnoses()}
                        </Section>

                        <Section title="Locations">
                            {renderLocations()}
                        </Section>

                        <Section title="Hospitalization">
                            <FieldGroup>
                                {renderField('The location/organization from which the patient came before admission', <FhirReferenceDisplay references={encounter.hospitalization?.origin} />)}
                                {renderField('From where patient was admitted (physician referral, transfer)', <FhirCodeableConceptDisplay code={encounter.hospitalization?.admitSource} />, 'hospitalization.admitSource.coding[0].display')}
                                {renderField('The type of hospital re-admission that has occurred', <FhirCodeableConceptDisplay code={encounter.hospitalization?.reAdmission} />, 'hospitalization.reAdmission.coding[0].display')}
                                {renderField('Diet preferences reported by the patient', <FhirCodeableConceptDisplay code={encounter.hospitalization?.dietPreference} multiple={true} />)}
                                {renderField('Special courtesies (VIP, board member)', <FhirCodeableConceptDisplay code={encounter.hospitalization?.specialCourtesy} multiple={true} />)}
                                {renderField('Wheelchair, translator, stretcher, etc.', <FhirCodeableConceptDisplay code={encounter.hospitalization?.specialArrangement} multiple={true} />)}
                                {renderField('Location/organization to which the patient is discharged', <FhirReferenceDisplay references={encounter.hospitalization?.destination} />)}
                                {renderField('Category or kind of location after discharge', <FhirCodeableConceptDisplay code={encounter.hospitalization?.dischargeDisposition} />, 'hospitalization.dischargeDisposition.coding[0].display')}
                            </FieldGroup>
                        </Section>

                        <Section title="Status History">
                            {renderStatusHistory()}
                        </Section>

                        <Section title="Class History">
                            {renderClassHistory()}
                        </Section>
                    </FieldHistoryScope>
                </FieldHistoryProvider>
            </DialogContent>
        </Dialog>
    );
};

export default FhirEncounterDisplay;