Save RPMRepositoryFile only once for multiple enable calls

Author: dhilst

src/services/files.cpp

@@ -65,13 +65,6 @@ KeyFile::KeyFile(const std::filesystem::path& path)
     m_impl->loadFromFile(path);
 }
 
-// KeyFile::KeyFile(const std::string& input)
-//     : m_impl(std::make_unique<KeyFile::Impl>(Glib::KeyFile::create(),
-//     std::filesystem::path()))
-// {
-//     m_impl->loadFromData(input);
-// }
-
 KeyFile::~KeyFile() = default;
 
 namespace {

src/services/repos.cpp

@@ -5,7 +5,6 @@
 
 #include <algorithm>
 #include <filesystem>
-#include <fstream>
 #include <memory>
 #include <ranges>
 #include <sstream>
@@ -33,6 +32,7 @@ using cloyster::services::repos::IRepository;
 
 namespace cloyster::services::repos {
 
+// Represents a debian repository file 
 class DebianRepository : public IRepository {
 private:
     std::string m_type; // "deb" or "deb-src"
@@ -61,6 +61,7 @@ class DebianRepository : public IRepository {
     void source(const std::filesystem::path value) override;
 };
 
+// Represents a RPM Repository inside a repository file (/etc/yum.repo.d/*.repo)
 class RPMRepository final : public IRepository {
     std::string m_id;
     bool m_enabled = true;
@@ -126,16 +127,14 @@ class RPMRepository final : public IRepository {
     {
         return other.id() == id();
     }
-
-
 };
 
 namespace {
 
 // Easy conversions for string
 std::ostream& operator<<(std::ostream& ostr, const RPMRepository& repo)
 {
-    ostr << "RPMRepository( ";
+    ostr << "RPMRepository(";
     ostr << repo.id() << " ";
     ostr << repo.name() << " ";
     ostr << repo.baseurl().value_or("") << " ";
@@ -155,7 +154,7 @@ std::string toString(const T& input)
 
 };
 
-// Parses .repo files
+// Parses RPM .repo files
 class RPMRepositoryParser final {
 public:
     static void parse(
@@ -213,8 +212,6 @@ class RPMRepositoryParser final {
         }
 
         file.save();
-
-        LOG_DEBUG("UNPARSE FILE {}:\n{}", path.string(), file.toData());
     }
 };
 static_assert(IsParser<RPMRepositoryParser, std::filesystem::path,
@@ -232,6 +229,11 @@ class RPMRepositoryFile final {
         m_parser.parse(m_path, m_repos);
     }
 
+    const auto& path()
+    {
+        return m_path;
+    }
+
     auto& repos()
     {
         return m_repos;
@@ -325,11 +327,27 @@ class RPMRepoManager final {
         repofile->save();
     }
 
+    void enable(const auto& repo, auto& repofile, bool value)
+    {
+        LOG_DEBUG("Enabling/Disabling[{}] RPM repo {}",
+                  value, repo);
+        repofile->repo(repo)->enabled(value);
+    }
+
     // Enable/disable multiple repositories by name
     void enable(const std::vector<std::string>& repos, bool value)
     {
+        auto byIdPtr = [](const std::shared_ptr<RPMRepositoryFile>& rptr) {
+            return std::hash<std::string>{}(rptr->path());
+        };
+        std::unordered_set<std::shared_ptr<RPMRepositoryFile>, decltype(byIdPtr)> toSave;
         for (const auto& repo : repos) {
-            enable(repo, value);
+            auto& rfile = m_filesIdx.at(repo);
+            toSave.emplace(rfile);
+            enable(repo, rfile, value);
+        }
+        for (const auto& repoFil : toSave) {
+            repoFil->save();
         }
     }
 
@@ -531,7 +549,29 @@ void RepoManager::enable(const std::string& repoid)
 
 void RepoManager::enable(const std::vector<std::string>& repos)
 {
-    std::ranges::for_each(repos, [&](const auto& repoid) { enable(repoid); });
+    if (cloyster::dryRun) {
+        LOG_WARN("Dry Run: Would enable these repos: {}",
+                 fmt::join(repos, ","));
+        return;
+    }
+    try {
+        switch (m_os.getPackageType()) {
+            case OS::PackageType::RPM:
+                m_impl->rpm.enable(repos, true);
+                break;
+            default:
+                throw std::logic_error("Not implemented");
+
+        }
+    } catch (const std::out_of_range&) {
+        LOG_ERROR("Trying to enable unknown repository {}, "
+                  "failed because the repository was not found.",
+            fmt::join(repos, ","));
+
+        for (const auto& [id, _] : m_repos) {
+            LOG_ERROR("Repository available: {}", id);
+        }
+    }
 }
 
 void RepoManager::disable(const std::string& repoid)
@@ -563,8 +603,29 @@ void RepoManager::disable(const std::string& repoid)
 
 void RepoManager::disable(const std::vector<std::string>& repos)
 {
-    std::ranges::for_each(
-        repos, [&](const auto& repoid) { disable(repoid); });
+    if (cloyster::dryRun) {
+        LOG_WARN("Dry Run: Would enable repository {}", fmt::join(repos, ","));
+        return;
+    }
+
+    try {
+        switch (m_os.getPackageType()) {
+            case OS::PackageType::RPM:
+                m_impl->rpm.enable(repos, false);
+                break;
+            default:
+                throw std::logic_error("Not implemented");
+
+        }
+    } catch (const std::out_of_range&) {
+        LOG_ERROR("Trying to disable unknown repository {}, "
+                  "failed because the repository was not found.",
+            fmt::join(repos, ","));
+
+        for (const auto& [id, _] : m_repos) {
+            LOG_ERROR("Repository available: {}", id);
+        }
+    }
 }
 
 void RepoManager::install(const std::filesystem::path& path)