/*--------------------------------------------------------------------------------------------- * Copyright (c) Microsoft Corporation. All rights reserved. * Licensed under the MIT License. See License.txt in the project root for license information. *--------------------------------------------------------------------------------------------*/ import { LinkedList } from '../../../../base/common/linkedList.js'; import { Position } from '../../../common/core/position.js'; import { Range } from '../../../common/core/range.js'; import { ITextModel } from '../../../common/model.js'; import { SelectionRange, SelectionRangeProvider } from '../../../common/languages.js'; export class BracketSelectionRangeProvider implements SelectionRangeProvider { async provideSelectionRanges(model: ITextModel, positions: Position[]): Promise { const result: SelectionRange[][] = []; for (const position of positions) { const bucket: SelectionRange[] = []; result.push(bucket); const ranges = new Map>(); await new Promise(resolve => BracketSelectionRangeProvider._bracketsRightYield(resolve, 0, model, position, ranges)); await new Promise(resolve => BracketSelectionRangeProvider._bracketsLeftYield(resolve, 0, model, position, ranges, bucket)); } return result; } public static _maxDuration = 30; private static readonly _maxRounds = 2; private static _bracketsRightYield(resolve: () => void, round: number, model: ITextModel, pos: Position, ranges: Map>): void { const counts = new Map(); const t1 = Date.now(); while (true) { if (round >= BracketSelectionRangeProvider._maxRounds) { resolve(); break; } if (!pos) { resolve(); break; } const bracket = model.bracketPairs.findNextBracket(pos); if (!bracket) { resolve(); break; } const d = Date.now() - t1; if (d > BracketSelectionRangeProvider._maxDuration) { setTimeout(() => BracketSelectionRangeProvider._bracketsRightYield(resolve, round + 1, model, pos, ranges)); break; } if (bracket.bracketInfo.isOpeningBracket) { const key = bracket.bracketInfo.bracketText; // wait for closing const val = counts.has(key) ? counts.get(key)! : 0; counts.set(key, val + 1); } else { const key = bracket.bracketInfo.getOpeningBrackets()[0].bracketText; // process closing let val = counts.has(key) ? counts.get(key)! : 0; val -= 1; counts.set(key, Math.max(0, val)); if (val < 0) { let list = ranges.get(key); if (!list) { list = new LinkedList(); ranges.set(key, list); } list.push(bracket.range); } } pos = bracket.range.getEndPosition(); } } private static _bracketsLeftYield(resolve: () => void, round: number, model: ITextModel, pos: Position, ranges: Map>, bucket: SelectionRange[]): void { const counts = new Map(); const t1 = Date.now(); while (true) { if (round >= BracketSelectionRangeProvider._maxRounds && ranges.size === 0) { resolve(); break; } if (!pos) { resolve(); break; } const bracket = model.bracketPairs.findPrevBracket(pos); if (!bracket) { resolve(); break; } const d = Date.now() - t1; if (d > BracketSelectionRangeProvider._maxDuration) { setTimeout(() => BracketSelectionRangeProvider._bracketsLeftYield(resolve, round + 1, model, pos, ranges, bucket)); break; } if (!bracket.bracketInfo.isOpeningBracket) { const key = bracket.bracketInfo.getOpeningBrackets()[0].bracketText; // wait for opening const val = counts.has(key) ? counts.get(key)! : 0; counts.set(key, val + 1); } else { const key = bracket.bracketInfo.bracketText; // opening let val = counts.has(key) ? counts.get(key)! : 0; val -= 1; counts.set(key, Math.max(0, val)); if (val < 0) { const list = ranges.get(key); if (list) { const closing = list.shift(); if (list.size === 0) { ranges.delete(key); } const innerBracket = Range.fromPositions(bracket.range.getEndPosition(), closing!.getStartPosition()); const outerBracket = Range.fromPositions(bracket.range.getStartPosition(), closing!.getEndPosition()); bucket.push({ range: innerBracket }); bucket.push({ range: outerBracket }); BracketSelectionRangeProvider._addBracketLeading(model, outerBracket, bucket); } } } pos = bracket.range.getStartPosition(); } } private static _addBracketLeading(model: ITextModel, bracket: Range, bucket: SelectionRange[]): void { if (bracket.startLineNumber === bracket.endLineNumber) { return; } // xxxxxxxx { // // } const startLine = bracket.startLineNumber; const column = model.getLineFirstNonWhitespaceColumn(startLine); if (column !== 0 && column !== bracket.startColumn) { bucket.push({ range: Range.fromPositions(new Position(startLine, column), bracket.getEndPosition()) }); bucket.push({ range: Range.fromPositions(new Position(startLine, 1), bracket.getEndPosition()) }); } // xxxxxxxx // { // // } const aboveLine = startLine - 1; if (aboveLine > 0) { const column = model.getLineFirstNonWhitespaceColumn(aboveLine); if (column === bracket.startColumn && column !== model.getLineLastNonWhitespaceColumn(aboveLine)) { bucket.push({ range: Range.fromPositions(new Position(aboveLine, column), bracket.getEndPosition()) }); bucket.push({ range: Range.fromPositions(new Position(aboveLine, 1), bracket.getEndPosition()) }); } } } }