Merge remote-tracking branch 'origin/dnf'

1 files changed, 62 insertions, 0 deletions

diff --git a/src/packages/dnf.rs b/src/packages/dnf.rs
new file mode 100644
index 0000000..6f451ee
--- /dev/null
+++ b/src/packages/dnf.rs
@@ -0,0 +1,62 @@
+use std::process::{Command, Stdio};
+
+use super::{Package, PackageList, PackageManager};
+
+use crate::error::{Error, Result};
+
+pub struct Dnf;
+
+impl PackageManager for Dnf {
+    fn get_installed(&self) -> Result<PackageList> {
+        let list = Command::new("dnf").args(["list", "installed"]).output().unwrap();
+        let explicit_list = Command::new("dnf").args(["repoquery", "--userinstalled"]).output().unwrap();
+        
+        let list_str = String::from_utf8(list.stdout).unwrap();
+        let ex_list_str = String::from_utf8(explicit_list.stdout).unwrap();
+
+
+        let mut pkgs: Vec<Package> = Vec::new();
+        let list_lines: Vec<&str> = list_str.split('\n').collect();
+        // Pop first info line
+        let list_lines = &list_lines[1..list_lines.len()];
+        for pkg in list_lines {
+            if pkg.is_empty() {
+                continue;
+            };
+            let split: Vec<&str> = pkg.split_whitespace().collect();
+            if split.len() != 3 {
+                return Err(Error::UnknownOutput);
+            };
+
+            let explicit = ex_list_str.contains(pkg);
+
+            let Some(pkg_id) = split[0].split_once('.') else {
+                return Err(Error::UnknownOutput);
+            };
+            pkgs.push(Package {
+                id: pkg_id.0.to_string(),
+                version: split[1].to_string(),
+                explicit,
+            });
+        }
+
+        Ok(PackageList {
+            packages: pkgs,
+            manager: super::Manager::Dnf,
+        })
+    }
+
+    fn install(&self, pkgs: Vec<super::Package>) -> Result<()> { 
+        let mut args = vec!["dnf".to_string(), "install".to_string(), "--assumeyes".to_string()];
+
+        for pkg in pkgs {
+            args.push(pkg.id);
+        }
+        Command::new("sudo")
+            .stdout(Stdio::inherit())
+            .args(args)
+            .spawn()?
+            .wait_with_output()?;
+        Ok(())
+    }
+}