use std::path::Path;
use alint_core::{
Context, Error, Level, PerFileRule, Result, Rule, RuleSpec, Scope, ScopeFilter, Violation,
};
use serde::Deserialize;
#[derive(Debug, Deserialize)]
struct Options {
max_lines: u64,
}
#[derive(Debug)]
pub struct FileMaxLinesRule {
id: String,
level: Level,
policy_url: Option<String>,
message: Option<String>,
scope: Scope,
scope_filter: Option<ScopeFilter>,
max_lines: u64,
}
impl Rule for FileMaxLinesRule {
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>> {
let mut violations = Vec::new();
for entry in ctx.index.files() {
if !self.scope.matches(&entry.path) {
continue;
}
if let Some(filter) = &self.scope_filter
&& !filter.matches(&entry.path, ctx.index)
{
continue;
}
let full = ctx.root.join(&entry.path);
let Ok(bytes) = std::fs::read(&full) else {
continue;
};
violations.extend(self.evaluate_file(ctx, &entry.path, &bytes)?);
}
Ok(violations)
}
fn as_per_file(&self) -> Option<&dyn PerFileRule> {
Some(self)
}
fn scope_filter(&self) -> Option<&ScopeFilter> {
self.scope_filter.as_ref()
}
}
impl PerFileRule for FileMaxLinesRule {
fn path_scope(&self) -> &Scope {
&self.scope
}
fn evaluate_file(
&self,
_ctx: &Context<'_>,
path: &Path,
bytes: &[u8],
) -> Result<Vec<Violation>> {
let lines = count_lines(bytes);
if lines <= self.max_lines {
return Ok(Vec::new());
}
let msg = self.message.clone().unwrap_or_else(|| {
format!(
"file has {} line(s); at most {} allowed",
lines, self.max_lines,
)
});
Ok(vec![
Violation::new(msg).with_path(std::sync::Arc::<Path>::from(path)),
])
}
}
fn count_lines(bytes: &[u8]) -> u64 {
if bytes.is_empty() {
return 0;
}
#[allow(clippy::naive_bytecount)]
let newlines = bytes.iter().filter(|&&b| b == b'\n').count() as u64;
let trailing_unterminated = u64::from(!bytes.ends_with(b"\n"));
newlines + trailing_unterminated
}
pub fn build(spec: &RuleSpec) -> Result<Box<dyn Rule>> {
let Some(paths) = &spec.paths else {
return Err(Error::rule_config(
&spec.id,
"file_max_lines requires a `paths` field",
));
};
let opts: Options = spec
.deserialize_options()
.map_err(|e| Error::rule_config(&spec.id, format!("invalid options: {e}")))?;
Ok(Box::new(FileMaxLinesRule {
id: spec.id.clone(),
level: spec.level,
policy_url: spec.policy_url.clone(),
message: spec.message.clone(),
scope: Scope::from_paths_spec(paths)?,
scope_filter: spec.parse_scope_filter()?,
max_lines: opts.max_lines,
}))
}
#[cfg(test)]
mod tests {
use super::count_lines;
#[test]
fn empty_file_is_zero_lines() {
assert_eq!(count_lines(b""), 0);
}
#[test]
fn matches_min_lines_semantics() {
assert_eq!(count_lines(b"a\n"), 1);
assert_eq!(count_lines(b"a\nb\n"), 2);
assert_eq!(count_lines(b"a\nb"), 2);
assert_eq!(count_lines(b"\n\n"), 2);
assert_eq!(count_lines(b"a\n\nb\n"), 3);
}
}