import React from 'react';
import { Immunization } from 'fhir/r4';
import { FhirCodeableConceptDisplay } from '../../utils/fhirCodeableConcept';
import { FhirCodeDisplay } from '../../utils/fhirCode';
import { FhirQuantityDisplay } from '../../utils/fhirQuantity';
import { FhirAnnotationDisplay } from '../../utils/fhirAnnotation';
import { formatDate, 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 FhirImmunizationDisplayProps {
    immunization: Immunization;
    isOpen: boolean;
    onClose: () => void;
    title?: string;
}

export const FhirImmunizationDisplay: React.FC<FhirImmunizationDisplayProps> = ({
                                                                                    immunization,
                                                                                    isOpen,
                                                                                    onClose,
                                                                                    title = "Immunization Details"
                                                                                }) => {
    const renderDoseNumberValue = (protocol: any) => {
        if (protocol.doseNumberPositiveInt) {
            return protocol.doseNumberPositiveInt.toString();
        }
        if (protocol.doseNumberString) {
            return protocol.doseNumberString;
        }
        return '-';
    };

    const renderSeriesDosesValue = (protocol: any) => {
        if (protocol.seriesDosesPositiveInt) {
            return protocol.seriesDosesPositiveInt.toString();
        }
        if (protocol.seriesDosesString) {
            return protocol.seriesDosesString;
        }
        return '-';
    };

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

        return (
            <div className="space-y-4">
                {immunization.performer.map((performer, index) => (
                    <Card key={index}>
                        <CardHeader>
                            <h6 className="text-sm font-semibold">Performer {index + 1}</h6>
                        </CardHeader>
                        <CardContent>
                            <div className="bg-surface">
                                <FieldGroup>
                                    {renderField('What type of performance was done', <FhirCodeableConceptDisplay code={performer.function} />)}
                                    {renderField('Who performed event', <FhirReferenceDisplay references={performer.actor} />)}
                                </FieldGroup>
                            </div>
                        </CardContent>
                    </Card>
                ))}
            </div>
        );
    };

    const renderReactions = () => {
        if (!immunization.reaction || immunization.reaction.length === 0) {
            return (
                <p className="text-sm italic text-muted-foreground">
                    No reactions recorded
                </p>
            );
        }

        return (
            <div className="space-y-4">
                {immunization.reaction.map((reaction, index) => (
                    <Card key={index}>
                        <CardHeader>
                            <h6 className="text-sm font-semibold">Reaction {index + 1}</h6>
                        </CardHeader>
                        <CardContent>
                            <div className="bg-surface">
                                <FieldGroup>
                                    {renderField('Date', formatDateTime(reaction.date))}
                                    {renderField('Additional information on reaction', <FhirReferenceDisplay references={reaction.detail} />)}
                                    {renderField('Reported', reaction.reported === true ? 'Yes' : 'No')}
                                </FieldGroup>
                            </div>
                        </CardContent>
                    </Card>
                ))}
            </div>
        );
    };

    const renderProtocolApplied = () => {
        if (!immunization.protocolApplied || immunization.protocolApplied.length === 0) {
            return (
                <p className="text-sm italic text-muted-foreground">
                    No protocols recorded
                </p>
            );
        }

        return (
            <div className="space-y-4">
                {immunization.protocolApplied.map((protocol, index) => (
                    <Card key={index}>
                        <CardHeader>
                            <h6 className="text-sm font-semibold">Protocol {index + 1}</h6>
                        </CardHeader>
                        <CardContent>
                            <div className="bg-surface">
                                <FieldGroup>
                                    {renderField('Name of vaccine series', protocol.series)}
                                    {renderField('Vaccine preventable disease being targeted', <FhirCodeableConceptDisplay code={protocol.targetDisease} multiple={true} />)}
                                    {renderField('Dose number within series', renderDoseNumberValue(protocol))}
                                    {renderField('Recommended number of doses for immunity', renderSeriesDosesValue(protocol))}
                                    {renderField('Authority', <FhirReferenceDisplay references={protocol.authority} />)}
                                </FieldGroup>
                            </div>
                        </CardContent>
                    </Card>
                ))}
            </div>
        );
    };

    const renderEducation = () => {
        if (!immunization.education || immunization.education.length === 0) {
            return (
                <p className="text-sm italic text-muted-foreground">
                    No educational materials recorded
                </p>
            );
        }

        return (
            <div className="space-y-4">
                {immunization.education.map((education, index) => (
                    <Card key={index}>
                        <CardHeader>
                            <h6 className="text-sm font-semibold">Educational Material {index + 1}</h6>
                        </CardHeader>
                        <CardContent>
                            <div className="bg-surface">
                                <FieldGroup>
                                    {renderField('Type of educational material', education.documentType)}
                                    {renderField('Reference to educational material', education.reference ? (
                                        <a href={education.reference} target="_blank" rel="noopener noreferrer"
                                           className="text-sky-700 underline">
                                            {education.reference}
                                        </a>
                                    ) : '-')}
                                    {renderField('Educational material publication date', formatDateTime(education.publicationDate))}
                                    {renderField('Educational material presentation date', formatDateTime(education.presentationDate))}
                                </FieldGroup>
                            </div>
                        </CardContent>
                    </Card>
                ))}
            </div>
        );
    };

    // `immunization` 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
    // `immunization.*` so a non-Immunization/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="Immunization" id={immunization.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={immunization} />
                            </div>
                            <ShowFieldHistoryToggle />
            <PrintRecord/>
                        </DialogHeader>

                        <FieldGroup>
                            {renderField('Vaccine product administered', <FhirCodeableConceptDisplay code={immunization.vaccineCode} />, 'vaccineCode.coding[0].display')}
                            {renderField('Status', <FhirCodeDisplay code={immunization.status} />, 'status')}
                            {immunization.status === 'not-done' && renderField('Reason not done', <FhirCodeableConceptDisplay code={immunization.statusReason} />, 'statusReason.coding[0].display')}
                            {renderField('Vaccine lot number', immunization.lotNumber, 'lotNumber')}
                            {renderField('Manufacturer', immunization.manufacturer?.display || immunization.manufacturer?.identifier?.value, 'manufacturer')}
                            {renderField('Vaccine expiration date', formatDate(immunization.expirationDate), 'expirationDate')}
                            {renderField('Vaccine administration date', formatDateTime(immunization.occurrenceDateTime), 'occurrenceDateTime')}
                            {renderField('When the immunization was first captured in the subject\'s record', formatDateTime(immunization.recorded), 'recorded')}
                            {renderField('Primary Source', immunization.primarySource === true ? 'Yes' : 'No', 'primarySource')}
                            {immunization.primarySource === false && renderField('Source of a secondarily reported record', <FhirCodeableConceptDisplay code={immunization.reportOrigin} />, 'reportOrigin.coding[0].display')}
                            {renderField('Encounter immunization was part of', <FhirReferenceDisplay references={immunization.encounter} />)}
                            {renderField('Where immunization occurred', <FhirReferenceDisplay references={immunization.location} />)}
                        </FieldGroup>

                        <Section title="Administration">
                            <FieldGroup>
                                {renderField('Body site vaccine was administered', <FhirCodeableConceptDisplay code={immunization.site} />, 'site.coding[0].display')}
                                {renderField('How vaccine entered body', <FhirCodeableConceptDisplay code={immunization.route} />, 'route.coding[0].display')}
                                {renderField('Dose Quantity', <FhirQuantityDisplay quantity={immunization.doseQuantity} />, 'doseQuantity.value')}
                                {renderField('Subpotent Dose', immunization.isSubpotent === true ? 'Yes' : 'No', 'isSubpotent')}
                                {immunization.isSubpotent && renderField('Reason for being subpotent', <FhirCodeableConceptDisplay code={immunization.subpotentReason} multiple={true} />)}
                            </FieldGroup>
                        </Section>

                        <Section title="Reasons">
                            <FieldGroup>
                                {renderField('Why immunization occurred', <FhirCodeableConceptDisplay code={immunization.reasonCode} multiple={true} />)}
                                {renderField('Patient eligibility for a vaccination program', <FhirCodeableConceptDisplay code={immunization.programEligibility} multiple={true} />)}
                                {renderField('Funding source for the vaccine', <FhirCodeableConceptDisplay code={immunization.fundingSource} />, 'fundingSource.coding[0].display')}
                                {renderField('Why immunization occurred (References)', <FhirReferenceDisplay references={immunization.reasonReference} />)}
                            </FieldGroup>
                        </Section>

                        <Section title="Protocol Applied">
                            {renderProtocolApplied()}
                        </Section>

                        <Section title="Reactions">
                            {renderReactions()}
                        </Section>

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

                        <Section title="Educational Materials">
                            {renderEducation()}
                        </Section>

                        <Section title="Notes">
                            <FieldGroup>
                                {renderField('Additional immunization notes', <FhirAnnotationDisplay annotation={immunization.note} multiple={true} />)}
                            </FieldGroup>
                        </Section>
                    </FieldHistoryScope>
                </FieldHistoryProvider>
            </DialogContent>
        </Dialog>
    );
};

export default FhirImmunizationDisplay;