use std::path::Path;
use alint_core::{
Context, Error, FixSpec, Fixer, Level, PerFileRule, Result, Rule, RuleSpec, Scope, Violation,
eval_per_file,
};
use crate::fixers::FileAppendFinalNewlineFixer;
#[derive(Debug)]
pub struct FinalNewlineRule {
id: String,
level: Level,
policy_url: Option<String>,
message: Option<String>,
scope: Scope,
fixer: Option<FileAppendFinalNewlineFixer>,
}
impl Rule for FinalNewlineRule {
alint_core::rule_common_impl!();
fn evaluate(&self, ctx: &Context<'_>) -> Result<Vec<Violation>> {
eval_per_file(self, ctx)
}
fn fixer(&self) -> Option<&dyn Fixer> {
self.fixer.as_ref().map(|f| f as &dyn Fixer)
}
fn as_per_file(&self) -> Option<&dyn PerFileRule> {
Some(self)
}
}
impl PerFileRule for FinalNewlineRule {
fn path_scope(&self) -> &Scope {
&self.scope
}
fn evaluate_file(
&self,
_ctx: &Context<'_>,
path: &Path,
bytes: &[u8],
) -> Result<Vec<Violation>> {
if bytes.is_empty() {
return Ok(Vec::new());
}
if bytes.last().copied() == Some(b'\n') {
return Ok(Vec::new());
}
let msg = self
.message
.clone()
.unwrap_or_else(|| "file does not end with a newline".to_string());
Ok(vec![
Violation::new(msg).with_path(std::sync::Arc::<Path>::from(path)),
])
}
fn max_bytes_needed(&self) -> Option<usize> {
Some(1)
}
}
pub fn build(spec: &RuleSpec) -> Result<Box<dyn Rule>> {
let _paths = spec
.paths
.as_ref()
.ok_or_else(|| Error::rule_config(&spec.id, "final_newline requires a `paths` field"))?;
let scope = Scope::from_spec(spec)?;
let fixer = match &spec.fix {
Some(FixSpec::FileAppendFinalNewline { .. }) => Some(FileAppendFinalNewlineFixer),
Some(other) => {
return Err(Error::rule_config(
&spec.id,
format!(
"fix.{} is not compatible with final_newline",
other.op_name()
),
));
}
None => None,
};
Ok(Box::new(FinalNewlineRule {
id: spec.id.clone(),
level: spec.level,
policy_url: spec.policy_url.clone(),
message: spec.message.clone(),
scope,
fixer,
}))
}
#[cfg(test)]
mod tests {
fn lacks_final_newline(bytes: &[u8]) -> bool {
!bytes.is_empty() && bytes.last().copied() != Some(b'\n')
}
#[test]
fn file_ending_in_lf_is_fine() {
assert!(!lacks_final_newline(b"hello\n"));
}
#[test]
fn file_ending_in_crlf_is_fine() {
assert!(!lacks_final_newline(b"hello\r\n"));
}
#[test]
fn file_missing_trailing_newline_flagged() {
assert!(lacks_final_newline(b"hello"));
}
#[test]
fn empty_file_is_ok() {
assert!(!lacks_final_newline(b""));
}
#[test]
fn single_newline_is_ok() {
assert!(!lacks_final_newline(b"\n"));
}
}