use nom::{
    bytes::complete::{tag, take_while1},
    combinator::map,
    sequence::tuple,
    IResult,
};

use super::{
    combinator::{
        blank_lines, colon_token, l_bracket_token, r_bracket_token, trim_line_end, GreenElement,
        NodeBuilder,
    },
    input::Input,
    keyword::affiliated_keyword_nodes,
    SyntaxKind,
};

#[cfg_attr(
  feature = "tracing",
  tracing::instrument(level = "debug", skip(input), fields(input = input.s))
)]
pub fn fn_def_node(input: Input) -> IResult<Input, GreenElement, ()> {
    let mut parser = map(
        tuple((
            affiliated_keyword_nodes,
            l_bracket_token,
            tag("fn"),
            colon_token,
            take_while1(|c: char| c.is_ascii_alphanumeric() || c == '-' || c == '_'),
            r_bracket_token,
            trim_line_end,
            blank_lines,
        )),
        |(
            affiliated_keywords,
            l_bracket,
            fn_,
            colon,
            label,
            r_bracket,
            (content, ws_, nl),
            post_blank,
        )| {
            let mut b = NodeBuilder::new();
            b.children.extend(affiliated_keywords);
            b.push(l_bracket);
            b.text(fn_);
            b.push(colon);
            b.text(label);
            b.push(r_bracket);
            b.text(content);
            b.ws(ws_);
            b.nl(nl);
            b.children.extend(post_blank);
            b.finish(SyntaxKind::FN_DEF)
        },
    );
    crate::lossless_parser!(parser, input)
}

#[test]
fn parse() {
    use crate::ParseConfig;
    use crate::{ast::FnDef, tests::to_ast};

    let to_fn_def = to_ast::<FnDef>(fn_def_node);

    insta::assert_debug_snapshot!(
         to_fn_def("[fn:1] https://orgmode.org").syntax,
         @r###"
    FN_DEF@0..26
      L_BRACKET@0..1 "["
      TEXT@1..3 "fn"
      COLON@3..4 ":"
      TEXT@4..5 "1"
      R_BRACKET@5..6 "]"
      TEXT@6..26 " https://orgmode.org"
    "###
    );

    insta::assert_debug_snapshot!(
         to_fn_def("[fn:word_1] https://orgmode.org").syntax,
         @r###"
    FN_DEF@0..31
      L_BRACKET@0..1 "["
      TEXT@1..3 "fn"
      COLON@3..4 ":"
      TEXT@4..10 "word_1"
      R_BRACKET@10..11 "]"
      TEXT@11..31 " https://orgmode.org"
    "###
    );

    insta::assert_debug_snapshot!(
         to_fn_def("[fn:WORD-1] https://orgmode.org").syntax,
         @r###"
    FN_DEF@0..31
      L_BRACKET@0..1 "["
      TEXT@1..3 "fn"
      COLON@3..4 ":"
      TEXT@4..10 "WORD-1"
      R_BRACKET@10..11 "]"
      TEXT@11..31 " https://orgmode.org"
    "###
    );

    insta::assert_debug_snapshot!(
         to_fn_def("[fn:WORD]").syntax,
         @r###"
    FN_DEF@0..9
      L_BRACKET@0..1 "["
      TEXT@1..3 "fn"
      COLON@3..4 ":"
      TEXT@4..8 "WORD"
      R_BRACKET@8..9 "]"
    "###
    );

    insta::assert_debug_snapshot!(
         to_fn_def("[fn:1] In particular, the parser requires stars at column 0 to be\n").syntax,
         @r###"
    FN_DEF@0..66
      L_BRACKET@0..1 "["
      TEXT@1..3 "fn"
      COLON@3..4 ":"
      TEXT@4..5 "1"
      R_BRACKET@5..6 "]"
      TEXT@6..65 " In particular, the p ..."
      NEW_LINE@65..66 "\n"
    "###
    );

    let config = &ParseConfig::default();

    assert!(fn_def_node(("[fn:] https://orgmode.org", config).into()).is_err());
    assert!(fn_def_node(("[fn:wor d] https://orgmode.org", config).into()).is_err());
    assert!(fn_def_node(("[fn:WORD https://orgmode.org", config).into()).is_err());

    insta::assert_debug_snapshot!(
         to_fn_def("#+ATTR_poi: 1\n[fn:WORD-1] https://orgmode.org").syntax,
         @r###"
    FN_DEF@0..45
      AFFILIATED_KEYWORD@0..14
        HASH_PLUS@0..2 "#+"
        TEXT@2..10 "ATTR_poi"
        COLON@10..11 ":"
        TEXT@11..13 " 1"
        NEW_LINE@13..14 "\n"
      L_BRACKET@14..15 "["
      TEXT@15..17 "fn"
      COLON@17..18 ":"
      TEXT@18..24 "WORD-1"
      R_BRACKET@24..25 "]"
      TEXT@25..45 " https://orgmode.org"
    "###
    );
}