fallow_engine/

guard.rs

//! Typed guard report assembly for pre-edit architecture guidance.

use std::fmt;
use std::path::{Component, Path};

use fallow_config::{ResolvedBoundaryConfig, ResolvedConfig, RulePackRule, RulePackRuleKind};
use fallow_types::guard::{
    GuardBoundary, GuardFileReport, GuardPolicyRule, GuardReport, GuardSeverities, GuardZone,
};
use rustc_hash::FxHashSet;

/// Error returned when a guard target cannot be represented safely.
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum GuardError {
    /// The requested target is outside the resolved project root.
    OutsideRoot(String),
}

impl fmt::Display for GuardError {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::OutsideRoot(path) => write!(f, "guard target is outside project root: {path}"),
        }
    }
}

impl std::error::Error for GuardError {}

/// Build a typed guard report for one or more target files.
///
/// Paths may be project-relative or absolute under `config.root`. Returned
/// paths are project-root-relative and use forward slashes.
///
/// # Errors
///
/// Returns [`GuardError::OutsideRoot`] for absolute paths outside the project
/// root or relative paths containing parent-directory traversal.
pub fn build_guard_report(
    config: &ResolvedConfig,
    files: &[String],
) -> Result<GuardReport, GuardError> {
    let scopes = compile_rule_scopes(config);
    let mut reports = Vec::with_capacity(files.len());
    for file in files {
        reports.push(build_file_report(config, &scopes, file)?);
    }
    Ok(GuardReport { files: reports })
}

