interface implementation

1 files changed, 60 insertions, 0 deletions

diff --git a/src/errors.js b/src/errors.js
new file mode 100644
index 0000000..65b31a7
--- /dev/null
+++ b/src/errors.js
@@ -0,0 +1,60 @@
+const vscode = require('vscode');
+
+const fnRegex = /^(\t*)(.*)err\s?:?=.+?$/;
+
+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,
+		];
+	}
+}
+
+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
+			}
+		);
+	}
+};
+
+module.exports = {
+    ErrorsWrapper, wrapError
+}
\ No newline at end of file