#include "pch.h" #include "vcpkglib.h" #include "vcpkg_Files.h" #include "Paragraphs.h" #include "metrics.h" #include "vcpkg_Strings.h" namespace vcpkg { static StatusParagraphs load_current_database(const fs::path& vcpkg_dir_status_file, const fs::path& vcpkg_dir_status_file_old) { if (!fs::exists(vcpkg_dir_status_file)) { if (!fs::exists(vcpkg_dir_status_file_old)) { // no status file, use empty db return StatusParagraphs(); } fs::rename(vcpkg_dir_status_file_old, vcpkg_dir_status_file); } auto text = Files::read_contents(vcpkg_dir_status_file).get_or_throw(); auto pghs = Paragraphs::parse_paragraphs(text); std::vector> status_pghs; for (auto&& p : pghs) { status_pghs.push_back(std::make_unique(p)); } return StatusParagraphs(std::move(status_pghs)); } StatusParagraphs database_load_check(const vcpkg_paths& paths) { auto updates_dir = paths.vcpkg_dir_updates; std::error_code ec; fs::create_directory(paths.installed, ec); fs::create_directory(paths.vcpkg_dir, ec); fs::create_directory(paths.vcpkg_dir_info, ec); fs::create_directory(updates_dir, ec); const fs::path& status_file = paths.vcpkg_dir_status_file; const fs::path status_file_old = status_file.parent_path() / "status-old"; const fs::path status_file_new = status_file.parent_path() / "status-new"; StatusParagraphs current_status_db = load_current_database(status_file, status_file_old); auto b = fs::directory_iterator(updates_dir); auto e = fs::directory_iterator(); if (b == e) { // updates directory is empty, control file is up-to-date. return current_status_db; } for (; b != e; ++b) { if (!fs::is_regular_file(b->status())) continue; if (b->path().filename() == "incomplete") continue; auto text = Files::read_contents(b->path()).get_or_throw(); auto pghs = Paragraphs::parse_paragraphs(text); for (auto&& p : pghs) { current_status_db.insert(std::make_unique(p)); } } std::fstream(status_file_new, std::ios_base::out | std::ios_base::binary | std::ios_base::trunc) << current_status_db; if (fs::exists(status_file_old)) fs::remove(status_file_old); if (fs::exists(status_file)) fs::rename(status_file, status_file_old); fs::rename(status_file_new, status_file); fs::remove(status_file_old); b = fs::directory_iterator(updates_dir); for (; b != e; ++b) { if (!fs::is_regular_file(b->status())) continue; fs::remove(b->path()); } return current_status_db; } void write_update(const vcpkg_paths& paths, const StatusParagraph& p) { static int update_id = 0; auto my_update_id = update_id++; auto tmp_update_filename = paths.vcpkg_dir_updates / "incomplete"; auto update_filename = paths.vcpkg_dir_updates / std::to_string(my_update_id); std::fstream fs(tmp_update_filename, std::ios_base::out | std::ios_base::binary | std::ios_base::trunc); fs << p; fs.close(); fs::rename(tmp_update_filename, update_filename); } static void upgrade_to_slash_terminated_sorted_format(std::vector* lines, const fs::path& listfile_path) { static bool was_tracked = false; if (lines->empty()) { return; } if (lines->at(0).back() == '/') { return; // File already in the new format } if (!was_tracked) { was_tracked = true; TrackProperty("listfile", "update to new format"); } // The files are sorted such that directories are placed just before the files they contain // (They are not necessarily sorted alphabetically, e.g. libflac) // Therefore we can detect the entries that represent directories by comparing every element with the next one // and checking if the next has a slash immediately after the current one's length for (size_t i = 0; i < lines->size() - 1; i++) { std::string& current_string = lines->at(i); const std::string& next_string = lines->at(i + 1); const size_t potential_slash_char_index = current_string.length(); // Make sure the index exists first if (next_string.size() > potential_slash_char_index && next_string.at(potential_slash_char_index) == '/') { current_string += '/'; // Mark as a directory } } // After suffixing the directories with a slash, we can now sort. // We cannot sort before adding the suffixes because the following (actual example): /* x86-windows/include/FLAC <<<<<< This would be separated from its group due to sorting x86-windows/include/FLAC/all.h x86-windows/include/FLAC/assert.h x86-windows/include/FLAC/callback.h x86-windows/include/FLAC++ x86-windows/include/FLAC++/all.h x86-windows/include/FLAC++/decoder.h x86-windows/include/FLAC++/encoder.h * x86-windows/include/FLAC/ <<<<<< This will now be kept with its group when sorting x86-windows/include/FLAC/all.h x86-windows/include/FLAC/assert.h x86-windows/include/FLAC/callback.h x86-windows/include/FLAC++/ x86-windows/include/FLAC++/all.h x86-windows/include/FLAC++/decoder.h x86-windows/include/FLAC++/encoder.h */ // Note that after sorting, the FLAC++/ group will be placed before the FLAC/ group // The new format is lexicographically sorted std::sort(lines->begin(), lines->end()); // Replace the listfile on disk const fs::path updated_listfile_path = listfile_path.generic_string() + "_updated"; Files::write_all_lines(updated_listfile_path, *lines); fs::rename(updated_listfile_path, listfile_path); } std::vector get_installed_files(const vcpkg_paths& paths, const StatusParagraphs& status_db) { std::vector installed_files; for (const std::unique_ptr& pgh : status_db) { if (pgh->state != install_state_t::installed) { continue; } const fs::path listfile_path = paths.listfile_path(pgh->package); std::vector installed_files_of_current_pgh = Files::read_all_lines(listfile_path).get_or_throw(); Strings::trim_all_and_remove_whitespace_strings(&installed_files_of_current_pgh); upgrade_to_slash_terminated_sorted_format(&installed_files_of_current_pgh, listfile_path); // Remove the directories installed_files_of_current_pgh.erase( std::remove_if(installed_files_of_current_pgh.begin(), installed_files_of_current_pgh.end(), [](const std::string& file) -> bool { return file.back() == '/'; } ), installed_files_of_current_pgh.end()); StatusParagraph_and_associated_files pgh_and_files = { *pgh, ImmutableSortedVector::create(std::move(installed_files_of_current_pgh)) }; installed_files.push_back(std::move(pgh_and_files)); } return installed_files; } CMakeVariable::CMakeVariable(const std::wstring& varname, const wchar_t* varvalue) : s(Strings::wformat(LR"("-D%s=%s")", varname, varvalue)) { } CMakeVariable::CMakeVariable(const std::wstring& varname, const std::string& varvalue) : CMakeVariable(varname, Strings::utf8_to_utf16(varvalue).c_str()) { } CMakeVariable::CMakeVariable(const std::wstring& varname, const std::wstring& varvalue) : CMakeVariable(varname, varvalue.c_str()) {} CMakeVariable::CMakeVariable(const std::wstring& varname, const fs::path& path) : CMakeVariable(varname, path.generic_wstring()) {} std::wstring make_cmake_cmd(const fs::path& cmake_exe, const fs::path& cmake_script, const std::vector& pass_variables) { std::wstring cmd_cmake_pass_variables = Strings::wjoin(L" ", pass_variables, [](auto&& v) { return v.s; }); return Strings::wformat(LR"("%s" %s -P "%s")", cmake_exe.native(), cmd_cmake_pass_variables, cmake_script.generic_wstring()); } }