fn build_file_report(
    config: &ResolvedConfig,
    scopes: &[RuleScope<'_>],
    input: &str,
) -> Result<GuardFileReport, GuardError> {
    let rel_path = normalize_target_path(config, input)?;
    let full_path = config.root.join(&rel_path);
    let rules = config.resolve_rules_for_path(&full_path);
    let zone_name = config.boundaries.classify_zone(&rel_path);
    let zone = zone_name.and_then(|name| guard_zone(&config.boundaries, name));
    let boundary = guard_boundary(&config.boundaries, &rel_path, zone_name);
    let notes = guard_notes(config, zone_name, boundary.coverage_required);

    Ok(GuardFileReport {
        exists: full_path.exists(),
        boundary,
        policy_rules: guard_policy_rules(scopes, &rel_path, zone_name, rules.policy_violation),
        severities: GuardSeverities {
            boundary_violation: rules.boundary_violation.to_string(),
            policy_violation: rules.policy_violation.to_string(),
        },
        path: rel_path,
        zone,
        notes,
    })
}

fn normalize_target_path(config: &ResolvedConfig, input: &str) -> Result<String, GuardError> {
    let normalized = input.replace('\\', "/");
    let path = Path::new(&normalized);
    if looks_windows_absolute(&normalized) && !path.is_absolute() {
        return Err(GuardError::OutsideRoot(input.to_string()));
    }
    let relative = if path.is_absolute() {
        path.strip_prefix(&config.root)
            .map_err(|_| GuardError::OutsideRoot(input.to_string()))?
    } else {
        path
    };
    normalize_relative_path(relative, input)
}

fn looks_windows_absolute(path: &str) -> bool {
    let bytes = path.as_bytes();
    bytes.len() >= 3 && bytes[1] == b':' && bytes[2] == b'/'
}

fn normalize_relative_path(path: &Path, original: &str) -> Result<String, GuardError> {
    let mut parts = Vec::new();
    for component in path.components() {
        match component {
            Component::CurDir => {}
            Component::Normal(part) => parts.push(part.to_string_lossy().replace('\\', "/")),
            Component::ParentDir | Component::RootDir | Component::Prefix(_) => {
                return Err(GuardError::OutsideRoot(original.to_string()));
            }
        }
    }
    Ok(parts.join("/"))
}

fn guard_zone(boundaries: &ResolvedBoundaryConfig, name: &str) -> Option<GuardZone> {
    boundaries
        .zones
        .iter()
        .find(|zone| zone.name == name)
        .map(|zone| GuardZone {
            name: zone.name.clone(),
            patterns: zone.patterns.clone(),
        })
}

fn guard_boundary(
    boundaries: &ResolvedBoundaryConfig,
    rel_path: &str,
    zone_name: Option<&str>,
) -> GuardBoundary {
    let configured = boundaries_configured(boundaries);
    let coverage_required = zone_name.is_none()
        && boundaries.coverage.require_all_files
        && !boundaries.allows_unmatched(rel_path);

    let Some(zone_name) = zone_name else {
        return GuardBoundary {
            configured,
            unrestricted: true,
            allowed_zones: Vec::new(),
            allowed_type_only_zones: Vec::new(),
            forbidden_calls: Vec::new(),
            coverage_required,
        };
    };

    let forbidden_calls = boundaries
        .calls_forbidden_by_zone
        .get(zone_name)
        .cloned()
        .unwrap_or_default();
    let Some(rule) = boundaries
        .rules
        .iter()
        .find(|rule| rule.from_zone == zone_name)
    else {
        return GuardBoundary {
            configured,
            unrestricted: true,
            allowed_zones: Vec::new(),
            allowed_type_only_zones: Vec::new(),
            forbidden_calls,
            coverage_required,
        };
    };

    let mut allowed_zones = vec![zone_name.to_string()];
    allowed_zones.extend(rule.allowed_zones.iter().cloned());
    allowed_zones.sort();
    allowed_zones.dedup();

    GuardBoundary {
        configured,
        unrestricted: false,
        allowed_zones,
        allowed_type_only_zones: rule.allow_type_only_zones.clone(),
        forbidden_calls,
        coverage_required,
    }
}

fn guard_notes(
    config: &ResolvedConfig,
    zone_name: Option<&str>,
    coverage_required: bool,
) -> Vec<String> {
    let mut notes = Vec::new();
    if boundaries_configured(&config.boundaries) && zone_name.is_none() {
        notes.push(
            "Files outside every zone are unrestricted for import and call checks.".to_string(),
        );
        if coverage_required {
            notes.push(
                "boundaries.coverage.requireAllFiles is enabled: reachable files with no zone are reported as boundary-coverage violations."
                    .to_string(),
            );
        }
    }
    if !boundaries_configured(&config.boundaries) && config.rule_packs.is_empty() {
        notes.push("No boundary zones or rule packs are configured.".to_string());
    }
    if zone_name.is_some() {
        notes.push("Same-zone imports are always allowed.".to_string());
    }
    notes
}

fn boundaries_configured(boundaries: &ResolvedBoundaryConfig) -> bool {
    !boundaries.zones.is_empty() || !boundaries.logical_groups.is_empty()
}

fn guard_policy_rules(
    scopes: &[RuleScope<'_>],
    rel_path: &str,
    zone: Option<&str>,
    master_severity: fallow_config::Severity,
) -> Vec<GuardPolicyRule> {
    if master_severity == fallow_config::Severity::Off {
        return Vec::new();
    }

    scopes
        .iter()
        .filter(|scope| {
            compiled_scope_applies(&scope.files, &scope.exclude, &scope.zones, rel_path, zone)
        })
        .filter_map(|scope| guard_policy_rule(scope.pack, scope.rule, master_severity))
        .collect()
}

/// One rule-pack rule with its file and zone scope compiled once per report.
struct RuleScope<'a> {
    pack: &'a str,
    rule: &'a RulePackRule,
    files: Vec<globset::GlobMatcher>,
    exclude: Vec<globset::GlobMatcher>,
    zones: FxHashSet<String>,
}

fn compile_rule_scopes(config: &ResolvedConfig) -> Vec<RuleScope<'_>> {
    config
        .rule_packs
        .iter()
        .flat_map(|pack| {
            pack.rules.iter().map(move |rule| RuleScope {
                pack: pack.name.as_str(),
                rule,
                files: compile_scope_globs(&rule.files),
                exclude: compile_scope_globs(&rule.exclude),
                zones: rule.zones.iter().cloned().collect(),
            })
        })
        .collect()
}

fn compile_scope_globs(patterns: &[String]) -> Vec<globset::GlobMatcher> {
    patterns
        .iter()
        .filter_map(|pattern| globset::Glob::new(pattern).ok())
        .map(|glob| glob.compile_matcher())
        .collect()
}

