Install package command

package.json

@@ -170,7 +170,7 @@
       },
       {
         "command": "go.show.commands",
-        "title": "Go: Show Other Commands...",
+        "title": "Go: Show All Commands...",
         "description": "Shows all commands from the Go extension in the wuick pick"
       },
       {
@@ -190,7 +190,7 @@
       },
       {
         "command": "go.lint.package",
-        "title": "Go: Lint Package",
+        "title": "Go: Lint Current Package",
         "description": "Run linter in the package of the current file."
       },
       {
@@ -200,7 +200,7 @@
       },
       {
         "command": "go.vet.package",
-        "title": "Go: Vet Package",
+        "title": "Go: Vet Current Package",
         "description": "Run go vet in the package of the current file."
       },
       {
@@ -210,13 +210,18 @@
       },
       {
         "command": "go.build.package",
-        "title": "Go: Build Package",
+        "title": "Go: Build Current Package",
         "description": "Build the package of the current file."
       },
       {
         "command": "go.build.workspace",
         "title": "Go: Build Workspace",
         "description": "Build the current workspace."
+      },
+      {
+        "command": "go.install.package",
+        "title": "Go: Install Current Package",
+        "description": "Install the current package."
       }
     ],
     "debuggers": [

src/goBuild.ts

@@ -13,7 +13,11 @@ import { getCurrentGoWorkspaceFromGOPATH } from './goPath';
 export function buildCode(buildWorkspace?: boolean) {
 	let editor = vscode.window.activeTextEditor;
 	if (!editor && !buildWorkspace) {
-		vscode.window.showInformationMessage('No editor is active, cant find current package to build');
+		vscode.window.showInformationMessage('No editor is active, cannot find current package to build');
+		return;
+	}
+	if (editor.document.languageId !== 'go' && !buildWorkspace) {
+		vscode.window.showInformationMessage('File in the active editor is not a Go file, cannot find current package to build');
 		return;
 	}
 

src/goInstall.ts

@@ -0,0 +1,46 @@
+import path = require('path');
+import vscode = require('vscode');
+import { getToolsEnvVars, getCurrentGoPath } from './util';
+import { outputChannel } from './goStatus';
+import { getCurrentGoWorkspaceFromGOPATH, getGoRuntimePath } from './goPath';
+import cp = require('child_process');
+
+export function installCurrentPackage() {
+	let editor = vscode.window.activeTextEditor;
+	if (!editor) {
+		vscode.window.showInformationMessage('No editor is active, cannot find current package to install');
+		return;
+	}
+	if (editor.document.languageId !== 'go') {
+		vscode.window.showInformationMessage('File in the active editor is not a Go file, cannot find current package to install');
+		return;
+	}
+
+	let goRuntimePath = getGoRuntimePath();
+	if (!goRuntimePath) {
+		vscode.window.showInformationMessage('Cannot find "go" binary. Update PATH or GOROOT appropriately');
+		return;
+	}
+
+	const env = Object.assign({}, getToolsEnvVars());
+	const cwd = path.dirname(editor.document.uri.fsPath);
+	const goConfig = vscode.workspace.getConfiguration('go', editor.document.uri);
+	const args = ['install'];
+
+	if (goConfig['buildTags']) {
+		args.push('-tags', '"' + goConfig['buildTags'] + '"');
+	}
+
+	// Find the right importPath instead of directly using `.`. Fixes https://github.com/Microsoft/vscode-go/issues/846
+	const currentGoWorkspace = getCurrentGoWorkspaceFromGOPATH(getCurrentGoPath(), cwd);
+	const importPath = currentGoWorkspace ? cwd.substr(currentGoWorkspace.length + 1) : '.';
+	args.push(importPath);
+
+	outputChannel.clear();
+	outputChannel.show();
+	outputChannel.appendLine(`Installing ${importPath === '.' ? 'current package' : importPath}`);
+
+	cp.execFile(goRuntimePath, args, { env, cwd }, (err, stdout, stderr) => {
+		outputChannel.appendLine(err ? `Installation failed: ${stderr}` : `Installation successful`);
+	});
+}

src/goLint.ts

@@ -9,7 +9,11 @@ import { outputChannel } from './goStatus';
 export function lintCode(lintWorkspace?: boolean) {
 	let editor = vscode.window.activeTextEditor;
 	if (!editor && !lintWorkspace) {
-		vscode.window.showInformationMessage('No editor is active, cant find current package to lint');
+		vscode.window.showInformationMessage('No editor is active, cannot find current package to lint');
+		return;
+	}
+	if (editor.document.languageId !== 'go' && !lintWorkspace) {
+		vscode.window.showInformationMessage('File in the active editor is not a Go file, cannot find current package to lint');
 		return;
 	}
 

src/goMain.ts

@@ -44,6 +44,7 @@ import { playgroundCommand } from './goPlayground';
 import { lintCode } from './goLint';
 import { vetCode } from './goVet';
 import { buildCode } from './goBuild';
+import { installCurrentPackage } from './goInstall';
 
 export let errorDiagnosticCollection: vscode.DiagnosticCollection;
 export let warningDiagnosticCollection: vscode.DiagnosticCollection;
@@ -304,6 +305,8 @@ export function activate(ctx: vscode.ExtensionContext): void {
 
 	ctx.subscriptions.push(vscode.commands.registerCommand('go.build.workspace', () => buildCode(true)));
 
+	ctx.subscriptions.push(vscode.commands.registerCommand('go.install.package', installCurrentPackage));
+
 	vscode.languages.setLanguageConfiguration(GO_MODE.language, {
 		indentationRules: {
 			decreaseIndentPattern: /^\s*(\bcase\b.*:|\bdefault\b:|}[),]?|\)[,]?)$/,

src/goVet.ts

@@ -9,7 +9,11 @@ import { outputChannel } from './goStatus';
 export function vetCode(vetWorkspace?: boolean) {
 	let editor = vscode.window.activeTextEditor;
 	if (!editor && !vetWorkspace) {
-		vscode.window.showInformationMessage('No editor is active, cant find current package to vet');
+		vscode.window.showInformationMessage('No editor is active, cannot find current package to vet');
+		return;
+	}
+	if (editor.document.languageId !== 'go' && !vetWorkspace) {
+		vscode.window.showInformationMessage('File in the active editor is not a Go file, cannot find current package to vet');
 		return;
 	}