import { useState } from 'react';
import { Item, ItemGroup } from "../../../../components/ui/item-group";
import { useNavigate } from 'react-router-dom';
import { Icon } from "../../../../components/ui/icon";
import { Flex, FlexItem, FlexBlock } from '../../../../components/ui/flex';
import { Text } from '../../../../components/ui/text';
import { Spacer } from '../../../../components/ui/spacer';
// `calendar` and `shield` still feed WordPress icon components on each row - a separate sweep.
import { Pencil, Plus, Calendar as calendar, Shield as shield } from 'lucide-react';
import { Immunization } from 'fhir/r4';
import { AppSpinner, FhirImmunizationDisplay, ScrollableTab } from '../../../../components';
import { ResourceListStatus } from '../../../../components/ui/resource-list-status';
import { formatFhirCodeableConcept } from '../../../../utils/fhirCodeableConcept.tsx';
import { formatDate } from '../../../../utils/dateUtils.ts';
import {
    useFhirImmunizationBroadcast,
    useFhirImmunizations,
    useFhirImmunizationTargetedBroadcast
} from '../../../../hooks';
import { useAuthServiceContext, useImmunizationServiceContext } from '../../../../services';
import '../medicationRecords.css';
import { SourcePractice } from '../../../../components/ui/source-practice';
import { RecordPracticeNote, recordPracticeId, useRecordPracticeView, useRetireOnPracticeChange } from '../../../../contexts/RecordPracticeContext';

const Immunizations = () => {
    const navigate = useNavigate();
    const { userRole, USER_ROLES } = useAuthServiceContext();
    const { getImmunizationDisplay } = useImmunizationServiceContext();
    const { loading, error, clearError, immunizationsList, attempted } = useFhirImmunizations();
    const [selectedImmunization, setSelectedImmunization] = useState<Immunization | null>(null);
    const [showDisplayModal, setShowDisplayModal] = useState<boolean>(false);
    // The record's practice context: its own immunizations under a named practice; the empty state speaks for the whole record.
    const practiceView = useRecordPracticeView(immunizationsList, recordPracticeId);
    useRetireOnPracticeChange(() => { setShowDisplayModal(false); setSelectedImmunization(null); });

    useFhirImmunizationBroadcast();
    useFhirImmunizationTargetedBroadcast({
        immunization: selectedImmunization || undefined,
        isOpen: showDisplayModal,
        onUpdate: (updatedImmunization) => {
            setSelectedImmunization(updatedImmunization);
        },
        onDelete: () => {
            setShowDisplayModal(false);
            setSelectedImmunization(null);
        }
    });

    const viewImmunization = (immunization: Immunization) => {
        setSelectedImmunization(immunization);
        setShowDisplayModal(true);
    };

    const editImmunization = (event: React.MouseEvent, immunization: Immunization) => {
        event.stopPropagation();
        navigate(immunization.id as string, { relative: 'path' });
    };

    const addImmunization = () => {
        navigate('new', { relative: 'path' });
    };

    const isAuthorizedToEdit = () => {
        return userRole && [USER_ROLES.PRACTITIONER].includes(userRole as string);
    };

    const getStatusDisplay = (immunization: Immunization): { text: string, color?: string } => {
        if (!immunization) return { text: 'Unknown' };

        switch (immunization.status) {
            case 'completed':
                return { text: 'Completed', color: '#00a32a' };
            case 'entered-in-error':
                return { text: 'Error', color: '#cc1818' };
            case 'not-done':
                return { text: 'Not Done', color: '#cc1818' };
            default:
                return { text: immunization.status.charAt(0).toUpperCase() + immunization.status.slice(1).replace('-', ' ') };
        }
    };

    return (
        <div>

            <Flex align="center" justify="space-between">
                <FlexBlock>
                </FlexBlock>
                
                {isAuthorizedToEdit() && (
                    <FlexItem>
                        <button
                            type="button"
                            className="btn btn-blue"
                            onClick={addImmunization}
                        >
                            <Plus />
                            Add
                        </button>
                    </FlexItem>
                )}
            </Flex>
            <Spacer margin="8"/>

            <ResourceListStatus
                loading={loading}
                error={error}
                isEmpty={immunizationsList.length === 0}

                /* - without this the empty message is a claim about the patient's
 record made before the server was ever asked. */
                attempted={attempted}
                resourceLabel="immunizations"
                emptyMessage="No immunization records found."
                onDismissError={clearError}
            />

            <>
                {loading ? (
                    <AppSpinner/>
                ) : (
                    <ScrollableTab>
                        <RecordPracticeNote view={practiceView} />
                        {practiceView.shown.length !== 0 && (
                            <ItemGroup isBordered isSeparated isRounded size='large'>
                                {practiceView.shown.map((immunization: Immunization) => {
                                    const displayTitle = getImmunizationDisplay(immunization);
                                    const { text: statusText, color: statusColor } = getStatusDisplay(immunization);
                                    const site = formatFhirCodeableConcept(immunization.site);
                                    const route = formatFhirCodeableConcept(immunization.route);
                                    const occurrenceDate = immunization.occurrenceDateTime ? formatDate(immunization.occurrenceDateTime) : '';

                                    return (
                                        <Item key={immunization.id}
                                              onClick={() => viewImmunization(immunization)}>
                                            <Flex>
                                                <FlexItem>
                                                    <Icon icon={shield}
                                                          size={24}
                                                          className="ahr-record-row__icon"
                                                          style={{ color: statusColor }}/>
                                                </FlexItem>
                                                <FlexBlock>
                                                    {/* Wraps: on a narrow row Edit moves below the whole title rather than being squeezed past the card's edge. */}
                                                    <Flex align="center" justify="space-between" wrap>
                                                        <FlexItem className="ahr-record-row__main">
                                                            <Text className="ahr-record-row__title">
                                                                {displayTitle}
                                                            </Text>
                                                            {(site || route) && (
                                                                <Text variant="muted" className="ahr-record-row__line">
                                                                    {site && `Site: ${site}`}
                                                                    {site && route && ' • '}
                                                                    {route && `Route: ${route}`}
                                                                </Text>
                                                            )}
                                                            {occurrenceDate && (
                                                                <Text variant="muted" className="ahr-record-row__line">
                                                                    <Icon icon={calendar}
                                                                          size={16}
                                                                          className="ahr-record-row__line-icon"/>
                                                                    {occurrenceDate}
                                                                </Text>
                                                            )}
                                                            <Text variant="muted" className="ahr-record-row__line"><SourcePractice resource={immunization} /></Text>
                                                        </FlexItem>
                                                        {isAuthorizedToEdit() && (
                                                            <FlexItem className="ahr-record-row__action">
                                                                <button
                                                                    type="button"
                                                                    className="btn btn-blue-ghost-sm"
                                                                    onClick={(event: React.MouseEvent) => editImmunization(event, immunization)}
                                                                >
                                                                    <Pencil />
                                                                    Edit
                                                                </button>
                                                            </FlexItem>
                                                        )}
                                                    </Flex>
                                                </FlexBlock>
                                            </Flex>
                                        </Item>
                                    );
                                })}
                            </ItemGroup>
                        )}
                    </ScrollableTab>
                )}
            </>

            {selectedImmunization && (
                <FhirImmunizationDisplay
                    immunization={selectedImmunization}
                    isOpen={showDisplayModal}
                    onClose={() => {
                        setShowDisplayModal(false);
                        setSelectedImmunization(null);
                    }}
                />
            )}
        </div>
    );
};

export default Immunizations;