-
Notifications
You must be signed in to change notification settings - Fork 1
/
extension.js
103 lines (84 loc) · 2.77 KB
/
extension.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
const vscode = require('vscode');
const vbspretty = require('./src/vbspretty');
const path = require('path');
const {
Position,
Range,
} = require('vscode');
const contributions = vscode.workspace.getConfiguration('vbaFormatter');
const indentCharValue = '\t';
const breakLineCharValue = '\n';
const levelValue = contributions.get('level');
const breakOnSeperatorValue = contributions.get('breakOnSeperator');
const removeCommentsValue = contributions.get('removeComments');
const editor = vscode.window.activeTextEditor;
/**
* @param {vscode.ExtensionContext} context
*/
function activate(context) {
vscode.window.showInformationMessage('VBA Beautifier is now active!');
let buttonActivation = vscode.commands.registerTextEditorCommand('extension.pretty', (editor) => {
let document = editor.document
prepareDocument(document)
});
context.subscriptions.push(buttonActivation);
let formatFunction = vscode.languages.registerDocumentFormattingEditProvider('vbs', {
provideDocumentFormattingEdits: (document) => {
prepareDocument(document)
}
});
context.subscriptions.push(formatFunction);
}
function prepareDocument(document) {
if (editor && editor.document.languageId === 'vbs' || editor.document.languageId === 'vb') {
var inFile = document.fileName;
const documentText = document.getText();
const fileExtension = getFileExtension(inFile);
const start = new Position(getStartLine(documentText, fileExtension), 0);
const end = new Position(document.lineCount + 1, 0);
const range = new Range(start, end);
const sourceFile = document.getText(range);
vscode.window.showInformationMessage('Beautifying');
let outFile = vbspretty({
level: levelValue,
indentChar: indentCharValue,
breakLineChar: breakLineCharValue,
breakOnSeperator: breakOnSeperatorValue,
removeComments: removeCommentsValue,
source: sourceFile,
});
console.log(outFile);
const edit = new vscode.WorkspaceEdit();
edit.replace(document.uri, range, outFile);
return vscode.workspace.applyEdit(edit)
} else {
vscode.window.showInformationMessage('Not a Visual Basic or VB Script file!');
}
}
function getStartLine(text, fileExtension) {
if (fileExtension === '.cls') {
const lineNumber = findLineNumber(text, 'Attribute VB_Exposed');
return lineNumber;
} else if (fileExtension === '.bas') {
const lineNumber = findLineNumber(text, 'Attribute VB_Name');
return lineNumber;
}
else {
return 0;
}
}
function findLineNumber(text, searchText) {
const lines = text.split('\n');
const lineNumber = lines.findIndex(line => line.includes(searchText)) + 1;
return lineNumber;
}
function getFileExtension(fileName) {
return path.extname(fileName);
}
function deactivate() {
vscode.window.showInformationMessage('deactivated');
}
module.exports = {
activate,
deactivate
}