nushell/src/parser/parse/parser.rs

#![allow(unused)]

use crate::parser::parse::{
    call_node::*, flag::*, operator::*, pipeline::*, token_tree::*, token_tree_builder::*,
    tokens::*, unit::*,
};
use crate::{Span, Tagged};
use nom;
use nom::branch::*;
use nom::bytes::complete::*;
use nom::character::complete::*;
use nom::combinator::*;
use nom::multi::*;
use nom::sequence::*;

use log::trace;
use nom::dbg;
use nom::*;
use nom::{AsBytes, FindSubstring, IResult, InputLength, InputTake, Slice};
use nom5_locate::{position, LocatedSpan};
use std::fmt::Debug;
use std::str::FromStr;

pub type NomSpan<'a> = LocatedSpan<&'a str>;

pub fn nom_input(s: &'a str) -> NomSpan<'a> {
    LocatedSpan::new(s)
}

macro_rules! operator {
    ($name:tt : $token:tt ) => {
        pub fn $name(input: NomSpan) -> IResult<NomSpan, TokenNode> {
            let start = input.offset;
            let (input, tag) = tag(stringify!($token))(input)?;
            let end = input.offset;

            Ok((
                input,
                TokenTreeBuilder::spanned_op(tag.fragment, (start, end)),
            ))
        }
    };
}

operator! { gt:  >  }
operator! { lt:  <  }
operator! { gte: >= }
operator! { lte: <= }
operator! { eq:  == }
operator! { neq: != }

fn trace_step<'a, T: Debug>(
    input: NomSpan<'a>,
    name: &str,
    block: impl FnOnce(NomSpan<'a>) -> IResult<NomSpan<'a>, T>,
) -> IResult<NomSpan<'a>, T> {
    trace!(target: "nu::lite_parse", "+ before {} @ {:?}", name, input);
    match block(input) {
        Ok((input, result)) => {
            trace!(target: "nu::lite_parse", "after {} @ {:?} -> {:?}", name, input, result);
            Ok((input, result))
        }

        Err(e) => {
            trace!(target: "nu::lite_parse", "- failed {} :: {:?}", name, e);
            Err(e)
        }
    }
}

pub fn raw_integer(input: NomSpan) -> IResult<NomSpan, Tagged<i64>> {
    let start = input.offset;
    trace_step(input, "raw_integer", move |input| {
        let (input, neg) = opt(tag("-"))(input)?;
        let (input, num) = digit1(input)?;
        let end = input.offset;

        Ok((
            input,
            Tagged::from_simple_spanned_item(int(num.fragment, neg), (start, end)),
        ))
    })
}

pub fn operator(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "operator", |input| {
        let (input, operator) = alt((gte, lte, neq, gt, lt, eq))(input)?;

        Ok((input, operator))
    })
}

pub fn dq_string(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "dq_string", |input| {
        let start = input.offset;
        let (input, _) = char('"')(input)?;
        let start1 = input.offset;
        let (input, _) = many0(none_of("\""))(input)?;
        let end1 = input.offset;
        let (input, _) = char('"')(input)?;
        let end = input.offset;
        Ok((
            input,
            TokenTreeBuilder::spanned_string((start1, end1), (start, end)),
        ))
    })
}

pub fn sq_string(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "sq_string", move |input| {
        let start = input.offset;
        let (input, _) = char('\'')(input)?;
        let start1 = input.offset;
        let (input, _) = many0(none_of("\'"))(input)?;
        let end1 = input.offset;
        let (input, _) = char('\'')(input)?;
        let end = input.offset;

        Ok((
            input,
            TokenTreeBuilder::spanned_string((start1, end1), (start, end)),
        ))
    })
}

pub fn string(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "string", move |input| {
        alt((sq_string, dq_string))(input)
    })
}

pub fn external(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "external", move |input| {
        let start = input.offset;
        let (input, _) = tag("^")(input)?;
        let (input, bare) = take_while(is_bare_char)(input)?;
        let end = input.offset;

        Ok((
            input,
            TokenTreeBuilder::spanned_external(bare, (start, end)),
        ))
    })
}

