[frontend] Apply commons to custom dashboards

fellowseb · fellowseb · commit 8d49d901ed06 · 2026-04-20T01:33:40.000+02:00
diff --git a/opencti-platform/opencti-front/src/private/components/workspaces/dashboards/Dashboard.jsx b/opencti-platform/opencti-front/src/private/components/workspaces/dashboards/Dashboard.jsx
@@ -1,20 +1,21 @@
-import { useEffect, useMemo, useState } from 'react';
-import * as R from 'ramda';
 import { graphql, useFragment } from 'react-relay';
-import ReactGridLayout, { useContainerWidth } from 'react-grid-layout';
-import { v4 as uuid } from 'uuid';
-import { Stack, Box } from '@mui/material';
-import { useTheme } from '@mui/styles';
-import DashboardTimeFilters from './DashboardTimeFilters';
+import Stack from '@mui/material/Stack';
+import DashboardTimeFilters from '../../../../components/dashboard/DashboardTimeFilters';
 import WorkspaceHeader from '../workspaceHeader/WorkspaceHeader';
-import { commitMutation, handleError } from '../../../../relay/environment';
+import { commitMutation, handleError, fetchQuery } from '../../../../relay/environment';
 import { workspaceMutationFieldPatch } from '../WorkspaceEditionOverview';
 import useGranted, { EXPLORE_EXUPDATE } from '../../../../utils/hooks/useGranted';
-import WorkspaceWidgetPopover from './WorkspaceWidgetPopover';
-import { fromB64, toB64 } from '../../../../utils/String';
-import { deserializeDashboardManifestForFrontend, serializeDashboardManifestForBackend } from '../../../../utils/filters/filtersUtils';
 import useApiMutation from '../../../../utils/hooks/useApiMutation';
