Internal: Adjusted to use the new API.

ckeditor · Mar 10, 2020 · 92545d0 · 92545d0
1 parent f952d62
commit 92545d0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/commands/setheadercolumncommand.js b/src/commands/setheadercolumncommand.js
@@ -10,7 +10,7 @@
 import Command from '@ckeditor/ckeditor5-core/src/command';
 
 import { updateNumericAttribute } from './utils';
-import { getTableCellsInSelection } from '../tableselection/utils';
+import { getSelectionAffectedTableCells } from '../utils';
 
 /**
  * The header column command.
@@ -33,7 +33,7 @@ export default class SetHeaderColumnCommand extends Command {
 	 */
 	refresh() {
 		const model = this.editor.model;
-		const selectedCells = getTableCellsInSelection( model.document.selection, true );
+		const selectedCells = getSelectionAffectedTableCells( model.document.selection );
 		const isInTable = selectedCells.length > 0;
 
 		this.isEnabled = isInTable;
@@ -65,7 +65,7 @@ export default class SetHeaderColumnCommand extends Command {
 		const model = this.editor.model;
 		const tableUtils = this.editor.plugins.get( 'TableUtils' );
 
-		const selectedCells = getTableCellsInSelection( model.document.selection, true );
+		const selectedCells = getSelectionAffectedTableCells( model.document.selection );
 		const firstCell = selectedCells[ 0 ];
 		const lastCell = selectedCells[ selectedCells.length - 1 ];
 		const tableRow = firstCell.parent;

diff --git a/src/commands/setheaderrowcommand.js b/src/commands/setheaderrowcommand.js
@@ -10,7 +10,7 @@
 import Command from '@ckeditor/ckeditor5-core/src/command';
 
 import { createEmptyTableCell, updateNumericAttribute } from './utils';
-import { getTableCellsInSelection } from '../tableselection/utils';
+import { getSelectionAffectedTableCells } from '../utils';
 import TableWalker from '../tablewalker';
 
 /**
@@ -33,7 +33,7 @@ export default class SetHeaderRowCommand extends Command {
 	 */
 	refresh() {
 		const model = this.editor.model;
-		const selectedCells = getTableCellsInSelection( model.document.selection, true );
+		const selectedCells = getSelectionAffectedTableCells( model.document.selection );
 		const isInTable = selectedCells.length > 0;
 
 		this.isEnabled = isInTable;
@@ -64,7 +64,7 @@ export default class SetHeaderRowCommand extends Command {
 	execute( options = {} ) {
 		const model = this.editor.model;
 
-		const selectedCells = getTableCellsInSelection( model.document.selection, true );
+		const selectedCells = getSelectionAffectedTableCells( model.document.selection );
 		const firstCell = selectedCells[ 0 ];
 		const lastCell = selectedCells[ selectedCells.length - 1 ];
 		const table = firstCell.parent.parent;