add setCellSelection instance method

Author: huangshuwei

packages/ve-table/src/index.jsx

@@ -7,6 +7,8 @@ import {
     getContextmenuBodyOptionCollection,
     createEmptyRowData,
     isContextmenuPanelClicked,
+    getRowKey,
+    getColumnByColkey,
 } from "./util";
 import {
     getValByUnit,
@@ -1585,27 +1587,15 @@ export default {
 
         // cell selection by click
         cellSelectionByClick({ rowData, column }) {
-            const { rowKeyFieldName, cellSelectionOption } = this;
+            const { rowKeyFieldName } = this;
 
-            const rowKey = rowData[rowKeyFieldName];
+            const rowKey = getRowKey(rowData, rowKeyFieldName);
 
-            // update cell selection
-            if (
-                !(
-                    cellSelectionOption &&
-                    isBoolean(cellSelectionOption.enable) &&
-                    cellSelectionOption.enable === false
-                )
-            ) {
-                if (!isEmptyValue(rowKey) && !isEmptyValue(column.key)) {
-                    this.cellSelectionKeyChange({
-                        rowKey,
-                        colKey: column.key,
-                    });
-
-                    this.columnToVisible(column);
-                }
-            }
+            // set cell selection and column to visible
+            this[INSTANCE_METHODS.SET_CELL_SELECTION]({
+                rowKey,
+                colKey: column.key,
+            });
         },
 
         /*
@@ -1802,6 +1792,31 @@ export default {
             }
         },
 
+        /*
+        set cell selection and column to visible
+        */
+        [INSTANCE_METHODS.SET_CELL_SELECTION]({ rowKey, colKey }) {
+            const { cellSelectionOption } = this;
+
+            if (
+                cellSelectionOption &&
+                isBoolean(cellSelectionOption.enable) &&
+                cellSelectionOption.enable === false
+            ) {
+                return false;
+            }
+
+            if (!isEmptyValue(rowKey) && !isEmptyValue(colKey)) {
+                this.cellSelectionKeyChange({
+                    rowKey,
+                    colKey,
+                });
+
+                const column = getColumnByColkey(colKey, this.colgroups);
+                this.columnToVisible(column);
+            }
+        },
+
         // hide columns by keys
         [INSTANCE_METHODS.HIDE_COLUMNS_BY_KEYS](keys) {
             if (!isEmptyArray(keys)) {

packages/ve-table/src/util/constant.js

@@ -134,6 +134,8 @@ export const INSTANCE_METHODS = {
     STOP_EDITING_CELL: "stopEditingCell",
     // set highlight row
     SET_HIGHLIGHT_ROW: "setHighlightRow",
+    // set cell selection
+    SET_CELL_SELECTION: "setCellSelection",
     // hide columns by keys
     HIDE_COLUMNS_BY_KEYS: "hideColumnsByKeys",
     // show columns by keys

tests/unit/specs/ve-table-cell-selection.spec.js

@@ -573,6 +573,28 @@ describe("veTable cell selection", () => {
         expect(firstCell.find(".ve-table-cell-selection").exists()).toBe(false);
     });
 
+    it("table instance: setCellSelection method", async () => {
+        const wrapper = mount(veTable, {
+            propsData: {
+                columns: COLUMNS,
+                tableData: TABLE_DATA,
+                rowKeyFieldName: "rowKey",
+            },
+        });
+
+        wrapper.vm.setCellSelection({ rowKey: "2", colKey: "a" });
+
+        await later();
+
+        const selectionTd = wrapper
+            .findAll(".ve-table-body-tr")
+            .at(1)
+            .findAll(".ve-table-body-td")
+            .at(0);
+
+        expect(selectionTd.classes()).toContain("ve-table-cell-selection");
+    });
+
     /* it("virtual scroll keyboard events", async () => {
         const mockFn = jest.fn();