src/goVulncheck: add "Current Module" option

When users need to scan vulnerabilities of a module located not in the
workspace root, they can use this option - which infer the module
root directory by running `go env GOMOD`.

Change-Id: I70126de30fc4cc08a7207ab4a74a12f6290ccba1
Reviewed-on: https://go-review.googlesource.com/c/vscode-go/+/429476
Reviewed-by: Suzy Mueller <[email protected]>
Run-TryBot: Hyang-Ah Hana Kim <[email protected]>
Reviewed-by: Jamal Carvalho <[email protected]>
TryBot-Result: kokoro <[email protected]>

Author: hyangah

src/goVulncheck.ts

@@ -223,7 +223,7 @@ export class VulncheckProvider {
 	}
 
 	private async runInternal(goCtx: GoExtensionContext) {
-		const pick = await vscode.window.showQuickPick(['Current Package', 'Workspace']);
+		const pick = await vscode.window.showQuickPick(['Current Package', 'Current Module', 'Workspace']);
 		let dir, pattern: string;
 		const document = vscode.window.activeTextEditor?.document;
 		switch (pick) {
@@ -241,6 +241,14 @@ export class VulncheckProvider {
 				dir = path.dirname(document.fileName);
 				pattern = '.';
 				break;
+			case 'Current Module':
+				dir = await moduleDir(document);
+				if (!dir) {
+					vscode.window.showErrorMessage('vulncheck error: no current module');
+					return;
+				}
+				pattern = './...';
+				break;
 			case 'Workspace':
 				dir = await this.activeDir();
 				pattern = './...';
@@ -308,6 +316,18 @@ export class VulncheckProvider {
 	}
 }
 
+async function moduleDir(document: vscode.TextDocument | undefined) {
+	const docDir = document && document.fileName && path.dirname(document.fileName);
+	if (!docDir) {
+		return;
+	}
+	const modFile = await getGoModFile(vscode.Uri.file(docDir));
+	if (!modFile) {
+		return;
+	}
+	return path.dirname(modFile);
+}
+
 // run `gopls vulncheck`.
 export async function vulncheck(
 	goCtx: GoExtensionContext,