summaryrefslogtreecommitdiffstats
path: root/crates/driver/src/parse.rs
blob: aad93609f939064a5a9b78181f2399a601f8c296 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
use nom::{
	bytes::complete::{tag, take_while1},
	combinator::{all_consuming, opt},
	error::ParseError,
	Err, IResult, InputLength, Parser,
};

#[derive(Debug, Clone, Copy)]
pub struct Task<'a> {
	pub recipe: &'a str,
	pub task: &'a str,
	pub host: Option<&'a str>,
	pub target: Option<&'a str>,
}

#[derive(Debug, Clone, Copy)]
pub struct TaskFlags {
	pub force_run: bool,
}

fn is_name_char(c: char) -> bool {
	matches!(c, 'a'..='z' | 'A' ..='Z' | '0'..='9' | '_' | '-')
}

fn name(input: &str) -> IResult<&str, &str> {
	take_while1(is_name_char)(input)
}

fn task_id(input: &str) -> IResult<&str, (&str, &str)> {
	let (input, recipe) = name(input)?;
	let (input, _) = tag(":")(input)?;
	let (input, task) = name(input)?;
	Ok((input, (recipe, task)))
}

fn task_arg_target(input: &str) -> IResult<&str, &str> {
	let (input, _) = tag(":")(input)?;
	let (input, target) = name(input)?;
	Ok((input, target))
}

fn task_args(input: &str) -> IResult<&str, (Option<&str>, Option<&str>)> {
	let (input, _) = tag("/")(input)?;
	let (input, host) = opt(name)(input)?;
	let (input, target) = opt(task_arg_target)(input)?;

	Ok((input, (host, target)))
}

fn task(input: &str) -> IResult<&str, Task> {
	let (input, (recipe, task)) = task_id(input)?;
	let (input, args) = opt(task_args)(input)?;

	let (host, target) = args.unwrap_or_default();

	Ok((
		input,
		Task {
			recipe,
			task,
			host,
			target,
		},
	))
}

fn task_flags(input: &str) -> IResult<&str, TaskFlags> {
	let (input, force_run) = opt(tag("+"))(input)?;

	Ok((
		input,
		TaskFlags {
			force_run: force_run.is_some(),
		},
	))
}

fn task_with_flags(input: &str) -> IResult<&str, (Task, TaskFlags)> {
	let (input, task) = task(input)?;
	let (input, flags) = task_flags(input)?;

	Ok((input, (task, flags)))
}

fn parse_all<I, O, E: ParseError<I>, F>(f: F, input: I) -> Result<O, Err<E>>
where
	I: InputLength,
	F: Parser<I, O, E>,
{
	all_consuming(f)(input).map(|(_, result)| result)
}

pub fn parse_task_with_flags(input: &str) -> Option<(Task, TaskFlags)> {
	parse_all(task_with_flags, input).ok()
}