#![allow(dead_code)]

mod to_tokens;
mod token_stream;

pub mod runtime;
pub use to_tokens::*;
pub use token_stream::*;

/// The whole point.
///
/// Performs variable interpolation against the input and produces it as
/// [`TokenStream`].
///
/// # Interpolation
///
/// Variable interpolation is done with `#var` (similar to `$var` in
/// `macro_rules!` macros). This grabs the `var` variable that is currently in
/// scope and inserts it in that location in the output tokens. Any type
/// implementing the [`ToTokens`] trait can be interpolated. This includes most
/// Rust primitive types.
///
/// [`ToTokens`]: trait.ToTokens.html
///
/// Repetition is done using `#(...)*` or `#(...),*` again similar to
/// `macro_rules!`. This iterates through the elements of any variable
/// interpolated within the repetition and inserts a copy of the repetition body
/// for each one. The variables in an interpolation may be a `Vec`, slice,
/// `BTreeSet`, or any `Iterator`.
///
/// - `#(#var)*` — no separators
/// - `#(#var),*` — the character before the asterisk is used as a separator
/// - `#( struct #var; )*` — the repetition can contain other tokens
/// - `#( #k => println!("{}", #v), )*` — even multiple interpolations
#[macro_export]
#[doc(hidden)]
macro_rules! quote {
    () => {
        $crate::tokens::TokenStream::new()
    };
    ($($tt:tt)*) => {{
        let mut _s = $crate::tokens::TokenStream::new();
        $crate::quote_each_token!(_s $($tt)*);
        _s
    }};
}

pub use quote;

/// Formatting macro for constructing a `TokenStream`.
///
/// <br>
///
/// # Syntax
///
/// Syntax is copied from the [`format!`] macro, supporting both positional and
/// named arguments.
#[macro_export]
#[doc(hidden)]
macro_rules!  format_token {
    ($($fmt:tt)*) => {
        $crate::TokenStream::from(format!($($fmt)*))
    };
}

// Extract the names of all #metavariables and pass them to the $call macro.
//
// in:   pounded_var_names!(then!(...) a #b c #( #d )* #e)
// out:  then!(... b);
//       then!(... d);
//       then!(... e);
#[macro_export]
#[doc(hidden)]
macro_rules! pounded_var_names {
    ($call:ident! $extra:tt $($tts:tt)*) => {
        $crate::pounded_var_names_with_context!($call! $extra
            (@ $($tts)*)
            ($($tts)* @)
        )
    };
}

#[macro_export]
#[doc(hidden)]
macro_rules! pounded_var_names_with_context {
    ($call:ident! $extra:tt ($($b1:tt)*) ($($curr:tt)*)) => {
        $(
            $crate::pounded_var_with_context!($call! $extra $b1 $curr);
        )*
    };
}

#[macro_export]
#[doc(hidden)]
macro_rules! pounded_var_with_context {
    ($call:ident! $extra:tt $b1:tt ( $($inner:tt)* )) => {
        $crate::pounded_var_names!($call! $extra $($inner)*);
    };

    ($call:ident! $extra:tt $b1:tt [ $($inner:tt)* ]) => {
        $crate::pounded_var_names!($call! $extra $($inner)*);
    };

    ($call:ident! $extra:tt $b1:tt { $($inner:tt)* }) => {
        $crate::pounded_var_names!($call! $extra $($inner)*);
    };

    ($call:ident!($($extra:tt)*) # $var:ident) => {
        $crate::$call!($($extra)* $var);
    };

    ($call:ident! $extra:tt $b1:tt $curr:tt) => {};
}

#[macro_export]
#[doc(hidden)]
macro_rules! quote_bind_into_iter {
    ($has_iter:ident $var:ident) => {
        // `mut` may be unused if $var occurs multiple times in the list.
        #[allow(unused_mut)]
        let (mut $var, i) = $var.quote_into_iter();
        let $has_iter = $has_iter | i;
    };
}

