use alint_core::{Context, Error, FixSpec, Fixer, Level, Result, Rule, RuleSpec, Scope, Violation};
use serde::Deserialize;
use crate::fixers::{FileCollapseBlankLinesFixer, line_is_blank};
#[derive(Debug, Deserialize)]
#[serde(deny_unknown_fields)]
struct Options {
max: u32,
}
#[derive(Debug)]
pub struct MaxConsecutiveBlankLinesRule {
id: String,
level: Level,
policy_url: Option<String>,
message: Option<String>,
scope: Scope,
max: u32,
fixer: Option<FileCollapseBlankLinesFixer>,
}
impl Rule for MaxConsecutiveBlankLinesRule {
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;
}
let full = ctx.root.join(&entry.path);
let Ok(bytes) = std::fs::read(&full) else {
continue;
};
let Ok(text) = std::str::from_utf8(&bytes) else {
continue;
};
if let Some(line_no) = first_over_limit(text, self.max) {
let msg = self
.message
.clone()
.unwrap_or_else(|| format!("more than {} consecutive blank line(s)", self.max));
violations.push(
Violation::new(msg)
.with_path(entry.path.clone())
.with_location(line_no, 1),
);
}
}
Ok(violations)
}
fn fixer(&self) -> Option<&dyn Fixer> {
self.fixer.as_ref().map(|f| f as &dyn Fixer)
}
}
fn first_over_limit(text: &str, max: u32) -> Option<usize> {
let mut blank_run: u32 = 0;
let mut remaining = text;
let mut line_no: usize = 0;
loop {
let (body, has_ending, rest) = match remaining.find('\n') {
Some(i) => {
let before = &remaining[..i];
let body = before.strip_suffix('\r').unwrap_or(before);
(body, true, &remaining[i + 1..])
}
None => (remaining, false, ""),
};
if !has_ending && body.is_empty() {
return None;
}
line_no += 1;
if line_is_blank(body) {
blank_run += 1;
if blank_run > max {
return Some(line_no);
}
} else {
blank_run = 0;
}
if !has_ending {
return None;
}
remaining = rest;
}
}
pub fn build(spec: &RuleSpec) -> Result<Box<dyn Rule>> {
let paths = spec.paths.as_ref().ok_or_else(|| {
Error::rule_config(
&spec.id,
"max_consecutive_blank_lines requires a `paths` field",
)
})?;
let opts: Options = spec
.deserialize_options()
.map_err(|e| Error::rule_config(&spec.id, format!("invalid options: {e}")))?;
let fixer = match &spec.fix {
Some(FixSpec::FileCollapseBlankLines { .. }) => {
Some(FileCollapseBlankLinesFixer::new(opts.max))
}
Some(other) => {
return Err(Error::rule_config(
&spec.id,
format!(
"fix.{} is not compatible with max_consecutive_blank_lines",
other.op_name()
),
));
}
None => None,
};
Ok(Box::new(MaxConsecutiveBlankLinesRule {
id: spec.id.clone(),
level: spec.level,
policy_url: spec.policy_url.clone(),
message: spec.message.clone(),
scope: Scope::from_paths_spec(paths)?,
max: opts.max,
fixer,
}))
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn within_limit_is_ok() {
assert_eq!(first_over_limit("a\n\nb\n", 1), None);
}
#[test]
fn one_over_limit_is_flagged() {
assert_eq!(first_over_limit("a\n\n\nb\n", 1), Some(3));
}
#[test]
fn max_zero_flags_any_blank() {
assert_eq!(first_over_limit("a\n\nb\n", 0), Some(2));
}
#[test]
fn trailing_newline_is_not_a_blank_line() {
assert_eq!(first_over_limit("a\n", 0), None);
}
#[test]
fn whitespace_only_line_counts_as_blank() {
assert_eq!(first_over_limit("a\n \n\t\nb\n", 1), Some(3));
}
#[test]
fn crlf_endings_are_counted() {
assert_eq!(first_over_limit("a\r\n\r\n\r\n\r\nb\r\n", 1), Some(3));
}
}