cargo-cooldown 0.3.1

//! Allow-rule parsing and merging for package-specific cooldown exceptions.

use std::collections::HashMap;

use serde::Deserialize;

/// Root allow-rule object embedded in `cooldown.toml`.
#[derive(Debug, Default, Clone, Deserialize, PartialEq, Eq)]
#[serde(deny_unknown_fields)]
pub struct AllowRules {
    #[serde(default)]
    pub allow: AllowSection,
}

/// TOML section that contains global, package, and exact-version rules.
#[derive(Debug, Default, Clone, Deserialize, PartialEq, Eq)]
#[serde(deny_unknown_fields)]
pub struct AllowSection {
    #[serde(default)]
    pub exact: Vec<AllowExact>,
    #[serde(default)]
    pub package: Vec<AllowPackage>,
    pub global: Option<AllowGlobal>,
}

/// Allows one exact crate version to bypass cooldown.
#[derive(Debug, Clone, Deserialize, PartialEq, Eq)]
#[serde(deny_unknown_fields)]
pub struct AllowExact {
    #[serde(rename = "crate")]
    pub crate_name: String,
    pub version: String,
}

/// Overrides the cooldown window for one crate name.
#[derive(Debug, Clone, Deserialize, PartialEq, Eq)]
#[serde(deny_unknown_fields)]
pub struct AllowPackage {
    #[serde(rename = "crate")]
    pub crate_name: String,
    #[serde(default)]
    pub minutes: Option<u64>,
    #[serde(rename = "min-publish-age")]
    pub min_publish_age: Option<String>,
    #[serde(skip)]
    pub(crate) min_publish_age_seconds: Option<u64>,
}

/// Provides a lower default cooldown window for all registry crates.
#[derive(Debug, Clone, Deserialize, PartialEq, Eq)]
#[serde(deny_unknown_fields)]
pub struct AllowGlobal {
    #[serde(default)]
    pub minutes: Option<u64>,
}

impl AllowRules {
    #[cfg(test)]
    pub fn merged(base: &Self, overlay: &Self) -> Self {
        let mut merged = base.clone();
        merged.merge_from(overlay);
        merged
    }

    pub fn merge_from(&mut self, overlay: &Self) {
        self.allow.merge_from(&overlay.allow);
    }

    pub fn is_exact_allowed(&self, name: &str, version: &str) -> bool {
        self.allow
            .exact
            .iter()
            .any(|entry| entry.crate_name == name && entry.version == version)
    }

    pub fn per_crate_min_publish_age_seconds(&self) -> HashMap<String, u64> {
        self.allow
            .package
            .iter()
            .filter_map(|pkg| {
                pkg.effective_min_publish_age_seconds()
                    .map(|seconds| (pkg.crate_name.clone(), seconds))
            })
            .collect()
    }

    pub fn global_minutes(&self) -> Option<u64> {
        self.allow.global.as_ref().and_then(|global| global.minutes)
    }

    #[cfg(test)]
    pub fn effective_minutes_for(&self, name: &str, default_minutes: u64) -> u64 {
        let mut effective = default_minutes;
        if let Some(global) = self.global_minutes() {
            effective = effective.min(global);
        }
        if let Some(rule) = self.allow.package.iter().find(|pkg| pkg.crate_name == name)
            && let Some(minutes) = rule.minutes
        {
            effective = effective.min(minutes);
        }
        effective
    }

    #[cfg(test)]
    pub fn effective_min_publish_age_seconds_for(&self, name: &str, default_seconds: u64) -> u64 {
        let mut effective = default_seconds;
        if let Some(global) = self.global_minutes() {
            effective = effective.min(minutes_to_seconds_saturating(global));
        }
        if let Some(rule) = self.allow.package.iter().find(|pkg| pkg.crate_name == name)
            && let Some(seconds) = rule.effective_min_publish_age_seconds()
        {
            effective = effective.min(seconds);
        }
        effective
    }
}

