allow specifying lifetimes for cpi

lang/syn/src/codegen/program/cpi.rs

@@ -70,9 +70,52 @@ pub fn generate(program: &Program) -> proc_macro2::TokenStream {
                 let sighash_arr = sighash(SIGHASH_GLOBAL_NAMESPACE, name);
                 let sighash_tts: proc_macro2::TokenStream =
                     format!("{:?}", sighash_arr).parse().unwrap();
+
+                let (generics, generics_def) = {
+                    let ty = &*ix.ctx.raw_arg.ty;
+                    let generics = match ty {
+                        syn::Type::Path(ty_path) => {
+                            let segment = &ty_path.path.segments[0];
+                            match &segment.arguments {
+                                syn::PathArguments::AngleBracketed(bracket_args) => {
+                                    &bracket_args.args
+                                }
+                                _ => panic!("Invalid context"),
+                            }
+                        },
+                        _ => panic!("Invalid context"),
+                    };
+                    // If the user doesn't provide lifetimes, make this
+                    // assumption about the type.
+                    if generics.len() == 1 {
+                        (
+                            quote! {
+                                '_, '_, '_, 'info, #accounts_ident<'info>
+                            },
+                            quote! {
+                                <'info>
+                            }
+                        )
+                    }
+                    // User provided lifetimes, so use them.
+                    else {
+                        let generics_def = &ix.raw_method.sig.generics;
+                        (
+                            quote! {
+                                #generics
+                            },
+                            quote! {
+                                #generics_def
+                            }
+                        )
+                    }
+                };
+
+                println!("GENERICS: {:?}", crate::parser::tts_to_string(&generics));
+
                 quote! {
-                    pub fn #method_name<'a, 'b, 'c, 'info>(
-                        ctx: anchor_lang::context::CpiContext<'a, 'b, 'c, 'info, #accounts_ident<'info>>,
+                    pub fn #method_name #generics_def(
+                        ctx: anchor_lang::context::CpiContext<#generics>,
                         #(#args),*
                     ) -> ProgramResult {
                         let ix = {

lang/syn/src/lib.rs

@@ -87,6 +87,7 @@ pub struct Ix {
     pub args: Vec<IxArg>,
     // The ident for the struct deriving Accounts.
     pub anchor_ident: Ident,
+    pub ctx: IxArg,
 }
 
 #[derive(Debug)]

lang/syn/src/parser/program/instructions.rs

@@ -29,6 +29,7 @@ pub fn parse(program_mod: &syn::ItemMod) -> ParseResult<(Vec<Ix>, Option<Fallbac
                 ident: method.sig.ident.clone(),
                 args,
                 anchor_ident,
+                ctx,
             })
         })
         .collect::<ParseResult<Vec<Ix>>>()?;