select new row/column

microsoft · Sep 21, 2023 · 5a864e9 · 5a864e9
1 parent e808fbf
commit 5a864e9
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/packages/roosterjs-editor-plugins/lib/plugins/TableResize/editors/TableInserter.ts b/packages/roosterjs-editor-plugins/lib/plugins/TableResize/editors/TableInserter.ts
@@ -1,7 +1,7 @@
 import Disposable from '../../../pluginUtils/Disposable';
 import TableEditFeature from './TableEditorFeature';
 import { createElement, getIntersectedRect, normalizeRect, VTable } from 'roosterjs-editor-dom';
-import { CreateElementData, IEditor, TableOperation } from 'roosterjs-editor-types';
+import { CreateElementData, IEditor, TableOperation, TableSelection } from 'roosterjs-editor-types';
 
 const INSERTER_COLOR = '#4A4A4A';
 const INSERTER_COLOR_DARK_MODE = 'white';
@@ -118,6 +118,21 @@ class TableInsertHandler implements Disposable {
         vtable.writeBack();
 
         this.onInsert(vtable.table);
+
+        // Select newly inserted row or column
+        if (vtable.row != undefined && vtable.col != undefined && vtable.cells) {
+            const inserted: TableSelection = this.isHorizontal
+                ? {
+                      firstCell: { x: 0, y: vtable.row + 1 },
+                      lastCell: { x: vtable.cells[vtable.row].length - 1, y: vtable.row + 1 },
+                  }
+                : {
+                      firstCell: { x: vtable.col + 1, y: 0 },
+                      lastCell: { x: vtable.col + 1, y: vtable.cells.length - 1 },
+                  };
+
+            this.editor.select(vtable.table, inserted);
+        }
     };
 }