add getRangeCellSelection instance method

Author: huangshuwei

packages/ve-table/src/index.jsx

@@ -3284,6 +3284,37 @@ export default {
             }
         },
 
+        /*
+        get range cell selection
+        */
+        [INSTANCE_METHODS.GET_RANGE_CELL_SELECTION]() {
+            const {
+                cellSelectionData,
+                cellSelectionRangeData,
+                allRowKeys,
+                colgroups,
+            } = this;
+
+            const { rowKey, colKey } = cellSelectionData.currentCell;
+
+            if (!isEmptyValue(rowKey) && !isEmptyValue(colKey)) {
+                let selectionRangeKeys = getSelectionRangeKeys({
+                    cellSelectionRangeData,
+                });
+
+                let selectionRangeIndexes = getSelectionRangeIndexes({
+                    cellSelectionRangeData,
+                    colgroups,
+                    allRowKeys,
+                });
+
+                return {
+                    selectionRangeKeys,
+                    selectionRangeIndexes,
+                };
+            }
+        },
+
         /*
         set all cell selection and column to visible
         */

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

@@ -198,6 +198,8 @@ export const INSTANCE_METHODS = {
     SET_CELL_SELECTION: "setCellSelection",
     // set range cell selection
     SET_RANGE_CELL_SELECTION: "setRangeCellSelection",
+    // get range cell selection
+    GET_RANGE_CELL_SELECTION: "getRangeCellSelection",
     // set all cell selection
     SET_ALL_CELL_SELECTION: "setAllCellSelection",
     // hide columns by keys

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

@@ -629,6 +629,77 @@ describe("veTable cell selection", () => {
         });
     });
 
+    it("table instance: setAllCellSelection method", async () => {
+        const wrapper = mount(veTable, {
+            propsData: {
+                columns: COLUMNS,
+                tableData: TABLE_DATA,
+                rowKeyFieldName: "rowKey",
+            },
+        });
+
+        wrapper.vm.setAllCellSelection();
+
+        await later();
+
+        expect(wrapper.vm.cellSelectionRangeData).toEqual({
+            bottomRowKey: "5",
+            leftColKey: "b",
+            rightColKey: "d",
+            topRowKey: "1",
+        });
+    });
+
+    it("table instance: getRangeCellSelection method", async () => {
+        const wrapper = mount(veTable, {
+            propsData: {
+                columns: COLUMNS,
+                tableData: TABLE_DATA,
+                rowKeyFieldName: "rowKey",
+            },
+        });
+
+        wrapper.vm.setRangeCellSelection({
+            startRowKey: "2",
+            startColKey: "a",
+            endRowKey: "5",
+            endColKey: "c",
+            isScrollToStartCell: true,
+        });
+
+        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");
+        expect(wrapper.vm.cellSelectionRangeData).toEqual({
+            bottomRowKey: "5",
+            leftColKey: "a",
+            rightColKey: "c",
+            topRowKey: "2",
+        });
+
+        const rangeCellSelection = wrapper.vm.getRangeCellSelection();
+        expect(rangeCellSelection).toEqual({
+            selectionRangeIndexes: {
+                endColIndex: 2,
+                endRowIndex: 4,
+                startColIndex: 0,
+                startRowIndex: 1,
+            },
+            selectionRangeKeys: {
+                endColKey: "c",
+                endRowKey: "5",
+                startColKey: "a",
+                startRowKey: "2",
+            },
+        });
+    });
+
     /* it("virtual scroll keyboard events", async () => {
         const mockFn = jest.fn();