vcpkg/toolsrc/src/commands_install.cpp

336 lines
15 KiB
C++
Raw Normal View History

2017-01-28 04:49:09 +08:00
#include "pch.h"
2016-09-19 11:50:08 +08:00
#include "vcpkg_Commands.h"
2017-01-26 11:32:50 +08:00
#include "vcpkglib.h"
2016-09-19 11:50:08 +08:00
#include "metrics.h"
#include "vcpkg_Files.h"
#include "vcpkg_System.h"
#include "vcpkg_Dependencies.h"
#include "vcpkg_Input.h"
#include "vcpkg_Util.h"
2017-02-28 07:52:57 +08:00
#include "Paragraphs.h"
2016-09-19 11:50:08 +08:00
namespace vcpkg::Commands::Install
2016-09-19 11:50:08 +08:00
{
using Dependencies::PackageSpecWithInstallPlan;
using Dependencies::RequestType;
2017-04-04 07:22:32 +08:00
using Dependencies::InstallPlanType;
2016-11-16 03:56:46 +08:00
2017-04-04 07:29:11 +08:00
static void install_and_write_listfile(const VcpkgPaths& paths, const BinaryParagraph& bpgh)
{
2016-12-15 08:51:05 +08:00
std::vector<std::string> output;
2016-12-15 08:51:05 +08:00
const fs::path package_prefix_path = paths.package_dir(bpgh.spec);
const size_t prefix_length = package_prefix_path.native().size();
2017-04-04 06:02:45 +08:00
const Triplet& target_triplet = bpgh.spec.target_triplet();
const std::string& triplet_subfolder = target_triplet.canonical_name();
const fs::path triplet_subfolder_path = paths.installed / triplet_subfolder;
std::error_code ec;
fs::create_directory(triplet_subfolder_path, ec);
output.push_back(Strings::format(R"(%s/)", triplet_subfolder));
for (auto it = fs::recursive_directory_iterator(package_prefix_path); it != fs::recursive_directory_iterator(); ++it)
{
const std::string filename = it->path().filename().generic_string();
if (fs::is_regular_file(it->status()) && (_stricmp(filename.c_str(), "CONTROL") == 0 || _stricmp(filename.c_str(), "BUILD_INFO") == 0))
{
// Do not copy the control file
continue;
}
2016-12-15 08:51:05 +08:00
const std::string suffix = it->path().generic_u8string().substr(prefix_length + 1);
const fs::path target = triplet_subfolder_path / suffix;
auto status = it->status(ec);
if (ec)
{
2017-04-04 07:31:00 +08:00
System::println(System::Color::error, "failed: %s: %s", it->path().u8string(), ec.message());
continue;
}
2016-12-15 08:51:05 +08:00
if (fs::is_directory(status))
{
fs::create_directory(target, ec);
if (ec)
{
2017-04-04 07:31:00 +08:00
System::println(System::Color::error, "failed: %s: %s", target.u8string(), ec.message());
}
2016-12-15 08:51:05 +08:00
// Trailing backslash for directories
output.push_back(Strings::format(R"(%s/%s/)", triplet_subfolder, suffix));
2016-12-15 08:51:05 +08:00
continue;
}
2016-12-15 08:51:05 +08:00
if (fs::is_regular_file(status))
{
if (fs::exists(target))
{
2017-04-04 07:31:00 +08:00
System::println(System::Color::warning, "File %s was already present and will be overwritten", target.u8string(), ec.message());
}
fs::copy_file(*it, target, fs::copy_options::overwrite_existing, ec);
if (ec)
{
2017-04-04 07:31:00 +08:00
System::println(System::Color::error, "failed: %s: %s", target.u8string(), ec.message());
}
output.push_back(Strings::format(R"(%s/%s)", triplet_subfolder, suffix));
2016-12-15 08:51:05 +08:00
continue;
}
2016-12-15 08:51:05 +08:00
if (!fs::status_known(status))
{
2017-04-04 07:31:00 +08:00
System::println(System::Color::error, "failed: %s: unknown status", it->path().u8string());
2016-12-15 08:51:05 +08:00
continue;
}
2016-12-15 08:51:05 +08:00
2017-04-04 07:31:00 +08:00
System::println(System::Color::error, "failed: %s: cannot handle file type", it->path().u8string());
}
2017-02-16 11:19:19 +08:00
std::sort(output.begin(), output.end());
2016-12-16 10:24:06 +08:00
Files::write_all_lines(paths.listfile_path(bpgh), output);
}
static void remove_first_n_chars(std::vector<std::string>* strings, const size_t n)
{
for (std::string& s : *strings)
{
s.erase(0, n);
}
};
static std::vector<std::string> extract_files_in_triplet(const std::vector<StatusParagraphAndAssociatedFiles>& pgh_and_files, const Triplet& triplet)
{
std::vector<std::string> output;
for (const StatusParagraphAndAssociatedFiles& t : pgh_and_files)
{
if (t.pgh.package.spec.target_triplet() != triplet)
{
continue;
}
output.insert(output.end(), t.files.begin(), t.files.end());
}
std::sort(output.begin(), output.end());
return output;
}
static SortedVector<std::string> build_list_of_package_files(const fs::path& package_dir)
{
const std::vector<fs::path> package_file_paths = Files::recursive_find_all_files_in_dir(package_dir);
const size_t package_remove_char_count = package_dir.generic_string().size() + 1; // +1 for the slash
auto package_files = Util::fmap(package_file_paths, [package_remove_char_count](const fs::path& path)
{
std::string as_string = path.generic_string();
as_string.erase(0, package_remove_char_count);
return std::move(as_string);
});
return SortedVector<std::string>(std::move(package_files));
2017-01-06 07:23:08 +08:00
}
static SortedVector<std::string> build_list_of_installed_files(const std::vector<StatusParagraphAndAssociatedFiles>& pgh_and_files, const Triplet& triplet)
2017-01-06 07:23:08 +08:00
{
std::vector<std::string> installed_files = extract_files_in_triplet(pgh_and_files, triplet);
const size_t installed_remove_char_count = triplet.canonical_name().size() + 1; // +1 for the slash
remove_first_n_chars(&installed_files, installed_remove_char_count);
2017-01-06 07:23:08 +08:00
return SortedVector<std::string>(std::move(installed_files));
2017-01-06 07:23:08 +08:00
}
static void print_plan(const std::vector<PackageSpecWithInstallPlan>& plan)
{
std::vector<const PackageSpecWithInstallPlan*> already_installed;
std::vector<const PackageSpecWithInstallPlan*> build_and_install;
std::vector<const PackageSpecWithInstallPlan*> install;
for (const PackageSpecWithInstallPlan& i : plan)
{
switch (i.plan.plan_type)
{
case InstallPlanType::ALREADY_INSTALLED:
already_installed.push_back(&i);
continue;
case InstallPlanType::BUILD_AND_INSTALL:
build_and_install.push_back(&i);
continue;
case InstallPlanType::INSTALL:
install.push_back(&i);
continue;
default:
Checks::unreachable(VCPKG_LINE_INFO);
}
}
auto print_lambda = [](const PackageSpecWithInstallPlan* p) { return to_output_string(p->plan.request_type, p->spec.to_string()); };
if (!already_installed.empty())
{
std::sort(already_installed.begin(), already_installed.end(), &PackageSpecWithInstallPlan::compare_by_name);
System::println("The following packages are already installed:\n%s", Strings::join("\n", already_installed, print_lambda));
}
if (!build_and_install.empty())
{
std::sort(build_and_install.begin(), build_and_install.end(), &PackageSpecWithInstallPlan::compare_by_name);
System::println("The following packages will be built and installed:\n%s", Strings::join("\n", build_and_install, print_lambda));
}
if (!install.empty())
{
std::sort(install.begin(), install.end(), &PackageSpecWithInstallPlan::compare_by_name);
System::println("The following packages will be installed:\n%s", Strings::join("\n", install, print_lambda));
}
}
2017-04-04 07:29:11 +08:00
void install_package(const VcpkgPaths& paths, const BinaryParagraph& binary_paragraph, StatusParagraphs* status_db)
2017-01-06 07:23:08 +08:00
{
const fs::path package_dir = paths.package_dir(binary_paragraph.spec);
2017-04-04 06:02:45 +08:00
const Triplet& triplet = binary_paragraph.spec.target_triplet();
const std::vector<StatusParagraphAndAssociatedFiles> pgh_and_files = get_installed_files(paths, *status_db);
2017-01-06 07:23:08 +08:00
const SortedVector<std::string> package_files = build_list_of_package_files(package_dir);
const SortedVector<std::string> installed_files = build_list_of_installed_files(pgh_and_files, triplet);
std::vector<std::string> intersection;
std::set_intersection(package_files.begin(), package_files.end(),
installed_files.begin(), installed_files.end(),
std::back_inserter(intersection));
if (!intersection.empty())
{
const fs::path triplet_install_path = paths.installed / triplet.canonical_name();
2017-04-04 07:31:00 +08:00
System::println(System::Color::error, "The following files are already installed in %s and are in conflict with %s",
triplet_install_path.generic_string(),
2017-04-08 07:17:54 +08:00
binary_paragraph.spec);
2017-01-31 11:37:07 +08:00
System::print("\n ");
2017-03-09 14:33:28 +08:00
System::println(Strings::join("\n ", intersection));
2017-01-31 11:37:07 +08:00
System::println("");
Checks::exit_fail(VCPKG_LINE_INFO);
}
StatusParagraph spgh;
spgh.package = binary_paragraph;
spgh.want = Want::INSTALL;
spgh.state = InstallState::HALF_INSTALLED;
for (auto&& dep : spgh.package.depends)
{
if (status_db->find_installed(dep, spgh.package.spec.target_triplet()) == status_db->end())
{
Checks::unreachable(VCPKG_LINE_INFO);
}
}
write_update(paths, spgh);
status_db->insert(std::make_unique<StatusParagraph>(spgh));
install_and_write_listfile(paths, spgh.package);
spgh.state = InstallState::INSTALLED;
write_update(paths, spgh);
status_db->insert(std::make_unique<StatusParagraph>(spgh));
}
2017-04-04 07:29:11 +08:00
void perform_and_exit(const VcpkgCmdArguments& args, const VcpkgPaths& paths, const Triplet& default_target_triplet)
2016-09-19 11:50:08 +08:00
{
static const std::string OPTION_DRY_RUN = "--dry-run";
// input sanitization
static const std::string example = Commands::Help::create_example_string("install zlib zlib:x64-windows curl boost");
args.check_min_arg_count(1, example);
auto specs = Util::fmap(args.command_arguments, [&](auto&& arg)
{
auto spec = Input::check_and_get_package_spec(arg, default_target_triplet, example);
Input::check_triplet(spec.target_triplet(), paths);
return spec;
});
const std::unordered_set<std::string> options = args.check_and_get_optional_command_arguments({ OPTION_DRY_RUN });
const bool dryRun = options.find(OPTION_DRY_RUN) != options.cend();
// create the plan
StatusParagraphs status_db = database_load_check(paths);
std::vector<PackageSpecWithInstallPlan> install_plan = Dependencies::create_install_plan(paths, specs, status_db);
Checks::check_exit(VCPKG_LINE_INFO, !install_plan.empty(), "Install plan cannot be empty");
// log the plan
std::string specs_string = install_plan[0].spec.to_string();
for (size_t i = 1; i < install_plan.size(); ++i)
2016-09-19 11:50:08 +08:00
{
specs_string.push_back(',');
specs_string.append(install_plan[i].spec.to_string());
2016-09-19 11:50:08 +08:00
}
2017-04-04 05:41:36 +08:00
Metrics::track_property("installplan", specs_string);
2016-09-19 11:50:08 +08:00
print_plan(install_plan);
const bool has_non_user_requested_packages = std::find_if(install_plan.cbegin(), install_plan.cend(), [](const PackageSpecWithInstallPlan& package)-> bool
{
return package.plan.request_type != RequestType::USER_REQUESTED;
}) != install_plan.cend();
if (has_non_user_requested_packages)
{
System::println(System::Color::warning, "Additional packages (*) need to be installed to complete this operation.");
}
if (dryRun)
{
Checks::exit_success(VCPKG_LINE_INFO);
}
// execute the plan
for (const PackageSpecWithInstallPlan& action : install_plan)
2016-09-19 11:50:08 +08:00
{
2017-04-08 07:17:54 +08:00
const std::string display_name = action.spec.to_string();
2017-04-08 07:01:09 +08:00
2016-09-19 11:50:08 +08:00
try
{
2017-04-08 06:17:10 +08:00
switch (action.plan.plan_type)
2016-09-19 11:50:08 +08:00
{
2017-04-08 06:17:10 +08:00
case InstallPlanType::ALREADY_INSTALLED:
2017-04-08 07:08:32 +08:00
System::println(System::Color::success, "Package %s is already installed", display_name);
2017-04-08 06:17:10 +08:00
break;
case InstallPlanType::BUILD_AND_INSTALL:
{
2017-04-08 07:01:09 +08:00
System::println("Building package %s... ", display_name);
2017-04-08 06:17:10 +08:00
const Build::BuildResult result = Commands::Build::build_package(action.plan.source_pgh.value_or_exit(VCPKG_LINE_INFO),
action.spec,
paths,
paths.port_dir(action.spec),
status_db);
if (result != Build::BuildResult::SUCCEEDED)
{
System::println(System::Color::error, Build::create_error_message(result, action.spec));
System::println(Build::create_user_troubleshooting_message(action.spec));
Checks::exit_fail(VCPKG_LINE_INFO);
}
2017-04-08 07:01:09 +08:00
System::println(System::Color::success, "Building package %s... done", display_name);
2017-04-08 06:17:10 +08:00
const BinaryParagraph bpgh = Paragraphs::try_load_cached_package(paths, action.spec).value_or_exit(VCPKG_LINE_INFO);
2017-04-08 07:01:09 +08:00
System::println("Installing package %s... ", display_name);
2017-04-08 06:17:10 +08:00
install_package(paths, bpgh, &status_db);
2017-04-08 07:01:09 +08:00
System::println(System::Color::success, "Installing package %s... done", display_name);
2017-04-08 06:48:22 +08:00
break;
2017-04-08 06:17:10 +08:00
}
case InstallPlanType::INSTALL:
2017-04-08 07:01:09 +08:00
System::println("Installing package %s... ", display_name);
2017-04-08 06:17:10 +08:00
install_package(paths, action.plan.binary_pgh.value_or_exit(VCPKG_LINE_INFO), &status_db);
2017-04-08 07:01:09 +08:00
System::println(System::Color::success, "Installing package %s... done", display_name);
2017-04-08 06:17:10 +08:00
break;
2017-04-08 06:20:04 +08:00
case InstallPlanType::UNKNOWN:
2017-04-08 06:17:10 +08:00
default:
Checks::unreachable(VCPKG_LINE_INFO);
2016-09-19 11:50:08 +08:00
}
}
catch (const std::exception& e)
{
2017-04-08 07:17:54 +08:00
System::println(System::Color::error, "Error: Could not install package %s: %s", action.spec, e.what());
Checks::exit_fail(VCPKG_LINE_INFO);
2016-09-19 11:50:08 +08:00
}
}
Checks::exit_success(VCPKG_LINE_INFO);
2016-09-19 11:50:08 +08:00
}
}