[pbs-devel] [PATCH v3 proxmox-apt 03/10] add functions to check for Proxmox repositories

[Fabian Ebner <f.ebner@proxmox.com>]

Signed-off-by: Fabian Ebner <f.ebner@proxmox.com>
---

No changes from v2.

 src/repositories/check.rs | 42 +++++++++++++++++++++++++++++++++++++++
 src/repositories/mod.rs   | 21 ++++++++++++++++++++
 tests/repositories.rs     | 29 ++++++++++++++++++++++++++-
 3 files changed, 91 insertions(+), 1 deletion(-)

diff --git a/src/repositories/check.rs b/src/repositories/check.rs
index 87fbbac..d0656cd 100644
--- a/src/repositories/check.rs
+++ b/src/repositories/check.rs
@@ -44,4 +44,46 @@ impl APTRepository {
 
         Ok(())
     }
+
+    /// Checks if the repository is the no-subscription repository of the specified
+    /// Proxmox product.
+    pub fn is_no_subscription(&self, product: &str) -> bool {
+        let base_uri = "http://download.proxmox.com/debian";
+        let no_subscription_uri = format!("{}/{}", base_uri, product);
+        let no_subscription_component = format!("{}-no-subscription", product);
+
+        if self
+            .uris
+            .iter()
+            .any(|uri| uri.trim_end_matches('/') == no_subscription_uri)
+        {
+            return self
+                .components
+                .iter()
+                .any(|comp| *comp == no_subscription_component);
+        } else {
+            false
+        }
+    }
+
+    /// Checks if the repository is the enterprise repository of the specified
+    /// Proxmox product.
+    pub fn is_enterprise(&self, product: &str) -> bool {
+        let base_uri = "https://enterprise.proxmox.com/debian";
+        let enterprise_uri = format!("{}/{}", base_uri, product);
+        let enterprise_component = format!("{}-enterprise", product);
+
+        if self
+            .uris
+            .iter()
+            .any(|uri| uri.trim_end_matches('/') == enterprise_uri)
+        {
+            return self
+                .components
+                .iter()
+                .any(|comp| *comp == enterprise_component);
+        } else {
+            false
+        }
+    }
 }
diff --git a/src/repositories/mod.rs b/src/repositories/mod.rs
index 5c446af..bd0af1d 100644
--- a/src/repositories/mod.rs
+++ b/src/repositories/mod.rs
@@ -107,6 +107,27 @@ fn check_filename<P: AsRef<Path>>(
     Ok(Some((file_type, path_string)))
 }
 
+/// Checks if the enterprise repository for the specified Proxmox product is
+/// configured and enabled.
+pub fn enterprise_repository_enabled<A: AsRef<APTRepository>>(repos: &[A], product: &str) -> bool {
+    repos.iter().any(|repo| {
+        let repo = repo.as_ref();
+        repo.enabled && repo.is_enterprise(product)
+    })
+}
+
+/// Checks if the no-subscription repository for the specified Proxmox product
+/// is configured and enabled.
+pub fn no_subscription_repository_enabled<A: AsRef<APTRepository>>(
+    repos: &[A],
+    product: &str,
+) -> bool {
+    repos.iter().any(|repo| {
+        let repo = repo.as_ref();
+        repo.enabled && repo.is_no_subscription(product)
+    })
+}
+
 /// Similar to [`repositories_from_files`], but for a single file, and adds the
 /// parsed repositories onto the provided vector instead. Another difference is
 /// that it doesn't call [`basic_check`](check::basic_check).
diff --git a/tests/repositories.rs b/tests/repositories.rs
index 020e133..deb2f6a 100644
--- a/tests/repositories.rs
+++ b/tests/repositories.rs
@@ -3,7 +3,10 @@ use std::path::PathBuf;
 
 use anyhow::{bail, format_err, Error};
 
-use proxmox_apt::repositories::{repositories_from_files, write_repositories};
+use proxmox_apt::repositories::{
+    enterprise_repository_enabled, no_subscription_repository_enabled, repositories_from_files,
+    write_repositories,
+};
 use proxmox_apt::types::APTRepository;
 
 #[test]
@@ -71,3 +74,27 @@ fn test_parse_write() -> Result<(), Error> {
 
     Ok(())
 }
+
+#[test]
+fn test_proxmox_repositories() -> Result<(), Error> {
+    let test_dir = std::env::current_dir()?.join("tests");
+    let read_dir = test_dir.join("sources.list.d");
+
+    let pve_list = read_dir.join("pve.list");
+    let repos = repositories_from_files(&vec![pve_list])?;
+
+    assert_eq!(false, enterprise_repository_enabled(&repos, "pbs"));
+    assert_eq!(false, enterprise_repository_enabled(&repos, "pve"));
+    assert_eq!(false, no_subscription_repository_enabled(&repos, "pmg"));
+    assert_eq!(true, no_subscription_repository_enabled(&repos, "pve"));
+
+    let pbs_list = read_dir.join("pbs-enterprise.list");
+    let repos = repositories_from_files(&vec![pbs_list])?;
+
+    assert_eq!(true, enterprise_repository_enabled(&repos, "pbs"));
+    assert_eq!(false, enterprise_repository_enabled(&repos, "pve"));
+    assert_eq!(false, no_subscription_repository_enabled(&repos, "pmg"));
+    assert_eq!(false, no_subscription_repository_enabled(&repos, "pve"));
+
+    Ok(())
+}
-- 
2.20.1