fix: missing flow and types

src/core_modules/capture-core/components/WidgetRelatedStages/RelatedStagesActions/RelatedStagesActions.component.js

@@ -71,7 +71,6 @@ export const RelatedStagesActionsPlain = ({
             linkMode: action,
         }));
     };
-
     const canAddNewEventToStage = useCanAddNewEventToStage(programStage, linkableEvents);
 
     if (!programStage) {

src/core_modules/capture-core/components/WidgetRelatedStages/hooks/useCanAddNewEventToStage.js

@@ -1,13 +1,18 @@
+// @flow
 import { useMemo } from 'react';
 import { useSelector } from 'react-redux';
+import { ProgramStage } from '../../../metaData';
+import type { LinkableEvent } from '../RelatedStagesActions/RelatedStagesActions.types';
 
-export const useCanAddNewEventToStage = (programStage, existingRelatedEvents) => {
+export const useCanAddNewEventToStage = (programStage: ?ProgramStage, existingRelatedEvents: LinkableEvent[]) => {
+    console.log('programStage', programStage);
+    console.log('existingRelatedEvents', existingRelatedEvents);
     const hiddenProgramStages = useSelector(({ rulesEffectsHiddenProgramStageDesc }) =>
         rulesEffectsHiddenProgramStageDesc?.['enrollmentEvent-newEvent'],
     );
 
     return useMemo(() => {
-        const isProgramStageHidden = hiddenProgramStages?.[programStage.id];
+        const isProgramStageHidden = hiddenProgramStages?.[programStage?.id];
 
         if (isProgramStageHidden) { return false; }