[Lens] Hide column in table

x-pack/plugins/lens/public/datatable_visualization/components/columns.tsx

@@ -9,7 +9,7 @@ import { i18n } from '@kbn/i18n';
 import { EuiDataGridColumn, EuiDataGridColumnCellActionProps } from '@elastic/eui';
 import type { Datatable, DatatableColumnMeta } from 'src/plugins/expressions';
 import type { FormatFactory } from '../../types';
-import type { DatatableColumns } from './types';
+import { ColumnConfig } from './table_basic';
 
 export const createGridColumns = (
   bucketColumns: string[],
@@ -22,10 +22,11 @@ export const createGridColumns = (
     negate?: boolean
   ) => void,
   isReadOnly: boolean,
-  columnConfig: DatatableColumns & { type: 'lens_datatable_columns' },
+  columnConfig: ColumnConfig,
   visibleColumns: string[],
   formatFactory: FormatFactory,
-  onColumnResize: (eventData: { columnId: string; width: number | undefined }) => void
+  onColumnResize: (eventData: { columnId: string; width: number | undefined }) => void,
+  onColumnHide: (eventData: { columnId: string }) => void
 ) => {
   const columnsReverseLookup = table.columns.reduce<
     Record<string, { name: string; index: number; meta?: DatatableColumnMeta }>
@@ -133,8 +134,7 @@ export const createGridColumns = (
         ]
       : undefined;
 
-    const initialWidth = columnConfig.columnWidth?.find(({ columnId }) => columnId === field)
-      ?.width;
+    const initialWidth = columnConfig.columns.find(({ columnId }) => columnId === field)?.width;
 
     const columnDefinition: EuiDataGridColumn = {
       id: field,
@@ -173,6 +173,16 @@ export const createGridColumns = (
                 'data-test-subj': 'lensDatatableResetWidth',
                 isDisabled: initialWidth == null,
               },
+              {
+                color: 'text',
+                size: 'xs',
+                onClick: () => onColumnHide({ columnId: field }),
+                iconType: 'empty',
+                label: i18n.translate('xpack.lens.table.hide.hideLabel', {
+                  defaultMessage: 'Hide',
+                }),
+                'data-test-subj': 'lensDatatableHide',
+              },
             ],
       },
     };

x-pack/plugins/lens/public/datatable_visualization/components/constants.ts

@@ -6,3 +6,4 @@
 
 export const LENS_EDIT_SORT_ACTION = 'sort';
 export const LENS_EDIT_RESIZE_ACTION = 'resize';
+export const LENS_TOGGLE_ACTION = 'toggle';

x-pack/plugins/lens/public/datatable_visualization/components/dimension_editor.tsx

@@ -0,0 +1,57 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+
+import React from 'react';
+import { i18n } from '@kbn/i18n';
+import { EuiSwitch, EuiFormRow } from '@elastic/eui';
+import { VisualizationDimensionEditorProps } from '../../types';
+import { DatatableVisualizationState } from '../visualization';
+
+export function TableDimensionEditor(
+  props: VisualizationDimensionEditorProps<DatatableVisualizationState>
+) {
+  const { state, setState, accessor } = props;
+  const column = state.columns.find((c) => c.columnId === accessor);
+
+  if (!column) {
+    return null;
+  }
+
+  return (
+    <EuiFormRow
+      label={i18n.translate('xpack.lens.table.columnVisibilityLabel', {
+        defaultMessage: 'Column hidden in table',
+      })}
+      display="columnCompressedSwitch"
+    >
+      <EuiSwitch
+        compressed
+        label={i18n.translate('xpack.lens.table.columnVisibilityLabel', {
+          defaultMessage: 'Column hidden in table',
+        })}
+        showLabel={false}
+        data-test-subj="lns-table-column-hidden"
+        checked={Boolean(column?.hidden)}
+        onChange={() => {
+          const newState = {
+            ...state,
+            columns: state.columns.map((currentColumn) => {
+              if (currentColumn.columnId === accessor) {
+                return {
+                  ...currentColumn,
+                  hidden: !column.hidden,
+                };
+              } else {
+                return currentColumn;
+              }
+            }),
+          };
+          setState(newState);
+        }}
+      />
+    </EuiFormRow>
+  );
+}

x-pack/plugins/lens/public/datatable_visualization/components/table_actions.test.ts

@@ -12,17 +12,19 @@ import {
   createGridFilterHandler,
   createGridResizeHandler,
   createGridSortingConfig,
+  createGridHideHandler,
 } from './table_actions';
-import { DatatableColumns, LensGridDirection } from './types';
+import { LensGridDirection } from './types';
+import { ColumnConfig } from './table_basic';
 