fn compiled_scope_applies(
    files: &[globset::GlobMatcher],
    exclude: &[globset::GlobMatcher],
    zones: &FxHashSet<String>,
    relative: &str,
    zone: Option<&str>,
) -> bool {
    (files.is_empty() || files.iter().any(|matcher| matcher.is_match(relative)))
        && !exclude.iter().any(|matcher| matcher.is_match(relative))
        && (zones.is_empty() || zone.is_some_and(|zone| zones.contains(zone)))
}

fn guard_policy_rule(
    pack: &str,
    rule: &RulePackRule,
    master_severity: fallow_config::Severity,
) -> Option<GuardPolicyRule> {
    let severity = rule.severity.unwrap_or(master_severity);
    if severity == fallow_config::Severity::Off {
        return None;
    }

    Some(GuardPolicyRule {
        pack: pack.to_string(),
        rule_id: rule.id.clone(),
        kind: rule_kind(rule.kind).to_string(),
        patterns: rule_patterns(rule),
        allowed_files: rule.allowed_files.clone(),
        proof_kinds: rule.proof_kinds.clone(),
        message: rule.message.clone(),
        severity: severity.to_string(),
        suppress_token: format!("policy-violation:{pack}/{}", rule.id),
    })
}

const fn rule_kind(kind: RulePackRuleKind) -> &'static str {
    match kind {
        RulePackRuleKind::BannedCall => "banned-call",
        RulePackRuleKind::BannedImport => "banned-import",
        RulePackRuleKind::BannedEffect => "banned-effect",
        RulePackRuleKind::BannedExport => "banned-export",
        RulePackRuleKind::GdpProofProducer => "gdp-proof-producer",
    }
}

