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);
	}
}