131 lines
4.4 KiB
Java
131 lines
4.4 KiB
Java
/*
|
||
* To change this license header, choose License Headers in Project Properties.
|
||
* To change this template file, choose Tools | Templates
|
||
* and open the template in the editor.
|
||
*/
|
||
package common;
|
||
|
||
/**
|
||
*
|
||
* @author tange
|
||
*/
|
||
import java.io.BufferedReader;
|
||
import java.io.File;
|
||
import java.io.IOException;
|
||
import java.io.InputStreamReader;
|
||
import java.util.ArrayList;
|
||
import java.util.List;
|
||
import java.util.Scanner;
|
||
import java.util.logging.Level;
|
||
import java.util.logging.Logger;
|
||
|
||
public class VSWhereUtil {
|
||
/**
|
||
* 从指定的起始目录开始，逐层向上查找 vswhere.exe
|
||
*
|
||
* @param startDir 起始目录
|
||
* @return vswhere.exe 的绝对路径
|
||
* @throws RuntimeException 如果未找到 vswhere.exe
|
||
*/
|
||
private static String findVswhereExe(File startDir) {
|
||
File currentDir = startDir;
|
||
|
||
while (currentDir != null) {
|
||
File vswhereFile = new File(currentDir, "vswhere.exe");
|
||
if (vswhereFile.exists() && vswhereFile.isFile()) {
|
||
return vswhereFile.getAbsolutePath();
|
||
}
|
||
currentDir = currentDir.getParentFile(); // 向上一级目录
|
||
}
|
||
|
||
throw new RuntimeException("未找到 vswhere.exe，请确保该文件存在于程序目录或上层目录中。");
|
||
}
|
||
private static String getVcvarsPath() throws IOException, InterruptedException {
|
||
String vswhereDir=findVswhereExe(new File(System.getProperty("user.dir")));
|
||
ProcessBuilder processBuilder = new ProcessBuilder(vswhereDir,
|
||
"-latest",
|
||
"-products", "*",
|
||
"installationPath",
|
||
"-find", "**/vcvars64.bat");
|
||
processBuilder.redirectErrorStream(true); // 合并标准错误输出
|
||
Process process = processBuilder.start();
|
||
List<String> output = new ArrayList<>();
|
||
try (Scanner scanner = new Scanner(process.getInputStream())) {
|
||
while (scanner.hasNextLine()) {
|
||
output.add(scanner.nextLine());
|
||
}
|
||
}
|
||
process.waitFor();
|
||
if (!output.isEmpty()) {
|
||
return output.get(0);
|
||
}
|
||
return null;
|
||
}
|
||
|
||
|
||
|
||
public static String findClExePath() {
|
||
// 获取当前工作目录
|
||
File currentDir = new File(System.getProperty("user.dir"));
|
||
// 查找 vswhere.exe 路径
|
||
String vswherePath = findVswhereExe(currentDir);
|
||
|
||
String clExeDir = null;
|
||
try {
|
||
// 构建命令行调用 vswhere.exe
|
||
ProcessBuilder processBuilder = new ProcessBuilder(vswherePath,
|
||
"-latest",
|
||
"-products", "*",
|
||
"-requires", "Microsoft.VisualStudio.Component.VC.Tools.x86.x64",
|
||
"-find", "**/Hostx64/x64/cl.exe");
|
||
|
||
// 设置工作目录为当前程序目录
|
||
processBuilder.directory(new File(System.getProperty("user.dir")));
|
||
|
||
// 启动进程
|
||
Process process = processBuilder.start();
|
||
|
||
// 读取命令行输出
|
||
BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
|
||
String line;
|
||
|
||
while ((line = reader.readLine()) != null) {
|
||
if (line.endsWith("cl.exe")) {
|
||
clExeDir = line.substring(0, line.lastIndexOf("cl.exe"));
|
||
break;
|
||
}
|
||
}
|
||
|
||
reader.close();
|
||
process.waitFor();
|
||
} catch (Exception e) {
|
||
e.printStackTrace();
|
||
}
|
||
|
||
return clExeDir;
|
||
}
|
||
public static String getMsvcCompilerCommand(){
|
||
try {
|
||
return "cmd.exe /c call \""+getVcvarsPath()+"\" && \"$compilerPath$/cl.exe\" \"$sourceFile$\" /O2 /Fe:\"$exeFile$\"";
|
||
} catch (IOException ex) {
|
||
Logger.getLogger(VSWhereUtil.class.getName()).log(Level.SEVERE, null, ex);
|
||
} catch (InterruptedException ex) {
|
||
Logger.getLogger(VSWhereUtil.class.getName()).log(Level.SEVERE, null, ex);
|
||
}
|
||
return null;
|
||
}
|
||
public static String getMsvcLinkCommand(){
|
||
return "cmd.exe /c";
|
||
}
|
||
public static void main(String[] args) {
|
||
String clPath = findClExePath();
|
||
if (clPath != null) {
|
||
System.out.println("cl.exe 目录路径: " + clPath);
|
||
System.out.println(getMsvcCompilerCommand());
|
||
|
||
} else {
|
||
System.out.println("未找到 cl.exe 路径");
|
||
}
|
||
}
|
||
}
|