import React, { useState, useEffect } from 'react';
import { usePublicQuestionnaireIds } from '../../hooks/usePublicQuestionnaireIds';
import { TextareaControl } from "../ui/textarea-control";
import { Card, CardHeader, CardBody, CardDivider } from '../ui/wp-card';
import { Elevation } from '../ui/elevation';
import { InputControl } from '../ui/input-control';
import { DismissibleAlert } from '../ui/dismissible-alert';
import { Heading } from '../ui/heading';
import { Spacer } from '../ui/spacer';
import { TabPanel } from '../ui/tab-panel';
import { Disabled } from '../ui/disabled';
import { useFormContext, Controller, useFieldArray } from 'react-hook-form';
import {
    IValueSet,
    useAuthServiceContext,
    useValueSetServiceContext
} from '../../services';
import {
    FhirAnnotation,
    FhirCode,
    FhirCodeableConcept,
    FhirPeriod,
    FhirReference,
    FhirReferenceCollection
} from '../../components';
import {
    useFhirEncounters,
    useFhirLocations,
    useFhirQuestionnaires,
    useFhirTaskBroadcast,
    useFhirTasks
} from '../../hooks';
import { FhirResource } from 'fhir/r4';

interface FhirTaskProps {
    elevated?: boolean;
    disabled?: boolean;
    enableInlineActions?: boolean;
}

