use std::collections::HashMap;
use std::process::Stdio;
use async_trait::async_trait;
use eyre::bail;
use super::{InstallOpts, PackageRequest, PackageState, PackageStatus, SystemPackageManager};
use crate::result::Result;
use crate::system::sudo;
pub struct DnfManager {}
impl DnfManager {
pub fn new() -> Self {
Self {}
}
}
fn parse_rpm_query(output: &str, requests: &[PackageRequest]) -> Vec<PackageStatus> {
let mut installed: HashMap<&str, &str> = HashMap::new();
for line in output.lines() {
if let Some((name, version)) = line.split_once('\t') {
installed.insert(name, version);
}
}
requests
.iter()
.map(|req| {
let state = match installed.get(req.name.as_str()) {
Some(version) => match &req.version {
Some(requested)
if *version != requested
&& !version.starts_with(&format!("{requested}-")) =>
{
PackageState::VersionMismatch {
installed: version.to_string(),
}
}
_ => PackageState::Installed {
version: version.to_string(),
},
},
None => PackageState::Missing,
};
PackageStatus {
request: req.clone(),
state,
}
})
.collect()
}
#[async_trait]
impl SystemPackageManager for DnfManager {
fn name(&self) -> &'static str {
"dnf"
}
fn is_available(&self) -> bool {
cfg!(target_os = "linux") && crate::file::which("dnf").is_some()
}
fn unavailable_reason(&self) -> String {
if cfg!(target_os = "linux") {
"dnf not found".to_string()
} else {
"only available on linux".to_string()
}
}
async fn installed(&self, pkgs: &[PackageRequest]) -> Result<Vec<PackageStatus>> {
if pkgs.is_empty() {
return Ok(vec![]);
}
let mut args = vec![
"-q".to_string(),
"--qf".to_string(),
"%{NAME}\\t%{VERSION}-%{RELEASE}\\n".to_string(),
];
args.extend(pkgs.iter().map(|p| p.name.clone()));
debug!("$ rpm {}", args.join(" "));
let output = tokio::process::Command::new("rpm")
.args(&args)
.stdin(Stdio::null())
.stdout(Stdio::piped())
.stderr(Stdio::piped())
.output()
.await?;
let stderr = String::from_utf8_lossy(&output.stderr);
if !output.status.success()
&& !stderr.is_empty()
&& !stderr.lines().all(|l| {
l.trim().is_empty() || l.contains("is not installed") || l.contains("no packages")
})
{
bail!("rpm -q failed: {}", stderr.trim());
}
let stdout = String::from_utf8_lossy(&output.stdout);
Ok(parse_rpm_query(&stdout, pkgs))
}
async fn install(&self, pkgs: &[PackageRequest], opts: &InstallOpts) -> Result<()> {
let mut args = vec!["install".to_string(), "-y".to_string()];
if opts.update {
args.push("--refresh".to_string());
}
args.push("--".to_string());
args.extend(pkgs.iter().map(|p| match &p.version {
Some(v) => format!("{}-{v}", p.name),
None => p.name.clone(),
}));
if opts.dry_run {
miseprintln!("{}", sudo::argv("dnf", &args).join(" "));
return Ok(());
}
sudo::run("dnf", &args, &[])
}
async fn upgrade(&self, pkgs: &[PackageRequest], opts: &InstallOpts) -> Result<()> {
let mut args = vec![
"upgrade".to_string(),
"-y".to_string(),
"--refresh".to_string(),
"--".to_string(),
];
args.extend(pkgs.iter().map(|p| match &p.version {
Some(v) => format!("{}-{v}", p.name),
None => p.name.clone(),
}));
if opts.dry_run {
miseprintln!("{}", sudo::argv("dnf", &args).join(" "));
return Ok(());
}
sudo::run("dnf", &args, &[])
}
}
#[cfg(test)]
mod tests {
use super::*;
fn req(name: &str, version: Option<&str>) -> PackageRequest {
PackageRequest {
name: name.to_string(),
version: version.map(str::to_string),
}
}
#[test]
fn test_parse_rpm_query() {
let requests = vec![
req("bc", None),
req("nonexistent", None),
req("bash", Some("5.2.26-3.fc40")),
req("glib2-devel", None),
req("zsh", Some("5.8-1.fc40")),
req("tmux", Some("3.4")),
];
let output = "bc\t1.07.1-14.fc39\npackage nonexistent is not installed\nbash\t5.2.26-3.fc40\nglib2\t2.80.0-1.fc40\nzsh\t5.9-2.fc40\ntmux\t3.4-3.fc40\n";
let statuses = parse_rpm_query(output, &requests);
assert_eq!(
statuses[0].state,
PackageState::Installed {
version: "1.07.1-14.fc39".to_string()
}
);
assert_eq!(statuses[1].state, PackageState::Missing);
assert_eq!(
statuses[2].state,
PackageState::Installed {
version: "5.2.26-3.fc40".to_string()
}
);
assert_eq!(statuses[3].state, PackageState::Missing);
assert_eq!(
statuses[4].state,
PackageState::VersionMismatch {
installed: "5.9-2.fc40".to_string()
}
);
assert_eq!(
statuses[5].state,
PackageState::Installed {
version: "3.4-3.fc40".to_string()
}
);
}
}