import React from 'react';
import { ClinicalImpression } from 'fhir/r4';
import { FhirCodeableConceptDisplay } from '../../utils/fhirCodeableConcept';
import { FhirCodeDisplay } from '../../utils/fhirCode';
import { FhirPeriodDisplay } from '../../utils/fhirPeriod';
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 FhirClinicalImpressionDisplayProps {
    clinicalImpression: ClinicalImpression;
    isOpen: boolean;
    onClose: () => void;
    title?: string;
}

export const FhirClinicalImpressionDisplay: React.FC<FhirClinicalImpressionDisplayProps> = ({
                                                                                                clinicalImpression,
                                                                                                isOpen,
                                                                                                onClose,
                                                                                                title = "Clinical Impression Details"
                                                                                            }) => {
    const renderEffectiveValue = () => {
        if (clinicalImpression.effectiveDateTime) {
            return formatDateTime(clinicalImpression.effectiveDateTime);
        }
        if (clinicalImpression.effectivePeriod) {
            return <FhirPeriodDisplay period={clinicalImpression.effectivePeriod} />;
        }
        return '-';
    };

    const renderFindings = () => {
        if (!clinicalImpression.finding || clinicalImpression.finding.length === 0) {
            return (
                <p className="text-sm italic text-muted-foreground">
                    No findings recorded
                </p>
            );
        }

        return (
            <div className="space-y-4">
                {clinicalImpression.finding.map((finding, index) => (
                    <Card key={index}>
                        <CardHeader>
                            <h5 className="text-sm font-semibold">Finding {index + 1}</h5>
                        </CardHeader>
                        <CardContent>
                            <div className="bg-surface">
                                <FieldGroup>
                                    {renderField('What was found (Code)', <FhirCodeableConceptDisplay code={finding.itemCodeableConcept} />)}
                                    {renderField('What was found (Reference)', <FhirReferenceDisplay references={finding.itemReference} />)}
                                    {renderField('Basis for finding', finding.basis)}
                                </FieldGroup>
                            </div>
                        </CardContent>
                    </Card>
                ))}
            </div>
        );
    };

    const renderInvestigations = () => {
        if (!clinicalImpression.investigation || clinicalImpression.investigation.length === 0) {
            return (
                <p className="text-sm italic text-muted-foreground">
                    No investigations recorded
                </p>
            );
        }

        return (
            <div className="space-y-4">
                {clinicalImpression.investigation.map((investigation, index) => (
                    <Card key={index}>
                        <CardHeader>
                            <h5 className="text-sm font-semibold">Investigation {index + 1}</h5>
                        </CardHeader>
                        <CardContent>
                            <div className="bg-surface">
                                <FieldGroup>
                                    {renderField('Investigation Code', <FhirCodeableConceptDisplay code={investigation.code} />)}
                                    {renderField('Investigation Items', <FhirReferenceDisplay references={investigation.item} />)}
                                </FieldGroup>
                            </div>
                        </CardContent>
                    </Card>
                ))}
            </div>
        );
    };

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

                        <FieldGroup>
                            {renderField('Status', <FhirCodeDisplay code={clinicalImpression.status} />, 'status')}
                            {renderField('Clinician performing assessment', <FhirReferenceDisplay references={clinicalImpression.assessor} />)}
                            {renderField('Why/how assessment was performed', clinicalImpression.description, 'description')}
                            {renderField('When assessment was documented', formatDateTime(clinicalImpression.date), 'date')}
                            {renderField(
                                'Time of assessment',
                                renderEffectiveValue(),
                                clinicalImpression.effectiveDateTime ? 'effectiveDateTime' : undefined,
                            )}
                            {renderField('Summary', clinicalImpression.summary, 'summary')}
                            {renderField('Encounter', <FhirReferenceDisplay references={clinicalImpression.encounter} />)}
                            {renderField('Previous assessment', <FhirReferenceDisplay references={clinicalImpression.previous} />)}
                        </FieldGroup>

                        <Section title="Problems">
                            <FieldGroup>
                                {renderField('Relevant impressions of patient state', <FhirReferenceDisplay references={clinicalImpression.problem} />)}
                            </FieldGroup>
                        </Section>

                        <Section title="Findings">
                            {renderFindings()}
                        </Section>

                        <Section title="Investigations">
                            {renderInvestigations()}
                        </Section>

                        <Section title="Prognosis">
                            <FieldGroup>
                                {renderField('Estimate of likely outcome', <FhirCodeableConceptDisplay code={clinicalImpression.prognosisCodeableConcept} multiple={true} />)}
                                {renderField('RiskAssessment expressing likely outcome', <FhirReferenceDisplay references={clinicalImpression.prognosisReference} />)}
                            </FieldGroup>
                        </Section>

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

export default FhirClinicalImpressionDisplay;