mirror of
https://github.com/CachyOS/New-Cli-Installer.git
synced 2025-01-23 22:42:31 +08:00
74 lines
2.7 KiB
C++
74 lines
2.7 KiB
C++
|
#include "chwd_profiles.hpp"
|
||
|
|
#include "utils.hpp"
|
||
|
|
|
||
|
|
#include <algorithm> // for any_of, sort, for_each
|
||
|
|
|
||
|
|
#include <fmt/compile.h>
|
||
|
|
#include <fmt/format.h>
|
||
|
|
|
||
|
|
#define TOML_EXCEPTIONS 0 // disable exceptions
|
||
|
|
#include <toml++/toml.h>
|
||
|
|
|
||
|
|
#if defined(__clang__)
|
||
|
|
#pragma clang diagnostic push
|
||
|
|
#pragma clang diagnostic ignored "-Wold-style-cast"
|
||
|
|
#elif defined(__GNUC__)
|
||
|
|
#pragma GCC diagnostic push
|
||
|
|
#pragma GCC diagnostic ignored "-Wuseless-cast"
|
||
|
|
#pragma GCC diagnostic ignored "-Wold-style-cast"
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include <range/v3/algorithm/any_of.hpp>
|
||
|
|
#include <range/v3/algorithm/for_each.hpp>
|
||
|
|
#include <range/v3/view/filter.hpp>
|
||
|
|
|
||
|
|
#if defined(__clang__)
|
||
|
|
#pragma clang diagnostic pop
|
||
|
|
#elif defined(__GNUC__)
|
||
|
|
#pragma GCC diagnostic pop
|
||
|
|
#endif
|
||
|
|
|
||
|
|
namespace detail::chwd {
|
||
|
|
|
||
|
|
auto get_all_profile_names(std::string_view profile_type) noexcept -> std::optional<std::vector<std::string>> {
|
||
|
|
const auto& file_path = fmt::format(FMT_COMPILE("/var/lib/chwd/db/pci/{}/profiles.toml"), profile_type);
|
||
|
|
toml::parse_result config = toml::parse_file(file_path);
|
||
|
|
/* clang-format off */
|
||
|
|
if (config.failed()) { return {}; }
|
||
|
|
/* clang-format on */
|
||
|
|
|
||
|
|
std::vector<std::string> profile_names{};
|
||
|
|
for (auto&& [key, value] : config.table()) {
|
||
|
|
if (value.is_table()) {
|
||
|
|
for (auto&& [nested_key, nested_value] : *value.as_table()) {
|
||
|
|
/* clang-format off */
|
||
|
|
if (!nested_value.is_table()) { continue; }
|
||
|
|
/* clang-format on */
|
||
|
|
|
||
|
|
const auto& nested_profile_name = fmt::format(FMT_COMPILE("{}.{}"), std::string_view{key}, std::string_view{nested_key});
|
||
|
|
profile_names.push_back(nested_profile_name);
|
||
|
|
}
|
||
|
|
profile_names.push_back(std::string(key));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return profile_names;
|
||
|
|
}
|
||
|
|
|
||
|
|
auto get_available_profile_names(std::string_view profile_type) noexcept -> std::optional<std::vector<std::string>> {
|
||
|
|
auto&& all_profile_names = get_all_profile_names(profile_type);
|
||
|
|
/* clang-format off */
|
||
|
|
if (!all_profile_names.has_value()) { return {}; }
|
||
|
|
/* clang-format on */
|
||
|
|
|
||
|
|
const auto& available_profile_names = utils::make_multiline(utils::exec("chwd --list -d | grep Name | awk '{print $4}'"));
|
||
|
|
|
||
|
|
std::vector<std::string> filtered_profile_names{};
|
||
|
|
const auto& functor = [available_profile_names](auto&& profile_name) { return ranges::any_of(available_profile_names, [profile_name](auto&& available_profile) { return available_profile == profile_name; }); };
|
||
|
|
|
||
|
|
ranges::for_each(all_profile_names.value() | ranges::views::filter(functor), [&](auto&& rng) { filtered_profile_names.push_back(std::forward<decltype(rng)>(rng)); });
|
||
|
|
return filtered_profile_names;
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace detail::chwd
|