summaryrefslogtreecommitdiffstats
path: root/crates/driver/src/main.rs
blob: 0e5c629c1e5f37463115d8ca6b079cc692edc75d (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
mod args;
mod context;
mod driver;
mod parse;
mod paths;
mod pin;
mod recipe;
mod resolve;
mod task;
mod template;

use clap::Parser;

use runner::Runner;

#[derive(Parser)]
#[clap(version = clap::crate_version!())]
struct Opts {
	/// Allow N jobs at once.
	/// Defaults to the number of available CPUs
	#[clap(short, long)]
	jobs: Option<usize>,
	/// The tasks to run
	#[clap(name = "task", required = true)]
	tasks: Vec<String>,
}

fn main() {
	let opts: Opts = Opts::parse();

	let runner = unsafe { Runner::new(&runner::Options { jobs: opts.jobs }) }.unwrap();

	let ctx = context::Context::new(
		recipe::read_recipes("examples/recipes").unwrap(),
		pin::read_pins("examples/pins.yml").unwrap(),
	)
	.unwrap();

	let mut rsv = resolve::Resolver::new(&ctx);

	for task in opts.tasks {
		let task_ref = match ctx.parse(&task) {
			Ok(task_ref) => task_ref,
			Err(err) => {
				eprintln!("{}", err);
				std::process::exit(1);
			}
		};
		let errors = rsv.add_goal(&task_ref);
		if !errors.is_empty() {
			for error in errors {
				eprintln!("{}", error);
			}
			std::process::exit(1);
		}
	}
	let taskset = rsv.into_taskset();
	let mut driver = driver::Driver::new(&ctx, taskset).unwrap();
	if let Err(error) = driver.run(&runner) {
		eprintln!("{}", error);
		std::process::exit(1);
	}
}