cpm_rs/
input_parser.rs

1use std::fs;
2use regex::Regex;
3
4use crate::customtask::CustomTask;
5
6/**
7 * Pattern for task names.
8 */
9static TASK_NAME_PATTERN: &str
10	= r"a-zA-Z-_.0-9";
11
12/**
13 * Pattern for dependency names.
14 * NAME_PAT should be replaced by TASK_NAME_PATTERN.
15 */
16static DEPENDENCY_NAME_PATTERN: &str
17	= r"([NAME_PAT]+)";
18
19/**
20 * Pattern for task definition.
21 * LIST_PAT should be replaced by TASK_LIST_PATTERN.
22 * NAME_PAT should be replaced by TASK_NAME_PATTERN.
23 */
24static TASK_DEFINITION_PATTERN: &str
25	= r"([NAME_PAT]+)\s*\(([-+]?\d+)\)(\s+after\s+\[([NAME_PAT\s,]*)\])?";
26
27pub fn parse_input_file(filename: &String) -> Result<Vec<CustomTask<i64>>, String> {
28	let task_def_str = TASK_DEFINITION_PATTERN.to_string()
29		.replace("NAME_PAT", TASK_NAME_PATTERN);
30	let dependency_def_str = DEPENDENCY_NAME_PATTERN.to_string()
31		.replace("NAME_PAT", TASK_NAME_PATTERN);
32
33	let task_definition: Regex = Regex::new(task_def_str.as_str()).unwrap();
34	let dependency_definition: Regex = Regex::new(dependency_def_str.as_str()).unwrap();
35
36	let mut task_list: Vec<CustomTask<i64>> = vec!{};
37
38	//println!("Input file: {}", filename);
39	let contents: String;
40	match fs::read_to_string(filename) {
41		Ok(file_content) => {contents = file_content;},
42		Err(e) => { return Err(format!("Could not read file: {}\n\r{}", filename, e)); },
43	};
44
45	let mut leftover = contents.clone();
46
47	for cap in task_definition.captures_iter(&contents) {
48		let id = cap[1].to_string();
49		let duration: u32;
50		match cap[2].parse::<u32>() {
51			Ok(dur) => duration = dur,
52			Err(_) => {
53				return Err(find_pattern_error(&cap[2], contents.clone()));
54			},
55		}
56
57		let dependency_str = cap.get(4).map_or("", |m| m.as_str());
58		let mut dependencies: Vec<String> = vec!{};
59		for dep_name in dependency_definition.captures_iter(&dependency_str) {
60			dependencies.push(dep_name[1].to_string());
61		}
62
63		let task = CustomTask::new(id, duration.into(), dependencies);
64		//println!("Task: {:?}", task);
65		leftover = leftover.replacen(&cap[0], "", 1);
66		task_list.push(task);
67	}
68	if leftover.replace("\n", "").len() > 0 {
69		return Err(find_pattern_error(&leftover, contents.clone()));
70	}
71	Ok(task_list)
72}
73
74pub fn find_pattern_error(error_str: &str, contents: String) -> String {
75	let output: String;
76	let error_lines: Vec<&str> = error_str.split('\n').collect();
77	for error_line in error_lines {
78		if error_line.len() > 0 {
79			//println!("Error line: {}", error_line);
80			let pos = contents.find(error_line).unwrap();
81			let (left_split, _) = contents.split_at(pos);
82			let lines = left_split.split('\n').collect::<Vec<&str>>();
83			let line_num = lines.len();
84			let column = lines.last().unwrap().len() + 1;
85			output = format!("line {}, column {}\n", line_num, column);
86			return output;
87		}
88	}
89	return format!("failed to parse errorous lines.");
90}
91
cpm_rs/input_parser.rs

cpm_rs/
input_parser.rs
