feignhttp-codegen 0.6.0

use crate::enu::ArgType;
use crate::func::FnArg;
use proc_macro::TokenStream;
use quote::{ToTokens, quote};
use std::collections::HashMap;
use std::str::FromStr;
use syn::{Attribute, Field, Lit, PatType, Token, Type, parse::Parse};

/// Parse url and return url token stream.
/// A URL can be an expression.
pub fn parse_url_stream(attr: &TokenStream) -> syn::Result<proc_macro2::TokenStream> {
    let attr_str = attr.to_string();
    if attr_str.is_empty() {
        return Err(syn::Error::new(
            proc_macro2::Span::call_site(),
            "no metadata assign",
        ));
    }

    let attrs = attr_str.split(",");
    let exprs: Vec<&str> = attrs.into_iter().map(|u| u).collect();
    // The default starting expression is the URL.
    if let Some(expr_str) = exprs.first() {
        let expr_str = expr_str.trim();
        let url_expr = syn::parse::<syn::Expr>(TokenStream::from_str(expr_str).unwrap())?;
        let url_stream = parse_url(&url_expr)?;

        // Skip the url.
        for i in 1..exprs.len() {
            let exp_str = exprs[i].trim();
            let url_expr = syn::parse::<syn::Expr>(TokenStream::from_str(exp_str).unwrap())?;

            match url_expr {
                // An assignment path: path = xxx.
                syn::Expr::Assign(assign) => {
                    let left = &assign.left;
                    if let syn::Expr::Path(ref k) = **left {
                        let key = k.path.segments.last().unwrap().ident.to_string();
                        if key == "path" {
                            let right = &assign.right;
                            // Url + path.
                            return Ok(quote!(#url_stream .to_string() + #right));
                        }
                    }
                }
                _ => {}
            }
        }
        return Ok(quote!(#url_stream .to_string()));
    }
    Err(syn::Error::new(
        proc_macro2::Span::call_site(),
        "no metadata assign",
    ))
}

/// Parse and validate the url.
pub fn parse_url(url_expr: &syn::Expr) -> syn::Result<proc_macro2::TokenStream> {
    return match url_expr {
        // An assignment url: url = xxx.
        syn::Expr::Assign(assign) => {
            let left = &assign.left;
            if let syn::Expr::Path(ref k) = **left {
                let key = k.path.segments.last().unwrap().ident.to_string();
                if key != "url" {
                    return Err(syn::Error::new(
                        proc_macro2::Span::call_site(),
                        "metadata url not specified",
                    ));
                }
            }
            let right = &assign.right;
            Ok(right.to_token_stream())
        }
        // A literal url: `"http://xxx"`.
        syn::Expr::Lit(lit) => Ok(lit.to_token_stream()),
        // A variable url: URL.
        syn::Expr::Path(path) => Ok(path.to_token_stream()),
        syn::Expr::Field(field) => Ok(field.to_token_stream()),
        _ => Err(syn::Error::new(
            proc_macro2::Span::call_site(),
            "metadata url is invalid",
        )),
    };
}

pub fn remove_url_attr(attr: &str) -> String {
    let attrs = attr.split(",");
    let mut exprs: Vec<&str> = attrs.into_iter().map(|u| u).collect();
    if exprs.len() <= 0 {
        return "".into();
    }
    exprs.remove(0);
    return exprs.join(",");
}

#[allow(dead_code)]
struct Metas(Vec<Meta>);

impl Parse for Metas {
    fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
        let mut out = Vec::new();

        loop {
            if input.is_empty() {
                break;
            }

            out.push(input.parse()?);
        }
        Ok(Self(out))
    }
}
struct Meta(String, String);
impl Parse for Meta {
    fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
        let k: syn::Ident = input.parse()?;
        input.parse::<Token![=]>()?;
        let v: syn::Expr = input.parse()?;
        input.parse::<Option<Token![,]>>()?;
        Ok(Self(k.to_string(), expr_to_string(v)))
    }
}

fn expr_to_string(exp: syn::Expr) -> String {
    match exp {
        syn::Expr::Lit(l) => return lit_to_string(l.lit),
        _ => {}
    }
    exp.into_token_stream().to_string()
}

fn lit_to_string(lit: Lit) -> String {
    match lit {
        Lit::Str(s) => s.value(),
        Lit::Int(n) => n.to_string(),
        Lit::Float(f) => f.to_string(),
        Lit::Bool(b) => b.value().to_string(),
        Lit::Verbatim(v) => v.to_string(),
        _ => String::new(),
    }
}

pub fn _parse_exprs_attribute(att: &Attribute) -> syn::Result<HashMap<String, String>> {
    let metas = att.parse_args::<Metas>()?;

    let map: HashMap<String, String> = metas.0.into_iter().map(|x| (x.0, x.1)).collect();
    Ok(map)
}

/// Parse assignment expression like n1=v1, n2=v2, etc.
pub fn parse_exprs(attr_str: &str) -> HashMap<String, String> {
    let mut expr_map = HashMap::new();
    if attr_str == "" {
        return expr_map;
    }

    let attrs = attr_str.split(",");
    let exprs: Vec<&str> = attrs.into_iter().map(|u| u).collect();
    for exp_str in exprs.into_iter() {
        let expr = match syn::parse_str::<Meta>(exp_str) {
            Ok(x) => x,
            Err(_) => return HashMap::new(),
        };
        expr_map.insert(expr.0, expr.1);
    }

    expr_map
}

pub fn parse_args_from_sig(sig: &mut syn::Signature) -> syn::Result<Vec<FnArg>> {
    let iter = sig
        .inputs
        .iter_mut()
        .flat_map(|x| match x {
            syn::FnArg::Typed(y) => Some(y),
            _ => None,
        })
        .map(|x| x.into());
    parse_args(iter, Some(ArgType::QUERY))
}

pub fn parse_args_from_struct(item_struct: &mut syn::DataStruct) -> syn::Result<Vec<FnArg>> {
    let iter = item_struct.fields.iter_mut().map::<PType, _>(|x| x.into());
    parse_args(iter, None)
}

struct PType<'a> {
    ty: &'a Type,
    name: String,
    attrs: &'a mut Vec<Attribute>,
}

impl<'a> From<&'a mut PatType> for PType<'a> {
    fn from(this: &'a mut PatType) -> Self {
        Self {
            ty: &this.ty,
            name: this.pat.to_token_stream().to_string(),
            attrs: &mut this.attrs,
        }
    }
}
impl<'a> From<&'a mut Field> for PType<'a> {
    fn from(this: &'a mut Field) -> Self {
        Self {
            ty: &this.ty,
            name: this.ident.to_token_stream().to_string(),
            attrs: &mut this.attrs,
        }
    }
}