fn rule_patterns(rule: &RulePackRule) -> Vec<String> {
    match rule.kind {
        RulePackRuleKind::BannedCall => rule.callees.clone(),
        RulePackRuleKind::BannedImport => rule.specifiers.clone(),
        RulePackRuleKind::BannedEffect => rule
            .effects
            .iter()
            .map(|effect| effect.as_str().to_string())
            .collect(),
        RulePackRuleKind::BannedExport => rule.exports.clone(),
        RulePackRuleKind::GdpProofProducer => vec!["@gdp-ts/core.defineProof".to_string()],
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use fallow_config::{
        BoundaryCallsConfig, BoundaryConfig, BoundaryCoverageConfig, BoundaryRule, BoundaryZone,
        EffectKind, FallowConfig, ForbiddenCallRule, ForbiddenCallee, OutputFormat, RulePackDef,
        RulePackRule, RulePackRuleKind, RulesConfig, Severity,
    };
    use std::fs;

    fn rule(id: &str, kind: RulePackRuleKind) -> RulePackRule {
        RulePackRule {
            id: id.to_string(),
            kind,
            callees: Vec::new(),
            specifiers: Vec::new(),
            effects: Vec::new(),
            exports: Vec::new(),
            allowed_files: Vec::new(),
            proof_kinds: Vec::new(),
            ignore_type_only: false,
            files: Vec::new(),
            exclude: Vec::new(),
            zones: Vec::new(),
            message: None,
            severity: None,
        }
    }

    fn pack(rules: Vec<RulePackRule>) -> RulePackDef {
        RulePackDef {
            schema: None,
            version: 1,
            name: "team-policy".to_string(),
            description: None,
            rules,
        }
    }

    fn resolve(root: &Path, configure: impl FnOnce(&mut FallowConfig)) -> ResolvedConfig {
        let mut config = FallowConfig {
            rules: RulesConfig {
                policy_violation: Severity::Warn,
                ..RulesConfig::default()
            },
            ..FallowConfig::default()
        };
        configure(&mut config);
        config.resolve(root.to_path_buf(), OutputFormat::Json, 1, true, true, None)
    }

    #[test]
    fn zoned_file_reports_allow_rule_and_forbidden_call() {
        let temp = tempfile::tempdir().expect("tempdir");
        fs::create_dir_all(temp.path().join("src/domain")).expect("create dir");
        fs::write(temp.path().join("src/domain/user.ts"), "").expect("write file");
        let config = resolve(temp.path(), |config| {
            config.boundaries = BoundaryConfig {
                zones: vec![
                    BoundaryZone {
                        name: "domain".to_string(),
                        patterns: vec!["src/domain/**".to_string()],
                        auto_discover: Vec::new(),
                        root: None,
                    },
                    BoundaryZone {
                        name: "shared".to_string(),
                        patterns: vec!["src/shared/**".to_string()],
                        auto_discover: Vec::new(),
                        root: None,
                    },
                ],
                rules: vec![BoundaryRule {
                    from: "domain".to_string(),
                    allow: vec!["shared".to_string()],
                    allow_type_only: vec!["ui".to_string()],
                }],
                calls: BoundaryCallsConfig {
                    forbidden: vec![ForbiddenCallRule {
                        from: "domain".to_string(),
                        callee: ForbiddenCallee::Single("child_process.*".to_string()),
                    }],
                },
                ..BoundaryConfig::default()
            };
        });

        let report =
            build_guard_report(&config, &["src/domain/user.ts".to_string()]).expect("report");
        let file = &report.files[0];

        assert!(file.exists);
        assert_eq!(
            file.zone.as_ref().map(|zone| zone.name.as_str()),
            Some("domain")
        );
        assert!(!file.boundary.unrestricted);
        assert_eq!(file.boundary.allowed_zones, vec!["domain", "shared"]);
        assert_eq!(file.boundary.allowed_type_only_zones, vec!["ui"]);
        assert_eq!(file.boundary.forbidden_calls, vec!["child_process.*"]);
        assert!(file.notes.iter().any(|note| note.contains("Same-zone")));
    }

    fn required_coverage_config(root: &Path) -> ResolvedConfig {
        resolve(root, |config| {
            config.boundaries = BoundaryConfig {
                zones: vec![BoundaryZone {
                    name: "domain".to_string(),
                    patterns: vec!["src/domain/**".to_string()],
                    auto_discover: Vec::new(),
                    root: None,
                }],
                coverage: BoundaryCoverageConfig {
                    require_all_files: true,
                    allow_unmatched: vec!["src/generated/**".to_string()],
                },
                ..BoundaryConfig::default()
            };
        })
    }

    #[test]
    fn unzoned_file_reports_required_coverage() {
        let temp = tempfile::tempdir().expect("tempdir");
        let config = required_coverage_config(temp.path());

        let report =
            build_guard_report(&config, &["src/ui/button.ts".to_string()]).expect("report");
        let file = &report.files[0];

        assert!(file.zone.is_none());
        assert!(file.boundary.unrestricted);
        assert!(file.boundary.coverage_required);
        assert!(
            file.notes
                .iter()
                .any(|note| note.contains("outside every zone"))
        );

        let allowed =
            build_guard_report(&config, &["src/generated/client.ts".to_string()]).expect("report");
        assert!(!allowed.files[0].boundary.coverage_required);
    }

    #[test]
    fn required_coverage_notes_state_the_requirement() {
        let temp = tempfile::tempdir().expect("tempdir");
        let config = required_coverage_config(temp.path());

        let report =
            build_guard_report(&config, &["src/ui/button.ts".to_string()]).expect("report");
        let notes = &report.files[0].notes;

        assert!(
            notes
                .iter()
                .any(|note| note.contains("requireAllFiles") && note.contains("boundary-coverage")),
            "unzoned file under required coverage must state the requirement: {notes:?}"
        );
        assert!(
            !notes
                .iter()
                .any(|note| note.contains("unrestricted for boundary checks")),
            "the unrestricted note must not claim to cover the coverage check: {notes:?}"
        );

        let allowed =
            build_guard_report(&config, &["src/generated/client.ts".to_string()]).expect("report");
        assert!(
            !allowed.files[0]
                .notes
                .iter()
                .any(|note| note.contains("requireAllFiles")),
            "allowUnmatched paths must not state a coverage requirement"
        );
    }

    #[test]
    fn pack_rule_scope_filters_policy_rules() {
        let temp = tempfile::tempdir().expect("tempdir");
        let mut domain_rule = rule("pure-domain", RulePackRuleKind::BannedEffect);
        domain_rule.effects = vec![EffectKind::Network];
        domain_rule.files = vec!["src/domain/**".to_string()];
        let mut excluded_rule = rule("no-generated-process", RulePackRuleKind::BannedCall);
        excluded_rule.callees = vec!["child_process.*".to_string()];
        excluded_rule.exclude = vec!["src/domain/**".to_string()];
        let mut config = resolve(temp.path(), |_| {});
        config.rule_packs = vec![pack(vec![domain_rule, excluded_rule])];

        let report =
            build_guard_report(&config, &["src/domain/user.ts".to_string()]).expect("report");
        let rules = &report.files[0].policy_rules;

        assert_eq!(rules.len(), 1);
        assert_eq!(rules[0].rule_id, "pure-domain");
        assert_eq!(rules[0].kind, "banned-effect");
        assert_eq!(rules[0].patterns, vec!["network"]);
        assert_eq!(
            rules[0].suppress_token,
            "policy-violation:team-policy/pure-domain"
        );
        assert_eq!(rules[0].severity, "warn");
    }

    #[test]
    fn compiled_rule_scopes_match_individual_file_reports() {
        let temp = tempfile::tempdir().expect("tempdir");
        let mut config = resolve(temp.path(), |config| {
            config.boundaries = BoundaryConfig {
                zones: vec![
                    BoundaryZone {
                        name: "domain".to_string(),
                        patterns: vec!["src/domain/**".to_string()],
                        auto_discover: Vec::new(),
                        root: None,
                    },
                    BoundaryZone {
                        name: "app".to_string(),
                        patterns: vec!["src/app/**".to_string()],
                        auto_discover: Vec::new(),
                        root: None,
                    },
                ],
                ..BoundaryConfig::default()
            };
        });
        let mut domain_rule = rule("domain-only", RulePackRuleKind::BannedImport);
        domain_rule.files = vec!["src/domain/**".to_string()];
        domain_rule.exclude = vec!["src/domain/generated/**".to_string()];
        let mut app_rule = rule("app-zone", RulePackRuleKind::BannedCall);
        app_rule.zones = vec!["app".to_string()];
        let mut invalid_glob_rule = rule("invalid-glob", RulePackRuleKind::BannedExport);
        invalid_glob_rule.files = vec!["[".to_string()];
        config.rule_packs = vec![pack(vec![domain_rule, app_rule, invalid_glob_rule])];

        let files = vec![
            "src/domain/user.ts".to_string(),
            "src/domain/generated/client.ts".to_string(),
            "src/app/page.ts".to_string(),
            "src/other.ts".to_string(),
        ];
        let batch = build_guard_report(&config, &files).expect("batch report");
        let individual = files
            .iter()
            .flat_map(|file| {
                build_guard_report(&config, std::slice::from_ref(file))
                    .expect("individual report")
                    .files
            })
            .collect::<Vec<_>>();

        assert_eq!(
            serde_json::to_value(&batch.files).expect("serialize batch reports"),
            serde_json::to_value(&individual).expect("serialize individual reports")
        );
        let rule_ids = batch
            .files
            .iter()
            .map(|file| {
                file.policy_rules
                    .iter()
                    .map(|rule| rule.rule_id.as_str())
                    .collect::<Vec<_>>()
            })
            .collect::<Vec<_>>();
        assert_eq!(
            rule_ids,
            vec![
                vec!["domain-only", "invalid-glob"],
                vec!["invalid-glob"],
                vec!["app-zone", "invalid-glob"],
                vec!["invalid-glob"],
            ]
        );
    }

    #[test]
    fn nonexistent_target_reports_exists_false() {
        let temp = tempfile::tempdir().expect("tempdir");
        let config = resolve(temp.path(), |_| {});

        let report = build_guard_report(&config, &["src/missing.ts".to_string()]).expect("report");

        assert_eq!(report.files[0].path, "src/missing.ts");
        assert!(!report.files[0].exists);
    }

    #[test]
    fn path_outside_root_errors() {
        let temp = tempfile::tempdir().expect("tempdir");
        let config = resolve(temp.path(), |_| {});

        let err = build_guard_report(&config, &["../outside.ts".to_string()]).unwrap_err();

        assert!(matches!(err, GuardError::OutsideRoot(_)));
    }
}