-import DashboardViz from './DashboardViz';
+import DashboardContent from '../../../../components/dashboard/DashboardContent';
+import useDashboard from '../../../../components/dashboard/useDashboard';
+
+const dashboardExportWidgetQuery = graphql`
+  query DashboardWidgetExportQuery($id: String!, $widgetId: ID!) {
+    workspace(id: $id) {
+      toWidgetExport(widgetId: $widgetId)
+    }
+  }
+`;
 
 const dashboardLayoutMutation = graphql`
   mutation DashboardLayoutMutation($id: ID!, $input: [EditInput!]!) {
@@ -54,142 +55,40 @@ const dashboardFragment = graphql`
   }
 `;
 
-const DashboardComponent = ({ data, noToolbar = false }) => {
-  const theme = useTheme();
-  const [commitWidgetImportMutation] = useApiMutation(dashboardImportWidgetMutation);
+const onExportWidget = async (id, widget) => {
+  const data = await fetchQuery(dashboardExportWidgetQuery, { id, widgetId: widget.id })
+    .toPromise();
+  return data.workspace?.toWidgetExport;
+};
 
+const DashboardComponent = ({ data, noToolbar = false }) => {
   const workspace = useFragment(dashboardFragment, data);
-  const { width, containerRef } = useContainerWidth();
-
-  const [deleting, setDeleting] = useState(false);
-  const [idToResize, setIdToResize] = useState();
-  const handleResize = (updatedWidget) => setIdToResize(updatedWidget);
+  const [commitWidgetImportMutation] = useApiMutation(dashboardImportWidgetMutation);
 
   const userHasEditAccess = workspace.currentUserAccessRight === 'admin'
     || workspace.currentUserAccessRight === 'edit';
   const userHasUpdateCapa = useGranted([EXPLORE_EXUPDATE]);
   const userCanEdit = userHasEditAccess && userHasUpdateCapa;
 
-  useEffect(() => {
-    const timeout = setTimeout(() => {
-      window.dispatchEvent(new Event('resize'));
-    }, 1200);
-    return () => {
-      clearTimeout(timeout);
-    };
-  }, []);
-
-  // Map of widget layouts, refreshed when workspace is updated (thanks to useMemo below).
-  // We use a local map of layouts to avoid a lot of computation when only changing position
-  // or dimension of widgets.
-  const [widgetsLayouts, setWidgetsLayouts] = useState({});
-
-  // Deserialized manifest, refreshed when workspace is updated.
-  const manifest = useMemo(() => {
-    return workspace.manifest && workspace.manifest.length > 0
-      ? deserializeDashboardManifestForFrontend(fromB64(workspace.manifest))
-      : { widgets: {}, config: {} };
-  }, [workspace.manifest]);
-
-  // Array of all widgets, refreshed when workspace is updated.
-  const widgetsArray = useMemo(() => {
-    return Object.values(manifest.widgets);
-  }, [manifest]);
-
-  useEffect(() => {
-    setWidgetsLayouts(
-      widgetsArray.reduce((res, widget) => {
-        res[widget.id] = widget.layout;
-        return res;
-      }, {}),
-    );
-  }, [widgetsArray]);
-
-  /**
-   * Merge a manifest with some layouts and transform it in base64.
-   *
-   * @param newManifest Manifest to merge with local changes and stringify.
-   * @param layouts Local layout changes.
-   * @returns {string} Manifest in B64.
-   */
-  const prepareManifest = (newManifest, layouts) => {
-    // Need to sync manifest with local layouts before sending for update.
-    // A desync occurs when resizing or moving a widget because in those cases
-    // we skip a complete reload to avoid performance issue.
-    const syncWidgets = Object.values(newManifest.widgets).reduce((res, widget) => {
-      const localLayout = layouts[widget.id];
-      res[widget.id] = {
-        ...widget,
-        layout: localLayout || widget.layout,
-      };
-      return res;
-    }, {});
-    const manifestToSave = {
-      ...newManifest,
-      widgets: syncWidgets,
-    };
-
-    const strManifest = serializeDashboardManifestForBackend(manifestToSave);
-    return toB64(strManifest);
-  };
-
-  const saveManifest = (newManifest, opts = { layouts: widgetsLayouts, noRefresh: false }) => {
-    const { layouts, noRefresh } = opts;
-    const newManifestEncoded = prepareManifest(newManifest, layouts);
-    // Sometimes (in case of layout adjustment) we do not want to re-fetch
-    // all the manifest because widgets data is still the same, and it's costly
-    // in performance.
+  const onSave = (id, newManifestEncoded, noRefresh, onCompleted) => {
     const mutation = noRefresh ? dashboardLayoutMutation : workspaceMutationFieldPatch;
-    if (workspace.manifest !== newManifestEncoded) {
-      commitMutation({
-        mutation,
-        variables: {
-          id: workspace.id,
-          input: {
-            key: 'manifest',
-            value: newManifestEncoded,
-          },
-        },
-        onCompleted: () => {
-          setDeleting(false);
+    commitMutation({
+      mutation,
+      variables: {
+        id,
+        input: {
+          key: 'manifest',
+          value: newManifestEncoded,
         },
-      });
-    }
-  };
-
-  const handleDateChange = (type, value) => {
-    let newManifest = R.assoc(
-      'config',
-      R.assoc(type, value === 'none' ? null : value, manifest.config),
-      manifest,
-    );
-    if (type === 'relativeDate' && value !== 'none') {
-      newManifest = R.assoc(
-        'config',
-        R.assoc('startDate', null, newManifest.config),
-        newManifest,
-      );
-      newManifest = R.assoc(
-        'config',
-        R.assoc('endDate', null, newManifest.config),
-        newManifest,
-      );
-    }
-    saveManifest(newManifest);
-  };
-
-  const getNextRow = () => {
-    return widgetsArray.reduce((max, { layout }) => {
-      const widgetEndRow = layout.y + layout.h;
-      return widgetEndRow > max ? widgetEndRow : max;
-    }, 0);
+      },
+      onCompleted,
+    });
   };
 
-  const importWidget = (widgetConfig) => {
-    const manifestEncoded = prepareManifest(manifest, widgetsLayouts);
+  const onImportWidget = (id, widgetConfig, manifestEncoded) => {
     commitWidgetImportMutation({
       variables: {
-        id: workspace.id,
+        id,
         input: {
           importType: 'widget',
           file: widgetConfig,
@@ -202,134 +101,32 @@ const DashboardComponent = ({ data, noToolbar = false }) => {
     });
   };
 
-  const handleAddWidget = (widgetConfig) => {
-    saveManifest({
-      ...manifest,
-      widgets: {
-        ...manifest.widgets,
-        [widgetConfig.id]: {
-          ...widgetConfig,
-          layout: {
-            i: widgetConfig.id,
-            x: 0,
-            y: getNextRow(),
-            w: 4,
-            h: 2,
-          },
-        },
-      },
-    });
-  };
-
-  const handleUpdateWidget = (widgetManifest) => {
-    const newManifest = {
-      ...manifest,
-      widgets: { ...manifest.widgets, [widgetManifest.id]: widgetManifest },
-    };
-    saveManifest(newManifest);
-  };
-
-  const handleDeleteWidget = (widgetId) => {
-    setDeleting(true);
-    const newWidgets = { ...manifest.widgets };
-    delete newWidgets[widgetId];
-    saveManifest({
-      ...manifest,
-      widgets: newWidgets,
-    });
-  };
-
-  const handleDuplicateWidget = (widgetToDuplicate) => {
-    handleAddWidget({
-      ...widgetToDuplicate,
-      id: uuid(),
-    });
-  };
-
-  const onLayoutChange = (layouts) => {
-    if (deleting) return;
-
-    const newLayouts = layouts.reduce((res, layout) => {
-      res[layout.i] = layout;
-      return res;
-    }, {});
-
-    if (R.equals(newLayouts, widgetsLayouts)) return; // ⛔ prevent loop
-
-    setWidgetsLayouts(newLayouts);
-    saveManifest(manifest, { layouts: newLayouts, noRefresh: true });
-  };
-
+  const helpers = useDashboard({ entity: workspace, onSave, onImportWidget, onExportWidget });
+  const { handleAddWidget, handleImportWidget, handleDateChange, config } = helpers;
   return (
-    <Box
-      id="container"
-      ref={containerRef}
-      sx={{
-        '& .react-grid-item.react-grid-placeholder': {
-          border: `2px solid ${theme.palette.primary.main}`,
-          borderRadius: 1,
-        },
-      }}
-    >
-      <Stack gap={2}>
-        {!noToolbar && (
-          <Stack gap={1}>
-            <WorkspaceHeader
-              handleAddWidget={handleAddWidget}
-              handleImportWidget={importWidget}
-              data={workspace}
-              variant="dashboard"
-            />
-            <DashboardTimeFilters
-              currentUserAccessRight={workspace.currentUserAccessRight}
-              config={manifest.config}
-              handleDateChange={handleDateChange}
-            />
-          </Stack>
-        )}
-        <ReactGridLayout
-          className="layout"
-          width={width}
-          layout={Object.values(widgetsLayouts)}
-          gridConfig={{ margin: [20, 20], rowHeight: 50, cols: 12, containerPadding: [0, 0] }}
-          dragConfig={{ enabled: userCanEdit ? !noToolbar : false, cancel: '.noDrag' }}
-          resizeConfig={{ enabled: userCanEdit ? !noToolbar : false }}
-          onLayoutChange={userCanEdit && !noToolbar ? onLayoutChange : () => true}
-          onResizeStart={userCanEdit ? (_, { i }) => handleResize(i) : undefined}
-          onResizeStop={userCanEdit ? handleResize : undefined}
-        >
-          {widgetsArray.map((widget) => {
-            if (!widgetsLayouts[widget.id]) return null;
-            const popover = userCanEdit && !noToolbar && (
-              <WorkspaceWidgetPopover
-                widget={widget}
-                workspace={workspace}
-                onUpdate={handleUpdateWidget}
-                onDuplicate={handleDuplicateWidget}
-                onDelete={() => handleDeleteWidget(widget.id)}
-              />
-            );
-
-            return (
-              <div
-                key={widget.id}
-                style={{
-                  display: 'relative',
-                }}
-              >
-                {widget.id === idToResize ? <div /> : (
-                  <DashboardViz
-                    widget={widget}
-                    config={manifest.config}
-                    popover={popover}
-                  />
-                )}
-              </div>
-            );
-          })}
-        </ReactGridLayout>
-      </Stack>
-    </Box>
+    <Stack gap={2}>
+      {!noToolbar && (
+        <Stack gap={1}>
+          <WorkspaceHeader
+            handleAddWidget={handleAddWidget}
+            handleImportWidget={handleImportWidget}
+            data={workspace}
+            variant="dashboard"
+          />
+          <DashboardTimeFilters
+            currentUserAccessRight={workspace.currentUserAccessRight}
+            config={config}
+            handleDateChange={handleDateChange}
+          />
+        </Stack>
+      )
+      }
+      <DashboardContent
+        helpers={helpers}
+        isEditable={userCanEdit}
+        dashboardEntity={workspace}
+      />
+    </Stack>
   );
 };
 