#[macro_export]
#[doc(hidden)]
macro_rules! quote_bind_next_or_break {
    ($var:ident) => {
        let $var = match $var.next() {
            Some(_x) => $crate::tokens::runtime::RepInterp(_x),
            None => break,
        };
    };
}

#[macro_export]
#[doc(hidden)]
macro_rules! quote_each_token {
    ($tokens:ident $($tts:tt)*) => {
        $crate::quote_tokens_with_context!($tokens
            (@ @ @ @ @ @ $($tts)*)
            (@ @ @ @ @ $($tts)* @)
            (@ @ @ @ $($tts)* @ @)
            (@ @ @ $(($tts))* @ @ @)
            (@ @ $($tts)* @ @ @ @)
            (@ $($tts)* @ @ @ @ @)
            ($($tts)* @ @ @ @ @ @)
        );
    };
}

#[macro_export]
#[doc(hidden)]
macro_rules! quote_tokens_with_context {
    ($tokens:ident
        ($($b3:tt)*) ($($b2:tt)*) ($($b1:tt)*)
        ($($curr:tt)*)
        ($($a1:tt)*) ($($a2:tt)*) ($($a3:tt)*)
    ) => {
        $(
            $crate::quote_token_with_context!($tokens $b3 $b2 $b1 $curr $a1 $a2 $a3);
        )*
    };
}