impl AllowPackage {
    fn effective_min_publish_age_seconds(&self) -> Option<u64> {
        self.min_publish_age_seconds
            .or_else(|| self.minutes.map(minutes_to_seconds_saturating))
    }
}

fn minutes_to_seconds_saturating(minutes: u64) -> u64 {
    minutes.saturating_mul(60)
}

impl AllowSection {
    pub fn merge_from(&mut self, overlay: &Self) {
        if let Some(global) = &overlay.global {
            self.global = Some(global.clone());
        }

        for package in &overlay.package {
            if let Some(existing) = self
                .package
                .iter_mut()
                .find(|existing| existing.crate_name == package.crate_name)
            {
                *existing = package.clone();
            } else {
                self.package.push(package.clone());
            }
        }

        for exact in &overlay.exact {
            if self.exact.iter().any(|existing| {
                existing.crate_name == exact.crate_name && existing.version == exact.version
            }) {
                continue;
            }
            self.exact.push(exact.clone());
        }
    }
}

/// Unit tests for allow-rule merging and lookup semantics.
#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn merged_allow_rules_deduplicates_exact_and_overrides_package_minutes() {
        let base = AllowRules {
            allow: AllowSection {
                exact: vec![AllowExact {
                    crate_name: "foo".to_string(),
                    version: "1.2.3".to_string(),
                }],
                package: vec![AllowPackage {
                    crate_name: "bar".to_string(),
                    minutes: Some(10),
                    min_publish_age: None,
                    min_publish_age_seconds: None,
                }],
                global: Some(AllowGlobal { minutes: Some(60) }),
            },
        };
        let overlay = AllowRules {
            allow: AllowSection {
                exact: vec![
                    AllowExact {
                        crate_name: "foo".to_string(),
                        version: "1.2.3".to_string(),
                    },
                    AllowExact {
                        crate_name: "baz".to_string(),
                        version: "9.9.9".to_string(),
                    },
                ],
                package: vec![AllowPackage {
                    crate_name: "bar".to_string(),
                    minutes: Some(5),
                    min_publish_age: None,
                    min_publish_age_seconds: None,
                }],
                global: Some(AllowGlobal { minutes: Some(30) }),
            },
        };

        let merged = AllowRules::merged(&base, &overlay);

        assert!(merged.is_exact_allowed("foo", "1.2.3"));
        assert!(merged.is_exact_allowed("baz", "9.9.9"));
        assert_eq!(merged.allow.exact.len(), 2);
        assert_eq!(merged.effective_minutes_for("bar", 90), 5);
        assert_eq!(merged.global_minutes(), Some(30));
    }

    #[test]
    fn package_rule_can_disable_cooldown_for_one_crate() {
        let allow_rules = AllowRules {
            allow: AllowSection {
                exact: Vec::new(),
                package: vec![AllowPackage {
                    crate_name: "tokio".to_string(),
                    minutes: Some(0),
                    min_publish_age: None,
                    min_publish_age_seconds: None,
                }],
                global: Some(AllowGlobal {
                    minutes: Some(1440),
                }),
            },
        };

        assert_eq!(allow_rules.effective_minutes_for("tokio", 1440), 0);
        assert_eq!(allow_rules.effective_minutes_for("serde", 1440), 1440);
    }

    #[test]
    fn package_rule_can_disable_cooldown_with_min_publish_age() {
        let allow_rules = AllowRules {
            allow: AllowSection {
                exact: Vec::new(),
                package: vec![AllowPackage {
                    crate_name: "tokio".to_string(),
                    minutes: None,
                    min_publish_age: Some("0".to_string()),
                    min_publish_age_seconds: Some(0),
                }],
                global: Some(AllowGlobal {
                    minutes: Some(1440),
                }),
            },
        };

        assert_eq!(
            allow_rules.effective_min_publish_age_seconds_for("tokio", 86_400),
            0
        );
        assert_eq!(
            allow_rules.effective_min_publish_age_seconds_for("serde", 86_400),
            86_400
        );
    }
}