export const FhirTask: React.FC<FhirTaskProps> = ({ elevated = true, disabled = false, enableInlineActions = true }) => {
    const { control } = useFormContext();
    const { userRole, USER_ROLES } = useAuthServiceContext();
    const { fetchValueSet, valueSetUrls } = useValueSetServiceContext();

    const [loading, setLoading] = useState(false);

    // value sets
    const [taskStatus, setTaskStatus] = useState<IValueSet[]>([]);
    const [taskIntent, setTaskIntent] = useState<IValueSet[]>([]);
    const [requestPriority, setRequestPriority] = useState<IValueSet[]>([]);
    // const [taskCode, setTaskCode] = useState<IValueSet[]>([]);
    const [performerRole, setPerformerRole] = useState<IValueSet[]>([]);
    const [resourceTypes, setResourceTypes] = useState<IValueSet[]>([]);

    const isPatient = [USER_ROLES.PATIENT, USER_ROLES.RELATED_PERSON].includes(userRole as string);
    const spacerMargin = '24px';

    useFhirTaskBroadcast();
    const { tasksList } = useFhirTasks();
    const { encountersList } = useFhirEncounters();
    const { locationsList } = useFhirLocations();
    const { questionnaireList } = useFhirQuestionnaires({ pageSize: 100 });
    // A public form is website content, never a task's questionnaire; unknown public forms offer nothing.
    const { publicIds, publicRead } = usePublicQuestionnaireIds();
    const taskQuestionnaires = publicRead === 'ready'
        ? questionnaireList.filter((questionnaire) => !questionnaire.id || !publicIds.has(questionnaire.id))
        : [];

    const outputFieldArray = useFieldArray({ control, name: 'output' });
    const focusFieldArray = useFieldArray({ control, name: 'focus' });
    const restrictionRecipientFieldArray = useFieldArray({ control, name: 'restriction.recipient' });

    useEffect(() => {
        const loadValueSets = async () => {
            setLoading(true);
            try {
                const [
                    taskStatusValueSet,
                    taskIntentValueSet,
                    requestPriorityValueSet,
                    // taskCodeValueSet,
                    performerRoleValueSet,
                    resourceTypesValueSet
                ] = await Promise.all([
                    fetchValueSet(valueSetUrls.taskStatus),
                    fetchValueSet(valueSetUrls.taskIntent),
                    fetchValueSet(valueSetUrls.requestPriority),
                    // fetchValueSet(valueSetUrls.taskCode),
                    fetchValueSet(valueSetUrls.performerRole),
                    fetchValueSet(valueSetUrls.resourceTypes)
                ]);

                setTaskStatus(taskStatusValueSet);
                setTaskIntent(taskIntentValueSet);
                setRequestPriority(requestPriorityValueSet);
                // setTaskCode(taskCodeValueSet);
                setPerformerRole(performerRoleValueSet);
                setResourceTypes(resourceTypesValueSet);
            } catch (err) {
                console.error('Error loading value sets:', err);
            } finally {
                setLoading(false);
            }
        };

        loadValueSets();
    }, [fetchValueSet, valueSetUrls]);

    return (
        <Card style={elevated ? { boxShadow: 'none' } : {}}>
            {elevated && <Elevation value={5}/>}
            <CardHeader>
                <Heading level={5}>Details</Heading>
            </CardHeader>
            <Disabled isDisabled={isPatient}>
                <CardBody>
                    <Spacer margin={spacerMargin}/>

                    {/* status */}
                    <FhirCode label="Task Status"
                              context="status"
                              options={taskStatus}
                              required={true}
                              loading={loading}
                              disabled={disabled}/>

                    <Spacer margin={spacerMargin}/>

                    {/* statusReason */}
                    {/*<FhirCodeableConcept label="Reason for current status"*/}
                    {/*                     context="statusReason"*/}
                    {/*                     options={[]}/>*/}

                    {/*<Spacer margin={spacerMargin}/>*/}

                    {/* businessStatus */}
                    {/*<FhirCodeableConcept label="Business Status"*/}
                    {/*                     context="businessStatus"*/}
                    {/*                     options={[]}/>*/}

                    {/*<Spacer margin={spacerMargin}/>*/}

                    {/* intent */}
                    <FhirCode label="Task Intent"
                              context="intent"
                              options={taskIntent}
                              required={true}
                              loading={loading}
                              disabled={disabled}/>

                    <Spacer margin={spacerMargin}/>

                    {/* priority */}
                    <FhirCode label="Priority"
                              context="priority"
                              options={requestPriority}
                              loading={loading}
                              disabled={disabled}/>

                    <Spacer margin={spacerMargin}/>

                    {/* code */}
                    {/*<FhirCodeableConcept label="Task Type"*/}
                    {/*                     context="code"*/}
                    {/*                     options={taskCode}/>*/}

                    <Spacer margin={spacerMargin}/>

                    {/* focus */}
                    <FhirReferenceCollection label="What task is acting on"
                                             references={focusFieldArray}
                                             context="focus"
                                             referenceTypes={[
                                                 { name: 'Questionnaire', key: 'Questionnaire' }
                                             ]}
                                             getReferenceList={(referenceType: string): FhirResource[] => {
                                                 switch (referenceType) {
                                                     case 'Questionnaire':
                                                         return taskQuestionnaires;
                                                     default:
                                                         return [];
                                                 }
                                             }}
                                             required={true}
                                             multiple={false}
                                             disabled={disabled}
                                             readonly={isPatient}
                                             enableInlineActions={enableInlineActions}/>

                    <Spacer margin={spacerMargin}/>

                    {/* description */}
                    <Controller key={'description'}
                                name={'description'}
                                control={control}
                                disabled={disabled}
                                rules={{ required: true }}
                                render={({ field, fieldState }) => (
                                    <>
                                        <TextareaControl
                                                         {...field}
                                                         label="Description *"
                                                         rows={5}/>

                                        {fieldState.error && (
                                            <DismissibleAlert status="error">
                                                Description is required
                                            </DismissibleAlert>
                                        )}
                                    </>
                                )}/>

                    <Spacer margin={spacerMargin}/>

                    {/* performerType */}
                    <FhirCodeableConcept label="Requested performer"
                                         context="performerType"
                                         options={performerRole}
                                         multiple={true}
                                         loading={loading}
                                         disabled={disabled}/>

                    <Spacer margin={spacerMargin}/>

                    {/* partOf */}
                    <FhirReference label="Part of Composite task"
                                   context="partOf"
                                   referenceType="Task"
                                   referenceList={tasksList}
                                   enableInlineActions={enableInlineActions}
                                   multiple={true}
                                   required={false}
                                   disabled={disabled}/>

                    <Spacer margin={spacerMargin}/>

                    {/* encounter */}
                    <FhirReference label="Healthcare event during which this task originated"
                                   context="encounter"
                                   referenceType="Encounter"
                                   referenceList={encountersList}
                                   enableInlineActions={enableInlineActions}
                                   required={false}
                                   disabled={disabled}/>

                    <Spacer margin={spacerMargin}/>

                    {/* location */}
                    <FhirReference label="Where task occurs"
                                   context="location"
                                   referenceType="Location"
                                   referenceList={locationsList}
                                   enableInlineActions={enableInlineActions}
                                   required={false}
                                   disabled={disabled}/>

                    <Spacer margin={spacerMargin}/>

                    {/* authoredOn */}
                    <Controller key="authoredOn"
                                name="authoredOn"
                                control={control}
                                disabled={disabled}
                                render={({ field }) => (
                                    <InputControl
                                                  {...field}
                                                  label="Task Creation Date"
                                                  type="datetime-local"/>
                                )}/>

                    <Spacer margin={spacerMargin}/>

                    {/* lastModified */}
                    <Controller key="lastModified"
                                name="lastModified"
                                control={control}
                                disabled={disabled}
                                render={({ field }) => (
                                    <InputControl
                                                  {...field}
                                                  label="Task Last Modified Date"
                                                  type="datetime-local"/>
                                )}/>

                    <Spacer margin={spacerMargin}/>

                    {/* executionPeriod */}
                    <FhirPeriod label="Start and end time of execution"
                                context="executionPeriod"
                                disabled={disabled}
                                readonly={isPatient}/>

                    <Spacer margin={spacerMargin}/>
                </CardBody>
            </Disabled>
            <CardDivider/>
            <CardBody>
                <TabPanel initialTabName="notes"
                          tabs={[
                              // { name: 'restrictions', title: 'Restrictions' },
                              { name: 'notes', title: 'Notes' }
                          ]}>
                    {(tab) => {
                        // restrictions
                        if (tab.name === 'restrictions') {
                            return (
                                <Disabled isDisabled={isPatient}>
                                    <div style={{ paddingTop: '16px' }}>
                                        <Spacer margin={spacerMargin}/>

                                        {/* repetitions */}
                                        <Controller key="restriction.repetitions"
                                                    name="restriction.repetitions"
                                                    control={control}
                                                    disabled={disabled}
                                                    render={({ field }) => (
                                                        <InputControl
                                                                      {...field}
                                                                      label="How many times to repeat"
                                                                      type="number"
                                                                      min="0"
                                                                      placeholder="Repetitions"/>
                                                    )}/>

                                        <Spacer margin={spacerMargin}/>

                                        {/* restriction.period */}
                                        <FhirPeriod label="When fulfillment sought"
                                                    context="restriction.period"
                                                    disabled={disabled}/>

                                        <Spacer margin={spacerMargin}/>

                                        {/* recipients */}
                                    </div>
                                </Disabled>
                            );
                        }

                        // notes
                        if (tab.name === 'notes') {
                            return (
                                <Disabled isDisabled={isPatient}>
                                    <div style={{ paddingTop: '16px' }}>
                                        <Spacer margin={spacerMargin}/>

                                        {/* note */}
                                        <FhirAnnotation label="Comments made about the task"
                                                        context="note"
                                                        disabled={disabled}
                                                        readonly={isPatient}/>

                                        <Spacer margin={spacerMargin}/>
                                    </div>
                                </Disabled>
                            );
                        }
                    }}
                </TabPanel>
            </CardBody>
        </Card>
    );
};

export default FhirTask;