#[macro_export]
#[doc(hidden)]
macro_rules! quote_token_with_context {
    ($tokens:ident $b3:tt $b2:tt $b1:tt @ $a1:tt $a2:tt $a3:tt) => {};

    ($tokens:ident $b3:tt $b2:tt $b1:tt (#) ( $($inner:tt)* ) * $a3:tt) => {{
        use $crate::tokens::runtime::ext::*;
        let has_iter = $crate::tokens::runtime::ThereIsNoIteratorInRepetition;
        $crate::pounded_var_names!(quote_bind_into_iter!(has_iter) () $($inner)*);
        let _: $crate::tokens::runtime::HasIterator = has_iter;
        // This is `while true` instead of `loop` because if there are no
        // iterators used inside of this repetition then the body would not
        // contain any `break`, so the compiler would emit unreachable code
        // warnings on anything below the loop. We use has_iter to detect and
        // fail to compile when there are no iterators, so here we just work
        // around the unneeded extra warning.
        while true {
            $crate::pounded_var_names!(quote_bind_next_or_break!() () $($inner)*);
            $crate::quote_each_token!($tokens $($inner)*);
        }
    }};
    ($tokens:ident $b3:tt $b2:tt # (( $($inner:tt)* )) * $a2:tt $a3:tt) => {};
    ($tokens:ident $b3:tt # ( $($inner:tt)* ) (*) $a1:tt $a2:tt $a3:tt) => {};

    ($tokens:ident $b3:tt $b2:tt $b1:tt (#) ( $($inner:tt)* ) $sep:tt *) => {{
        use $crate::tokens::runtime::ext::*;
        let mut _i = 0usize;
        let has_iter = $crate::tokens::runtime::ThereIsNoIteratorInRepetition;
        $crate::pounded_var_names!(quote_bind_into_iter!(has_iter) () $($inner)*);
        let _: $crate::tokens::runtime::HasIterator = has_iter;
        while true {
            $crate::pounded_var_names!(quote_bind_next_or_break!() () $($inner)*);
            if _i > 0 {
                $crate::quote_token!($tokens $sep);
            }
            _i += 1;
            $crate::quote_each_token!($tokens $($inner)*);
        }
    }};
    ($tokens:ident $b3:tt $b2:tt # (( $($inner:tt)* )) $sep:tt * $a3:tt) => {};
    ($tokens:ident $b3:tt # ( $($inner:tt)* ) ($sep:tt) * $a2:tt $a3:tt) => {};
    ($tokens:ident # ( $($inner:tt)* ) * (*) $a1:tt $a2:tt $a3:tt) => {
        // https://github.com/dtolnay/quote/issues/130
        $crate::quote_token!($tokens *);
    };
    ($tokens:ident # ( $($inner:tt)* ) $sep:tt (*) $a1:tt $a2:tt $a3:tt) => {};

    ($tokens:ident $b3:tt $b2:tt $b1:tt (#) $var:ident $a2:tt $a3:tt) => {
        $crate::tokens::ToTokens::to_tokens(&$var, &mut $tokens);
    };
    ($tokens:ident $b3:tt $b2:tt # ($var:ident) $a1:tt $a2:tt $a3:tt) => {};
    ($tokens:ident $b3:tt $b2:tt $b1:tt ($curr:tt) $a1:tt $a2:tt $a3:tt) => {
        $crate::quote_token!($tokens $curr);
    };
}

#[macro_export]
#[doc(hidden)]
macro_rules! quote_token {
    ($tokens:ident ( $($inner:tt)* )) => {
        $crate::tokens::runtime::push_group(
            &mut $tokens,
            $crate::tokens::Delimiter::Parenthesis,
            $crate::quote!($($inner)*),
        );
    };

    ($tokens:ident [ $($inner:tt)* ]) => {
        $crate::tokens::runtime::push_group(
            &mut $tokens,
            $crate::tokens::Delimiter::Bracket,
            $crate::quote!($($inner)*),
        );
    };

    ($tokens:ident { $($inner:tt)* }) => {
        $crate::tokens::runtime::push_group(
            &mut $tokens,
            $crate::tokens::Delimiter::Brace,
            $crate::quote!($($inner)*),
        );
    };

    ($tokens:ident +) => {
        $crate::tokens::runtime::push_add(&mut $tokens);
    };

    ($tokens:ident +=) => {
        $crate::tokens::runtime::push_add_eq(&mut $tokens);
    };

    ($tokens:ident &) => {
        $crate::tokens::runtime::push_and(&mut $tokens);
    };

    ($tokens:ident &&) => {
        $crate::tokens::runtime::push_and_and(&mut $tokens);
    };

    ($tokens:ident &=) => {
        $crate::tokens::runtime::push_and_eq(&mut $tokens);
    };

    ($tokens:ident @) => {
        $crate::tokens::runtime::push_at(&mut $tokens);
    };

    ($tokens:ident !) => {
        $crate::tokens::runtime::push_bang(&mut $tokens);
    };

    ($tokens:ident ^) => {
        $crate::tokens::runtime::push_caret(&mut $tokens);
    };

    ($tokens:ident ^=) => {
        $crate::tokens::runtime::push_caret_eq(&mut $tokens);
    };

    ($tokens:ident :) => {
        $crate::tokens::runtime::push_colon(&mut $tokens);
    };

    ($tokens:ident ::) => {
        $crate::tokens::runtime::push_colon2(&mut $tokens);
    };

    ($tokens:ident ,) => {
        $crate::tokens::runtime::push_comma(&mut $tokens);
    };

    ($tokens:ident /) => {
        $crate::tokens::runtime::push_div(&mut $tokens);
    };

    ($tokens:ident /=) => {
        $crate::tokens::runtime::push_div_eq(&mut $tokens);
    };

    ($tokens:ident .) => {
        $crate::tokens::runtime::push_dot(&mut $tokens);
    };

    ($tokens:ident ..) => {
        $crate::tokens::runtime::push_dot2(&mut $tokens);
    };

    ($tokens:ident ...) => {
        $crate::tokens::runtime::push_dot3(&mut $tokens);
    };

    ($tokens:ident ..=) => {
        $crate::tokens::runtime::push_dot_dot_eq(&mut $tokens);
    };

    ($tokens:ident =) => {
        $crate::tokens::runtime::push_eq(&mut $tokens);
    };

    ($tokens:ident ==) => {
        $crate::tokens::runtime::push_eq_eq(&mut $tokens);
    };

    ($tokens:ident >=) => {
        $crate::tokens::runtime::push_ge(&mut $tokens);
    };

    ($tokens:ident >) => {
        $crate::tokens::runtime::push_gt(&mut $tokens);
    };

    ($tokens:ident <=) => {
        $crate::tokens::runtime::push_le(&mut $tokens);
    };

    ($tokens:ident <) => {
        $crate::tokens::runtime::push_lt(&mut $tokens);
    };

    ($tokens:ident *=) => {
        $crate::tokens::runtime::push_mul_eq(&mut $tokens);
    };

    ($tokens:ident !=) => {
        $crate::tokens::runtime::push_ne(&mut $tokens);
    };

    ($tokens:ident |) => {
        $crate::tokens::runtime::push_or(&mut $tokens);
    };

    ($tokens:ident |=) => {
        $crate::tokens::runtime::push_or_eq(&mut $tokens);
    };

    ($tokens:ident ||) => {
        $crate::tokens::runtime::push_or_or(&mut $tokens);
    };

    ($tokens:ident #) => {
        $crate::tokens::runtime::push_pound(&mut $tokens);
    };

    ($tokens:ident ?) => {
        $crate::tokens::runtime::push_question(&mut $tokens);
    };

    ($tokens:ident ->) => {
        $crate::tokens::runtime::push_rarrow(&mut $tokens);
    };

    ($tokens:ident <-) => {
        $crate::tokens::runtime::push_larrow(&mut $tokens);
    };

    ($tokens:ident %) => {
        $crate::tokens::runtime::push_rem(&mut $tokens);
    };

    ($tokens:ident %=) => {
        $crate::tokens::runtime::push_rem_eq(&mut $tokens);
    };

    ($tokens:ident =>) => {
        $crate::tokens::runtime::push_fat_arrow(&mut $tokens);
    };

    ($tokens:ident ;) => {
        $crate::tokens::runtime::push_semi(&mut $tokens);
    };

    ($tokens:ident <<) => {
        $crate::tokens::runtime::push_shl(&mut $tokens);
    };

    ($tokens:ident <<=) => {
        $crate::tokens::runtime::push_shl_eq(&mut $tokens);
    };

    ($tokens:ident >>) => {
        $crate::tokens::runtime::push_shr(&mut $tokens);
    };

    ($tokens:ident >>=) => {
        $crate::tokens::runtime::push_shr_eq(&mut $tokens);
    };

    ($tokens:ident *) => {
        $crate::tokens::runtime::push_star(&mut $tokens);
    };

    ($tokens:ident -) => {
        $crate::tokens::runtime::push_sub(&mut $tokens);
    };

    ($tokens:ident -=) => {
        $crate::tokens::runtime::push_sub_eq(&mut $tokens);
    };

    ($tokens:ident $ident:ident) => {
        $crate::tokens::runtime::push_ident(&mut $tokens, stringify!($ident));
    };

    ($tokens:ident $other:tt) => {
        $crate::tokens::runtime::parse(&mut $tokens, stringify!($other));
    };
}

pub fn to_ident(name: &str) -> TokenStream {
    // keywords list based on https://doc.rust-lang.org/reference/keywords.html
    match name {
        "abstract" | "as" | "become" | "box" | "break" | "const" | "continue" | "crate" | "do" | "else" | "enum" | "extern" | "false" | "final" | "fn" | "for" | "if" | "impl" | "in" | "let" | "loop" | "macro" | "match" | "mod" | "move" | "mut" | "override" | "priv" | "pub" | "ref" | "return" | "static" | "struct" | "super" | "trait" | "true" | "type" | "typeof" | "unsafe" | "unsized" | "use" | "virtual" | "where" | "while" | "yield" | "try" | "async" | "await" | "dyn" => format!("r#{name}").into(),
        "Self" | "self" => format!("{name}_").into(),
        "_" => "unused".into(),
        _ => crate::trim_tick(name).into(),
    }
}