diff --git a/src/plugins/vis_type_table/public/components/table_vis_component.tsx b/src/plugins/vis_type_table/public/components/table_vis_component.tsx index a9b8c6a8ba36..6819caae308c 100644 --- a/src/plugins/vis_type_table/public/components/table_vis_component.tsx +++ b/src/plugins/vis_type_table/public/components/table_vis_component.tsx @@ -3,12 +3,13 @@ * SPDX-License-Identifier: Apache-2.0 */ -import React, { useState, useCallback, useMemo } from 'react'; +import React, { useState, useCallback, useMemo, useRef, useEffect } from 'react'; +import { isEqual } from 'lodash'; import { EuiDataGridProps, EuiDataGrid } from '@elastic/eui'; import { IInterpreterRenderHandlers } from 'src/plugins/expressions'; import { Table } from '../table_vis_response_handler'; -import { TableVisConfig } from '../types'; +import { TableVisConfig, ColumnWidth } from '../types'; import { getDataGridColumns } from './table_vis_grid_columns'; interface TableVisComponentProps { @@ -45,7 +46,56 @@ export const TableVisComponent = ({ table, visConfig, handlers }: TableVisCompon }) as EuiDataGridProps['renderCellValue']; }, [rows, pagination.pageIndex, pagination.pageSize]); - const dataGridColumns = getDataGridColumns(table, visConfig, handlers); + // resize column + const [columnsWidth, setColumnsWidth] = useState( + handlers.uiState.get('vis.columnsWidth') || [] + ); + const curColumnsWidth = useRef<{ + columnsWidth: ColumnWidth[]; + }>({ + columnsWidth: handlers.uiState?.get('vis.columnsWidth'), + }); + + const onColumnResize: EuiDataGridProps['onColumnResize'] = useCallback( + ({ columnId, width }) => { + setColumnsWidth((prevState) => { + const nextColIndex = columns.findIndex((c) => c.id === columnId); + const prevColIndex = prevState.findIndex((c) => c.colIndex === nextColIndex); + const nextState = [...prevState]; + const updatedColWidth = { colIndex: nextColIndex, width }; + + // if updated column index is not found, then add it to nextState + // else reset it in nextState + if (prevColIndex < 0) nextState.push(updatedColWidth); + else nextState[prevColIndex] = updatedColWidth; + + // update uiState + handlers.uiState?.set('vis.columnsWidth', nextState); + return nextState; + }); + }, + [columns, setColumnsWidth, handlers.uiState] + ); + + useEffect(() => { + const updateTable = () => { + const updatedVisState = handlers.uiState?.getChanges()?.vis; + if (!isEqual(updatedVisState?.columnsWidth, curColumnsWidth.current.columnsWidth)) { + curColumnsWidth.current.columnsWidth = updatedVisState?.columnsWidth; + setColumnsWidth(updatedVisState?.columnsWidth || []); + } + }; + + if (handlers.uiState) { + handlers.uiState.on('change', updateTable); + } + + return () => { + handlers.uiState?.off('change', updateTable); + }; + }, [handlers.uiState]); + + const dataGridColumns = getDataGridColumns(table, visConfig, handlers, columnsWidth); return ( ); }; diff --git a/src/plugins/vis_type_table/public/components/table_vis_grid_columns.tsx b/src/plugins/vis_type_table/public/components/table_vis_grid_columns.tsx index 5541e39d018c..b1a9443fe059 100644 --- a/src/plugins/vis_type_table/public/components/table_vis_grid_columns.tsx +++ b/src/plugins/vis_type_table/public/components/table_vis_grid_columns.tsx @@ -8,13 +8,14 @@ import { i18n } from '@osd/i18n'; import { EuiDataGridColumn, EuiDataGridColumnCellActionProps } from '@elastic/eui'; import { IInterpreterRenderHandlers } from 'src/plugins/expressions'; import { Table } from '../table_vis_response_handler'; -import { TableVisConfig } from '../types'; +import { TableVisConfig, ColumnWidth } from '../types'; import { convertToFormattedData } from '../utils'; export const getDataGridColumns = ( table: Table, visConfig: TableVisConfig, - handlers: IInterpreterRenderHandlers + handlers: IInterpreterRenderHandlers, + columnsWidth: ColumnWidth[] ) => { const { formattedRows, formattedColumns } = convertToFormattedData(table, visConfig); @@ -117,6 +118,8 @@ export const getDataGridColumns = ( ] : undefined; + const initialWidth = columnsWidth.find((c) => c.colIndex === colIndex); + const dataGridColumn: EuiDataGridColumn = { id: col.id, display: col.title, @@ -138,6 +141,9 @@ export const getDataGridColumns = ( }, cellActions, }; + if (initialWidth) { + dataGridColumn.initialWidth = initialWidth.width; + } return dataGridColumn; }); }; diff --git a/src/plugins/vis_type_table/public/types.ts b/src/plugins/vis_type_table/public/types.ts index 170683719461..b3a377065140 100644 --- a/src/plugins/vis_type_table/public/types.ts +++ b/src/plugins/vis_type_table/public/types.ts @@ -64,3 +64,8 @@ export interface FormattedColumn { sumTotal?: number; total?: number; } + +export interface ColumnWidth { + colIndex: number; + width: number; +}