feat: selected cell highlight

packages/s2-core/__tests__/unit/interaction/base-interaction/click/data-cell-click-spec.ts

@@ -8,11 +8,13 @@ import type { S2Options } from '@/common/interface';
 import type { SpreadSheet } from '@/sheet-type';
 import {
   HOVER_FOCUS_DURATION,
+  type InteractionCellSelectedHighlightType,
   InteractionName,
   InteractionStateName,
   InterceptType,
   S2Event,
 } from '@/common/constant';
+import type { Node } from '@/facet/layout/node';
 
 jest.mock('@/interaction/event-controller');
 
@@ -50,6 +52,7 @@ describe('Interaction Data Cell Click Tests', () => {
     expect(s2.interaction.getState()).toEqual({
       cells: [mockCellInfo.mockCellMeta],
       stateName: InteractionStateName.SELECTED,
+      onUpdateCells: expect.any(Function),
     });
     expect(s2.showTooltipWithInfo).toHaveBeenCalled();
   });
@@ -68,6 +71,7 @@ describe('Interaction Data Cell Click Tests', () => {
     expect(s2.interaction.getState()).toEqual({
       cells: [mockCellInfo.mockCellMeta],
       stateName: InteractionStateName.SELECTED,
+      onUpdateCells: expect.any(Function),
     });
   });
 
@@ -160,4 +164,66 @@ describe('Interaction Data Cell Click Tests', () => {
     expect(s2.interaction.isHoverFocusState()).toBeFalsy();
     expect(clearHoverTimerSpy).toHaveBeenCalledTimes(2);
   });
+
+  test('should highlight the column header cell when data cell clicked', () => {
+    const headerCellId0 = 'header-0';
+    const headerCellId1 = 'header-1';
+    const columnNode: Array<Partial<Node>> = [
+      {
+        belongsCell: {
+          getMeta: () => ({
+            id: headerCellId0,
+            colIndex: -1,
+            rowIndex: -1,
+          }),
+        } as any,
+        id: headerCellId0,
+      },
+      {
+        belongsCell: {
+          getMeta: () => ({
+            id: headerCellId1,
+            colIndex: -1,
+            rowIndex: -1,
+          }),
+        } as any,
+        id: headerCellId1,
+      },
+    ];
+    s2.getColumnNodes = jest.fn(() => columnNode) as any;
+    s2.getRowNodes = jest.fn(() => []);
+
+    const firstDataCellInfo = createMockCellInfo(
+      `${headerCellId0}[&]first-data-cell`,
+    );
+    s2.getCell = () => firstDataCellInfo.mockCell as any;
+
+    s2.setOptions({
+      interaction: {
+        selectedCellHighlight: {
+          colHeader: true,
+        } as InteractionCellSelectedHighlightType,
+      },
+    });
+
+    const mockHeaderCellInfo = createMockCellInfo(headerCellId0, {
+      colIndex: columnNode[0].belongsCell.getMeta().colIndex,
+      rowIndex: columnNode[0].belongsCell.getMeta().rowIndex,
+    });
+
+    s2.interaction.getAllColHeaderCells = jest.fn();
+    s2.interaction.updateCells = jest.fn();
+
+    s2.emit(S2Event.DATA_CELL_CLICK, {
+      stopPropagation() {},
+    } as unknown as GEvent);
+
+    expect(s2.interaction.getState()).toEqual({
+      cells: [firstDataCellInfo.mockCellMeta, mockHeaderCellInfo.mockCellMeta],
+      stateName: InteractionStateName.SELECTED,
+      onUpdateCells: expect.any(Function),
+    });
+    expect(s2.interaction.getAllColHeaderCells).toHaveBeenCalled();
+    expect(s2.interaction.updateCells).toHaveBeenCalled();
+  });
 });

packages/s2-core/__tests__/unit/interaction/brush-selection/data-brush-selection-spec.ts

