virtue-next 0.1.3

use crate::error::Error;
use crate::prelude::Delimiter;
use crate::prelude::Group;
use crate::prelude::Ident;
use crate::prelude::Punct;
use crate::prelude::TokenTree;
use std::iter::Peekable;
/// Documentation for `assume_group`.
///
/// # Panics
///
/// Panics if internal invariants are violated.

pub fn assume_group(t: Option<TokenTree>) -> Group {
    match t {
        | Some(TokenTree::Group(group)) => group,
        | _ => unreachable!(),
    }
}
/// Documentation for `assume_ident`.
///
/// # Panics
///
/// Panics if internal invariants are violated.
pub fn assume_ident(t: Option<TokenTree>) -> Ident {
    match t {
        | Some(TokenTree::Ident(ident)) => ident,
        | _ => unreachable!(),
    }
}
/// Documentation for `assume_punct`.
///
/// # Panics
///
/// Panics if internal invariants are violated.
pub fn assume_punct(
    t: Option<TokenTree>,
    punct: char,
) -> Punct {
    match t {
        | Some(TokenTree::Punct(p)) => {
            debug_assert_eq!(punct, p.as_char());
            p
        },
        | _ => unreachable!(),
    }
}

pub fn consume_ident(input: &mut Peekable<impl Iterator<Item = TokenTree>>) -> Option<Ident> {
    match input.peek() {
        | Some(TokenTree::Ident(_)) => Some(super::utils::assume_ident(input.next())),
        | Some(TokenTree::Group(group)) => {
            // When calling from a macro_rules!, sometimes an ident is defined as :
            // Group { delimiter: None, stream: TokenStream [Ident] }
            let mut stream = group.stream().into_iter();
            if let Some(TokenTree::Ident(i)) = stream.next()
                && stream.next().is_none()
            {
                let _ = input.next();
                return Some(i);
            }
            None
        },
        | _ => None,
    }
}
/// Documentation for `consume_punct_if`.
///
/// # Panics
///
/// Panics if internal invariants are violated.

pub fn consume_punct_if(
    input: &mut Peekable<impl Iterator<Item = TokenTree>>,
    punct: char,
) -> Option<Punct> {
    if let Some(TokenTree::Punct(p)) = input.peek()
        && p.as_char() == punct
    {
        let next_token = input.next();
        match next_token {
            | Some(TokenTree::Punct(p)) => return Some(p),
            | _ => unreachable!(),
        }
    }
    None
}

#[cfg(any(test, feature = "proc-macro2"))]
pub fn ident_eq(
    ident: &Ident,
    text: &str,
) -> bool {
    ident == text
}

#[cfg(not(any(test, feature = "proc-macro2")))]
pub fn ident_eq(
    ident: &Ident,
    text: &str,
) -> bool {
    ident.to_string() == text
}

fn check_if_arrow(
    tokens: &[TokenTree],
    punct: &Punct,
) -> bool {
    if punct.as_char() == '>'
        && let Some(TokenTree::Punct(previous_punct)) = tokens.last()
        && previous_punct.as_char() == '-'
    {
        return true;
    }
    false
}

const OPEN_BRACKETS: &[char] = &['<', '(', '[', '{'];
const CLOSING_BRACKETS: &[char] = &['>', ')', ']', '}'];
const BRACKET_DELIMITER: &[Option<Delimiter>] = &[
    None,
    Some(Delimiter::Parenthesis),
    Some(Delimiter::Bracket),
    Some(Delimiter::Brace),
];

/// Documentation for `read_tokens_until_punct`.
///
/// # Errors
///
/// Returns an error if parsing fails.
#[allow(clippy::useless_let_if_seq)]
#[allow(clippy::manual_let_else)]
pub fn read_tokens_until_punct(
    input: &mut Peekable<impl Iterator<Item = TokenTree>>,
    expected_puncts: &[char],
) -> Result<Vec<TokenTree>, Error> {
    let mut result = Vec::new();
    let mut open_brackets = Vec::<char>::new();
    'outer: loop {
        match input.peek() {
            | Some(TokenTree::Punct(punct)) => {
                if check_if_arrow(&result, punct) {
                    // do nothing
                } else if OPEN_BRACKETS.contains(&punct.as_char()) {
                    open_brackets.push(punct.as_char());
                } else if let Some(index) =
                    CLOSING_BRACKETS.iter().position(|c| c == &punct.as_char())
                {
                    let last_bracket = if let Some(bracket) = open_brackets.pop() {
                        bracket
                    } else {
                        if expected_puncts.contains(&punct.as_char()) {
                            break;
                        }
                        return Err(Error::InvalidRustSyntax {
                            span: punct.span(),
                            expected: format!(
                                "one of {:?}, got '{}'",
                                expected_puncts,
                                punct.as_char()
                            ),
                        });
                    };
                    let expected = OPEN_BRACKETS[index];
                    assert_eq!(
                        expected,
                        last_bracket,
                        "Unexpected closing bracket: found {}, expected {}",
                        punct.as_char(),
                        expected
                    );
                } else if expected_puncts.contains(&punct.as_char()) && open_brackets.is_empty() {
                    break;
                }
                result.push(input.next().unwrap());
            },
            | Some(TokenTree::Group(g)) if open_brackets.is_empty() => {
                for punct in expected_puncts {
                    if let Some(idx) = OPEN_BRACKETS.iter().position(|c| c == punct)
                        && let Some(delim) = BRACKET_DELIMITER[idx]
                        && delim == g.delimiter()
                    {
                        // we need to split on this delimiter
                        break 'outer;
                    }
                }
                result.push(input.next().unwrap());
            },
            | Some(_) => result.push(input.next().unwrap()),
            | None => {
                break;
            },
        }
    }
    Ok(result)
}