fn extract_name(attr: &Attribute) -> Option<String> {
    let metas = parse_attr_metas(attr)?;
    let nested_meta = metas.first()?;
    match nested_meta {
        NestedMeta::Lit(lit) => {
            if let syn::Lit::Str(lit) = lit {
                if !lit.value().is_empty() {
                    return Some(lit.value());
                }
            }
        }
        _ => {
            if let Some(name_value) = get_meta_str_value(nested_meta, "name") {
                return Some(name_value);
            }
        }
    }
    None
}

/// Parse function args.
fn parse_args<'a>(
    types: impl Iterator<Item = PType<'a>>,
    default_arg_type: Option<ArgType>,
) -> syn::Result<Vec<FnArg>> {
    let mut req_args: Vec<FnArg> = Vec::new();

    for pat_type in types {
        let name = pat_type.name;
        let ident = syn::Ident::new(&name.clone(), proc_macro2::Span::call_site());

        match &*pat_type.ty {
            syn::Type::Path(_) | syn::Type::Reference(_) | syn::Type::Array(_) => {}
            _ => {
                return Err(syn::Error::new_spanned(
                    quote!(),
                    "function args type must be like `std::slice::Iter`, `&std::slice::Iter` or `[T; n]`",
                ));
            }
        }

        let mut found_one = false;
        for (ty, attr) in pat_type
            .attrs
            .iter()
            .flat_map(|x| x.path().get_ident().map(|u| (u, x)))
            .flat_map(|(x, att)| ArgType::from_str(&x.to_string()).map(|u| (u, att)))
        {
            found_one = true;
            let name = extract_name(attr).unwrap_or_else(|| name.clone());
            let meta_map = parse_attr_meta_to_map(attr);

            req_args.push(FnArg {
                arg_type: ty,
                name,
                var: ident.clone(),
                var_type: pat_type.ty.clone(),
                meta_map,
            });
        }

        if let (Some(arg_type), false) = (&default_arg_type, found_one) {
            req_args.push(FnArg {
                arg_type: arg_type.clone(),
                name,
                var: ident,
                var_type: pat_type.ty.clone(),
                meta_map: HashMap::new(),
            });
        }

        pat_type.attrs.retain(|x| {
            if let Some(i) = x.path().get_ident() {
                let i = i.to_string();
                !i.as_str().parse::<ArgType>().is_ok()
            } else {
                true
            }
        });
    }

    Ok(req_args)
}

