|
|
@@ -0,0 +1,88 @@
|
|
|
+import { EditorSelection, EditorState, SelectionRange } from '@codemirror/state'
|
|
|
+import { syntaxTree } from '@codemirror/language'
|
|
|
+import { Tree } from '@lezer/common'
|
|
|
+import {
|
|
|
+ ancestorOfNodeWithType,
|
|
|
+ descendantsOfNodeWithType,
|
|
|
+} from '../../utils/tree-operations/ancestors'
|
|
|
+import { getMousedownSelection, selectionIntersects } from './selection'
|
|
|
+
|
|
|
+/**
|
|
|
+ * A custom extension that updates the selection in a transaction if the mouse pointer was used
|
|
|
+ * to position a cursor at the start or end of an argument (the cursor is placed inside the brace),
|
|
|
+ * or to drag a range across the whole range of an argument (the selection is placed inside the braces),
|
|
|
+ * when the selection was not already inside the command.
|
|
|
+ */
|
|
|
+export const selectDecoratedArgument = EditorState.transactionFilter.of(tr => {
|
|
|
+ if (tr.selection && tr.isUserEvent('select.pointer')) {
|
|
|
+ const tree = syntaxTree(tr.state)
|
|
|
+ let selection = tr.selection
|
|
|
+ const mousedownSelection = getMousedownSelection(tr.state)
|
|
|
+ let replaced = false
|
|
|
+ for (const [index, range] of selection.ranges.entries()) {
|
|
|
+ const replacementRange =
|
|
|
+ selectArgument(tree, range, mousedownSelection, 1) ||
|
|
|
+ selectArgument(tree, range, mousedownSelection, -1)
|
|
|
+ if (replacementRange) {
|
|
|
+ selection = selection.replaceRange(replacementRange, index)
|
|
|
+ replaced = true
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if (replaced) {
|
|
|
+ return [tr, { selection }]
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return tr
|
|
|
+})
|
|
|
+
|
|
|
+const selectArgument = (
|
|
|
+ tree: Tree,
|
|
|
+ range: SelectionRange,
|
|
|
+ mousedownSelection: EditorSelection | undefined,
|
|
|
+ side: -1 | 1
|
|
|
+): SelectionRange | undefined => {
|
|
|
+ const anchor = tree.resolveInner(range.anchor, side)
|
|
|
+
|
|
|
+ const ancestorCommand = ancestorOfNodeWithType(anchor, '$Command')
|
|
|
+ if (!ancestorCommand) {
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ const mousedownSelectionInside =
|
|
|
+ mousedownSelection !== undefined &&
|
|
|
+ selectionIntersects(mousedownSelection, ancestorCommand)
|
|
|
+ if (mousedownSelectionInside) {
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ const [inner] = descendantsOfNodeWithType(ancestorCommand, '$TextArgument')
|
|
|
+ if (!inner) {
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ if (side === 1) {
|
|
|
+ if (
|
|
|
+ range.anchor === inner.from + 1 ||
|
|
|
+ range.anchor === ancestorCommand.from
|
|
|
+ ) {
|
|
|
+ if (range.empty) {
|
|
|
+ // selecting at the start
|
|
|
+ return EditorSelection.cursor(inner.from + 1)
|
|
|
+ } else if (Math.abs(range.head - inner.to) < 2) {
|
|
|
+ // selecting from the start to the end
|
|
|
+ return EditorSelection.range(inner.from + 1, inner.to - 1)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ if (range.anchor === inner.to - 1 || range.anchor === ancestorCommand.to) {
|
|
|
+ if (range.empty) {
|
|
|
+ // selecting at the end
|
|
|
+ return EditorSelection.cursor(inner.to - 1)
|
|
|
+ } else if (Math.abs(range.head - ancestorCommand.from) < 2) {
|
|
|
+ // selecting from the end to the start
|
|
|
+ return EditorSelection.range(inner.to - 1, inner.from + 1)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
Alf Eaton