pub fn bare(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "bare", move |input| {
        let start = input.offset;
        let (input, _) = take_while1(is_start_bare_char)(input)?;
        let (input, _) = take_while(is_bare_char)(input)?;
        let end = input.offset;

        Ok((input, TokenTreeBuilder::spanned_bare((start, end))))
    })
}

pub fn var(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "var", move |input| {
        let start = input.offset;
        let (input, _) = tag("$")(input)?;
        let (input, bare) = member(input)?;
        let end = input.offset;

        Ok((
            input,
            TokenTreeBuilder::spanned_var(bare.span(), (start, end)),
        ))
    })
}

pub fn member(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "identifier", move |input| {
        let start = input.offset;
        let (input, _) = take_while1(is_id_start)(input)?;
        let (input, _) = take_while(is_id_continue)(input)?;

        let end = input.offset;

        Ok((input, TokenTreeBuilder::spanned_member((start, end))))
    })
}

pub fn flag(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "flag", move |input| {
        let start = input.offset;
        let (input, _) = tag("--")(input)?;
        let (input, bare) = bare(input)?;
        let end = input.offset;

        Ok((
            input,
            TokenTreeBuilder::spanned_flag(bare.span(), (start, end)),
        ))
    })
}

pub fn shorthand(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "shorthand", move |input| {
        let start = input.offset;
        let (input, _) = tag("-")(input)?;
        let (input, bare) = bare(input)?;
        let end = input.offset;

        Ok((
            input,
            TokenTreeBuilder::spanned_shorthand(bare.span(), (start, end)),
        ))
    })
}

pub fn raw_unit(input: NomSpan) -> IResult<NomSpan, Tagged<Unit>> {
    trace_step(input, "raw_unit", move |input| {
        let start = input.offset;
        let (input, unit) = alt((
            tag("B"),
            tag("b"),
            tag("KB"),
            tag("kb"),
            tag("Kb"),
            tag("K"),
            tag("k"),
            tag("MB"),
            tag("mb"),
            tag("Mb"),
            tag("GB"),
            tag("gb"),
            tag("Gb"),
            tag("TB"),
            tag("tb"),
            tag("Tb"),
            tag("PB"),
            tag("pb"),
            tag("Pb"),
        ))(input)?;
        let end = input.offset;

        Ok((
            input,
            Tagged::from_simple_spanned_item(Unit::from(unit.fragment), (start, end)),
        ))
    })
}

pub fn size(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "size", move |input| {
        let mut is_size = false;
        let start = input.offset;
        let (input, int) = raw_integer(input)?;
        if let Ok((input, Some(size))) = opt(raw_unit)(input) {
            let end = input.offset;

            // Check to make sure there is no trailing parseable characters
            if let Ok((input, Some(extra))) = opt(bare)(input) {
                return Err(nom::Err::Error((input, nom::error::ErrorKind::Char)));
            }

            Ok((
                input,
                TokenTreeBuilder::spanned_size((*int, *size), (start, end)),
            ))
        } else {
            let end = input.offset;

            // Check to make sure there is no trailing parseable characters
            if let Ok((input, Some(extra))) = opt(bare)(input) {
                return Err(nom::Err::Error((input, nom::error::ErrorKind::Char)));
            }

            Ok((input, TokenTreeBuilder::spanned_int((*int), (start, end))))
        }
    })
}

pub fn leaf(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "leaf", move |input| {
        let (input, node) =
            alt((size, string, operator, flag, shorthand, var, external, bare))(input)?;

        Ok((input, node))
    })
}

pub fn token_list(input: NomSpan) -> IResult<NomSpan, Vec<TokenNode>> {
    trace_step(input, "token_list", move |input| {
        let (input, first) = node(input)?;
        let (input, list) = many0(pair(space1, node))(input)?;

        Ok((input, make_token_list(None, first, list, None)))
    })
}

