sysml-v2-parser 0.20.1

//! Shared parsers for accept/send payload clauses and transition accept triggers.

use crate::ast::{ActionUsage, Expression, Node, PayloadClause, TransitionAccept};
use crate::parser::action::action_usage_body;
use crate::parser::expr::expression;
use crate::parser::lex::{name, qualified_name, starts_with_keyword, ws1, ws_and_comments};
use crate::parser::node_from_to;
use crate::parser::with_span;
use crate::parser::Input;
use nom::bytes::complete::tag;
use nom::combinator::opt;
use nom::sequence::preceded;
use nom::IResult;
use nom::Parser;

/// Required typed payload: `name : qualified_name`.
pub(crate) fn typed_payload_clause(input: Input<'_>) -> IResult<Input<'_>, PayloadClause> {
    let (input, (name_span, name)) = with_span(name).parse(input)?;
    let (input, _) = preceded(ws_and_comments, tag(&b":"[..])).parse(input)?;
    let (input, (type_span, type_name)) =
        preceded(ws_and_comments, with_span(qualified_name)).parse(input)?;
    Ok((
        input,
        PayloadClause {
            name,
            type_name: Some(type_name),
            name_span,
            type_span: Some(type_span),
        },
    ))
}

/// After `accept` keyword: `name : Type` or shorthand expression.
pub(crate) fn transition_accept(input: Input<'_>) -> IResult<Input<'_>, TransitionAccept> {
    let (input, _) = preceded(ws_and_comments, tag(&b"accept"[..])).parse(input)?;
    let (input, _) = ws1(input)?;
    let (input, expr_node) = expression(input)?;
    let (input, type_suffix) = opt(preceded(
        preceded(ws_and_comments, tag(&b":"[..])),
        preceded(ws_and_comments, with_span(qualified_name)),
    ))
    .parse(input)?;
    if let (Expression::FeatureRef(name), Some((type_span, type_name))) =
        (&expr_node.value, type_suffix)
    {
        return Ok((
            input,
            TransitionAccept::Payload(PayloadClause {
                name: name.clone(),
                type_name: Some(type_name),
                name_span: expr_node.span.clone(),
                type_span: Some(type_span),
            }),
        ));
    }
    Ok((input, TransitionAccept::Shorthand(expr_node)))
}

/// Standalone control-node statement: `accept|send` payload (`;` or body).
fn control_node_payload_stmt<'a>(
    input: Input<'a>,
    keyword: &'a [u8],
    control_name: &'static str,
) -> IResult<Input<'a>, Node<ActionUsage>> {
    let start = input;
    let (input, _) = ws_and_comments(input)?;
    let (input, _) = tag(keyword).parse(input)?;
    let (input, _) = ws1(input)?;
    let (input, payload) = typed_payload_clause(input)?;
    let name_span = payload.name_span.clone();
    let type_ref_span = payload.type_span.clone();
    let (input, _) = ws_and_comments(input)?;
    let (input, body) = action_usage_body(input)?;
    let (input, _) =
        nom::combinator::opt(preceded(ws_and_comments, tag(&b";"[..]))).parse(input)?;
    let (accept, send) = if keyword == b"accept" {
        (Some(payload), None)
    } else {
        (None, Some(payload))
    };
    Ok((
        input,
        node_from_to(
            start,
            input,
            ActionUsage {
                name: control_name.to_string(),
                type_name: String::new(),
                accept,
                send,
                body,
                name_span: Some(name_span),
                type_ref_span,
            },
        ),
    ))
}

pub(crate) fn control_node_action_usage(
    input: Input<'_>,
) -> IResult<Input<'_>, Node<ActionUsage>> {
    let (peek, _) = ws_and_comments(input)?;
    let frag = peek.fragment();
    if starts_with_keyword(frag, b"accept") {
        return control_node_payload_stmt(input, b"accept", "accept");
    }
    if starts_with_keyword(frag, b"send") {
        return control_node_payload_stmt(input, b"send", "send");
    }
    Err(nom::Err::Error(nom::error::Error::new(
        input,
        nom::error::ErrorKind::Tag,
    )))
}