81 lines
2.0 KiB
Rust
81 lines
2.0 KiB
Rust
use std::{env, path::{PathBuf, Path}, fs};
|
|
use comment_parser::CommentParser;
|
|
use ignore::Walk;
|
|
use regex::Regex;
|
|
|
|
// TODO: Add language support for tsx, jsx, lua
|
|
|
|
#[derive(Debug)]
|
|
struct Reminder {
|
|
file: PathBuf,
|
|
row: u32,
|
|
col: u32,
|
|
contents: String
|
|
}
|
|
|
|
static ALLOWED_VERBS: [&str; 3] = ["TODO", "FIXME", "BUG"];
|
|
|
|
fn get_row_and_column(contents: &str, substr: &str, from: usize) -> (u32, u32) {
|
|
let occurence = contents[from..].find(substr).unwrap() + from;
|
|
let row = (contents.chars().take(occurence).filter(|c| *c == '\n').count()+1) as u32;
|
|
let col = (occurence - contents[..occurence].rfind('\n').unwrap_or(0)) as u32;
|
|
|
|
(row, col)
|
|
}
|
|
|
|
fn list_reminders<P: AsRef<Path>>(path: P) -> Vec<Reminder>
|
|
{
|
|
let mut reminders = vec![];
|
|
let reminder_pattern: Regex = Regex::new(r"^[A-Z]+.*:").unwrap();
|
|
|
|
for result in Walk::new(path) {
|
|
if result.is_err() { continue; }
|
|
|
|
let entry = result.unwrap();
|
|
let path = entry.path();
|
|
if !path.is_file() { continue; }
|
|
|
|
let rules = comment_parser::get_syntax_from_path(path);
|
|
if rules.is_err() { continue; }
|
|
|
|
let rules = rules.unwrap();
|
|
let file_contents = fs::read_to_string(path).unwrap();
|
|
let parser = CommentParser::new(&file_contents, rules);
|
|
let mut search_from = 0;
|
|
for comment in parser {
|
|
let text = comment.text().trim_start();
|
|
for line in text.lines() {
|
|
if !reminder_pattern.is_match(line) { continue; }
|
|
|
|
let is_allowed = ALLOWED_VERBS.into_iter().any(|v| line.starts_with(v));
|
|
if !is_allowed { continue; }
|
|
|
|
let (row, col) = get_row_and_column(&file_contents, line, search_from);
|
|
reminders.push(Reminder {
|
|
file: path.to_path_buf(),
|
|
row,
|
|
col,
|
|
contents: line.into()
|
|
});
|
|
|
|
search_from += file_contents[search_from..].find(line).unwrap()+1;
|
|
}
|
|
}
|
|
}
|
|
|
|
reminders
|
|
}
|
|
|
|
fn main() {
|
|
let search_dir;
|
|
if let Some(path) = env::args().nth(1) {
|
|
search_dir = path;
|
|
} else {
|
|
search_dir = ".".to_string();
|
|
}
|
|
|
|
for reminder in list_reminders(search_dir) {
|
|
println!("{}:{}:{}:{}", reminder.file.display(), reminder.row, reminder.col, reminder.contents);
|
|
}
|
|
}
|