pub fn spaced_token_list(input: NomSpan) -> IResult<NomSpan, Vec<TokenNode>> {
    trace_step(input, "spaced_token_list", move |input| {
        let (input, sp_left) = opt(space1)(input)?;
        let (input, first) = node(input)?;
        let (input, list) = many0(pair(space1, node))(input)?;
        let (input, sp_right) = opt(space1)(input)?;

        Ok((input, make_token_list(sp_left, first, list, sp_right)))
    })
}

fn make_token_list(
    sp_left: Option<NomSpan>,
    first: TokenNode,
    list: Vec<(NomSpan, TokenNode)>,
    sp_right: Option<NomSpan>,
) -> Vec<TokenNode> {
    let mut nodes = vec![];

    if let Some(sp_left) = sp_left {
        nodes.push(TokenNode::Whitespace(Span::from(sp_left)));
    }

    nodes.push(first);

    for (ws, token) in list {
        nodes.push(TokenNode::Whitespace(Span::from(ws)));
        nodes.push(token);
    }

    if let Some(sp_right) = sp_right {
        nodes.push(TokenNode::Whitespace(Span::from(sp_right)));
    }

    nodes
}

pub fn whitespace(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "whitespace", move |input| {
        let left = input.offset;
        let (input, ws1) = space1(input)?;
        let right = input.offset;

        Ok((input, TokenTreeBuilder::spanned_ws((left, right))))
    })
}

pub fn delimited_paren(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "delimited_paren", move |input| {
        let left = input.offset;
        let (input, _) = char('(')(input)?;
        let (input, ws1) = opt(whitespace)(input)?;
        let (input, inner_items) = opt(token_list)(input)?;
        let (input, ws2) = opt(whitespace)(input)?;
        let (input, _) = char(')')(input)?;
        let right = input.offset;

        let mut items = vec![];

        if let Some(space) = ws1 {
            items.push(space);
        }

        if let Some(inner_items) = inner_items {
            items.extend(inner_items);
        }

        if let Some(space) = ws2 {
            items.push(space);
        }

        Ok((
            input,
            TokenTreeBuilder::spanned_parens(items, (left, right)),
        ))
    })
}

pub fn delimited_square(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "delimited_paren", move |input| {
        let left = input.offset;
        let (input, _) = char('[')(input)?;
        let (input, ws1) = opt(whitespace)(input)?;
        let (input, inner_items) = opt(token_list)(input)?;
        let (input, ws2) = opt(whitespace)(input)?;
        let (input, _) = char(']')(input)?;
        let right = input.offset;

        let mut items = vec![];

        if let Some(space) = ws1 {
            items.push(space);
        }

        if let Some(inner_items) = inner_items {
            items.extend(inner_items);
        }

        if let Some(space) = ws2 {
            items.push(space);
        }

        Ok((
            input,
            TokenTreeBuilder::spanned_square(items, (left, right)),
        ))
    })
}

pub fn delimited_brace(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "delimited_brace", move |input| {
        let left = input.offset;
        let (input, _) = char('{')(input)?;
        let (input, _) = opt(space1)(input)?;
        let (input, items) = opt(token_list)(input)?;
        let (input, _) = opt(space1)(input)?;
        let (input, _) = char('}')(input)?;
        let right = input.offset;

        Ok((
            input,
            TokenTreeBuilder::spanned_brace(items.unwrap_or_else(|| vec![]), (left, right)),
        ))
    })
}

pub fn raw_call(input: NomSpan) -> IResult<NomSpan, Tagged<CallNode>> {
    trace_step(input, "raw_call", move |input| {
        let left = input.offset;
        let (input, items) = token_list(input)?;
        let right = input.offset;

        Ok((input, TokenTreeBuilder::spanned_call(items, (left, right))))
    })
}

pub fn path(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "path", move |input| {
        let left = input.offset;
        let (input, head) = node1(input)?;
        let (input, _) = tag(".")(input)?;
        let (input, tail) = separated_list(tag("."), alt((member, string)))(input)?;
        let right = input.offset;

        Ok((
            input,
            TokenTreeBuilder::spanned_path((head, tail), (left, right)),
        ))
    })
}

pub fn node1(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "node1", alt((leaf, delimited_paren)))
}