/// Parse return type of function.
pub fn parse_return_type(sig: &syn::Signature) -> syn::Result<Vec<syn::Type>> {
    let output = &sig.output;
    let mut err_msg = "function must have a return value".to_string();
    if let syn::ReturnType::Type(.., t) = output {
        if let syn::Type::Path(ref t_path) = **t {
            if let Some(syn::PathSegment {
                ident,
                arguments:
                    syn::PathArguments::AngleBracketed(syn::AngleBracketedGenericArguments {
                        args, ..
                    }),
            }) = t_path.path.segments.last()
            {
                if ident.to_string() == "Result" {
                    let mut return_args = Vec::new();
                    for arg in args.iter() {
                        if let syn::GenericArgument::Type(t) = arg {
                            return_args.push(t.clone());
                        }
                    }
                    return Ok(return_args);
                }
            }
        }
        err_msg = "return value must be Result".to_string();
    }
    Err(syn::Error::new_spanned(&sig, err_msg))
}

/// Compatible with syn 1.0 `NestedMeta`.
pub enum NestedMeta {
    Meta(syn::Meta),
    Lit(syn::Lit),
}

impl Parse for NestedMeta {
    fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
        if input.peek(syn::Lit) {
            Ok(NestedMeta::Lit(input.parse()?))
        } else {
            Ok(NestedMeta::Meta(input.parse()?))
        }
    }
}

impl quote::ToTokens for NestedMeta {
    fn to_tokens(&self, tokens: &mut proc_macro2::TokenStream) {
        match self {
            NestedMeta::Meta(meta) => meta.to_tokens(tokens),
            NestedMeta::Lit(lit) => lit.to_tokens(tokens),
        }
    }
}

pub fn parse_attr_metas(attr: &syn::Attribute) -> Option<Vec<NestedMeta>> {
    if let syn::Meta::List(meta_list) = &attr.meta {
        let nested = meta_list.parse_args_with(
            syn::punctuated::Punctuated::<NestedMeta, syn::Token![,]>::parse_terminated,
        );
        if let Ok(nested) = nested {
            return Some(nested.into_iter().collect());
        }
    }
    None
}

pub fn get_meta_str_value(meta: &NestedMeta, name: &str) -> Option<String> {
    match meta {
        // A literal, like the `"name"` in `#[param(p = "name")]`.
        NestedMeta::Meta(syn::Meta::NameValue(name_value)) => {
            let key = name_value.path.segments.last().unwrap().ident.to_string();
            if key == name {
                if let syn::Expr::Lit(expr_lit) = &name_value.value {
                    if let syn::Lit::Str(lit) = &expr_lit.lit {
                        return Some(lit.value());
                    }
                }
            }
        }
        _ => {}
    }
    None
}

pub fn parse_attr_meta_to_map(attr: &syn::Attribute) -> HashMap<String, String> {
    let mut attr_map = HashMap::new();
    if let Some(metas) = parse_attr_metas(attr) {
        for meta in metas.into_iter() {
            match meta {
                // A literal, like the `xxx` in `#[get(p = xxx)]`.
                NestedMeta::Meta(syn::Meta::NameValue(name_value)) => {
                    let key = name_value.path.segments.last().unwrap().ident.to_string();
                    match &name_value.value {
                        syn::Expr::Lit(expr_lit) => match &expr_lit.lit {
                            syn::Lit::Str(s) => {
                                attr_map.insert(key, s.value());
                            }
                            syn::Lit::Int(i) => {
                                attr_map.insert(key, i.to_string());
                            }
                            syn::Lit::Float(f) => {
                                attr_map.insert(key, f.to_string());
                            }
                            syn::Lit::Bool(b) => {
                                attr_map.insert(key, format!("{}", b.value()));
                            }
                            _ => {}
                        },
                        _ => {}
                    }
                }
                _ => {}
            }
        }
    }
    attr_map
}