party 0.3.1

A command execution automator
Documentation 
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

//! `party run`
use std::path::Path;

use anyhow::bail;

use crate::{
    cli::RunArgs,
    parser::command_parser::CommandParser,
    party_command::{self, make_default_commands, PartyCommand},
    runner, schdeuler,
    util::{check_file_path, CHECK, DEFAULT_PARTY_CONF, HORSE, POPPER},
};

/// Implementation of `party run`
pub async fn run(run_args: RunArgs) -> anyhow::Result<()> {
    // Check if file exists only if provided via -f
    if let Some(file_path) = &run_args.file {
        check_file_path(file_path)?;
    }

    let file_path = run_args.file.unwrap_or(DEFAULT_PARTY_CONF.to_string());

    let commands = if Path::new(&file_path).exists() {
        let parser = CommandParser {
            path: file_path.to_string(),
        };
        let tasks = parser.parse()?;
        party_command::convert_toml_tasks(tasks.tasks)
    } else {
        make_default_commands()
    };

    match run_args.index {
        Some(index) => run_single_task(commands, index),
        None => run_all(commands).await,
    }
}

fn run_single_task(commands: Vec<PartyCommand>, index: usize) -> anyhow::Result<()> {
    println!("Staring party run {} {} {}", HORSE, HORSE, HORSE);

    let Some(command) = commands.get(index - 1) else {
        bail!(
            "Index {} out of range. Only {} tasks are available",
            index,
            commands.len()
        )
    };

    println!("Running a single task: {}", command);
    runner::run_single_command(command)?;
    println!(
        "{} {} Party run complete! {} {}",
        CHECK, POPPER, POPPER, CHECK
    );

    Ok(())
}

async fn run_all(commands: Vec<PartyCommand>) -> anyhow::Result<()> {
    println!("Staring party run {} {} {}", HORSE, HORSE, HORSE);

    let no_commands = commands.len();

    let batches = schdeuler::schedule_commands(commands);

    if batches.len() == no_commands {
        runner::run_sync_commands(batches, no_commands)?;
    } else {
        runner::run_async_commands(batches, no_commands).await?;
    }

    println!(
        "{} {} Party run complete! {} {}",
        CHECK, POPPER, POPPER, CHECK
    );

    Ok(())
}