fix(grid): only operate selected data which is visible

worktile · Feb 21, 2025 · 20563a2 · 20563a2
1 parent 684b952
commit 20563a2
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/packages/grid/src/services/selection.service.ts b/packages/grid/src/services/selection.service.ts
@@ -101,19 +101,19 @@ export class AITableGridSelectionService {
 
     selectCells(startCell: AIRecordFieldIdPath, endCell?: AIRecordFieldIdPath) {
         const [startRecordId, startFieldId] = startCell;
-        const records = this.aiTable.records();
-        const fields = this.aiTable.fields();
+        const records = this.aiTable.context!.linearRows();
+        const fields = AITable.getVisibleFields(this.aiTable);
         const selectedCells = new Set<string>();
 
         if (!endCell) {
             selectedCells.add(`${startRecordId}:${startFieldId}`);
         } else {
             const [endRecordId, endFieldId] = endCell;
 
-            const startRowIndex = records.findIndex((record) => record._id === startRecordId);
-            const endRowIndex = records.findIndex((record) => record._id === endRecordId);
-            const startColIndex = fields.findIndex((field) => field._id === startFieldId);
-            const endColIndex = fields.findIndex((field) => field._id === endFieldId);
+            const startRowIndex = this.aiTable.context!.visibleRowsIndexMap().get(startRecordId)!;
+            const endRowIndex = this.aiTable.context!.visibleRowsIndexMap().get(endRecordId)!;
+            const startColIndex = this.aiTable.context!.visibleColumnsIndexMap().get(startFieldId)!;
+            const endColIndex = this.aiTable.context!.visibleColumnsIndexMap().get(endFieldId)!;
 
             const minRowIndex = Math.min(startRowIndex, endRowIndex);
             const maxRowIndex = Math.max(startRowIndex, endRowIndex);