53 lines
2.2 KiB
TypeScript
53 lines
2.2 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';
|
||
import evaluate, { extract } from '@emmetio/math-expression';
|
||
import { DocumentStreamReader } from './bufferStream';
|
||
|
||
export function evaluateMathExpression(): Thenable<boolean> {
|
||
if (!vscode.window.activeTextEditor) {
|
||
vscode.window.showInformationMessage('No editor is active');
|
||
return Promise.resolve(false);
|
||
}
|
||
const editor = vscode.window.activeTextEditor;
|
||
const stream = new DocumentStreamReader(editor.document);
|
||
return editor.edit(editBuilder => {
|
||
editor.selections.forEach(selection => {
|
||
// startpos always comes before endpos
|
||
const startpos = selection.isReversed ? selection.active : selection.anchor;
|
||
const endpos = selection.isReversed ? selection.anchor : selection.active;
|
||
const selectionText = stream.substring(startpos, endpos);
|
||
|
||
try {
|
||
if (selectionText) {
|
||
// respect selections
|
||
const result = String(evaluate(selectionText));
|
||
editBuilder.replace(new vscode.Range(startpos, endpos), result);
|
||
} else {
|
||
// no selection made, extract expression from line
|
||
const lineToSelectionEnd = stream.substring(new vscode.Position(selection.end.line, 0), endpos);
|
||
const extractedIndices = extract(lineToSelectionEnd);
|
||
if (!extractedIndices) {
|
||
throw new Error('Invalid extracted indices');
|
||
}
|
||
const result = String(evaluate(lineToSelectionEnd.substr(extractedIndices[0], extractedIndices[1])));
|
||
const rangeToReplace = new vscode.Range(
|
||
new vscode.Position(selection.end.line, extractedIndices[0]),
|
||
new vscode.Position(selection.end.line, extractedIndices[1])
|
||
);
|
||
editBuilder.replace(rangeToReplace, result);
|
||
}
|
||
} catch (err) {
|
||
vscode.window.showErrorMessage('Could not evaluate expression');
|
||
// Ignore error since most likely it’s because of non-math expression
|
||
console.warn('Math evaluation error', err);
|
||
}
|
||
});
|
||
});
|
||
}
|