Add RPMRepoManager sketch implementation

Author: dhilst

include/cloysterhpc/services/repos.h

@@ -4,8 +4,8 @@
  */
 
 /* This class is set to be reimplemented */
-#ifndef CLOYSTERHPC_REPOS_H_DEPRECATED_
-#define CLOYSTERHPC_REPOS_H_DEPRECATED_
+#ifndef CLOYSTERHPC_REPOS_H_
+#define CLOYSTERHPC_REPOS_H_
 
 #include <filesystem>
 #include <optional>
@@ -48,9 +48,9 @@ class IRepository {
     virtual void uri(std::optional<std::string> value) = 0;
     virtual void source(std::filesystem::path value) = 0;
 
-    IRepository(const IRepository&) = delete;
+    IRepository(const IRepository&) = default;
     IRepository(IRepository&&) = default;
-    IRepository& operator=(const IRepository&) = delete;
+    IRepository& operator=(const IRepository&) = default;
     IRepository& operator=(IRepository&&) = default;
 
 protected:
@@ -78,7 +78,6 @@ class RepoManager {
 
 private:
     void loadFiles(const std::filesystem::path& basedir);
-    void loadCustom(inifile& file, const std::filesystem::path& path);
     void loadRPMRepos(const std::filesystem::path& source);
     auto loadDefaults();
 
@@ -101,4 +100,4 @@ class RepoManager {
 };
 
 };
-#endif // CLOYSTERHPC_REPOS_H_DEPRECATED_
+#endif // CLOYSTERHPC_REPOS_H_

src/services/repos.cpp

@@ -10,6 +10,7 @@
 #include <ranges>
 #include <sstream>
 #include <stdexcept>
+#include <unordered_map>
 #include <utility>
 
 #include <boost/algorithm/string.hpp>
@@ -71,9 +72,9 @@ class RPMRepository final : public IRepository {
 public:
     RPMRepository() = default;
     ~RPMRepository() override = default;
-    RPMRepository(const RPMRepository&) = delete;
+    RPMRepository(const RPMRepository&) = default;
     RPMRepository(RPMRepository&&) = default;
-    RPMRepository& operator=(const RPMRepository&) = delete;
+    RPMRepository& operator=(const RPMRepository&) = default;
     RPMRepository& operator=(RPMRepository&&) = default;
 
     [[nodiscard]] std::string id() const override { return m_id; };
@@ -130,19 +131,88 @@ class RPMRepositoryParser final {
 static_assert(IsParser<RPMRepositoryParser, std::filesystem::path,
     std::vector<RPMRepository>>);
 
+
 class RPMRepositoryFile final {
+    static constexpr auto m_parser = RPMRepositoryParser();
+    std::filesystem::path m_path;
+    std::unordered_map<std::string, RPMRepository> m_repos;
+public:
+    RPMRepositoryFile(auto path, auto repos)
+    : m_path(std::move(path))
+    , m_repos(std::move(repos))
+    {
+    }
+
+    const auto& repos() const
+    {
+        return m_repos;
+    }
+
+    auto& repo(const std::string& name) const
+    {
+        return repos().at(name);
+    }
+
+    auto reposList() const
+    {
+        return repos()
+            | std::views::transform([](const auto& pair)
+                                    { return pair.second; })
+            | std::ranges::to<std::vector>();
+    }
+
+    void save() const
+    {
+        m_parser.unparse(reposList(), m_path);
+    }
+};
+
+class RPMRepoManager final {
+    static constexpr auto m_parser = RPMRepositoryParser();
+    static constexpr auto m_basedir = "/etc/yum.repos.d/";
+    std::unordered_map<std::string, std::shared_ptr<RPMRepositoryFile>> m_filesIdx;
+
+    void addRepoFile(auto&& repofile)
+    {
+        for (auto& [repo, _] : repofile.repos()) {
+            m_filesIdx.insert({repo, std::make_shared<RPMRepositoryFile>(repofile)});
+        }
+    }
+
+    static auto vecToMap(std::vector<RPMRepository>&& repos)
+    {
+        std::unordered_map<std::string, RPMRepository> output;
+        for (auto&& repo : std::move(repos))
+        {
+            output.insert({repo.id(), std::move(repo)});
+        }
+        return output;
+    }
+
 public:
-    ~RPMRepositoryFile() = default;
-    RPMRepositoryFile() = default;
-    RPMRepositoryFile(const RPMRepositoryFile&) = delete;
-    RPMRepositoryFile& operator=(const RPMRepositoryFile&) = delete;
-    RPMRepositoryFile(RPMRepositoryFile&&) = delete;
-    RPMRepositoryFile& operator=(RPMRepositoryFile&&) = delete;
-    void save();
-    void load();
+
+    void install(const std::filesystem::path& source)
+    {
+        const auto& dest =  m_basedir / source.filename();
+        if (m_filesIdx.contains(dest)) {
+            LOG_WARN("Repository already installed {}, skipping", dest.string());
+            return;
+        }
+        cloyster::copyFile(source, dest);
+        std::vector<RPMRepository> repos;
+        m_parser.parse(dest, repos);
+        auto&& reposMap = vecToMap(std::move(repos));
+        auto&& repofile = RPMRepositoryFile(dest, reposMap);
+        addRepoFile(repofile);
+    }
+
+    void enable(const auto& repo, bool value)
+    {
+        auto& repofile = m_filesIdx.at(repo);
+        repofile.repo(repo).enable(value);
+        repofile.save();
+    }
 };
-static_assert(cloyster::concepts::IsSaveable<RPMRepositoryFile>);
-static_assert(cloyster::concepts::NotCopiableNotMoveable<RPMRepositoryFile>);
 
 }; // namespace cloyster::services::repos {
 
@@ -164,7 +234,6 @@ constexpr std::string_view CLOYSTER_REPO_EL10 = {
 #include "cloysterhpc/repos/el10/cloyster.repo"
 };
 
-// @TODO Make this a RPMRepositoryFile method
 void installFile(const std::filesystem::path& path, std::istream& data)
 {
     if (cloyster::dryRun) {
@@ -181,7 +250,7 @@ void installFile(const std::filesystem::path& path, std::istream& data)
     fil << data.rdbuf();
 }
 
-constexpr auto getDefaultPath(const auto& osinfo)
+constexpr auto getCloysterRepoPath(const auto& osinfo)
 {
     switch (osinfo.getPlatform()) {
         case OS::Platform::el8:
@@ -276,7 +345,7 @@ namespace cloyster::services::repos {
 
 auto RepoManager::loadDefaults()
 {
-    const auto& path = getDefaultPath(m_os);
+    const auto& path = getCloysterRepoPath(m_os);
     std::istringstream stream;
     switch (m_os.getPlatform()) {
         case OS::Platform::el8:
@@ -517,7 +586,6 @@ void RepoManager::install(const std::vector<std::filesystem::path>& paths)
     }
 }
 
-/// Private API
 
 std::vector<std::shared_ptr<const IRepository>> RepoManager::listRepos() const
 {