use alint_core::{Context, Error, Level, NestedRuleSpec, Result, Rule, RuleSpec, Scope, Violation};
use serde::Deserialize;
use crate::for_each_dir::{IterateMode, evaluate_for_each};
#[derive(Debug, Deserialize)]
#[serde(deny_unknown_fields)]
struct Options {
select: String,
require: Vec<NestedRuleSpec>,
}
#[derive(Debug)]
pub struct EveryMatchingHasRule {
id: String,
level: Level,
policy_url: Option<String>,
select_scope: Scope,
require: Vec<NestedRuleSpec>,
}
impl Rule for EveryMatchingHasRule {
fn id(&self) -> &str {
&self.id
}
fn level(&self) -> Level {
self.level
}
fn policy_url(&self) -> Option<&str> {
self.policy_url.as_deref()
}
fn evaluate(&self, ctx: &Context<'_>) -> Result<Vec<Violation>> {
evaluate_for_each(
&self.id,
self.level,
&self.select_scope,
&self.require,
ctx,
IterateMode::Both,
)
}
}
pub fn build(spec: &RuleSpec) -> Result<Box<dyn Rule>> {
let opts: Options = spec
.deserialize_options()
.map_err(|e| Error::rule_config(&spec.id, format!("invalid options: {e}")))?;
if opts.require.is_empty() {
return Err(Error::rule_config(
&spec.id,
"every_matching_has requires at least one nested rule under `require:`",
));
}
let select_scope = Scope::from_patterns(&[opts.select])?;
Ok(Box::new(EveryMatchingHasRule {
id: spec.id.clone(),
level: spec.level,
policy_url: spec.policy_url.clone(),
select_scope,
require: opts.require,
}))
}
#[cfg(test)]
mod tests {
use super::*;
use alint_core::{FileEntry, FileIndex, RuleRegistry};
use std::path::{Path, PathBuf};
fn index(entries: &[(&str, bool)]) -> FileIndex {
FileIndex {
entries: entries
.iter()
.map(|(p, is_dir)| FileEntry {
path: PathBuf::from(p),
is_dir: *is_dir,
size: 1,
})
.collect(),
}
}
fn registry() -> RuleRegistry {
crate::builtin_registry()
}
#[test]
fn iterates_both_files_and_dirs() {
let require: Vec<NestedRuleSpec> =
vec![serde_yaml_ng::from_str("kind: file_exists\npaths: \"{path}\"\n").unwrap()];
let r = EveryMatchingHasRule {
id: "t".into(),
level: Level::Error,
policy_url: None,
select_scope: Scope::from_patterns(&["packages/*".to_string()]).unwrap(),
require,
};
let idx = index(&[
("packages", true),
("packages/a", true),
("packages/x.md", false),
]);
let reg = registry();
let ctx = Context {
root: Path::new("/"),
index: &idx,
registry: Some(®),
facts: None,
vars: None,
};
let v = r.evaluate(&ctx).unwrap();
assert_eq!(v.len(), 1);
assert_eq!(v[0].path.as_deref(), Some(Path::new("packages/a")));
}
}