import { useEffect } from 'react';
import { Immunization } from 'fhir/r4';
import { usePatientServiceContext, useWebSocketServiceContext } from '../services';

interface UseImmunizationWebSocketOptions {
    immunization?: Immunization;
    immunizationId?: string;
    isOpen?: boolean;
    onUpdate: (immunization: Immunization) => void;
    onDelete: () => void;
}

export const useFhirImmunizationTargetedBroadcast = ({
                                                         immunization,
                                                         immunizationId,
                                                         isOpen = true,
                                                         onUpdate,
                                                         onDelete
                                                     }: UseImmunizationWebSocketOptions) => {
    const { subscribe, unsubscribe } = useWebSocketServiceContext();
    const { patientState } = usePatientServiceContext();

    useEffect(() => {
        if (!patientState || !immunization || !isOpen) {
            return;
        }

        const handleImmunizationEvent = (cloudEvent: any) => {
            const immunizationPayload: Immunization = cloudEvent.data;
            const eventType = cloudEvent.type;

            const targetId = immunizationId || immunization.id;

            if (immunizationPayload.id === targetId) {
                switch (eventType) {
                    case 'com.akashichealthrecords.fhir.immunization.updated':
                        onUpdate(immunizationPayload);
                        break;
                    case 'com.akashichealthrecords.fhir.immunization.deleted':
                        onDelete();
                        break;
                }
            }
        };

        subscribe('com.akashichealthrecords.fhir.immunization.updated', handleImmunizationEvent);
        subscribe('com.akashichealthrecords.fhir.immunization.deleted', handleImmunizationEvent);

        return () => {
            unsubscribe('com.akashichealthrecords.fhir.immunization.updated', handleImmunizationEvent);
            unsubscribe('com.akashichealthrecords.fhir.immunization.deleted', handleImmunizationEvent);
        };
    }, [patientState, immunization, immunizationId, isOpen, onUpdate, onDelete, subscribe, unsubscribe]);
};