Function p_allocation_request

crates/hyperqueue/src/client/resources.rs:12–41

()

use crate::worker::parser::{is_valid_resource_char, is_valid_starting_resource_char};

fn p_allocation_request() -> impl CharParser<AllocationRequest> {
    parse_exact_string("all")
        .map(|_| AllocationRequest::All)
        .or(parse_resource_amount()
            .padded()
            .then(
                choice((
                    just("compact!"),
                    just("compact"),
                    just("tight!"),
                    just("tight"),
                    just("scatter"),
                ))
                .or_not(),
            )
            .try_map(|(amount, policy), span| {
                let alloc = match policy {
                    None | Some("compact") => AllocationRequest::Compact(amount),
                    Some("compact!") => AllocationRequest::ForceCompact(amount),
                    Some("tight") => AllocationRequest::Tight(amount),
                    Some("tight!") => AllocationRequest::ForceTight(amount),
                    Some("scatter") => AllocationRequest::Scatter(amount),
                    _ => unreachable!(),
                };
                alloc
                    .validate()
                    .map_err(|e| ParseError::custom(span, e.to_string()))?;
                Ok(alloc)
            }))
}

fn p_resource_identifier() -> impl CharParser<String> {
    filter(|&c| is_valid_starting_resource_char(c))