import React from 'react';
import { MedicationAdministration } from 'fhir/r4';
import { FhirCodeableConceptDisplay } from '../../utils/fhirCodeableConcept';
import { FhirCodeDisplay } from '../../utils/fhirCode';
import { FhirPeriodDisplay } from '../../utils/fhirPeriod';
import { FhirQuantityDisplay } from '../../utils/fhirQuantity';
import { FhirRatioDisplay } from '../../utils/fhirRatio';
import { FhirAnnotationDisplay } from '../../utils/fhirAnnotation';
import { formatDateTime } from '../../utils/dateUtils';
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';

interface FhirMedicationAdministrationDisplayProps {
    medicationAdministration: MedicationAdministration;
    isOpen: boolean;
    onClose: () => void;
    title?: string;
}

export const FhirMedicationAdministrationDisplay: React.FC<FhirMedicationAdministrationDisplayProps> = ({
                                                                                                            medicationAdministration,
                                                                                                            isOpen,
                                                                                                            onClose,
                                                                                                            title = "Medication Administration Details"
                                                                                                        }) => {
    const renderEffectiveValue = () => {
        if (medicationAdministration.effectiveDateTime) {
            return formatDateTime(medicationAdministration.effectiveDateTime);
        }
        if (medicationAdministration.effectivePeriod) {
            return <FhirPeriodDisplay period={medicationAdministration.effectivePeriod} />;
        }
        return '-';
    };

    const renderMedicationValue = () => {
        if (medicationAdministration.medicationCodeableConcept) {
            return <FhirCodeableConceptDisplay code={medicationAdministration.medicationCodeableConcept} />;
        }
        if (medicationAdministration.medicationReference) {
            return <FhirReferenceDisplay references={medicationAdministration.medicationReference} />;
        }
        return '-';
    };

    const renderDosage = () => {
        if (!medicationAdministration.dosage) {
            return (
                <p className="text-sm italic text-muted-foreground">
                    No dosage information recorded
                </p>
            );
        }

        const dosage = medicationAdministration.dosage;

        return (
            <Card>
                <CardHeader>
                    <h5 className="text-sm font-semibold">Dosage Information</h5>
                </CardHeader>
                <CardContent>
                    <div className="bg-surface">
                        <FieldGroup>
                            {renderField('Rate (Ratio)', <FhirRatioDisplay ratio={dosage.rateRatio} />)}
                            {renderField('Rate (Quantity)', <FhirQuantityDisplay quantity={dosage.rateQuantity} />)}
                            {renderField('Dosage Instructions', dosage.text || '', 'dosage.text')}
                            {renderField('Body Site Administered To', <FhirCodeableConceptDisplay code={dosage.site} />)}
                            {renderField('Path of Substance Into Body', <FhirCodeableConceptDisplay code={dosage.route} />)}
                            {renderField('Method', <FhirCodeableConceptDisplay code={dosage.method} />)}
                            {renderField('Dose', <FhirQuantityDisplay quantity={dosage.dose} />)}
                        </FieldGroup>
                    </div>
                </CardContent>
            </Card>
        );
    };

    const renderPerformers = () => {
        if (!medicationAdministration.performer || medicationAdministration.performer.length === 0) {
            return (
                <p className="text-sm italic text-muted-foreground">
                    No performers recorded
                </p>
            );
        }

        return (
            <div className="space-y-4">
                {medicationAdministration.performer.map((performer, index) => (
                    <Card key={index}>
                        <CardHeader>
                            <h5 className="text-sm font-semibold">Performer {index + 1}</h5>
                        </CardHeader>
                        <CardContent>
                            <div className="bg-surface">
                                <FieldGroup>
                                    {renderField('Type of Performance', <FhirCodeableConceptDisplay code={performer.function} />)}
                                    {renderField('Who Performed the Medication Administration', <FhirReferenceDisplay references={performer.actor} />)}
                                </FieldGroup>
                            </div>
                        </CardContent>
                    </Card>
                ))}
            </div>
        );
    };

    // `medicationAdministration` is frequently a forced cast over a union of resource types (several
    // callers keep one `selectedItem`/`selected.resource` and multiplex it across many `Fhir*Display`
    // modals, gating only via `isOpen`) -- it can be the wrong resource shape, or undefined, whenever
    // `isOpen` is false. This guard must run before anything below touches `medicationAdministration.*`
    // so a non-MedicationAdministration/absent value while closed never throws (mirrors the original
    // `if (!isOpen) return null;` that ran before WP's `<Modal>` was ever constructed).
    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="MedicationAdministration" id={medicationAdministration.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={medicationAdministration} />
                            </div>
                            <ShowFieldHistoryToggle />
            <PrintRecord/>
                        </DialogHeader>

                        <FieldGroup>
                            {renderField('What was administered', renderMedicationValue(), medicationAdministration.medicationCodeableConcept ? 'medicationCodeableConcept.coding[0].display' : undefined)}
                            {renderField('Status', <FhirCodeDisplay code={medicationAdministration.status} />, 'status')}
                            {renderField('Reason administration not performed', <FhirCodeableConceptDisplay code={medicationAdministration.statusReason} multiple={true} />)}
                            {renderField('Type of medication usage', <FhirCodeableConceptDisplay code={medicationAdministration.category} />, 'category.coding[0].display')}
                            {renderField('Start and end time of administration', renderEffectiveValue(), medicationAdministration.effectiveDateTime ? 'effectiveDateTime' : undefined)}
                            {renderField('Request administration performed against', <FhirReferenceDisplay references={medicationAdministration.request} />)}
                            {renderField('Device used to administer', <FhirReferenceDisplay references={medicationAdministration.device} />)}
                            {renderField('Encounter or Episode of Care administered as part of', <FhirReferenceDisplay references={medicationAdministration.context} />)}
                        </FieldGroup>

                        <Section title="Dosage">
                            {renderDosage()}
                        </Section>

                        <Section title="Reasons">
                            <FieldGroup>
                                {renderField('Reason administration performed', <FhirCodeableConceptDisplay code={medicationAdministration.reasonCode} multiple={true} />)}
                                {renderField('Condition or observation that supports why the medication was administered', <FhirReferenceDisplay references={medicationAdministration.reasonReference} />)}
                            </FieldGroup>
                        </Section>

                        <Section title="Performers">
                            {renderPerformers()}
                        </Section>

                        <Section title="Part Of">
                            <FieldGroup>
                                {renderField('Part of referenced event', <FhirReferenceDisplay references={medicationAdministration.partOf} />)}
                            </FieldGroup>
                        </Section>

                        <Section title="Notes">
                            <FieldGroup>
                                {renderField('Information about the administration', <FhirAnnotationDisplay annotation={medicationAdministration.note} multiple={true} />)}
                            </FieldGroup>
                        </Section>
                    </FieldHistoryScope>
                </FieldHistoryProvider>
            </DialogContent>
        </Dialog>
    );
};

export default FhirMedicationAdministrationDisplay;