[Security Solution][Lens] New trigger actions for chart legends and table cell actions

src/plugins/chart_expressions/expression_partition_vis/public/components/partition_vis_component.test.tsx

@@ -77,6 +77,8 @@ describe('PartitionVisComponent', function () {
       syncColors: false,
       fireEvent: jest.fn(),
       renderComplete: jest.fn(),
+      interactive: true,
+      columnCellValueActions: [],
       services: {
         data: dataPluginMock.createStartContract(),
         fieldFormats: fieldFormatsServiceMock.createStartContract(),
@@ -172,6 +174,16 @@ describe('PartitionVisComponent', function () {
     });
   });
 
+  it('should render legend actions when it is interactive', async () => {
+    const component = shallow(<PartitionVisComponent {...wrapperProps} interactive={true} />);
+    expect(component.find(Settings).prop('legendAction')).toBeDefined();
+  });
+
+  it('should not render legend actions when it is not interactive', async () => {
+    const component = shallow(<PartitionVisComponent {...wrapperProps} interactive={false} />);
+    expect(component.find(Settings).prop('legendAction')).toBeUndefined();
+  });
+
   it('hides the legend if the legend toggle is clicked', async () => {
     const component = mountWithIntl(<PartitionVisComponent {...wrapperProps} />);
     await actWithTimeout(async () => {

src/plugins/chart_expressions/expression_partition_vis/public/components/partition_vis_component.tsx

@@ -68,7 +68,7 @@ import {
 } from './partition_vis_component.styles';
 import { ChartTypes } from '../../common/types';
 import { filterOutConfig } from '../utils/filter_out_config';
-import { FilterEvent, StartDeps } from '../types';
+import { ColumnCellValueActions, FilterEvent, StartDeps } from '../types';
 
 declare global {
   interface Window {
@@ -85,19 +85,23 @@ export interface PartitionVisComponentProps {
   uiState: PersistedState;
   fireEvent: IInterpreterRenderHandlers['event'];
   renderComplete: IInterpreterRenderHandlers['done'];
+  interactive: boolean;
   chartsThemeService: ChartsPluginSetup['theme'];
   palettesRegistry: PaletteRegistry;
   services: Pick<StartDeps, 'data' | 'fieldFormats'>;
   syncColors: boolean;
+  columnCellValueActions: ColumnCellValueActions;
 }
 
 const PartitionVisComponent = (props: PartitionVisComponentProps) => {
   const {
+    columnCellValueActions,
     visData: originalVisData,
     visParams: preVisParams,
     visType,
     services,
     syncColors,
+    interactive,
   } = props;
   const visParams = useMemo(() => filterOutConfig(visType, preVisParams), [preVisParams, visType]);
   const chartTheme = props.chartsThemeService.useChartsTheme();
@@ -313,6 +317,32 @@ const PartitionVisComponent = (props: PartitionVisComponentProps) => {
     ]
   );
 
+  const legendActions = useMemo(
+    () =>
+      interactive
+        ? getLegendActions(
+            canFilter,
+            getLegendActionEventData(visData),
+            handleLegendAction,
+            columnCellValueActions,
+            visParams,
+            visData,
+            services.data.actions,
+            services.fieldFormats
+          )
+        : undefined,
+    [
+      columnCellValueActions,
+      getLegendActionEventData,
+      handleLegendAction,
+      interactive,
+      services.data.actions,
+      services.fieldFormats,
+      visData,
+      visParams,
+    ]
+  );
+
   const rescaleFactor = useMemo(() => {
     const overallSum = visData.rows.reduce((sum, row) => sum + row[metricColumn.id], 0);
     const slices = visData.rows.map((row) => row[metricColumn.id] / overallSum);
@@ -446,15 +476,7 @@ const PartitionVisComponent = (props: PartitionVisComponentProps) => {
                     splitChartFormatter
                   );
                 }}
-                legendAction={getLegendActions(
-                  canFilter,
-                  getLegendActionEventData(visData),
-                  handleLegendAction,
-                  visParams,
-                  visData,
-                  services.data.actions,
-                  services.fieldFormats
-                )}
+                legendAction={legendActions}
                 theme={[
                   // Chart background should be transparent for the usage at Canvas.
                   { background: { color: 'transparent' } },

src/plugins/chart_expressions/expression_partition_vis/public/expression_renderers/partition_vis_renderer.test.tsx

@@ -0,0 +1,51 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+import { createMockPieParams, createMockVisData } from '../mocks';
+import { CellValueAction } from '../types';
+import { getColumnCellValueActions } from './partition_vis_renderer';
+
+const visParams = createMockPieParams();
+const visData = createMockVisData();
+
+const cellValueAction: CellValueAction = {
+  displayName: 'Test',
+  id: 'test',
+  iconType: 'test-icon',
+  execute: () => {},
+};
+
+describe('getColumnCellValueActions', () => {
+  it('should get column cellValue actions for each params bucket', async () => {
+    const result = await getColumnCellValueActions(visParams, visData, async () => [
+      cellValueAction,
+    ]);
+    expect(result).toHaveLength(visParams.dimensions.buckets?.length ?? 0);
+  });
+
+  it('should contain the cellValue actions', async () => {
+    const result = await getColumnCellValueActions(visParams, visData, async () => [
+      cellValueAction,
+      cellValueAction,
+    ]);
+    expect(result[0]).toEqual([cellValueAction, cellValueAction]);
+  });
+
+  it('should return empty array if no buckets', async () => {
+    const result = await getColumnCellValueActions(
+      { ...visParams, dimensions: { ...visParams.dimensions, buckets: undefined } },
+      visData,
+      async () => [cellValueAction]
+    );
+    expect(result).toEqual([]);
+  });
+
+  it('should return empty array if getCompatibleCellValueActions not passed', async () => {
+    const result = await getColumnCellValueActions(visParams, visData, undefined);
+    expect(result).toEqual([]);
+  });
+});

src/plugins/chart_expressions/expression_partition_vis/public/expression_renderers/partition_vis_renderer.tsx

@@ -11,14 +11,16 @@ import { render, unmountComponentAtNode } from 'react-dom';
 import { I18nProvider } from '@kbn/i18n-react';
 import { css } from '@emotion/react';
 import { i18n } from '@kbn/i18n';
-import { ExpressionRenderDefinition } from '@kbn/expressions-plugin/public';
+import { Datatable, ExpressionRenderDefinition } from '@kbn/expressions-plugin/public';
 import type { PersistedState } from '@kbn/visualizations-plugin/public';
 import { withSuspense } from '@kbn/presentation-util-plugin/public';
 import { KibanaThemeProvider } from '@kbn/kibana-react-plugin/public';
 import { METRIC_TYPE } from '@kbn/analytics';
+import { getColumnByAccessor } from '@kbn/visualizations-plugin/common/utils';
 import { VisTypePieDependencies } from '../plugin';
 import { PARTITION_VIS_RENDERER_NAME } from '../../common/constants';
-import { ChartTypes, RenderValue } from '../../common/types';
+import { CellValueAction, GetCompatibleCellValueActions } from '../types';
+import { ChartTypes, PartitionVisParams, RenderValue } from '../../common/types';
 // eslint-disable-next-line @kbn/imports/no_boundary_crossing
 import { extractContainerType, extractVisualizationType } from '../../../common';
 
@@ -42,6 +44,28 @@ const partitionVisRenderer = css({
   height: '100%',
 });
 
+/**
+ * Retrieves the compatible CELL_VALUE_TRIGGER actions indexed by column
+ **/
+export const getColumnCellValueActions = async (
+  visConfig: PartitionVisParams,
+  visData: Datatable,
+  getCompatibleCellValueActions: GetCompatibleCellValueActions | undefined
+) => {
+  if (!Array.isArray(visConfig.dimensions.buckets) || !getCompatibleCellValueActions) {
+    return [];
+  }
+  return Promise.all(
+    visConfig.dimensions.buckets.reduce<Array<Promise<CellValueAction[]>>>((acc, accessor) => {
+      const column = getColumnByAccessor(accessor, visData.columns);
+      if (column) {
+        acc.push(getCompatibleCellValueActions([{ columnMeta: column.meta }]));
+      }
+      return acc;
+    }, [])
+  );
+};
+
 export const getPartitionVisRenderer: (
   deps: VisTypePieDependencies
 ) => ExpressionRenderDefinition<RenderValue> = ({ getStartDeps }) => ({
@@ -60,6 +84,12 @@ export const getPartitionVisRenderer: (
       unmountComponentAtNode(domNode);
     });
 
+    const columnCellValueActions = await getColumnCellValueActions(
+      visConfig,
+      visData,
+      handlers.getCompatibleCellValueActions as GetCompatibleCellValueActions
+    );
+
     const renderComplete = () => {
       const executionContext = handlers.getExecutionContext();
       const containerType = extractContainerType(executionContext);
@@ -90,9 +120,11 @@ export const getPartitionVisRenderer: (
               visType={visConfig.isDonut ? ChartTypes.DONUT : visType}
               renderComplete={renderComplete}
               fireEvent={handlers.event}
+              interactive={handlers.isInteractive()}
               uiState={handlers.uiState as PersistedState}
               services={{ data: plugins.data, fieldFormats: plugins.fieldFormats }}
               syncColors={syncColors}
+              columnCellValueActions={columnCellValueActions}
             />
           </div>
         </KibanaThemeProvider>

src/plugins/chart_expressions/expression_partition_vis/public/types.ts

@@ -5,7 +5,7 @@
  * in compliance with, at your election, the Elastic License 2.0 or the Server
  * Side Public License, v 1.
  */
-import type { ValueClickContext } from '@kbn/embeddable-plugin/public';
+import type { CellValueContext, ValueClickContext } from '@kbn/embeddable-plugin/public';
 import { ChartsPluginSetup, ChartsPluginStart } from '@kbn/charts-plugin/public';
 import {
   Plugin as ExpressionsPublicPlugin,
@@ -35,3 +35,16 @@ export interface FilterEvent {
   name: 'filter';
   data: ValueClickContext['data'];
 }
+
+export interface CellValueAction {
+  id: string;
+  iconType: string;
+  displayName: string;
+  execute: (data: CellValueContext['data']) => void;
+}
+
+export type ColumnCellValueActions = CellValueAction[][];
+
+export type GetCompatibleCellValueActions = (
+  data: CellValueContext['data']
+) => Promise<CellValueAction[]>;

src/plugins/chart_expressions/expression_partition_vis/public/utils/filter_helpers.ts

@@ -125,3 +125,7 @@ export const getFilterEventData = (
     return acc;
   }, []);
 };
+
+export const getSeriesValueColumnIndex = (value: string, visData: Datatable): number => {
+  return visData.columns.findIndex(({ id }) => !!visData.rows.find((r) => r[id] === value));
+};