feat: add quota notification reception

src/components/study-container.jsx

@@ -38,7 +38,7 @@
     useIntlRef,
     useNotificationsListener,
     useSnackMessage,
-    BuildStatus,
+    BuildStatus, fetchUserQuotaState,
 } from '@gridsuite/commons-ui';
 import NetworkModificationTreeModel from './graph/network-modification-tree-model';
 import { getFirstNodeOfType } from './graph/util/model-functions';
@@ -493,6 +493,28 @@
         listenerCallbackMessage: handleEvent,
     });
 
+    const handleQuotaEvent = useCallback(
+        (event) => {
+            const eventData = parseEventData(event);
+            const quotaType = eventData.headers.quotaType;
+
+            fetchUserQuotaState(userName).then((response) => {
+                const currentComputationQuota = response[quotaType]
+                if (currentComputationQuota != null) {
+                    if (currentComputationQuota.current >= currentComputationQuota.max) {
+                        console.info('TODO: Quota reached', quotaType, currentComputationQuota);
+                    } else {
+                        console.info('TODO: Quota available', quotaType, currentComputationQuota);
+                    }
+                }
+            })
+            }
+        );
+
+    useNotificationsListener(NotificationsUrlKeys.QUOTA, {
+        listenerCallbackMessage: handleQuotaEvent,
+    });
+
     useEffect(() => {
         if (studyUuid) {
             websocketExpectedCloseRef.current = false;

src/hooks/use-notifications-url-generator.ts

@@ -20,6 +20,7 @@ const useNotificationsUrlGenerator = (): Partial<Record<NotificationsUrlKeys, st
     // The websocket API doesn't allow relative urls
     const wsBase = getWsBase();
     const studyUuid = useSelector((state: AppState) => state.studyUuid);
+    const userId = useSelector((state: AppState) => state.user?.profile.sub);
 
     // return a mapColumns with NOTIFICATIONS_URL_KEYS and undefined value if URL is not yet buildable (studyUuid)
     // it will be used to register listeners as soon as possible.
@@ -30,14 +31,17 @@ const useNotificationsUrlGenerator = (): Partial<Record<NotificationsUrlKeys, st
             [NotificationsUrlKeys.STUDY]: studyUuid
                 ? `${wsBase}${PREFIX_STUDY_NOTIFICATION_WS}/notify?studyUuid=${encodeURIComponent(studyUuid)}`
                 : undefined,
+            [NotificationsUrlKeys.QUOTA]: userId
+                ? `${wsBase}${PREFIX_STUDY_NOTIFICATION_WS}/quota?userId=${encodeURIComponent(userId)}`
+                : undefined,
             [NotificationsUrlKeys.DIRECTORY_DELETE_STUDY]: studyUuid
                 ? `${wsBase}${PREFIX_DIRECTORY_NOTIFICATION_WS}/notify?updateType=deleteElement&elementUuid=${encodeURIComponent(
                       studyUuid
                   )}`
                 : undefined,
             [NotificationsUrlKeys.DIRECTORY]: `${wsBase}${PREFIX_DIRECTORY_NOTIFICATION_WS}/notify?updateType=directories`,
         }),
-        [wsBase, studyUuid]
+        [wsBase, studyUuid, userId]
     );
 };