auberge 0.15.12

use std::collections::{BTreeMap, BTreeSet};
use std::fs;
use std::path::PathBuf;

fn ansible_dir() -> PathBuf {
    PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("ansible")
}

/// Every key of a handler except `name`, sorted and YAML-serialized, so two
/// definitions compare equal exactly when they declare the same thing —
/// whatever order the keys were written in.
fn definition(handler: &serde_yaml::Mapping) -> String {
    handler
        .iter()
        .filter(|(key, _)| *key != &serde_yaml::Value::from("name"))
        .map(|(key, value)| {
            let render =
                |v: &serde_yaml::Value| serde_yaml::to_string(v).unwrap().trim().to_string();
            (render(key), render(value))
        })
        .collect::<BTreeMap<_, _>>()
        .iter()
        .map(|(key, value)| format!("{key}: {value}"))
        .collect::<Vec<_>>()
        .join("\n")
}

/// A role's handlers, as name -> definition.
fn role_handlers(role: &str) -> BTreeMap<String, String> {
    let path = ansible_dir()
        .join("roles")
        .join(role)
        .join("handlers/main.yml");
    if !path.exists() {
        return BTreeMap::new();
    }
    let handlers: Option<Vec<serde_yaml::Mapping>> =
        serde_yaml::from_str(&fs::read_to_string(&path).unwrap())
            .unwrap_or_else(|e| panic!("{}: {e}", path.display()));

    handlers
        .unwrap_or_default()
        .iter()
        .map(|handler| {
            let name = handler
                .get(serde_yaml::Value::from("name"))
                .and_then(|name| name.as_str())
                .unwrap_or_else(|| panic!("{}: handler without a name", path.display()))
                .to_string();
            (name, definition(handler))
        })
        .collect()
}

/// Roles a role pulls in with `include_role` — their handlers join the play too.
fn included_roles(role: &str) -> BTreeSet<String> {
    let tasks = ansible_dir().join("roles").join(role).join("tasks");
    let mut included = BTreeSet::new();

    for entry in fs::read_dir(&tasks)
        .into_iter()
        .flatten()
        .filter_map(Result::ok)
    {
        let Ok(content) = fs::read_to_string(entry.path()) else {
            continue;
        };
        let lines: Vec<&str> = content.lines().collect();
        for (i, line) in lines.iter().enumerate() {
            if !line.contains("include_role:") {
                continue;
            }
            let name = lines[i + 1..]
                .iter()
                .find_map(|l| l.trim().strip_prefix("name:"))
                .unwrap_or_else(|| panic!("{role}: include_role without a name"));
            included.insert(name.trim().trim_matches('"').to_string());
        }
    }

    included
}

/// Every play in the repo, as (label, the roles it loads). `include_role` is
/// followed transitively: an included role's handlers land in the same play.
fn plays() -> Vec<(String, BTreeSet<String>)> {
    let mut plays = Vec::new();

    for entry in fs::read_dir(ansible_dir().join("playbooks"))
        .expect("ansible/playbooks must exist")
        .filter_map(Result::ok)
    {
        let path = entry.path();
        let name = path.file_name().unwrap().to_string_lossy().to_string();
        if !name.ends_with(".yml") || name.ends_with(".meta.yml") {
            continue;
        }
        let parsed: Vec<serde_yaml::Value> =
            serde_yaml::from_str(&fs::read_to_string(&path).unwrap())
                .unwrap_or_else(|e| panic!("{}: {e}", path.display()));

        for play in parsed {
            let listed = play.get("roles").and_then(|r| r.as_sequence());
            let mut pending: Vec<String> = listed
                .into_iter()
                .flatten()
                .map(|role| {
                    role.as_str()
                        .or_else(|| role.get("role").and_then(|r| r.as_str()))
                        .unwrap_or_else(|| panic!("{}: unreadable role entry", path.display()))
                        .to_string()
                })
                .collect();

            let mut roles = BTreeSet::new();
            while let Some(role) = pending.pop() {
                if roles.insert(role.clone()) {
                    pending.extend(included_roles(&role));
                }
            }

            let label = play
                .get("name")
                .and_then(|n| n.as_str())
                .unwrap_or("unnamed play");
            plays.push((format!("{name} ({label})"), roles));
        }
    }

    plays
}

/// Ansible handler names live in one global namespace per play: when two roles
/// loaded by the same play define the same name, the last one loaded wins and
/// shadows the rest. Sharing a name is only safe when both definitions declare
/// the same thing — a shadow with a divergent `when:` guard skips silently, so
/// the notify becomes a no-op rather than an error (issue #569: baikal's pool
/// restart resolved to yourls's handler, whose guard is false on a
/// baikal-scoped run, and /run/php/baikal-fpm.sock never appeared).
#[test]
fn test_handlers_shared_within_a_play_declare_the_same_thing() {
    let mut violations = Vec::new();

    for (play, roles) in plays() {
        let mut by_name: BTreeMap<String, BTreeMap<String, String>> = BTreeMap::new();
        for role in &roles {
            for (name, definition) in role_handlers(role) {
                by_name
                    .entry(name)
                    .or_default()
                    .insert(role.clone(), definition);
            }
        }

        for (name, by_role) in by_name {
            if by_role.values().collect::<BTreeSet<_>>().len() < 2 {
                continue;
            }
            let definitions: String = by_role
                .iter()
                .map(|(role, definition)| {
                    format!("\n    {role}: {}", definition.replace('\n', "; "))
                })
                .collect();
            violations.push(format!(
                "{play}: handler `{name}` declares something different per role, so the \
                 last role loaded shadows the others — give each one a role-scoped \
                 name:{definitions}"
            ));
        }
    }

    assert!(violations.is_empty(), "{}", violations.join("\n"));
}