/
selectcolumncommand.js
63 lines (52 loc) · 1.84 KB
/
selectcolumncommand.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
/**
* @license Copyright (c) 2003-2020, CKSource - Frederico Knabben. All rights reserved.
* For licensing, see LICENSE.md or https://ckeditor.com/legal/ckeditor-oss-license
*/
/**
* @module table/commands/selectcolumncommand
*/
import Command from '@ckeditor/ckeditor5-core/src/command';
import TableWalker from '../tablewalker';
import { getSelectionAffectedTableCells } from '../utils/selection';
/**
* The select column command.
*
* The command is registered by {@link module:table/tableediting~TableEditing} as the `'selectTableColumn'` editor command.
*
* To select the columns containing the selected cells, execute the command:
*
* editor.execute( 'selectTableColumn' );
*
* @extends module:core/command~Command
*/
export default class SelectColumnCommand extends Command {
/**
* @inheritDoc
*/
refresh() {
const selectedCells = getSelectionAffectedTableCells( this.editor.model.document.selection );
this.isEnabled = selectedCells.length > 0;
}
/**
* @inheritDoc
*/
execute() {
const model = this.editor.model;
const referenceCells = getSelectionAffectedTableCells( model.document.selection );
const firstCell = referenceCells[ 0 ];
const lastCell = referenceCells.pop();
const table = firstCell.findAncestor( 'table' );
const tableUtils = this.editor.plugins.get( 'TableUtils' );
const startLocation = tableUtils.getCellLocation( firstCell );
const endLocation = tableUtils.getCellLocation( lastCell );
const startColumn = Math.min( startLocation.column, endLocation.column );
const endColumn = Math.max( startLocation.column, endLocation.column );
const rangesToSelect = [];
for ( const cellInfo of new TableWalker( table, { startColumn, endColumn } ) ) {
rangesToSelect.push( model.createRangeOn( cellInfo.cell ) );
}
model.change( writer => {
writer.setSelection( rangesToSelect );
} );
}
}