This repository has been archived by the owner on Jul 15, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 646
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Use go list all to get pkgname to use in completions Fixes #647
- Loading branch information
1 parent
1b6efbe
commit 78125e6
Showing
4 changed files
with
137 additions
and
67 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,97 @@ | ||
| import vscode = require('vscode'); | ||
| import cp = require('child_process'); | ||
| import path = require('path'); | ||
| import { getGoRuntimePath } from './goPath'; | ||
| import { isVendorSupported, getCurrentGoWorkspaceFromGOPATH } from './util'; | ||
|
|
||
| let allPkgs = new Map<string, string>(); | ||
| let goListAllCompleted: boolean = false; | ||
|
|
||
| /** | ||
| * Runs go list all | ||
| * @returns Map<string, string> mapping between package import path and package name | ||
| */ | ||
| export function goListAll(): Promise<Map<string, string>> { | ||
| let goRuntimePath = getGoRuntimePath(); | ||
|
|
||
| if (!goRuntimePath) { | ||
| vscode.window.showInformationMessage('Cannot find "go" binary. Update PATH or GOROOT appropriately'); | ||
| return Promise.resolve(null); | ||
| } | ||
|
|
||
| if (goListAllCompleted) { | ||
| return Promise.resolve(allPkgs); | ||
| } | ||
| return new Promise<Map<string, string>>((resolve, reject) => { | ||
| cp.execFile(goRuntimePath, ['list', '-f', '{{.Name}};{{.ImportPath}}', 'all'], (err, stdout, stderr) => { | ||
| if (err) return reject(); | ||
| stdout.split('\n').forEach(pkgDetail => { | ||
| if (!pkgDetail || !pkgDetail.trim() || pkgDetail.indexOf(';') === -1) return; | ||
| let [pkgName, pkgPath] = pkgDetail.trim().split(';'); | ||
| if (pkgName !== 'main') { | ||
| allPkgs.set(pkgPath, pkgName); | ||
| } | ||
| }); | ||
| goListAllCompleted = true; | ||
| return resolve(allPkgs); | ||
| }); | ||
| }); | ||
| } | ||
|
|
||
| /** | ||
| * Returns mapping of import path and package name for packages | ||
| * @param filePath. Used to determine the right relative path for vendor pkgs | ||
| * @returns Map<string, string> mapping between package import path and package name | ||
| */ | ||
| export function getAllPackageDetails(filePath: string): Promise<Map<string, string>> { | ||
|
|
||
| return Promise.all([isVendorSupported(), goListAll()]).then(values => { | ||
| let isVendorSupported = values[0]; | ||
| let pkgs: Map<string, string> = values[1]; | ||
| let currentFileDirPath = path.dirname(filePath); | ||
| let currentWorkspace = getCurrentGoWorkspaceFromGOPATH(currentFileDirPath); | ||
| if (!isVendorSupported || !currentWorkspace) { | ||
| return pkgs; | ||
| } | ||
|
|
||
| let pkgMap = new Map<string, string>(); | ||
| pkgs.forEach((pkgName, pkgPath) => { | ||
| let relativePkgPath = getRelativePackagePath(currentFileDirPath, currentWorkspace, pkgPath); | ||
| if (relativePkgPath) { | ||
| pkgMap.set(relativePkgPath, pkgs.get(pkgPath)); | ||
| } | ||
| }); | ||
| return pkgMap; | ||
| }); | ||
|
|
||
| } | ||
|
|
||
| /** | ||
| * If given pkgPath is not vendor pkg, then the same pkgPath is returned | ||
| * Else, the import path for the vendor pkg relative to given filePath is returned. | ||
| */ | ||
| export function getRelativePackagePath(currentFileDirPath: string, currentWorkspace: string, pkgPath: string): string { | ||
| let magicVendorString = '/vendor/'; | ||
| let vendorIndex = pkgPath.indexOf(magicVendorString); | ||
| if (vendorIndex === -1) { | ||
| magicVendorString = 'vendor/'; | ||
| if (pkgPath.startsWith(magicVendorString)) { | ||
| vendorIndex = 0; | ||
| } | ||
| } | ||
| // Check if current file and the vendor pkg belong to the same root project | ||
| // If yes, then vendor pkg can be replaced with its relative path to the "vendor" folder | ||
| // If not, then the vendor pkg should not be allowed to be imported. | ||
| if (vendorIndex > -1) { | ||
| let rootProjectForVendorPkg = path.join(currentWorkspace, pkgPath.substr(0, vendorIndex)); | ||
| let relativePathForVendorPkg = pkgPath.substring(vendorIndex + magicVendorString.length); | ||
|
|
||
| if (relativePathForVendorPkg && currentFileDirPath.startsWith(rootProjectForVendorPkg)) { | ||
| return relativePathForVendorPkg; | ||
| } | ||
| return ''; | ||
| } | ||
|
|
||
| return pkgPath; | ||
| } | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters