vcpkg/toolsrc/src/vcpkg.cpp

250 lines
7.3 KiB
C++
Raw Normal View History

2016-09-19 11:50:08 +08:00
#define WIN32_LEAN_AND_MEAN
#include <Windows.h>
#include <iostream>
#include <fstream>
#include <memory>
#include <cassert>
#include "vcpkg_Commands.h"
#include "metrics.h"
#include <Shlobj.h>
#include "vcpkg_Files.h"
#include "vcpkg_System.h"
2016-10-01 07:54:07 +08:00
#include "vcpkg_Input.h"
2016-11-08 08:06:36 +08:00
#include "Paragraphs.h"
#include "vcpkg_Strings.h"
2016-09-19 11:50:08 +08:00
using namespace vcpkg;
bool g_debugging = false;
void invalid_command(const std::string& cmd)
{
System::println(System::color::error, "invalid command: %s", cmd);
Commands::Help::print_usage();
2016-09-19 11:50:08 +08:00
exit(EXIT_FAILURE);
}
static void inner(const vcpkg_cmd_arguments& args)
{
TrackProperty("command", args.command);
if (args.command.empty())
{
Commands::Help::print_usage();
2016-09-19 11:50:08 +08:00
exit(EXIT_FAILURE);
}
if (auto command_function = Commands::find(args.command, Commands::get_available_commands_type_c()))
2016-09-19 11:50:08 +08:00
{
return command_function(args);
}
fs::path vcpkg_root_dir;
if (args.vcpkg_root_dir != nullptr)
{
vcpkg_root_dir = fs::absolute(Strings::utf8_to_utf16(*args.vcpkg_root_dir));
}
else
{
const optional<std::wstring> vcpkg_root_dir_env = System::get_environmental_variable(L"VCPKG_ROOT");
if (vcpkg_root_dir_env)
{
vcpkg_root_dir = fs::absolute(*vcpkg_root_dir_env);
}
2016-09-19 11:50:08 +08:00
else
{
2016-09-22 12:43:59 +08:00
vcpkg_root_dir = Files::find_file_recursively_up(fs::absolute(System::get_exe_path_of_current_process()), ".vcpkg-root");
}
2016-09-19 11:50:08 +08:00
}
Checks::check_exit(!vcpkg_root_dir.empty(), "Error: Could not detect vcpkg-root.");
2016-09-19 11:50:08 +08:00
const expected<vcpkg_paths> expected_paths = vcpkg_paths::create(vcpkg_root_dir);
Checks::check_exit(!expected_paths.error_code(), "Error: Invalid vcpkg root directory %s: %s", vcpkg_root_dir.string(), expected_paths.error_code().message());
const vcpkg_paths paths = expected_paths.get_or_throw();
int exit_code = _wchdir(paths.root.c_str());
Checks::check_exit(exit_code == 0, "Changing the working dir failed");
if (auto command_function = Commands::find(args.command, Commands::get_available_commands_type_b()))
2016-09-19 11:50:08 +08:00
{
return command_function(args, paths);
}
triplet default_target_triplet;
if (args.target_triplet != nullptr)
{
default_target_triplet = triplet::from_canonical_name(*args.target_triplet);
}
else
{
const optional<std::wstring> vcpkg_default_triplet_env = System::get_environmental_variable(L"VCPKG_DEFAULT_TRIPLET");
if (vcpkg_default_triplet_env)
2016-09-19 11:50:08 +08:00
{
default_target_triplet = triplet::from_canonical_name(Strings::utf16_to_utf8(*vcpkg_default_triplet_env));
2016-09-19 11:50:08 +08:00
}
else
2016-09-19 11:50:08 +08:00
{
default_target_triplet = triplet::X86_WINDOWS;
2016-09-19 11:50:08 +08:00
}
}
2016-10-01 07:54:07 +08:00
Input::check_triplet(default_target_triplet, paths);
if (auto command_function = Commands::find(args.command, Commands::get_available_commands_type_a()))
2016-09-19 11:50:08 +08:00
{
return command_function(args, paths, default_target_triplet);
}
return invalid_command(args.command);
}
static void loadConfig()
{
fs::path localappdata;
{
// Config path in AppDataLocal
wchar_t* localappdatapath = nullptr;
if (S_OK != SHGetKnownFolderPath(FOLDERID_LocalAppData, 0, nullptr, &localappdatapath))
__fastfail(1);
localappdata = localappdatapath;
CoTaskMemFree(localappdatapath);
}
try
{
std::string config_contents = Files::read_contents(localappdata / "vcpkg" / "config").get_or_throw();
2016-09-19 11:50:08 +08:00
std::unordered_map<std::string, std::string> keys;
2016-11-08 08:06:36 +08:00
auto pghs = Paragraphs::parse_paragraphs(config_contents);
2016-09-19 11:50:08 +08:00
if (pghs.size() > 0)
keys = pghs[0];
for (size_t x = 1; x < pghs.size(); ++x)
{
for (auto&& p : pghs[x])
keys.insert(p);
}
auto user_id = keys["User-Id"];
auto user_time = keys["User-Since"];
Checks::check_throw(!user_id.empty() && !user_time.empty(), ""); // Use as goto to the catch statement
SetUserInformation(user_id, user_time);
return;
}
catch (...)
{
}
// config file not found, could not be read, or invalid
std::string user_id, user_time;
InitUserInformation(user_id, user_time);
SetUserInformation(user_id, user_time);
try
{
std::error_code ec;
fs::create_directory(localappdata / "vcpkg", ec);
std::ofstream(localappdata / "vcpkg" / "config", std::ios_base::out | std::ios_base::trunc)
<< "User-Id: " << user_id << "\n"
<< "User-Since: " << user_time << "\n";
}
catch (...)
{
}
}
static System::Stopwatch2 g_timer;
2016-09-19 11:50:08 +08:00
static std::string trim_path_from_command_line(const std::string& full_command_line)
{
Checks::check_exit(full_command_line.size() > 0, "Internal failure - cannot have empty command line");
if (full_command_line[0] == '"')
{
auto it = std::find(full_command_line.cbegin() + 1, full_command_line.cend(), '"');
if (it != full_command_line.cend()) // Skip over the quote
++it;
while (it != full_command_line.cend() && *it == ' ') // Skip over a space
++it;
return std::string(it, full_command_line.cend());
}
auto it = std::find(full_command_line.cbegin(), full_command_line.cend(), ' ');
while (it != full_command_line.cend() && *it == ' ')
++it;
return std::string(it, full_command_line.cend());
}
int wmain(const int argc, const wchar_t* const* const argv)
{
if (argc == 0)
std::abort();
g_timer.start();
atexit([]()
{
g_timer.stop();
TrackMetric("elapsed_us", g_timer.microseconds());
Flush();
});
TrackProperty("version", Commands::Version::version());
2016-09-19 11:50:08 +08:00
const std::string trimmed_command_line = trim_path_from_command_line(Strings::utf16_to_utf8(GetCommandLineW()));
TrackProperty("cmdline", trimmed_command_line);
loadConfig();
2016-12-13 06:03:13 +08:00
TrackProperty("sqmuser", GetSQMUser());
2016-09-19 11:50:08 +08:00
const vcpkg_cmd_arguments args = vcpkg_cmd_arguments::create_from_command_line(argc, argv);
2017-02-01 05:31:45 +08:00
if (args.printmetrics != opt_bool_t::UNSPECIFIED)
SetPrintMetrics(args.printmetrics == opt_bool_t::ENABLED);
if (args.sendmetrics != opt_bool_t::UNSPECIFIED)
SetSendMetrics(args.sendmetrics == opt_bool_t::ENABLED);
2016-09-19 11:50:08 +08:00
2017-02-01 05:31:45 +08:00
if (args.debug != opt_bool_t::UNSPECIFIED)
2016-09-19 11:50:08 +08:00
{
2017-02-01 05:31:45 +08:00
g_debugging = (args.debug == opt_bool_t::ENABLED);
2016-09-19 11:50:08 +08:00
}
if (g_debugging)
{
inner(args);
exit(EXIT_FAILURE);
}
std::string exc_msg;
try
{
inner(args);
exit(EXIT_FAILURE);
}
catch (std::exception& e)
{
exc_msg = e.what();
}
catch (...)
{
exc_msg = "unknown error(...)";
}
TrackProperty("error", exc_msg);
fflush(stdout);
System::print(
"vcpkg.exe has crashed.\n"
"Please send an email to:\n"
" %s\n"
"containing a brief summary of what you were trying to do and the following data blob:\n"
"\n"
"Version=%s\n"
"EXCEPTION='%s'\n"
"CMD=\n",
Commands::Contact::email(),
Commands::Version::version(),
exc_msg);
fflush(stdout);
2016-09-19 11:50:08 +08:00
for (int x = 0; x < argc; ++x)
System::println("%s|", Strings::utf16_to_utf8(argv[x]));
fflush(stdout);
2016-09-19 11:50:08 +08:00
}