Skip to content

Commit d70c3c4

Browse files
refackrefack
committed
Use minimal powershell script to extract VS2017 location and version via COM*
Work with all flavors of VS2017, even the minimal "C++ Build Tools" *https://blogs.msdn.microsoft.com/heaths/2016/09/15/changes-to-visual-studio-15-setup/
1 parent db63f3c commit d70c3c4

File tree

2 files changed

+206
-116
lines changed

2 files changed

+206
-116
lines changed

‎lib/find-vs2017.js‎

Lines changed: 30 additions & 116 deletions
Original file line numberDiff line numberDiff line change
@@ -1,138 +1,52 @@
11
module.exports = {
22
locateMsbuild: locateMsbuild,
3-
locateVS2017: locateVS2017,
4-
getVS2017WinSDKVersion: getVS2017WinSDKVersion,
53
setGypVS2017Env: setGypVS2017Env
64
}
75

86
var log = require('npmlog')
97
, fs = require('fs')
108
, path = require('path')
119
, cp = require('child_process')
12-
, win = process.platform == 'win32'
13-
, msgFormat = require('util').format
14-
, findPython = require('./find-python')
1510

16-
var vs2017_install_path
17-
, vs2017_win_sdk_ver
18-
19-
function run_locate(gyp, callback) {
20-
if (!win) {
21-
return callback(null, '', '')
22-
}
23-
24-
if (vs2017_install_path || vs2017_install_path === '') {
25-
return callback(null, vs2017_install_path, vs2017_win_sdk_ver)
26-
}
27-
28-
var python = gyp.opts.python || process.env.PYTHON || 'python2'
29-
, findvc_path = path.join(__dirname, '..', 'find_vs2017.py')
30-
31-
findPython(python, locate_vc);
32-
33-
function locate_vc(err, python_bin) {
34-
if (err) {
35-
return callback(err)
36-
}
37-
38-
log.verbose('find vs2017', 'obtaining vs2017 install path using script %s',
39-
findvc_path)
40-
cp.execFile(python_bin, [findvc_path], function(err, stdout, stderr) {
41-
if (err) {
42-
return callback(err)
43-
}
44-
if (stdout) {
45-
vs2017_install_path = stdout.split('\r\n')[0]
46-
log.verbose('find vs2017', 'found Visual Studio 2017 in %s', vs2017_install_path)
47-
get_sdk_version(python_bin)
48-
} else {
49-
log.verbose('find vs2017',
50-
'no valid Visual Studio 2017 installation found')
51-
vs2017_install_path = ''
52-
vs2017_win_sdk_ver = ''
53-
}
54-
})
55-
}
56-
57-
function get_sdk_version(python_bin) {
58-
log.verbose('find vs2017', 'obtaining installed Windows SDKs')
59-
cp.execFile(python_bin, [findvc_path, vs2017_install_path],
60-
function(err, stdout, stderr) {
61-
if (err) {
62-
return callback(err)
63-
}
64-
if (stdout) {
65-
vs2017_win_sdk_ver = stdout.split('\r\n')[0]
66-
log.verbose('find vs2017', 'found VS2017 WinSDK %s', vs2017_win_sdk_ver)
67-
} else {
68-
log.verbose('find vs2017', 'no installed sdks found')
69-
}
70-
71-
callback(null, vs2017_install_path, vs2017_win_sdk_ver)
72-
})
11+
var vs2017Setup
12+
13+
function tryVS7 (gyp) {
14+
if (vs2017Setup) return vs2017Setup;
15+
try {
16+
const psFile = path.join(__dirname, '..', 'tools', 'Get-VSConfig.ps1');
17+
const vsSetupRaw = cp.execSync('powershell ' + psFile).toString();
18+
if (!vsSetupRaw) return;
19+
const vsSetup = vsSetupRaw.split(/[\r|\n]/g).reduce((s, l) => {
20+
const lParts = l.split(': ');
21+
if (lParts.length > 1) s[lParts[0]] = lParts[1];
22+
return s;
23+
}, {});
24+
return vs2017Setup = vsSetup;
25+
} catch (e) {
26+
gyp.log.verbose('try VS7', 'Couldn\'t find VS2017 :(');
7327
}
74-
7528
}
7629

30+
7731
function locateMsbuild(gyp, callback) {
78-
run_locate(gyp, function(err, vs_path, sdk) {
79-
if (err) {
80-
return callback(err)
81-
}
82-
if (vs_path === '') {
83-
return callback()
84-
}
85-
var msbuild_location = path.join(vs_path, 'MSBuild',
86-
'15.0', 'Bin', 'MSBuild.exe')
87-
log.verbose('find vs2017', 'looking for msbuild in %s', msbuild_location)
88-
fs.access(msbuild_location, function(err) {
89-
callback(null, err ? null : msbuild_location)
90-
})
91-
})
92-
}
32+
var vsSetup = tryVS7(gyp)
33+
if (!vsSetup)return callback()
9334

94-
function locateVS2017(gyp, callback) {
95-
run_locate(gyp, function(err, vs_path, sdk) {
96-
if (err) {
97-
callback(err)
98-
} else {
99-
callback(null, vs_path === '' ? null : vs_path)
100-
}
35+
var msbuild_location = path.join(vsSetup.InstallationPath, 'MSBuild',
36+
'15.0', 'Bin', 'MSBuild.exe')
37+
log.verbose('find vs2017', 'looking for msbuild in %s', msbuild_location)
38+
fs.access(msbuild_location, function(err) {
39+
callback(null, err ? null : msbuild_location)
10140
})
10241
}
10342

104-
function getVS2017WinSDKVersion(gyp, callback) {
105-
run_locate(gyp, function(err, vs_path, sdk) {
106-
if (err) {
107-
callback(err)
108-
} else {
109-
callback(null, sdk === '' ? null : sdk)
110-
}
111-
})
112-
}
11343

11444
function setGypVS2017Env(gyp, callback) {
115-
locateVS2017(gyp, setPath)
116-
117-
function setPath(err, vs_path) {
118-
if (err) {
119-
return callback(err)
120-
}
121-
if (vs_path) {
122-
process.env['vs2017_install'] = vs_path
123-
getVS2017WinSDKVersion(gyp, setSDK)
124-
} else {
125-
callback()
126-
}
127-
}
45+
var vsSetup = tryVS7(gyp);
46+
if (!vsSetup)return callback()
12847

129-
function setSDK(err, sdk) {
130-
if (err) {
131-
return callback(err)
132-
}
133-
if (sdk) {
134-
process.env['vs2017_sdk'] = sdk
135-
}
136-
callback()
137-
}
48+
gyp.opts.msvs_version = '2017';
49+
process.env['vs2017_install'] = vsSetup.InstallationPath;
50+
process.env['vs2017_sdk'] = vsSetup.SDK.replace(/\d+$/, '0')
51+
callback();
13852
}

‎tools/Get-VSConfig.ps1‎

Lines changed: 176 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,176 @@
1+
$Source = @"
2+
using System;
3+
using System.Runtime.CompilerServices;
4+
using System.Runtime.InteropServices;
5+
using Microsoft.VisualStudio.Setup.Configuration;
6+
7+
8+
namespace Microsoft.VisualStudio.Setup.Configuration
9+
{
10+
[CompilerGenerated]
11+
[Flags]
12+
[TypeIdentifier("310100ba-5f84-4103-abe0-e8132ae862d9", "Microsoft.VisualStudio.Setup.Configuration.InstanceState")]
13+
[ComVisible(true)]
14+
public enum InstanceState : uint
15+
{
16+
None = 0,
17+
Local = 1,
18+
Registered = 2,
19+
NoRebootRequired = 4,
20+
NoErrors = 8,
21+
Complete = 4294967295
22+
}
23+
24+
[CompilerGenerated]
25+
[Guid("6380BCFF-41D3-4B2E-8B2E-BF8A6810C848")]
26+
[InterfaceType(ComInterfaceType.InterfaceIsIUnknown)]
27+
[TypeIdentifier]
28+
[ComImport]
29+
public interface IEnumSetupInstances
30+
{
31+
void Next([MarshalAs(UnmanagedType.U4)] [In] int celt,
32+
[MarshalAs(UnmanagedType.LPArray, ArraySubType = UnmanagedType.Interface)] [Out] ISetupInstance[] rgelt,
33+
[MarshalAs(UnmanagedType.U4)] out int pceltFetched);
34+
}
35+
36+
37+
[CompilerGenerated]
38+
[Guid("42843719-DB4C-46C2-8E7C-64F1816EFD5B")]
39+
[InterfaceType(ComInterfaceType.InterfaceIsIUnknown)]
40+
[TypeIdentifier]
41+
[ComImport]
42+
public interface ISetupConfiguration
43+
{
44+
}
45+
46+
[CompilerGenerated]
47+
[Guid("26AAB78C-4A60-49D6-AF3B-3C35BC93365D")]
48+
[InterfaceType(ComInterfaceType.InterfaceIsIUnknown)]
49+
[TypeIdentifier]
50+
[ComImport]
51+
public interface ISetupConfiguration2 : ISetupConfiguration
52+
{
53+
[SpecialName]
54+
[MethodImpl(MethodCodeType = MethodCodeType.Runtime)]
55+
void _VtblGap1_3();
56+
57+
[return: MarshalAs(UnmanagedType.Interface)]
58+
IEnumSetupInstances EnumAllInstances();
59+
}
60+
61+
62+
[CompilerGenerated]
63+
[Guid("B41463C3-8866-43B5-BC33-2B0676F7F42E")]
64+
[InterfaceType(ComInterfaceType.InterfaceIsIUnknown)]
65+
[TypeIdentifier]
66+
[ComImport]
67+
public interface ISetupInstance
68+
{
69+
[SpecialName]
70+
[MethodImpl(MethodCodeType = MethodCodeType.Runtime)]
71+
void _VtblGap1_4();
72+
73+
[return: MarshalAs(UnmanagedType.BStr)]
74+
string GetInstallationVersion();
75+
}
76+
77+
[CompilerGenerated]
78+
[Guid("89143C9A-05AF-49B0-B717-72E218A2185C")]
79+
[InterfaceType(ComInterfaceType.InterfaceIsIUnknown)]
80+
[TypeIdentifier]
81+
[ComImport]
82+
public interface ISetupInstance2 : ISetupInstance
83+
{
84+
[return: MarshalAs(UnmanagedType.BStr)]
85+
string GetInstanceId();
86+
87+
[SpecialName]
88+
[MethodImpl(MethodCodeType = MethodCodeType.Runtime)]
89+
void _VtblGap1_2();
90+
91+
[return: MarshalAs(UnmanagedType.BStr)]
92+
string GetInstallationPath();
93+
94+
[SpecialName]
95+
[MethodImpl(MethodCodeType = MethodCodeType.Runtime)]
96+
void _VtblGap2_4();
97+
98+
[return: MarshalAs(UnmanagedType.U4)]
99+
InstanceState GetState();
100+
101+
[return: MarshalAs(UnmanagedType.SafeArray, SafeArraySubType = VarEnum.VT_UNKNOWN)]
102+
ISetupPackageReference[] GetPackages();
103+
104+
ISetupPackageReference GetProduct();
105+
}
106+
107+
[CompilerGenerated]
108+
[Guid("DA8D8A16-B2B6-4487-A2F1-594CCCCD6BF5")]
109+
[InterfaceType(ComInterfaceType.InterfaceIsIUnknown)]
110+
[TypeIdentifier]
111+
[ComImport]
112+
public interface ISetupPackageReference
113+
{
114+
[return: MarshalAs(UnmanagedType.BStr)]
115+
string GetId();
116+
117+
[SpecialName]
118+
[MethodImpl(MethodCodeType = MethodCodeType.Runtime)]
119+
void _VtblGap1_4();
120+
121+
[return: MarshalAs(UnmanagedType.BStr)]
122+
string GetType();
123+
}
124+
}
125+
126+
public static class VisualStudioSetup
127+
{
128+
public static int Main()
129+
{
130+
try
131+
{
132+
var query =
133+
(ISetupConfiguration2)
134+
Activator.CreateInstance(Marshal.GetTypeFromCLSID(new Guid("177F0C4A-1CD3-4DE7-A32C-71DBBB9FA36D")));
135+
var enumSetupInstances = query.EnumAllInstances();
136+
var rgelt = new ISetupInstance2[1];
137+
int pceltFetched;
138+
do
139+
{
140+
// ISSUE: reference to a compiler-generated method
141+
enumSetupInstances.Next(1, rgelt, out pceltFetched);
142+
if (pceltFetched > 0)
143+
PrintInstance(rgelt[0]);
144+
} while (pceltFetched > 0);
145+
return 0;
146+
}
147+
catch (Exception ex)
148+
{
149+
Console.Error.WriteLine("Error 0x{0:x8}: {1}", ex.HResult, ex.Message);
150+
return ex.HResult;
151+
}
152+
}
153+
154+
155+
private static void PrintInstance(ISetupInstance2 setupInstance2)
156+
{
157+
Console.WriteLine("InstallationPath: {0}", setupInstance2.GetInstallationPath());
158+
Console.WriteLine("Product: {0}", setupInstance2.GetProduct().GetId());
159+
foreach (var package in setupInstance2.GetPackages())
160+
{
161+
if (package.GetType() != "Exe") continue;;
162+
var id = package.GetId();
163+
if (id.IndexOf("SDK", StringComparison.Ordinal) == -1) continue;
164+
var parts = id.Split('_');
165+
if (parts.Length < 2) continue;
166+
var sdkVer = parts[1];
167+
char[] chars = { '1', '0', '8' };
168+
if (sdkVer.IndexOfAny(chars) == -1) continue;
169+
Console.WriteLine("SDK: {0}", sdkVer);
170+
}
171+
Console.WriteLine();
172+
}
173+
}
174+
"@
175+
Add-Type -TypeDefinition $Source -Language CSharp
176+
[VisualStudioSetup]::Main()

0 commit comments

Comments
 (0)