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
|
// The module 'vscode' contains the VS Code extensibility API
// Import the module and reference it with the alias vscode in your code below
const vscode = require('vscode');
const fnRegex = /^(\t*)(.*)err\s?:?=.+?$/;
// This method is called when your extension is activated
// Your extension is activated the very first time the command is executed
/**
* @param {vscode.ExtensionContext} context
*/
function activate(context) {
let wrapErrorCommand = vscode.commands.registerCommand('gotools.wrap-error', wrapError);
context.subscriptions.push(
vscode.languages.registerCodeActionsProvider(
'go',
new ErrorsWrapper(),
{
providedCodeActionKinds: [
vscode.CodeActionKind.RefactorRewrite
]
}
)
);
context.subscriptions.push(wrapErrorCommand);
}
// This method is called when your extension is deactivated
function deactivate() { }
class ErrorsWrapper {
provideCodeActions(document, range) {
const editor = vscode.window.activeTextEditor;
if (!editor) {
return undefined;
}
const line = document.lineAt(editor.selection.start.line);
if (!fnRegex.test(line.text)) {
vscode.commands.executeCommand('setContext', 'allowWrapIferr', false);
return undefined;
}
vscode.commands.executeCommand('setContext', 'allowWrapIferr', true);
const action = new vscode.CodeAction('Add error checking', vscode.CodeActionKind.RefactorRewrite);
action.command = { command: 'gotools.wrap-error', title: 'Add error checking block', tooltip: '' };
return [
action,
];
}
}
module.exports = {
activate,
deactivate
}
const wrapError = () => {
const editor = vscode.window.activeTextEditor;
if (!editor) {
return;
}
const document = editor.document;
const line = document.lineAt(editor.selection.start.line);
const matches = line.text.match(fnRegex);
if (matches === null || matches.length === 0) {
return;
}
const extravars = matches[2].split(',').map(x => x.trim()).filter(x => x);
if (extravars.filter(x => x !== "_").length > 0) {
editor.insertSnippet(
new vscode.SnippetString(`\nif err != nil {\n\treturn \${1:nil, }\${2:err}\n}\n`),
new vscode.Position(line.range.end.line, line.range.end.character + line.firstNonWhitespaceCharacterIndex),
);
} else {
const tabs = matches[1];
const original = matches[0].trimStart()
editor.insertSnippet(
new vscode.SnippetString(`${tabs}if ${original}; err != nil {\n${tabs}\treturn \${2:nil, }\${3:err}\n${tabs}}\n`),
line.range,
{
undoStopBefore: true,
undoStopAfter: true
}
);
}
};
|
