pay-respects/rule_parser/src/lib.rs

use std::path::Path;

use proc_macro::TokenStream;

#[proc_macro]
pub fn parse_rules(input: TokenStream) -> TokenStream {
	let directory = input.to_string().trim_matches('"').to_owned();
	let rules = get_rules(directory);
	let string_hashmap = gen_string_hashmap(rules);

	string_hashmap.parse().unwrap()
}

#[derive(serde::Deserialize)]
struct Rule {
	command: String,
	match_output: Vec<MatchOutput>,
}

#[derive(serde::Deserialize)]
struct MatchOutput {
	pattern: Vec<String>,
	suggest: String,
}

fn get_rules(directory: String) -> Vec<Rule> {
	let files = std::fs::read_dir(directory).expect("Failed to read directory.");

	let mut rules = Vec::new();
	for file in files {
		let file = file.expect("Failed to read file.");
		let path = file.path();
		let path = path.to_str().expect("Failed to convert path to string.");

		let rule_file = parse_file(Path::new(path));
		rules.push(rule_file);
	}
	rules
}

fn gen_string_hashmap(rules: Vec<Rule>) -> String {
	let mut string_hashmap = String::from("HashMap::from([");
	for rule in rules {
		let command = rule.command.to_owned();
		string_hashmap.push_str(&format!("(\"{}\", vec![", command));
		for match_output in rule.match_output {
			let pattern = match_output.pattern;
			let suggest = match_output.suggest;
			string_hashmap.push_str(&format!("(vec![\"{}\"], \"{}\"),", pattern.join("\", \""), suggest));
		}
		string_hashmap.push_str("]),");
	}
	string_hashmap.push_str("])");
	string_hashmap
}

fn parse_file(file: &Path) -> Rule {
	let file = std::fs::read_to_string(file).expect("Failed to read file.");
	toml::from_str(&file).expect("Failed to parse toml.")
}