@@ -95,6 +95,12 @@ describe('Interaction Data Cell Brush Selection Tests', () => {
     mockSpreadSheetInstance.showTooltipWithInfo = jest.fn();
     mockRootInteraction.getPanelGroupAllDataCells = () =>
       panelGroupAllDataCells;
+    mockRootInteraction.getSelectedCellHighlight = () => ({
+      rowHeader: false,
+      colHeader: false,
+      rowCells: false,
+      colCells: false,
+    });
     mockSpreadSheetInstance.interaction = mockRootInteraction;
     mockSpreadSheetInstance.render();
     mockSpreadSheetInstance.facet.layoutResult.colLeafNodes = Array.from(
@@ -135,6 +141,12 @@ describe('Interaction Data Cell Brush Selection Tests', () => {
     mockSpreadSheetInstance.setOptions({
       interaction: { selectedCellHighlight: true },
     });
+    mockRootInteraction.getSelectedCellHighlight = () => ({
+      rowHeader: true,
+      colHeader: true,
+      rowCells: false,
+      colCells: false,
+    });
 
     brushSelectionInstance.getBrushRange = () => {
       return {

packages/s2-core/__tests__/unit/interaction/data-cell-multi-selection-spec.ts

@@ -109,6 +109,7 @@ describe('Interaction Data Cell Multi Selection Tests', () => {
       expect(s2.interaction.getState()).toEqual({
         cells: [mockCellA.mockCellMeta, mockCellB.mockCellMeta],
         stateName: InteractionStateName.SELECTED,
+        onUpdateCells: expect.any(Function),
       });
 
       expect(
@@ -151,6 +152,7 @@ describe('Interaction Data Cell Multi Selection Tests', () => {
       expect(s2.interaction.getState()).toEqual({
         cells: [mockCellB.mockCellMeta],
         stateName: InteractionStateName.SELECTED,
+        onUpdateCells: expect.any(Function),
       });
 
       expect(

packages/s2-core/__tests__/unit/utils/export/copy-spec.ts

@@ -308,6 +308,27 @@ describe('List Table Core Data Process', () => {
     const data = getSelectedData(sss);
     expect(data).toBe(convertString(newLineText));
   });
+
+  it('should copy row data when select data row cell', () => {
+    s2.setOptions({
+      interaction: {
+        selectedCellHighlight: {
+          rowCells: true,
+        },
+      },
+    });
+
+    const cell = s2.interaction
+      .getAllCells()
+      .filter(({ cellType }) => cellType === CellTypes.DATA_CELL)[0];
+
+    s2.interaction.changeState({
+      cells: [getCellMeta(cell)],
+      stateName: InteractionStateName.SELECTED,
+    });
+
+    expect(getSelectedData(s2).split('\t').length).toBe(5);
+  });
 });
 
 describe('Pivot Table Core Data Process', () => {

packages/s2-core/__tests__/util/helpers.ts

@@ -124,12 +124,13 @@ export function getGradient(
 
 export const createMockCellInfo = (
   cellId: string,
-  { colIndex = 0, rowIndex = 0 } = {},
+  { colIndex = 0, rowIndex = 0, colId = '0' } = {},
 ) => {
   const mockCellViewMeta: Partial<ViewMeta> = {
     id: cellId,
     colIndex,
     rowIndex,
+    colId,
     type: undefined,
     x: 0,
     y: 0,
@@ -154,6 +155,7 @@ export const createMockCellInfo = (
     'y',
     'update',
     'spreadsheet',
+    'colId',
   ]);
   const mockCell = {
     ...mockCellViewMeta,

packages/s2-core/src/cell/data-cell.ts

@@ -22,7 +22,11 @@ import type {
   ViewMetaIndexType,
 } from '../common/interface';
 import { getBorderPositionAndStyle, getMaxTextWidth } from '../utils/cell/cell';
-import { includeCell } from '../utils/cell/data-cell';
+import {
+  includeCell,
+  shouldUpdateBySelectedCellsHighlight,
+  updateBySelectedCellsHighlight,
+} from '../utils/cell/data-cell';
 import { getIconPositionCfg } from '../utils/condition/condition';
 import { renderLine, renderRect, updateShapeAttr } from '../utils/g-renders';
 import { EMPTY_PLACEHOLDER } from '../common/constant/basic';
@@ -80,6 +84,7 @@ export class DataCell extends BaseCell<ViewMeta> {
 
   protected handleSelect(cells: CellMeta[]) {
     const currentCellType = cells?.[0]?.type;
+
     switch (currentCellType) {
       // 列多选
       case CellTypes.COL_CELL:
@@ -91,7 +96,9 @@ export class DataCell extends BaseCell<ViewMeta> {
         break;
       // 单元格单选/多选
       case CellTypes.DATA_CELL:
-        if (includeCell(cells, this)) {
+        if (shouldUpdateBySelectedCellsHighlight(this.spreadsheet)) {
+          updateBySelectedCellsHighlight(cells, this, this.spreadsheet);
+        } else if (includeCell(cells, this)) {
           this.updateByState(InteractionStateName.SELECTED);
         } else if (
           this.spreadsheet.options.interaction.selectedCellsSpotlight
@@ -134,7 +141,7 @@ export class DataCell extends BaseCell<ViewMeta> {
 
   public update() {
     const stateName = this.spreadsheet.interaction.getCurrentStateName();
-    const cells = this.spreadsheet.interaction.getCells();
+    const cells = this.spreadsheet.interaction.getCells([CellTypes.DATA_CELL]);
 
     if (stateName === InteractionStateName.ALL_SELECTED) {
       this.updateByState(InteractionStateName.SELECTED);

packages/s2-core/src/cell/header-cell.ts

@@ -325,10 +325,11 @@ export abstract class HeaderCell extends BaseCell<Node> {
     }
   }
 
-  protected handleSelect(cells: CellMeta[], nodes: Node[]) {
+  protected handleSelect(cells: Array<CellMeta>, nodes: Node[]) {
     if (includeCell(cells, this)) {
       this.updateByState(InteractionStateName.SELECTED);
     }
+
     const selectedNodeIds = map(nodes, 'id');
     if (includes(selectedNodeIds, this.meta.id)) {
       this.updateByState(InteractionStateName.SELECTED);
@@ -382,7 +383,10 @@ export abstract class HeaderCell extends BaseCell<Node> {
   public update() {
     const { interaction } = this.spreadsheet;
     const stateInfo = interaction?.getState();
-    const cells = interaction?.getCells();
+    const cells = interaction?.getCells([
+      CellTypes.COL_CELL,
+      CellTypes.ROW_CELL,
+    ]);
 
     if (!first(cells)) {
       return;

packages/s2-core/src/common/constant/interaction.ts

@@ -112,3 +112,10 @@ export enum ScrollDirection {
   LEADING = 'leading',
   TRAILING = 'trailing',
 }
+
+export interface InteractionCellSelectedHighlightType {
+  rowHeader?: boolean; // 高亮行头
+  colHeader?: boolean; // 高亮列头
+  rowCells?: boolean; // 高亮选中单元格所在行
+  colCells?: boolean; // 高亮选中单元格所在列
+}

packages/s2-core/src/common/interface/basic.ts

@@ -515,3 +515,5 @@ export interface GridInfo {
   cols: number[];
   rows: number[];
 }
+
+export type RowData = Data | DataType;

packages/s2-core/src/common/interface/interaction.ts

@@ -4,6 +4,7 @@ import type {
   CellTypes,
   InterceptType,
   ScrollbarPositionType,
+  InteractionCellSelectedHighlightType,
 } from '../constant';
 import type {
   BaseCell,
@@ -35,6 +36,7 @@ export interface CellMeta {
   colIndex: number;
   rowIndex: number;
   type: CellTypes;
+  rowQuery?: Record<string, any>;
   [key: string]: unknown;
 }
 
@@ -169,7 +171,7 @@ export interface InteractionOptions {
   // https://developer.mozilla.org/zh-CN/docs/Web/API/EventTarget/addEventListener
   eventListenerOptions?: boolean | AddEventListenerOptions;
   // highlight col and row header for selected cell
-  selectedCellHighlight?: boolean;
+  selectedCellHighlight?: boolean | InteractionCellSelectedHighlightType;
   // https://developer.mozilla.org/en-US/docs/Web/CSS/overscroll-behavior
   overscrollBehavior?: 'auto' | 'none' | 'contain';
   /** ***********CUSTOM INTERACTION HOOKS**************** */

packages/s2-core/src/data-set/base-data-set.ts

@@ -23,6 +23,7 @@ import {
   getValueRangeState,
   setValueRangeState,
 } from '../utils/condition/state-controller';
+import type { CellMeta, RowData } from '../common';
 import type { CellDataParams, DataType } from './index';
 
 export abstract class BaseDataSet {
@@ -177,4 +178,10 @@ export abstract class BaseDataSet {
   public moreThanOneValue() {
     return this.fields?.values?.length > 1;
   }
+
+  /**
+   * get a row cells data including cell
+   * @param cells
+   */
+  public abstract getRowData(cells: CellMeta): RowData;
 }

packages/s2-core/src/data-set/pivot-data-set.ts

@@ -34,6 +34,7 @@ import type {
   PartDrillDownFieldInLevel,
   S2DataConfig,
   ViewMeta,
+  RowData,
 } from '../common/interface';
 import { Node } from '../facet/layout/node';
 import {
@@ -56,6 +57,7 @@ import {
 } from '../utils/dataset/pivot-data-set';
 import { calcActionByType } from '../utils/number-calculate';
 import { handleSortAction } from '../utils/sort-action';
+import type { CellMeta } from '../common';
 import { BaseDataSet } from './base-data-set';
 import type {
   CellDataParams,
@@ -618,4 +620,8 @@ export class PivotDataSet extends BaseDataSet {
   private isCustomMeasuresPosition(customValueOrder?: number) {
     return isNumber(customValueOrder);
   }
+
+  public getRowData(cell: CellMeta): RowData {
+    return this.getMultiData(cell.rowQuery);
+  }
 }

packages/s2-core/src/data-set/table-data-set.ts

@@ -1,6 +1,8 @@
 import { each, orderBy, filter, includes, isFunction, isObject } from 'lodash';
 import { isAscSort, isDescSort } from '..';
 import type { S2DataConfig } from '../common/interface';
+import type { CellMeta } from '../common';
+import type { RowData } from '../common/interface/basic';
 import type { CellDataParams, DataType } from './interface';
 import { BaseDataSet } from './base-data-set';
 
@@ -167,4 +169,8 @@ export class TableDataSet extends BaseDataSet {
   public getMultiData(query: DataType, isTotals?: boolean): DataType[] {
     return this.displayData;
   }
+
+  public getRowData(cell: CellMeta): RowData {
+    return this.getCellData({ query: { rowIndex: cell.rowIndex } });
+  }
 }

packages/s2-core/src/interaction/base-interaction/click/data-cell-click.ts

@@ -13,7 +13,8 @@ import type {
 } from '../../../common/interface';
 import {
   getCellMeta,
-  updateRowColCells,
+  getInteractionCells,
+  afterSelectDataCells,
 } from '../../../utils/interaction/select-event';
 import {
   getTooltipOptions,
@@ -30,7 +31,7 @@ export class DataCellClick extends BaseEvent implements BaseEventImplement {
     this.spreadsheet.on(S2Event.DATA_CELL_CLICK, (event: CanvasEvent) => {
       event.stopPropagation();
 
-      const { interaction, options } = this.spreadsheet;
+      const { interaction } = this.spreadsheet;
       interaction.clearHoverTimer();
 
       if (interaction.hasIntercepts([InterceptType.CLICK])) {
@@ -60,15 +61,12 @@ export class DataCellClick extends BaseEvent implements BaseEventImplement {
       }
 
       interaction.changeState({
-        cells: [getCellMeta(cell)],
+        cells: getInteractionCells(getCellMeta(cell), this.spreadsheet),
         stateName: InteractionStateName.SELECTED,
+        onUpdateCells: afterSelectDataCells,
       });
       this.spreadsheet.emit(S2Event.GLOBAL_SELECTED, [cell]);
       this.showTooltip(event, meta);
-
-      if (options.interaction.selectedCellHighlight) {
-        updateRowColCells(meta);
-      }
     });
   }