zoi-cli 1.25.2

//! The audit command checks for vulnerabilities in installed or available
//! packages.
use anyhow::Result;
use colored::Colorize;
use comfy_table::presets::UTF8_FULL;
use comfy_table::{Attribute, Cell, ContentArrangement, Table};
use semver::{Version, VersionReq};

use crate::pkg::{config, db, local, types};

/// Executes the audit command.
///
/// # Errors
///
/// Returns an error if the configuration cannot be read, if the database cannot
/// be accessed, or if local package information cannot be retrieved.
///
/// # Panics
///
/// This function does not explicitly panic, but underlying library calls might.
pub fn run(
    all: bool,
    registry_filter: Option<String>,
    repo_filter: Option<&str>
) -> Result<()> {
    if all {
        println!(
            "{} Listing all known vulnerabilities...",
            "::".bold().blue()
        );
    } else {
        println!(
            "{} Auditing installed packages for vulnerabilities...",
            "::".bold().blue()
        );
    }

    let config = config::read_config()?;
    let mut registries = Vec::new();
    if let Some(reg) = registry_filter {
        registries.push(reg);
    } else {
        if let Some(default) = &config.default_registry {
            registries.push(default.handle.clone());
        }
        for reg in &config.added_registries {
            registries.push(reg.handle.clone());
        }
    }

    let mut all_advisories = Vec::new();
    for handle in registries {
        if let Ok(advisories) = db::list_all_advisories(&handle) {
            for (adv, repo) in advisories {
                all_advisories.push((adv, repo, handle.clone()));
            }
        }
    }

    if let Some(rf) = repo_filter {
        all_advisories.retain(|(_, repo, _)| {
            if rf.contains('/') {
                repo == rf
            } else {
                repo.split('/').any(|part| part == rf)
            }
        });
    }

    if all_advisories.is_empty() {
        println!(
            "\n{}",
            "No vulnerabilities found matching your criteria.".green()
        );
        return Ok(());
    }

    if all {
        print_advisories_table(&all_advisories);
    } else {
        let installed = local::get_installed_packages()?;
        let mut vulnerable_installed = Vec::new();

        for manifest in installed {
            for (adv, repo, reg) in &all_advisories {
                let package_match = adv.package == manifest.name
                    && *repo == manifest.repo
                    && *reg == manifest.registry_handle;

                let sub_package_match =
                    match (&adv.sub_package, &manifest.sub_package) {
                        (Some(adv_sub), Some(man_sub)) => adv_sub == man_sub,
                        (None, _) => true,
                        (Some(_), None) => false
                    };

                if package_match
                    && sub_package_match
                    && let Ok(version) = Version::parse(&manifest.version)
                    && let Ok(req) = VersionReq::parse(&adv.affected_range)
                    && req.matches(&version)
                {
                    vulnerable_installed.push((adv.clone(), manifest.clone()));
                }
            }
        }

        if vulnerable_installed.is_empty() {
            println!(
                "\n{}",
                "No vulnerabilities found in installed packages.".green()
            );
        } else {
            let count = vulnerable_installed.len();
            println!(
                "\n{} Found {count} vulnerabilities in installed packages:",
                "Warning".red().bold(),
            );
            print_vulnerable_table(&vulnerable_installed);
        }
    }

    Ok(())
}

/// Prints a table of all matching advisories.
fn print_advisories_table(advisories: &[(types::Advisory, String, String)]) {
    let mut table = Table::new();
    table
        .load_style(UTF8_FULL)
        .set_content_arrangement(ContentArrangement::Dynamic)
        .set_header(vec![
            Cell::new("ID").add_attribute(Attribute::Bold),
            Cell::new("Package").add_attribute(Attribute::Bold),
            Cell::new("Severity").add_attribute(Attribute::Bold),
            Cell::new("Affected").add_attribute(Attribute::Bold),
            Cell::new("Fixed In").add_attribute(Attribute::Bold),
            Cell::new("Summary").add_attribute(Attribute::Bold),
        ]);

    for (adv, _, _) in advisories {
        let severity_cell = match adv.severity {
            types::Severity::Low => {
                Cell::new("Low").fg(comfy_table::Color::Blue)
            }
            types::Severity::Medium => {
                Cell::new("Medium").fg(comfy_table::Color::Yellow)
            }
            types::Severity::High => {
                Cell::new("High").fg(comfy_table::Color::Red)
            }
            types::Severity::Critical => Cell::new("Critical")
                .fg(comfy_table::Color::Magenta)
                .add_attribute(Attribute::Bold)
        };

        let package_display = adv.sub_package.as_ref().map_or_else(
            || adv.package.clone(),
            |sub| {
                let pkg = &adv.package;
                format!("{pkg}:{sub}")
            }
        );

        table.add_row(vec![
            Cell::new(&adv.id).fg(comfy_table::Color::Cyan),
            Cell::new(package_display),
            severity_cell,
            Cell::new(&adv.affected_range),
            Cell::new(adv.fixed_in.as_deref().unwrap_or("N/A"))
                .fg(comfy_table::Color::Green),
            Cell::new(&adv.summary),
        ]);
    }

    println!("{table}");
}

/// Prints a table of vulnerabilities found in installed packages.
fn print_vulnerable_table(
    vulnerable: &[(types::Advisory, types::InstallManifest)]
) {
    let mut table = Table::new();
    table
        .load_style(UTF8_FULL)
        .set_content_arrangement(ContentArrangement::Dynamic)
        .set_header(vec![
            Cell::new("Package").add_attribute(Attribute::Bold),
            Cell::new("Installed").add_attribute(Attribute::Bold),
            Cell::new("ID").add_attribute(Attribute::Bold),
            Cell::new("Severity").add_attribute(Attribute::Bold),
            Cell::new("Fixed In").add_attribute(Attribute::Bold),
            Cell::new("Summary").add_attribute(Attribute::Bold),
        ]);

    for (adv, manifest) in vulnerable {
        let severity_cell = match adv.severity {
            types::Severity::Low => {
                Cell::new("Low").fg(comfy_table::Color::Blue)
            }
            types::Severity::Medium => {
                Cell::new("Medium").fg(comfy_table::Color::Yellow)
            }
            types::Severity::High => {
                Cell::new("High").fg(comfy_table::Color::Red)
            }
            types::Severity::Critical => Cell::new("Critical")
                .fg(comfy_table::Color::Magenta)
                .add_attribute(Attribute::Bold)
        };

        let package_display = manifest.sub_package.as_ref().map_or_else(
            || manifest.name.clone(),
            |sub| {
                let name = &manifest.name;
                format!("{name}:{sub}")
            }
        );

        table.add_row(vec![
            Cell::new(package_display).fg(comfy_table::Color::Cyan),
            Cell::new(&manifest.version).fg(comfy_table::Color::Red),
            Cell::new(&adv.id).fg(comfy_table::Color::DarkGrey),
            severity_cell,
            Cell::new(adv.fixed_in.as_deref().unwrap_or("N/A"))
                .fg(comfy_table::Color::Green),
            Cell::new(&adv.summary),
        ]);
    }

    println!("{table}");
}