rbatis-macro-driver 4.8.0

use quote::quote;
use quote::ToTokens;
use syn::{FnArg, ItemFn, Pat, ReturnType};

//find and check method return type
pub(crate) fn find_return_type(target_fn: &ItemFn) -> proc_macro2::TokenStream {
    let mut return_ty = target_fn.sig.output.to_token_stream();
    match &target_fn.sig.output {
        ReturnType::Type(_, b) => {
            return_ty = b.to_token_stream();
        }
        _ => {}
    }
    let s = format!("{}", return_ty);
    if !s.contains(":: Result") && !s.starts_with("Result") {
        return_ty = quote! {
             rbatis::Result <#return_ty>
        };
    }
    return_ty
}

pub(crate) fn get_fn_args(target_fn: &ItemFn) -> Vec<Box<Pat>> {
    let mut fn_arg_name_vec = vec![];
    for arg in &target_fn.sig.inputs {
        match arg {
            FnArg::Typed(t) => {
                fn_arg_name_vec.push(t.pat.clone());
                //println!("arg_name {}", arg_name);
            }
            _ => {}
        }
    }
    fn_arg_name_vec
}

//find and check method return type
pub(crate) fn find_fn_body(target_fn: &ItemFn) -> proc_macro2::TokenStream {
    //del todos
    let mut target_fn = target_fn.clone();
    let mut new_stmts = vec![];
    for x in &target_fn.block.stmts {
        let token = x
            .to_token_stream()
            .to_string()
            .replace("\n", "")
            .replace(" ", "");
        if token.eq("todo!()") || token.eq("unimplemented!()") || token.contains("impled!()") {
            //nothing to do
        } else {
            new_stmts.push(x.to_owned());
        }
    }
    target_fn.block.stmts = new_stmts;
    target_fn.block.to_token_stream()
}

pub(crate) fn is_query(return_source: &str) -> bool {
    let is_select = !return_source.contains("ExecResult");
    return is_select;
}

pub(crate) fn is_rb_ref(ty_stream: &str) -> bool {
    if ty_stream.contains("RBatis")
        || ty_stream.contains("RBatisConnExecutor")
        || ty_stream.contains("RBatisTxExecutor")
        || ty_stream.contains("RBatisTxExecutorGuard")
        || ty_stream.contains("Executor")
    {
        return true;
    }
    false
}

/// Check if the return type is Page<T> or Result<Page<T>>
/// Returns Some(inner_type) if it is, None otherwise
pub(crate) fn is_page_return_type(return_str: &str) -> Option<String> {
    let s = return_str.replace(" ", "");

    // Handle Result<Page<T>> or rbatis::Result<Page<T>>
    if s.contains("Result<Page<") || s.contains("Result<Page<") {
        if let Some(start) = s.find("Page<") {
            let inner = extract_type_after(&s, start + 5)?;
            return Some(inner);
        }
    }

    // Handle Page<T> or rbatis::plugin::Page<T>
    if s.contains("Page<") || s.contains("::Page<") {
        if let Some(start) = s.find("Page<") {
            let inner = extract_type_after(&s, start + 5)?;
            return Some(inner);
        }
    }

    None
}

/// Extract type content between angle brackets
/// e.g. "Activity>" -> "Activity", "Activity," -> "Activity"
fn extract_type_after(s: &str, start: usize) -> Option<String> {
    let mut depth = 0;
    let mut end = start;

    for (i, c) in s[start..].chars().enumerate() {
        match c {
            '<' => depth += 1,
            '>' => {
                if depth == 0 {
                    end = start + i;
                    break;
                }
                depth -= 1;
            }
            ',' if depth == 0 => {
                end = start + i;
                break;
            }
            _ => {}
        }
    }

    if end > start {
        let inner = s[start..end].to_string();
        // Handle nested generics like Vec<T>
        if inner.contains('<') {
            Some(inner)
        } else {
            Some(inner.trim().to_string())
        }
    } else {
        None
    }
}

/// Check if a type string is IPageRequest, PageRequest, or a reference to them
pub(crate) fn is_page_request(ty_stream: &str) -> bool {
    let s = ty_stream.replace(" ", "");
    s.contains("IPageRequest") || s.contains("PageRequest")
}