refactor: markers as enum parsed from lines
All checks were successful
Test / test (push) Successful in 4s

Fold Marker and IssueMarker into an enum.
Add functions to Line to parse into a Marker.
This commit is contained in:
Paul Campbell 2024-09-18 15:28:17 +01:00
parent 1e2a6e1804
commit 3bdb914052
8 changed files with 51 additions and 79 deletions

1
.ignore Normal file
View file

@ -0,0 +1 @@
**/tests

View file

@ -2,7 +2,7 @@
use std::path::Path;
use anyhow::{Context, Result};
use model::{Config, FoundMarkers, Line, Marker, MarkerKind};
use model::{Config, FoundMarkers, Line, Marker};
use patterns::{issue_pattern, marker_pattern};
mod model;
@ -37,7 +37,7 @@ fn main() -> Result<()> {
}
}
println!("{found_markers:?}");
println!("{found_markers:#?}");
// TODO: add authentication when provided
// let issues = ureq::get(&api).call()?.into_string()?;
@ -59,35 +59,16 @@ fn scan_file(file: &Path, config: &Config, found_markers: &mut FoundMarkers) ->
.file_read_to_string(file)?
.lines()
.enumerate()
.filter_map(|(n, line)| prefix_match(n, line, file, config))
.for_each(|marker| {
println!("- {}", marker.line().value());
if let Some(issue) = config
.issue_pattern()
.find(marker.line().value())
.map(|issue| issue.as_str())
.and_then(|issue| issue.parse::<usize>().ok())
{
found_markers.add_issue_marker(marker.into_issue_marker(issue));
} else {
found_markers.add_marker(marker);
}
});
.map(|(n, line)| {
Line::builder()
.file(file.to_path_buf())
.num(n)
.value(line.to_owned())
.build()
})
.filter_map(|line| line.into_marker().ok())
.filter(|marker| !matches!(marker, Marker::Unmarked))
.for_each(|marker| found_markers.add_marker(marker));
Ok(())
}
fn prefix_match(num: usize, line: &str, path: &Path, config: &Config) -> Option<Marker> {
let find = config.prefix_pattern().find(line)?;
let kind = match find.as_str() {
"TODO" => Some(MarkerKind::Todo),
"FIXME" => Some(MarkerKind::Fixme),
_ => None,
}?;
Some(
Marker::builder()
.kind(kind)
.file(path.to_path_buf())
.line(Line::builder().num(num).value(line.to_owned()).build())
.build(),
)
}

View file

@ -1,18 +1,47 @@
//
#![allow(dead_code)]
use std::path::{Path, PathBuf};
use anyhow::{Context, Result};
use bon::Builder;
use crate::patterns::{issue_pattern, marker_pattern};
use super::Marker;
#[derive(Debug, Builder)]
pub struct Line {
file: PathBuf,
num: usize,
value: String,
}
impl Line {
pub fn file(&self) -> &Path {
&self.file
}
pub fn num(&self) -> usize {
self.num
}
pub fn value(&self) -> &str {
&self.value
}
pub fn into_marker(self) -> Result<Marker> {
if marker_pattern()?.find(&self.value).is_some() {
match issue_pattern()?.captures(&self.value) {
Some(capture) => {
let issue = capture
.name("ISSUE_NUMBER")
.context("ISSUE_NUMBER")?
.as_str()
.to_owned();
Ok(Marker::Valid(self, issue))
}
None => Ok(Marker::Invalid(self)),
}
} else {
Ok(Marker::Unmarked)
}
}
}

View file

@ -1,16 +1,12 @@
use super::{IssueMarker, Marker};
use super::Marker;
//
#[derive(Debug, Default)]
pub struct FoundMarkers {
markers: Vec<Marker>,
issue_markers: Vec<IssueMarker>,
}
impl FoundMarkers {
pub fn add_marker(&mut self, marker: Marker) {
self.markers.push(marker);
}
pub fn add_issue_marker(&mut self, issue_marker: IssueMarker) {
self.issue_markers.push(issue_marker);
}
}

View file

@ -1,7 +0,0 @@
//
/// What type of comment #[derive(Debug)]
#[derive(Debug)]
pub enum MarkerKind {
Todo,
Fixme,
}

View file

@ -1,37 +1,11 @@
//
#![allow(dead_code)]
use std::path::{Path, PathBuf};
use bon::Builder;
use crate::model::Line;
use super::{IssueMarker, MarkerKind};
/// Represents a TODO or FIXME comment that doesn't have any issue number
#[derive(Debug, Builder)]
pub struct Marker {
/// What type of marker
kind: MarkerKind,
/// Path of the file
file: PathBuf,
/// The line from the file
line: Line,
}
impl Marker {
pub fn kind(&self) -> &MarkerKind {
&self.kind
}
pub fn file(&self) -> &Path {
&self.file
}
pub fn line(&self) -> &Line {
&self.line
}
pub fn into_issue_marker(self, issue: usize) -> IssueMarker {
IssueMarker::builder().marker(self).issue(issue).build()
}
#[derive(Debug)]
pub enum Marker {
Unmarked,
Invalid(Line),
Valid(Line, String),
}

View file

@ -1,10 +1,7 @@
//
mod found;
mod issue;
mod kind;
mod marker;
pub use found::FoundMarkers;
pub use issue::IssueMarker;
pub use kind::MarkerKind;
pub use marker::Marker;

View file

@ -5,4 +5,5 @@ mod markers;
pub use config::Config;
pub use line::Line;
pub use markers::*;
pub use markers::FoundMarkers;
pub use markers::Marker;