typhoon-context-macro 0.1.0-alpha.16

use syn::{parse_quote, visit_mut::VisitMut, PathArguments};

pub struct LifetimeInjector;

impl VisitMut for LifetimeInjector {
    fn visit_generics_mut(&mut self, i: &mut syn::Generics) {
        i.params.push(parse_quote!('info));
    }

    fn visit_type_path_mut(&mut self, i: &mut syn::TypePath) {
        if let Some(seg) = i.path.segments.last_mut() {
            let ident = seg.ident.to_string();
            if ident.starts_with("Mut") || ident == "Option" {
                if let PathArguments::AngleBracketed(ref mut angle_args) = seg.arguments {
                    if let Some(first_arg) = angle_args.args.first_mut() {
                        self.visit_generic_argument_mut(first_arg);
                    }
                }
                return;
            }

            match seg.arguments {
                PathArguments::AngleBracketed(ref mut gen_args) => {
                    gen_args.args.insert(0, parse_quote!('info));
                }
                PathArguments::None => {
                    seg.arguments = PathArguments::AngleBracketed(parse_quote!(<'info>));
                }
                PathArguments::Parenthesized(_) => {}
            }

            if ident.starts_with("Signer") {
                self.visit_path_arguments_mut(&mut seg.arguments);
            }
        }
    }
}