pub fn node(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(
        input,
        "node",
        alt((
            path,
            leaf,
            delimited_paren,
            delimited_brace,
            delimited_square,
        )),
    )
}

pub fn pipeline(input: NomSpan) -> IResult<NomSpan, TokenNode> {
    trace_step(input, "pipeline", |input| {
        let start = input.offset;
        let (input, head) = opt(tuple((raw_call, opt(space1), opt(tag("|")))))(input)?;
        let (input, items) = trace_step(
            input,
            "many0",
            many0(tuple((opt(space1), raw_call, opt(space1), opt(tag("|"))))),
        )?;

        let (input, tail) = opt(space1)(input)?;
        let (input, newline) = opt(multispace1)(input)?;

        if input.input_len() != 0 {
            return Err(Err::Error(error_position!(
                input,
                nom::error::ErrorKind::Eof
            )));
        }

        let end = input.offset;

        Ok((
            input,
            TokenTreeBuilder::spanned_pipeline(
                (make_call_list(head, items), tail.map(Span::from)),
                (start, end),
            ),
        ))
    })
}

fn make_call_list(
    head: Option<(Tagged<CallNode>, Option<NomSpan>, Option<NomSpan>)>,
    items: Vec<(
        Option<NomSpan>,
        Tagged<CallNode>,
        Option<NomSpan>,
        Option<NomSpan>,
    )>,
) -> Vec<PipelineElement> {
    let mut out = vec![];

    if let Some(head) = head {
        let el = PipelineElement::new(None, head.0, head.1.map(Span::from), head.2.map(Span::from));
        out.push(el);
    }

    for (ws1, call, ws2, pipe) in items {
        let el = PipelineElement::new(
            ws1.map(Span::from),
            call,
            ws2.map(Span::from),
            pipe.map(Span::from),
        );
        out.push(el);
    }

    out
}

fn int<T>(frag: &str, neg: Option<T>) -> i64 {
    let int = FromStr::from_str(frag).unwrap();

    match neg {
        None => int,
        Some(_) => int * -1,
    }
}

fn is_start_bare_char(c: char) -> bool {
    match c {
        _ if c.is_alphabetic() => true,
        _ if c.is_numeric() => true,
        '.' => true,
        '\\' => true,
        '/' => true,
        '_' => true,
        '-' => true,
        '@' => true,
        '*' => true,
        '?' => true,
        '~' => true,
        '+' => true,
        _ => false,
    }
}

fn is_bare_char(c: char) -> bool {
    match c {
        _ if c.is_alphanumeric() => true,
        ':' => true,
        '.' => true,
        '\\' => true,
        '/' => true,
        '_' => true,
        '-' => true,
        '@' => true,
        '*' => true,
        '?' => true,
        '=' => true,
        '~' => true,
        '+' => true,
        _ => false,
    }
}

fn is_id_start(c: char) -> bool {
    unicode_xid::UnicodeXID::is_xid_start(c)
}

