code-server-2/lib/vscode/extensions/emmet/src/incrementDecrement.ts

115 lines
3.1 KiB
TypeScript

/*---------------------------------------------------------------------------------------------
* Copyright (c) Microsoft Corporation. All rights reserved.
* Licensed under the MIT License. See License.txt in the project root for license information.
*--------------------------------------------------------------------------------------------*/
/* Based on @sergeche's work in his emmet plugin */
import * as vscode from 'vscode';
const reNumber = /[0-9]/;
/**
* Incerement number under caret of given editor
*/
export function incrementDecrement(delta: number): Thenable<boolean> | undefined {
if (!vscode.window.activeTextEditor) {
vscode.window.showInformationMessage('No editor is active');
return;
}
const editor = vscode.window.activeTextEditor;
return editor.edit(editBuilder => {
editor.selections.forEach(selection => {
let rangeToReplace = locate(editor.document, selection.isReversed ? selection.anchor : selection.active);
if (!rangeToReplace) {
return;
}
const text = editor.document.getText(rangeToReplace);
if (isValidNumber(text)) {
editBuilder.replace(rangeToReplace, update(text, delta));
}
});
});
}
/**
* Updates given number with `delta` and returns string formatted according
* to original string format
*/
export function update(numString: string, delta: number): string {
let m: RegExpMatchArray | null;
let decimals = (m = numString.match(/\.(\d+)$/)) ? m[1].length : 1;
let output = String((parseFloat(numString) + delta).toFixed(decimals)).replace(/\.0+$/, '');
if (m = numString.match(/^\-?(0\d+)/)) {
// padded number: preserve padding
output = output.replace(/^(\-?)(\d+)/, (_, minus, prefix) =>
minus + '0'.repeat(Math.max(0, (m ? m[1].length : 0) - prefix.length)) + prefix);
}
if (/^\-?\./.test(numString)) {
// omit integer part
output = output.replace(/^(\-?)0+/, '$1');
}
return output;
}
/**
* Locates number from given position in the document
*
* @return Range of number or `undefined` if not found
*/
export function locate(document: vscode.TextDocument, pos: vscode.Position): vscode.Range | undefined {
const line = document.lineAt(pos.line).text;
let start = pos.character;
let end = pos.character;
let hadDot = false, hadMinus = false;
let ch;
while (start > 0) {
ch = line[--start];
if (ch === '-') {
hadMinus = true;
break;
} else if (ch === '.' && !hadDot) {
hadDot = true;
} else if (!reNumber.test(ch)) {
start++;
break;
}
}
if (line[end] === '-' && !hadMinus) {
end++;
}
while (end < line.length) {
ch = line[end++];
if (ch === '.' && !hadDot && reNumber.test(line[end])) {
// A dot must be followed by a number. Otherwise stop parsing
hadDot = true;
} else if (!reNumber.test(ch)) {
end--;
break;
}
}
// ensure that found range contains valid number
if (start !== end && isValidNumber(line.slice(start, end))) {
return new vscode.Range(pos.line, start, pos.line, end);
}
return;
}
/**
* Check if given string contains valid number
*/
function isValidNumber(str: string): boolean {
return str ? !isNaN(parseFloat(str)) : false;
}