feat: add running main func code lens

Author: vuon9

docs/settings.md

@@ -210,12 +210,14 @@ Feature level setting to enable/disable code lens for references and run/debug t
 | --- | --- |
 | `references` | If true, enables the references code lens. Uses guru. Recalculates when there is change to the document followed by scrolling. Unnecessary when using the language server; use the call graph feature instead. <br/> Default: `false` |
 | `runtest` | If true, enables code lens for running and debugging tests <br/> Default: `true` |
+| `runmain` | If true, enables code lens for running main func <br/> Default: `true` |
 
 Default:
 ```
 {
 	"references" :	false,
 	"runtest" :	true,
+	"runmain" :	true,
 }
 ```
 ### `go.formatFlags`
@@ -492,7 +494,7 @@ Allowed Options: `package`, `workspace`, `off`
 Default: `"package"`
 ### `gopls`
 
-Customize `gopls` behavior by specifying the gopls' settings in this section. For example, 
+Customize `gopls` behavior by specifying the gopls' settings in this section. For example,
 ```
 "gopls" : {
 	"build.directoryFilters": ["-node_modules"]

package.json

@@ -481,6 +481,11 @@
         "command": "go.global.resetState",
         "title": "Go: Reset Global State",
         "description": "Reset keys in global state to undefined."
+      },
+      {
+        "command": "go.runMain",
+        "title": "Go: Run main() func on file",
+        "description": "Run main() func on file"
       }
     ],
     "breakpoints": [
@@ -1579,12 +1584,18 @@
               "type": "boolean",
               "default": true,
               "description": "If true, enables code lens for running and debugging tests"
+            },
+            "runmain": {
+              "type": "boolean",
+              "default": true,
+              "description": "If true, enables code lens for running main func"
             }
           },
           "additionalProperties": false,
           "default": {
             "references": false,
-            "runtest": true
+            "runtest": true,
+            "runmain": true
           },
           "description": "Feature level setting to enable/disable code lens for references and run/debug tests",
           "scope": "resource"

src/goMain.ts

@@ -61,7 +61,7 @@ import { GO111MODULE, goModInit, isModSupported } from './goModules';
 import { playgroundCommand } from './goPlayground';
 import { GoReferencesCodeLensProvider } from './goReferencesCodelens';
 import { GoRunTestCodeLensProvider } from './goRunTestCodelens';
-import { GoMainCodeLensProvider } from './goMainCodelens';
+import { GoMainCodeLensProvider, runMainFunc } from './goMainCodelens';
 import { disposeGoStatusBar, expandGoStatusBar, outputChannel, updateGoStatusBar } from './goStatus';
 import {
 	debugPrevious,
@@ -566,6 +566,12 @@ If you would like additional configuration for diagnostics from gopls, please se
 		})
 	);
 
+	ctx.subscriptions.push(
+		vscode.commands.registerCommand('go.runMain', (args) => {
+			runMainFunc()
+		})
+	)
+
 	ctx.subscriptions.push(
 		vscode.commands.registerCommand('go.show.commands', () => {
 			const extCommands = getExtensionCommands();

src/goMainCodelens.ts

@@ -8,16 +8,18 @@
 'use strict';
 
 import vscode = require('vscode');
+import cp = require('child_process');
+
 import { CancellationToken, CodeLens, TextDocument } from 'vscode';
 import { getGoConfig } from './config';
 import { GoBaseCodeLensProvider } from './goBaseCodelens';
 import { GoDocumentSymbolProvider } from './goOutline';
-import { getBenchmarkFunctions, getTestFunctions } from './testUtils';
-
-const mainFuncRegx = /^main$/u;
+import { getBinPath } from './util';
+import { envPath, getCurrentGoRoot } from './utils/pathUtils';
+import { reject } from 'lodash';
 
 export class GoMainCodeLensProvider extends GoBaseCodeLensProvider {
-	private readonly mainRegex = /^main.+/;
+	private readonly mainRegex = /^main$/;
 
 	public async provideCodeLenses(document: TextDocument, token: CancellationToken): Promise<CodeLens[]> {
 		if (!this.enabled) {
@@ -52,7 +54,7 @@ export class GoMainCodeLensProvider extends GoBaseCodeLensProvider {
 		}
 		const children = symbol.children;
 
-		return children.find(sym => sym.kind === vscode.SymbolKind.Function && mainFuncRegx.test(sym.name));
+		return children.find(sym => sym.kind === vscode.SymbolKind.Function && this.mainRegex.test(sym.name));
 	}
 
 	private async getCodeLensForMainFunc(document: TextDocument, token: CancellationToken): Promise<CodeLens[]> {
@@ -64,13 +66,8 @@ export class GoMainCodeLensProvider extends GoBaseCodeLensProvider {
 
 			return [
 				new CodeLens(mainFunc.range, {
-					title: 'run',
-					command: 'go.main.run',
-					arguments: [{ functionName: mainFunc.name }]
-				}),
-				new CodeLens(mainFunc.range, {
-					title: 'package run',
-					command: 'go.main.package',
+					title: 'run main',
+					command: 'go.runMain',
 					arguments: [{ functionName: mainFunc.name }]
 				})
 			];
@@ -79,3 +76,46 @@ export class GoMainCodeLensProvider extends GoBaseCodeLensProvider {
 		return await mainPromise();
 	}
 }
+
+const mainFuncOutputChannel = vscode.window.createOutputChannel('Go Main');
+
+export async function runMainFunc() {
+	let outputChannel = mainFuncOutputChannel
+	const goRuntimePath = getBinPath('go');
+	if (!goRuntimePath) {
+		vscode.window.showErrorMessage(
+			`Failed to run "go run ." as the "go" binary cannot be found in either GOROOT(${getCurrentGoRoot}) or PATH(${envPath})`
+		);
+		return Promise.resolve(false);
+	}
+
+	const editor = vscode.window.activeTextEditor;
+	const documentUri = editor ? editor.document.uri : null;
+	const args = ['run', documentUri.path];
+
+	outputChannel.clear()
+	outputChannel.show(true)
+	outputChannel.appendLine(["Running main func: ", goRuntimePath, ...args].join(' '))
+
+	cp.execFile(
+		goRuntimePath,
+		args,
+		{ },
+		(err, stdout, stderr) => {
+			try {
+				if (err) {
+					outputChannel.appendLine(err.message);
+					return;
+				}
+				if (stdout) {
+					outputChannel.append(stdout);
+				}
+				if (stderr) {
+					outputChannel.append(stderr);
+				}
+			} catch (e) {
+				reject(e);
+			}
+		}
+	)
+}