fn is_id_continue(c: char) -> bool {
    unicode_xid::UnicodeXID::is_xid_continue(c)
        || match c {
            '-' => true,
            '?' => true,
            '!' => true,
            _ => false,
        }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::parser::parse::token_tree_builder::TokenTreeBuilder as b;
    use crate::parser::parse::token_tree_builder::{CurriedToken, TokenTreeBuilder};
    use pretty_assertions::assert_eq;

    macro_rules! assert_leaf {
        (parsers [ $($name:tt)* ] $input:tt -> $left:tt .. $right:tt { $kind:tt $parens:tt } ) => {
            $(
                assert_eq!(
                    apply($name, stringify!($name), $input),
                    token(RawToken::$kind $parens, $left, $right)
                );
            )*

            assert_eq!(
                apply(leaf, "leaf", $input),
                token(RawToken::$kind $parens, $left, $right)
            );

            assert_eq!(
                apply(leaf, "leaf", $input),
                token(RawToken::$kind $parens, $left, $right)
            );

            assert_eq!(
                apply(node, "node", $input),
                token(RawToken::$kind $parens, $left, $right)
            );
        };

        (parsers [ $($name:tt)* ] $input:tt -> $left:tt .. $right:tt { $kind:tt } ) => {
            $(
                assert_eq!(
                    apply($name, stringify!($name), $input),
                    token(RawToken::$kind, $left, $right)
                );
            )*
        }
    }

    #[test]
    fn test_integer() {
        assert_leaf! {
            parsers [ size ]
            "123" -> 0..3 { Integer(123) }
        }

        assert_leaf! {
            parsers [ size ]
            "-123" -> 0..4 { Integer(-123) }
        }
    }

    #[test]
    fn test_size() {
        assert_leaf! {
            parsers [ size ]
            "123MB" -> 0..5 { Size(123, Unit::MB) }
        }

        assert_leaf! {
            parsers [ size ]
            "10GB" -> 0..4 { Size(10, Unit::GB) }
        }
    }

    #[test]
    fn test_operator() {
        assert_eq!(apply(node, "node", ">"), build_token(b::op(">")));

        // assert_leaf! {
        //     parsers [ operator ]
        //     ">=" -> 0..2 { Operator(Operator::GreaterThanOrEqual) }
        // }

        // assert_leaf! {
        //     parsers [ operator ]
        //     "<" -> 0..1 { Operator(Operator::LessThan) }
        // }

        // assert_leaf! {
        //     parsers [ operator ]
        //     "<=" -> 0..2 { Operator(Operator::LessThanOrEqual) }
        // }

        // assert_leaf! {
        //     parsers [ operator ]
        //     "==" -> 0..2 { Operator(Operator::Equal) }
        // }

        // assert_leaf! {
        //     parsers [ operator ]
        //     "!=" -> 0..2 { Operator(Operator::NotEqual) }
        // }
    }

    #[test]
    fn test_string() {
        assert_leaf! {
            parsers [ string dq_string ]
            r#""hello world""# -> 0..13 { String(span(1, 12)) }
        }

        assert_leaf! {
            parsers [ string sq_string ]
            r"'hello world'" -> 0..13 { String(span(1, 12)) }
        }
    }

    #[test]
    fn test_bare() {
        assert_leaf! {
            parsers [ bare ]
            "hello" -> 0..5 { Bare }
        }

        assert_leaf! {
            parsers [ bare ]
            "chrome.exe" -> 0..10 { Bare }
        }

        assert_leaf! {
            parsers [ bare ]
            r"C:\windows\system.dll" -> 0..21 { Bare }
        }

        assert_leaf! {
            parsers [ bare ]
            r"C:\Code\-testing\my_tests.js" -> 0..28 { Bare }
        }
    }

    #[test]
    fn test_flag() {
        // assert_leaf! {
        //     parsers [ flag ]
        //     "--hello" -> 0..7 { Flag(Tagged::from_item(FlagKind::Longhand, span(2, 7))) }
        // }

        // assert_leaf! {
        //     parsers [ flag ]
        //     "--hello-world" -> 0..13 { Flag(Tagged::from_item(FlagKind::Longhand, span(2, 13))) }
        // }
    }

    #[test]
    fn test_shorthand() {
        // assert_leaf! {
        //     parsers [ shorthand ]
        //     "-alt" -> 0..4 { Flag(Tagged::from_item(FlagKind::Shorthand, span(1, 4))) }
        // }
    }

    #[test]
    fn test_variable() {
        assert_leaf! {
            parsers [ var ]
            "$it" -> 0..3 { Variable(span(1, 3)) }
        }

        assert_leaf! {
            parsers [ var ]
            "$name" -> 0..5 { Variable(span(1, 5)) }
        }
    }

    #[test]
    fn test_external() {
        assert_leaf! {
            parsers [ external ]
            "^ls" -> 0..3 { External(span(1, 3)) }
        }
    }

    #[test]
    fn test_delimited_paren() {
        assert_eq!(
            apply(node, "node", "(abc)"),
            build_token(b::parens(vec![b::bare("abc")]))
        );

        assert_eq!(
            apply(node, "node", "(  abc  )"),
            build_token(b::parens(vec![b::ws("  "), b::bare("abc"), b::ws("  ")]))
        );

        assert_eq!(
            apply(node, "node", "(  abc def )"),
            build_token(b::parens(vec![
                b::ws("  "),
                b::bare("abc"),
                b::sp(),
                b::bare("def"),
                b::sp()
            ]))
        );

        assert_eq!(
            apply(node, "node", "(  abc def 123 456GB )"),
            build_token(b::parens(vec![
                b::ws("  "),
                b::bare("abc"),
                b::sp(),
                b::bare("def"),
                b::sp(),
                b::int(123),
                b::sp(),
                b::size(456, "GB"),
                b::sp()
            ]))
        );
    }

    #[test]
    fn test_delimited_square() {
        assert_eq!(
            apply(node, "node", "[abc]"),
            build_token(b::square(vec![b::bare("abc")]))
        );

        assert_eq!(
            apply(node, "node", "[  abc  ]"),
            build_token(b::square(vec![b::ws("  "), b::bare("abc"), b::ws("  ")]))
        );

        assert_eq!(
            apply(node, "node", "[  abc def ]"),
            build_token(b::square(vec![
                b::ws("  "),
                b::bare("abc"),
                b::sp(),
                b::bare("def"),
                b::sp()
            ]))
        );

        assert_eq!(
            apply(node, "node", "[  abc def 123 456GB ]"),
            build_token(b::square(vec![
                b::ws("  "),
                b::bare("abc"),
                b::sp(),
                b::bare("def"),
                b::sp(),
                b::int(123),
                b::sp(),
                b::size(456, "GB"),
                b::sp()
            ]))
        );
    }

    #[test]
    fn test_path() {
        let _ = pretty_env_logger::try_init();
        assert_eq!(
            apply(node, "node", "$it.print"),
            build_token(b::path(b::var("it"), vec![b::member("print")]))
        );

        assert_eq!(
            apply(node, "node", "$head.part1.part2"),
            build_token(b::path(
                b::var("head"),
                vec![b::member("part1"), b::member("part2")]
            ))
        );

        assert_eq!(
            apply(node, "node", "( hello ).world"),
            build_token(b::path(
                b::parens(vec![b::sp(), b::bare("hello"), b::sp()]),
                vec![b::member("world")]
            ))
        );

        assert_eq!(
            apply(node, "node", "( hello ).\"world\""),
            build_token(b::path(
                b::parens(vec![b::sp(), b::bare("hello"), b::sp()],),
                vec![b::string("world")]
            ))
        );
    }

    #[test]
    fn test_nested_path() {
        assert_eq!(
            apply(
                node,
                "node",
                "( $it.is.\"great news\".right yep $yep ).\"world\""
            ),
            build_token(b::path(
                b::parens(vec![
                    b::sp(),
                    b::path(
                        b::var("it"),
                        vec![b::member("is"), b::string("great news"), b::member("right")]
                    ),
                    b::sp(),
                    b::bare("yep"),
                    b::sp(),
                    b::var("yep"),
                    b::sp()
                ]),
                vec![b::string("world")]
            ))
        )
    }

    #[test]
    fn test_smoke_single_command() {
        assert_eq!(
            apply(raw_call, "raw_call", "git add ."),
            build(b::call(
                b::bare("git"),
                vec![b::sp(), b::bare("add"), b::sp(), b::bare(".")]
            ))
        );

        assert_eq!(
            apply(raw_call, "raw_call", "open Cargo.toml"),
            build(b::call(
                b::bare("open"),
                vec![b::sp(), b::bare("Cargo.toml")]
            ))
        );

        assert_eq!(
            apply(raw_call, "raw_call", "select package.version"),
            build(b::call(
                b::bare("select"),
                vec![b::sp(), b::bare("package.version")]
            ))
        );

        assert_eq!(
            apply(raw_call, "raw_call", "echo $it"),
            build(b::call(b::bare("echo"), vec![b::sp(), b::var("it")]))
        );

        assert_eq!(
            apply(raw_call, "raw_call", "open Cargo.toml --raw"),
            build(b::call(
                b::bare("open"),
                vec![b::sp(), b::bare("Cargo.toml"), b::sp(), b::flag("raw")]
            ))
        );

        assert_eq!(
            apply(raw_call, "raw_call", "open Cargo.toml -r"),
            build(b::call(
                b::bare("open"),
                vec![b::sp(), b::bare("Cargo.toml"), b::sp(), b::shorthand("r")]
            ))
        );

        assert_eq!(
            apply(raw_call, "raw_call", "config --set tabs 2"),
            build(b::call(
                b::bare("config"),
                vec![
                    b::sp(),
                    b::flag("set"),
                    b::sp(),
                    b::bare("tabs"),
                    b::sp(),
                    b::int(2)
                ]
            ))
        );
    }

    #[test]
    fn test_smoke_pipeline() {
        let _ = pretty_env_logger::try_init();

        assert_eq!(
            apply(
                pipeline,
                "pipeline",
                r#"git branch --merged | split-row "`n" | where $it != "* master""#
            ),
            build_token(b::pipeline(vec![
                (
                    None,
                    b::call(
                        b::bare("git"),
                        vec![b::sp(), b::bare("branch"), b::sp(), b::flag("merged")]
                    ),
                    Some(" ")
                ),
                (
                    Some(" "),
                    b::call(b::bare("split-row"), vec![b::sp(), b::string("`n")]),
                    Some(" ")
                ),
                (
                    Some(" "),
                    b::call(
                        b::bare("where"),
                        vec![
                            b::sp(),
                            b::var("it"),
                            b::sp(),
                            b::op("!="),
                            b::sp(),
                            b::string("* master")
                        ]
                    ),
                    None
                )
            ]))
        );

        assert_eq!(
            apply(pipeline, "pipeline", "ls | where { $it.size > 100 }"),
            build_token(b::pipeline(vec![
                (None, b::call(b::bare("ls"), vec![]), Some(" ")),
                (
                    Some(" "),
                    b::call(
                        b::bare("where"),
                        vec![
                            b::sp(),
                            b::braced(vec![
                                b::path(b::var("it"), vec![b::member("size")]),
                                b::sp(),
                                b::op(">"),
                                b::sp(),
                                b::int(100)
                            ])
                        ]
                    ),
                    None
                )
            ]))
        )
    }

    fn apply<T>(
        f: impl Fn(NomSpan) -> Result<(NomSpan, T), nom::Err<(NomSpan, nom::error::ErrorKind)>>,
        desc: &str,
        string: &str,
    ) -> T {
        match f(NomSpan::new(string)) {
            Ok(v) => v.1,
            Err(other) => {
                println!("{:?}", other);
                println!("for {} @ {}", string, desc);
                panic!("No dice");
            }
        }
    }

    fn span(left: usize, right: usize) -> Span {
        Span::from((left, right))
    }

    fn delimited(
        delimiter: Delimiter,
        children: Vec<TokenNode>,
        left: usize,
        right: usize,
    ) -> TokenNode {
        let node = DelimitedNode::new(delimiter, children);
        let spanned = Tagged::from_simple_spanned_item(node, (left, right));
        TokenNode::Delimited(spanned)
    }

    fn path(head: TokenNode, tail: Vec<Token>, left: usize, right: usize) -> TokenNode {
        let node = PathNode::new(
            Box::new(head),
            tail.into_iter().map(TokenNode::Token).collect(),
        );
        let spanned = Tagged::from_simple_spanned_item(node, (left, right));
        TokenNode::Path(spanned)
    }

    fn leaf_token(token: RawToken, left: usize, right: usize) -> TokenNode {
        TokenNode::Token(Tagged::from_simple_spanned_item(token, (left, right)))
    }

    fn token(token: RawToken, left: usize, right: usize) -> TokenNode {
        TokenNode::Token(Tagged::from_simple_spanned_item(token, (left, right)))
    }

    fn build<T>(block: CurriedNode<T>) -> T {
        let mut builder = TokenTreeBuilder::new();
        block(&mut builder)
    }

    fn build_token(block: CurriedToken) -> TokenNode {
        let mut builder = TokenTreeBuilder::new();
        block(&mut builder)
    }
}