-function getDefaultConfig(): DatatableColumns & {
-  type: 'lens_datatable_columns';
-} {
+function getDefaultConfig(): ColumnConfig {
   return {
-    columnIds: [],
-    sortBy: '',
-    sortDirection: 'none',
-    type: 'lens_datatable_columns',
+    columns: [
+      { columnId: 'a', type: 'lens_datatable_column' },
+      { columnId: 'b', type: 'lens_datatable_column' },
+    ],
+    sortingColumnId: '',
+    sortingDirection: 'none',
   };
 }
 
@@ -205,24 +207,28 @@ describe('Table actions', () => {
 
       expect(setColumnConfig).toHaveBeenCalledWith({
         ...columnConfig,
-        columnWidth: [{ columnId: 'a', width: 100, type: 'lens_datatable_column_width' }],
+        columns: [
+          { columnId: 'a', width: 100, type: 'lens_datatable_column' },
+          {
+            columnId: 'b',
+            type: 'lens_datatable_column',
+          },
+        ],
       });
 
       expect(onEditAction).toHaveBeenCalledWith({ action: 'resize', columnId: 'a', width: 100 });
     });
 
     it('should pull out the table custom width from the local state when passing undefined', () => {
       const columnConfig = getDefaultConfig();
-      columnConfig.columnWidth = [
-        { columnId: 'a', width: 100, type: 'lens_datatable_column_width' },
-      ];
+      columnConfig.columns = [{ columnId: 'a', width: 100, type: 'lens_datatable_column' }];
 
       const resizer = createGridResizeHandler(columnConfig, setColumnConfig, onEditAction);
       resizer({ columnId: 'a', width: undefined });
 
       expect(setColumnConfig).toHaveBeenCalledWith({
         ...columnConfig,
-        columnWidth: [],
+        columns: [{ columnId: 'a', width: undefined, type: 'lens_datatable_column' }],
       });
 
       expect(onEditAction).toHaveBeenCalledWith({
@@ -232,4 +238,23 @@ describe('Table actions', () => {
       });
     });
   });
+  describe('Column hiding', () => {
+    const setColumnConfig = jest.fn();
+
+    it('should allow to hide column', () => {
+      const columnConfig = getDefaultConfig();
+      const hiding = createGridHideHandler(columnConfig, setColumnConfig, onEditAction);
+      hiding({ columnId: 'a' });
+
+      expect(setColumnConfig).toHaveBeenCalledWith({
+        ...columnConfig,
+        columns: [
+          { columnId: 'a', hidden: true, type: 'lens_datatable_column' },
+          { columnId: 'b', type: 'lens_datatable_column' },
+        ],
+      });
+
+      expect(onEditAction).toHaveBeenCalledWith({ action: 'toggle', columnId: 'a' });
+    });
+  });
 });

x-pack/plugins/lens/public/datatable_visualization/components/table_actions.ts

@@ -7,43 +7,30 @@ import type { EuiDataGridSorting } from '@elastic/eui';
 import type { Datatable } from 'src/plugins/expressions';
 import type { LensFilterEvent } from '../../types';
 import type {
-  DatatableColumns,
   LensGridDirection,
   LensResizeAction,
   LensSortAction,
+  LensToggleAction,
 } from './types';
+import { ColumnConfig } from './table_basic';
 
 import { desanitizeFilterContext } from '../../utils';
 
 export const createGridResizeHandler = (
-  columnConfig: DatatableColumns & {
-    type: 'lens_datatable_columns';
-  },
-  setColumnConfig: React.Dispatch<
-    React.SetStateAction<
-      DatatableColumns & {
-        type: 'lens_datatable_columns';
-      }
-    >
-  >,
+  columnConfig: ColumnConfig,
+  setColumnConfig: React.Dispatch<React.SetStateAction<ColumnConfig>>,
   onEditAction: (data: LensResizeAction['data']) => void
 ) => (eventData: { columnId: string; width: number | undefined }) => {
   // directly set the local state of the component to make sure the visualization re-renders immediately,
   // re-layouting and taking up all of the available space.
   setColumnConfig({
     ...columnConfig,
-    columnWidth: [
-      ...(columnConfig.columnWidth || []).filter(({ columnId }) => columnId !== eventData.columnId),
-      ...(eventData.width !== undefined
-        ? [
-            {
-              columnId: eventData.columnId,
-              width: eventData.width,
-              type: 'lens_datatable_column_width' as const,
-            },
-          ]
-        : []),
-    ],
+    columns: columnConfig.columns.map((column) => {
+      if (column.columnId === eventData.columnId) {
+        return { ...column, width: eventData.width };
+      }
+      return column;
+    }),
   });
   return onEditAction({
     action: 'resize',
@@ -52,6 +39,27 @@ export const createGridResizeHandler = (
   });
 };
 
+export const createGridHideHandler = (
+  columnConfig: ColumnConfig,
+  setColumnConfig: React.Dispatch<React.SetStateAction<ColumnConfig>>,
+  onEditAction: (data: LensToggleAction['data']) => void
+) => (eventData: { columnId: string }) => {
+  // directly set the local state of the component to make sure the visualization re-renders immediately
+  setColumnConfig({
+    ...columnConfig,
+    columns: columnConfig.columns.map((column) => {
+      if (column.columnId === eventData.columnId) {
+        return { ...column, hidden: true };
+      }
+      return column;
+    }),
+  });
+  return onEditAction({
+    action: 'toggle',
+    columnId: eventData.columnId,
+  });
+};
+
 export const createGridFilterHandler = (
   tableRef: React.MutableRefObject<Datatable>,
   onClickValue: (data: LensFilterEvent['data']) => void
@@ -83,7 +91,7 @@ export const createGridFilterHandler = (
 };
 
 export const createGridSortingConfig = (
-  sortBy: string,
+  sortBy: string | undefined,
   sortDirection: LensGridDirection,
   onEditAction: (data: LensSortAction['data']) => void
 ): EuiDataGridSorting => ({