//! Pattern matching transpilation

#![allow(clippy::missing_errors_doc)]

#![allow(clippy::only_used_in_recursion)]

use super::Transpiler;

use crate::frontend::ast::{Expr, MatchArm, Pattern};

use anyhow::Result;

use proc_macro2::TokenStream;

use quote::{format_ident, quote};

impl Transpiler {

    /// Transpiles match expressions

    ///

    /// # Examples

    ///

    /// ```

    /// use ruchy::{Transpiler, Parser};

    /// 

    /// let transpiler = Transpiler::new();

    /// let mut parser = Parser::new(r#"match x { 1 => "one", _ => "other" }"#);

    /// let ast = parser.parse().unwrap();

    /// 

    /// let result = transpiler.transpile(&ast).unwrap();

    /// let code = result.to_string();

    /// assert!(code.contains("match"));

    /// assert!(code.contains("1 =>"));

    /// assert!(code.contains("_ =>"));

    /// ```

    pub fn transpile_match(&self, expr: &Expr, arms: &[MatchArm]) -> Result<TokenStream> {

        let expr_tokens = self.transpile_expr(expr)
?0
;

        let mut arm_tokens = Vec::new();

        for 
arm13
 in arms {

            let pattern_tokens = self.transpile_pattern(&arm.pattern)
?0
;

            let body_tokens = self.transpile_expr(&arm.body)
?0
;

            // Handle pattern guards if present

            if let Some(
guard_expr1
) = &arm.guard {

                let guard_tokens = self.transpile_expr(guard_expr)
?0
;

                arm_tokens.push(quote! {

                    #pattern_tokens if #guard_tokens => #body_tokens

                });

            } else {

                arm_tokens.push(quote! {

                    #pattern_tokens => #body_tokens

                });

            }

        }

        Ok(quote! {

            match #expr_tokens {

                #(#arm_tokens,)*

            }

        })

    }

    /// Transpiles patterns

    ///

    /// # Examples

    ///

    /// ```

    /// use ruchy::{Transpiler, Parser};

    /// 

    /// // Wildcard pattern

    /// let transpiler = Transpiler::new();

    /// let mut parser = Parser::new("match x { _ => 42 }");

    /// let ast = parser.parse().unwrap();

    /// let result = transpiler.transpile(&ast).unwrap();

    /// assert!(result.to_string().contains("_ =>"));

    /// ```

    ///

    /// ```

    /// use ruchy::{Transpiler, Parser};

    /// 

    /// // Identifier pattern

    /// let transpiler = Transpiler::new();

    /// let mut parser = Parser::new("match x { y => y + 1 }");

    /// let ast = parser.parse().unwrap();

    /// let result = transpiler.transpile(&ast).unwrap();

    /// assert!(result.to_string().contains("y =>"));

    /// ```

    ///

    /// ```

    /// use ruchy::{Transpiler, Parser};

    /// 

    /// // Tuple pattern

    /// let transpiler = Transpiler::new();

    /// let mut parser = Parser::new("match pair { (a, b) => a + b }");

    /// let ast = parser.parse().unwrap();

    /// let result = transpiler.transpile(&ast).unwrap();

    /// assert!(result.to_string().contains("("));

    /// assert!(result.to_string().contains("a"));

    /// assert!(result.to_string().contains("b"));

    /// ```

    pub fn transpile_pattern(&self, pattern: &Pattern) -> Result<TokenStream> {

        match pattern {

            Pattern::Wildcard => Ok(quote! { _ }),

            Pattern::Literal(lit) => Ok(Self::transpile_literal(lit)),

            Pattern::Identifier(name) => {

                let ident = format_ident!("{}", name);

                Ok(quote! { #ident })

            }

            Pattern::QualifiedName(path) => {

                // Generate qualified path like Ordering::Less

                let 
segments1
: 
Vec<_>1
 = 
path.iter()1
.
map1
(|s| format_ident!("{}", s)).
collect1
();

                Ok(quote! { #(#segments)::* })

            }

            Pattern::Tuple(patterns) => {

                let pattern_tokens: Result<Vec<_>> =

                    
patterns.iter()4
.
map4
(|p| self.transpile_pattern(p)).
collect4
();

                let pattern_tokens = pattern_tokens
?0
;

                Ok(quote! { (#(#pattern_tokens),*) })

            }

            Pattern::List(patterns) => {

                if patterns.is_empty() {

                    Ok(quote! { [] })

                } else {

                    // Check for rest pattern

                    let 
has_rest4
 = 
patterns.iter()4
.
any4
(|p| matches!(p, Pattern::Rest | Pattern::RestNamed(_)));

                    if has_rest {

                        // Handle patterns with rest

                        let mut pattern_tokens = Vec::new();

                        for 
p6
 in patterns {

                            match p {

                                Pattern::Rest => pattern_tokens.push(quote! { .. }),

                                Pattern::RestNamed(name) => {

                                    let name_ident = format_ident!("{}", name);

                                    pattern_tokens.push(quote! { ..#name_ident });

                                }

                                _ => pattern_tokens.push(self.transpile_pattern(p)
?0
),

                            }

                        }

                        Ok(quote! { [#(#pattern_tokens),*] })

                    } else {

                        // Simple list pattern

                        let pattern_tokens: Result<Vec<_>> =

                            
patterns.iter()1
.
map1
(|p| self.transpile_pattern(p)).
collect1
();

                        let pattern_tokens = pattern_tokens
?0
;

                        Ok(quote! { [#(#pattern_tokens),*] })

                    }

                }

            }

            Pattern::Struct { name, fields, has_rest } => {

                let struct_name = format_ident!("{}", name);

                if fields.is_empty() {

                    Ok(quote! { #struct_name {} })

                } else {

                    let field_patterns: Result<Vec<_>> = fields

                        .iter()

                        .
map3
(|field| {

                            let field_ident = format_ident!("{}", field.name);

                            if let Some(ref pattern) = field.pattern {

                                let pattern_tokens = self.transpile_pattern(pattern)
?0
;

                                Ok(quote! { #field_ident: #pattern_tokens })

                            } else {

                                // Shorthand field pattern

                                Ok(quote! { #field_ident })

                            }

                        })

                        .collect();

                    let field_patterns = field_patterns
?0
;

                    if *has_rest {

                        Ok(quote! { #struct_name { #(#field_patterns),*, .. } })

                    } else {

                        Ok(quote! { #struct_name { #(#field_patterns),* } })

                    }

                }

            }

            Pattern::Or(patterns) => {

                let pattern_tokens: Result<Vec<_>> =

                    
patterns.iter()2
.
map2
(|p| self.transpile_pattern(p)).
collect2
();

                let pattern_tokens = pattern_tokens
?0
;

                // Join patterns with |

                let mut result = TokenStream::new();

                for (i, tokens) in 
pattern_tokens.iter()2
.
enumerate2
() {

                    if i > 0 {

                        result.extend(quote! { | });

                    
}2

                    result.extend(tokens.clone());

                }

                Ok(result)

            }

            Pattern::Range {

                start,

                end,

                inclusive,

            } => {

                let start_tokens = self.transpile_pattern(start)
?0
;

                let end_tokens = self.transpile_pattern(end)
?0
;

                if *inclusive {

                    Ok(quote! { #start_tokens..=#end_tokens })

                } else {

                    Ok(quote! { #start_tokens..#end_tokens })

                }

            }

            Pattern::Rest => Ok(quote! { .. }),

            Pattern::RestNamed(name) => {

                let name_ident = format_ident!("{}", name);

                Ok(quote! { ..#name_ident })

            }

            Pattern::Ok(pattern) => {

                let inner = self.transpile_pattern(pattern)?;

                Ok(quote! { Ok(#inner) })

            }

            Pattern::Err(pattern) => {

                let inner = self.transpile_pattern(pattern)?;

                Ok(quote! { Err(#inner) })

            }

            Pattern::Some(pattern) => {

                let inner = self.transpile_pattern(pattern)?;

                Ok(quote! { Some(#inner) })

            }

            Pattern::None => Ok(quote! { None }),

        }

    }

}

#[cfg(test)]

#[path = "patterns_tests.rs"]

mod tests;