62 lines
2.7 KiB
JavaScript
62 lines
2.7 KiB
JavaScript
/*---------------------------------------------------------------------------------------------
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
|
* Licensed under the MIT License. See License.txt in the project root for license information.
|
|
*--------------------------------------------------------------------------------------------*/
|
|
import { Range, SelectionRange } from '../jsonLanguageTypes';
|
|
import { createScanner } from 'jsonc-parser';
|
|
export function getSelectionRanges(document, positions, doc) {
|
|
function getSelectionRange(position) {
|
|
var offset = document.offsetAt(position);
|
|
var node = doc.getNodeFromOffset(offset, true);
|
|
var result = [];
|
|
while (node) {
|
|
switch (node.type) {
|
|
case 'string':
|
|
case 'object':
|
|
case 'array':
|
|
// range without ", [ or {
|
|
var cStart = node.offset + 1, cEnd = node.offset + node.length - 1;
|
|
if (cStart < cEnd && offset >= cStart && offset <= cEnd) {
|
|
result.push(newRange(cStart, cEnd));
|
|
}
|
|
result.push(newRange(node.offset, node.offset + node.length));
|
|
break;
|
|
case 'number':
|
|
case 'boolean':
|
|
case 'null':
|
|
case 'property':
|
|
result.push(newRange(node.offset, node.offset + node.length));
|
|
break;
|
|
}
|
|
if (node.type === 'property' || node.parent && node.parent.type === 'array') {
|
|
var afterCommaOffset = getOffsetAfterNextToken(node.offset + node.length, 5 /* CommaToken */);
|
|
if (afterCommaOffset !== -1) {
|
|
result.push(newRange(node.offset, afterCommaOffset));
|
|
}
|
|
}
|
|
node = node.parent;
|
|
}
|
|
var current = undefined;
|
|
for (var index = result.length - 1; index >= 0; index--) {
|
|
current = SelectionRange.create(result[index], current);
|
|
}
|
|
if (!current) {
|
|
current = SelectionRange.create(Range.create(position, position));
|
|
}
|
|
return current;
|
|
}
|
|
function newRange(start, end) {
|
|
return Range.create(document.positionAt(start), document.positionAt(end));
|
|
}
|
|
var scanner = createScanner(document.getText(), true);
|
|
function getOffsetAfterNextToken(offset, expectedToken) {
|
|
scanner.setPosition(offset);
|
|
var token = scanner.scan();
|
|
if (token === expectedToken) {
|
|
return scanner.getTokenOffset() + scanner.getTokenLength();
|
|
}
|
|
return -1;
|
|
}
|
|
return positions.map(getSelectionRange);
|
|
}
|