Move install_package() and deinstall_package() to the files of the

appropriate commands
This commit is contained in:
Alexander Karatarakis 2016-12-01 01:49:24 -08:00
parent 89aaf195fb
commit 6eac44c964
4 changed files with 279 additions and 279 deletions

View File

@ -9,8 +9,7 @@ namespace vcpkg
{
StatusParagraphs database_load_check(const vcpkg_paths& paths);
void install_package(const vcpkg_paths& paths, const BinaryParagraph& binary_paragraph, StatusParagraphs& status_db);
void deinstall_package(const vcpkg_paths& paths, const package_spec& spec, StatusParagraphs& status_db);
void write_update(const vcpkg_paths& paths, const StatusParagraph& p);
expected<SourceParagraph> try_load_port(const fs::path& control_path);

View File

@ -67,6 +67,138 @@ namespace vcpkg
// delete_directory(port_buildtrees_dir);
}
static void install_and_write_listfile(const vcpkg_paths& paths, const BinaryParagraph& bpgh)
{
std::fstream listfile(paths.listfile_path(bpgh), std::ios_base::out | std::ios_base::binary | std::ios_base::trunc);
auto package_prefix_path = paths.package_dir(bpgh.spec);
auto prefix_length = package_prefix_path.native().size();
const triplet& target_triplet = bpgh.spec.target_triplet();
const std::string& target_triplet_as_string = target_triplet.canonical_name();
std::error_code ec;
fs::create_directory(paths.installed / target_triplet_as_string, ec);
listfile << target_triplet << "\n";
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;
}
auto suffix = it->path().generic_u8string().substr(prefix_length + 1);
auto target = paths.installed / target_triplet_as_string / suffix;
auto status = it->status(ec);
if (ec)
{
System::println(System::color::error, "failed: %s: %s", it->path().u8string(), ec.message());
continue;
}
if (fs::is_directory(status))
{
fs::create_directory(target, ec);
if (ec)
{
System::println(System::color::error, "failed: %s: %s", target.u8string(), ec.message());
}
listfile << target_triplet << "/" << suffix << "\n";
}
else if (fs::is_regular_file(status))
{
fs::copy_file(*it, target, ec);
if (ec)
{
System::println(System::color::error, "failed: %s: %s", target.u8string(), ec.message());
}
listfile << target_triplet << "/" << suffix << "\n";
}
else if (!fs::status_known(status))
{
System::println(System::color::error, "failed: %s: unknown status", it->path().u8string());
}
else
System::println(System::color::error, "failed: %s: cannot handle file type", it->path().u8string());
}
listfile.close();
}
static std::map<std::string, fs::path> remove_first_n_chars_and_map(const std::vector<fs::path> absolute_paths, const size_t n)
{
std::map<std::string, fs::path> output;
for (const fs::path& absolute_path : absolute_paths)
{
std::string suffix = absolute_path.generic_string();
suffix.erase(0, n);
output.emplace(suffix, absolute_path);
}
return output;
}
static void print_map_values(const std::vector<std::string> keys, const std::map<std::string, fs::path>& map)
{
System::println("");
for (const std::string& key : keys)
{
System::println(" %s", map.at(key).generic_string());
}
System::println("");
}
static void install_package(const vcpkg_paths& paths, const BinaryParagraph& binary_paragraph, StatusParagraphs& status_db)
{
const fs::path package_dir = paths.package_dir(binary_paragraph.spec);
const std::vector<fs::path> package_files = Files::recursive_find_all_files_in_dir(package_dir);
const fs::path installed_dir = paths.installed / binary_paragraph.spec.target_triplet().canonical_name();
const std::vector<fs::path> installed_files = Files::recursive_find_all_files_in_dir(installed_dir);
const std::map<std::string, fs::path> package_files_relative_paths_to_absolute_paths = remove_first_n_chars_and_map(package_files, package_dir.generic_string().size() + 1);
const std::map<std::string, fs::path> installed_files_relative_paths_to_absolute_paths = remove_first_n_chars_and_map(installed_files, installed_dir.generic_string().size() + 1);
const std::vector<std::string> package_files_set = Maps::extract_keys(package_files_relative_paths_to_absolute_paths);
const std::vector<std::string> installed_files_set = Maps::extract_keys(installed_files_relative_paths_to_absolute_paths);
std::vector<std::string> intersection;
std::set_intersection(package_files_set.cbegin(), package_files_set.cend(),
installed_files_set.cbegin(), installed_files_set.cend(),
std::back_inserter(intersection));
if (!intersection.empty())
{
System::println(System::color::error, "The following files are already installed and are in conflict with %s:", binary_paragraph.spec);
print_map_values(intersection, installed_files_relative_paths_to_absolute_paths);
exit(EXIT_FAILURE);
}
StatusParagraph spgh;
spgh.package = binary_paragraph;
spgh.want = want_t::install;
spgh.state = install_state_t::half_installed;
for (auto&& dep : spgh.package.depends)
{
if (status_db.find_installed(dep, spgh.package.spec.target_triplet()) == status_db.end())
{
Checks::unreachable();
}
}
write_update(paths, spgh);
status_db.insert(std::make_unique<StatusParagraph>(spgh));
install_and_write_listfile(paths, spgh.package);
spgh.state = install_state_t::installed;
write_update(paths, spgh);
status_db.insert(std::make_unique<StatusParagraph>(spgh));
}
void install_command(const vcpkg_cmd_arguments& args, const vcpkg_paths& paths, const triplet& default_target_triplet)
{
static const std::string example = create_example_string("install zlib zlib:x64-windows curl boost");

View File

@ -2,6 +2,7 @@
#include "vcpkg.h"
#include "vcpkg_System.h"
#include "vcpkg_Input.h"
#include <fstream>
namespace vcpkg
{
@ -21,6 +22,150 @@ namespace vcpkg
}
}
enum class deinstall_plan
{
not_installed,
dependencies_not_satisfied,
should_deinstall
};
static deinstall_plan deinstall_package_plan(
const StatusParagraphs::iterator package_it,
const StatusParagraphs& status_db,
std::vector<const StatusParagraph*>& dependencies_out)
{
dependencies_out.clear();
if (package_it == status_db.end() || (*package_it)->state == install_state_t::not_installed)
{
return deinstall_plan::not_installed;
}
auto& pkg = (*package_it)->package;
for (auto&& inst_pkg : status_db)
{
if (inst_pkg->want != want_t::install)
continue;
if (inst_pkg->package.spec.target_triplet() != pkg.spec.target_triplet())
continue;
const auto& deps = inst_pkg->package.depends;
if (std::find(deps.begin(), deps.end(), pkg.spec.name()) != deps.end())
{
dependencies_out.push_back(inst_pkg.get());
}
}
if (!dependencies_out.empty())
return deinstall_plan::dependencies_not_satisfied;
return deinstall_plan::should_deinstall;
}
static void deinstall_package(const vcpkg_paths& paths, const package_spec& spec, StatusParagraphs& status_db)
{
auto package_it = status_db.find(spec.name(), spec.target_triplet());
if (package_it == status_db.end())
{
System::println(System::color::success, "Package %s is not installed", spec);
return;
}
auto& pkg = **package_it;
std::vector<const StatusParagraph*> deps;
auto plan = deinstall_package_plan(package_it, status_db, deps);
switch (plan)
{
case deinstall_plan::not_installed:
System::println(System::color::success, "Package %s is not installed", spec);
return;
case deinstall_plan::dependencies_not_satisfied:
System::println(System::color::error, "Error: Cannot remove package %s:", spec);
for (auto&& dep : deps)
{
System::println(" %s depends on %s", dep->package.displayname(), pkg.package.displayname());
}
exit(EXIT_FAILURE);
case deinstall_plan::should_deinstall:
break;
default:
Checks::unreachable();
}
pkg.want = want_t::purge;
pkg.state = install_state_t::half_installed;
write_update(paths, pkg);
std::fstream listfile(paths.listfile_path(pkg.package), std::ios_base::in | std::ios_base::binary);
if (listfile)
{
std::vector<fs::path> dirs_touched;
std::string suffix;
while (std::getline(listfile, suffix))
{
if (!suffix.empty() && suffix.back() == '\r')
suffix.pop_back();
std::error_code ec;
auto target = paths.installed / suffix;
auto status = fs::status(target, ec);
if (ec)
{
System::println(System::color::error, "failed: %s", ec.message());
continue;
}
if (fs::is_directory(status))
{
dirs_touched.push_back(target);
}
else if (fs::is_regular_file(status))
{
fs::remove(target, ec);
if (ec)
{
System::println(System::color::error, "failed: %s: %s", target.u8string(), ec.message());
}
}
else if (!fs::status_known(status))
{
System::println(System::color::warning, "Warning: unknown status: %s", target.u8string());
}
else
{
System::println(System::color::warning, "Warning: %s: cannot handle file type", target.u8string());
}
}
auto b = dirs_touched.rbegin();
auto e = dirs_touched.rend();
for (; b != e; ++b)
{
if (fs::directory_iterator(*b) == fs::directory_iterator())
{
std::error_code ec;
fs::remove(*b, ec);
if (ec)
{
System::println(System::color::error, "failed: %s", ec.message());
}
}
}
listfile.close();
fs::remove(paths.listfile_path(pkg.package));
}
pkg.state = install_state_t::not_installed;
write_update(paths, pkg);
System::println(System::color::success, "Package %s was successfully removed", pkg.package.displayname());
}
void remove_command(const vcpkg_cmd_arguments& args, const vcpkg_paths& paths, const triplet& default_target_triplet)
{
static const std::string example = create_example_string("remove zlib zlib:x64-windows curl boost");

View File

@ -108,7 +108,7 @@ static std::string get_fullpkgname_from_listfile(const fs::path& path)
return ret;
}
static void write_update(const vcpkg_paths& paths, const StatusParagraph& p)
void vcpkg::write_update(const vcpkg_paths& paths, const StatusParagraph& p)
{
static int update_id = 0;
auto my_update_id = update_id++;
@ -120,282 +120,6 @@ static void write_update(const vcpkg_paths& paths, const StatusParagraph& p)
fs::rename(tmp_update_filename, update_filename);
}
static void install_and_write_listfile(const vcpkg_paths& paths, const BinaryParagraph& bpgh)
{
std::fstream listfile(paths.listfile_path(bpgh), std::ios_base::out | std::ios_base::binary | std::ios_base::trunc);
auto package_prefix_path = paths.package_dir(bpgh.spec);
auto prefix_length = package_prefix_path.native().size();
const triplet& target_triplet = bpgh.spec.target_triplet();
const std::string& target_triplet_as_string = target_triplet.canonical_name();
std::error_code ec;
fs::create_directory(paths.installed / target_triplet_as_string, ec);
listfile << target_triplet << "\n";
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;
}
auto suffix = it->path().generic_u8string().substr(prefix_length + 1);
auto target = paths.installed / target_triplet_as_string / suffix;
auto status = it->status(ec);
if (ec)
{
System::println(System::color::error, "failed: %s: %s", it->path().u8string(), ec.message());
continue;
}
if (fs::is_directory(status))
{
fs::create_directory(target, ec);
if (ec)
{
System::println(System::color::error, "failed: %s: %s", target.u8string(), ec.message());
}
listfile << target_triplet << "/" << suffix << "\n";
}
else if (fs::is_regular_file(status))
{
fs::copy_file(*it, target, ec);
if (ec)
{
System::println(System::color::error, "failed: %s: %s", target.u8string(), ec.message());
}
listfile << target_triplet << "/" << suffix << "\n";
}
else if (!fs::status_known(status))
{
System::println(System::color::error, "failed: %s: unknown status", it->path().u8string());
}
else
System::println(System::color::error, "failed: %s: cannot handle file type", it->path().u8string());
}
listfile.close();
}
static std::map<std::string, fs::path> remove_first_n_chars_and_map(const std::vector<fs::path> absolute_paths, const size_t n)
{
std::map<std::string, fs::path> output;
for (const fs::path& absolute_path : absolute_paths)
{
std::string suffix = absolute_path.generic_string();
suffix.erase(0, n);
output.emplace(suffix, absolute_path);
}
return output;
}
static void print_map_values(const std::vector<std::string> keys, const std::map<std::string, fs::path>& map)
{
System::println("");
for (const std::string& key : keys)
{
System::println(" %s", map.at(key).generic_string());
}
System::println("");
}
void vcpkg::install_package(const vcpkg_paths& paths, const BinaryParagraph& binary_paragraph, StatusParagraphs& status_db)
{
const fs::path package_dir = paths.package_dir(binary_paragraph.spec);
const std::vector<fs::path> package_files = Files::recursive_find_all_files_in_dir(package_dir);
const fs::path installed_dir = paths.installed / binary_paragraph.spec.target_triplet().canonical_name();
const std::vector<fs::path> installed_files = Files::recursive_find_all_files_in_dir(installed_dir);
const std::map<std::string, fs::path> package_files_relative_paths_to_absolute_paths = remove_first_n_chars_and_map(package_files, package_dir.generic_string().size() + 1);
const std::map<std::string, fs::path> installed_files_relative_paths_to_absolute_paths = remove_first_n_chars_and_map(installed_files, installed_dir.generic_string().size() + 1);
const std::vector<std::string> package_files_set = Maps::extract_keys(package_files_relative_paths_to_absolute_paths);
const std::vector<std::string> installed_files_set = Maps::extract_keys(installed_files_relative_paths_to_absolute_paths);
std::vector<std::string> intersection;
std::set_intersection(package_files_set.cbegin(), package_files_set.cend(),
installed_files_set.cbegin(), installed_files_set.cend(),
std::back_inserter(intersection));
if (!intersection.empty())
{
System::println(System::color::error, "The following files are already installed and are in conflict with %s:", binary_paragraph.spec);
print_map_values(intersection, installed_files_relative_paths_to_absolute_paths);
exit(EXIT_FAILURE);
}
StatusParagraph spgh;
spgh.package = binary_paragraph;
spgh.want = want_t::install;
spgh.state = install_state_t::half_installed;
for (auto&& dep : spgh.package.depends)
{
if (status_db.find_installed(dep, spgh.package.spec.target_triplet()) == status_db.end())
{
Checks::unreachable();
}
}
write_update(paths, spgh);
status_db.insert(std::make_unique<StatusParagraph>(spgh));
install_and_write_listfile(paths, spgh.package);
spgh.state = install_state_t::installed;
write_update(paths, spgh);
status_db.insert(std::make_unique<StatusParagraph>(spgh));
}
enum class deinstall_plan
{
not_installed,
dependencies_not_satisfied,
should_deinstall
};
static deinstall_plan deinstall_package_plan(
const StatusParagraphs::iterator package_it,
const StatusParagraphs& status_db,
std::vector<const StatusParagraph*>& dependencies_out)
{
dependencies_out.clear();
if (package_it == status_db.end() || (*package_it)->state == install_state_t::not_installed)
{
return deinstall_plan::not_installed;
}
auto& pkg = (*package_it)->package;
for (auto&& inst_pkg : status_db)
{
if (inst_pkg->want != want_t::install)
continue;
if (inst_pkg->package.spec.target_triplet() != pkg.spec.target_triplet())
continue;
const auto& deps = inst_pkg->package.depends;
if (std::find(deps.begin(), deps.end(), pkg.spec.name()) != deps.end())
{
dependencies_out.push_back(inst_pkg.get());
}
}
if (!dependencies_out.empty())
return deinstall_plan::dependencies_not_satisfied;
return deinstall_plan::should_deinstall;
}
void vcpkg::deinstall_package(const vcpkg_paths& paths, const package_spec& spec, StatusParagraphs& status_db)
{
auto package_it = status_db.find(spec.name(), spec.target_triplet());
if (package_it == status_db.end())
{
System::println(System::color::success, "Package %s is not installed", spec);
return;
}
auto& pkg = **package_it;
std::vector<const StatusParagraph*> deps;
auto plan = deinstall_package_plan(package_it, status_db, deps);
switch (plan)
{
case deinstall_plan::not_installed:
System::println(System::color::success, "Package %s is not installed", spec);
return;
case deinstall_plan::dependencies_not_satisfied:
System::println(System::color::error, "Error: Cannot remove package %s:", spec);
for (auto&& dep : deps)
{
System::println(" %s depends on %s", dep->package.displayname(), pkg.package.displayname());
}
exit(EXIT_FAILURE);
case deinstall_plan::should_deinstall:
break;
default:
Checks::unreachable();
}
pkg.want = want_t::purge;
pkg.state = install_state_t::half_installed;
write_update(paths, pkg);
std::fstream listfile(paths.listfile_path(pkg.package), std::ios_base::in | std::ios_base::binary);
if (listfile)
{
std::vector<fs::path> dirs_touched;
std::string suffix;
while (std::getline(listfile, suffix))
{
if (!suffix.empty() && suffix.back() == '\r')
suffix.pop_back();
std::error_code ec;
auto target = paths.installed / suffix;
auto status = fs::status(target, ec);
if (ec)
{
System::println(System::color::error, "failed: %s", ec.message());
continue;
}
if (fs::is_directory(status))
{
dirs_touched.push_back(target);
}
else if (fs::is_regular_file(status))
{
fs::remove(target, ec);
if (ec)
{
System::println(System::color::error, "failed: %s: %s", target.u8string(), ec.message());
}
}
else if (!fs::status_known(status))
{
System::println(System::color::warning, "Warning: unknown status: %s", target.u8string());
}
else
{
System::println(System::color::warning, "Warning: %s: cannot handle file type", target.u8string());
}
}
auto b = dirs_touched.rbegin();
auto e = dirs_touched.rend();
for (; b != e; ++b)
{
if (fs::directory_iterator(*b) == fs::directory_iterator())
{
std::error_code ec;
fs::remove(*b, ec);
if (ec)
{
System::println(System::color::error, "failed: %s", ec.message());
}
}
}
listfile.close();
fs::remove(paths.listfile_path(pkg.package));
}
pkg.state = install_state_t::not_installed;
write_update(paths, pkg);
System::println(System::color::success, "Package %s was successfully removed", pkg.package.displayname());
}
expected<SourceParagraph> vcpkg::try_load_port(const fs::path& path)
{
try