1use codama_attributes::{AttributeContext, CodamaAttribute};
2use codama_errors::{CodamaError, CodamaResult};
3use codama_koroks::CrateKorok;
4use codama_stores::CrateStore;
5use proc_macro::TokenStream;
6use proc_macro2::TokenStream as TokenStream2;
7use quote::quote;
8
9#[proc_macro_derive(CodamaAccount, attributes(codama))]
10pub fn codama_account_derive(input: TokenStream) -> TokenStream {
11 codama_derive(input.into())
12 .unwrap_or_else(CodamaError::into_compile_error)
13 .into()
14}
15
16#[proc_macro_derive(CodamaAccounts, attributes(codama))]
17pub fn codama_accounts_derive(input: TokenStream) -> TokenStream {
18 codama_derive(input.into())
19 .unwrap_or_else(CodamaError::into_compile_error)
20 .into()
21}
22
23#[proc_macro_derive(CodamaInstruction, attributes(codama))]
24pub fn codama_instruction_derive(input: TokenStream) -> TokenStream {
25 codama_derive(input.into())
26 .unwrap_or_else(CodamaError::into_compile_error)
27 .into()
28}
29
30#[proc_macro_derive(CodamaInstructions, attributes(codama))]
31pub fn codama_instructions_derive(input: TokenStream) -> TokenStream {
32 codama_derive(input.into())
33 .unwrap_or_else(CodamaError::into_compile_error)
34 .into()
35}
36
37#[proc_macro_derive(CodamaType, attributes(codama))]
38pub fn codama_type_derive(input: TokenStream) -> TokenStream {
39 codama_derive(input.into())
40 .unwrap_or_else(CodamaError::into_compile_error)
41 .into()
42}
43
44fn codama_derive(input: TokenStream2) -> CodamaResult<TokenStream2> {
45 let store = CrateStore::hydrate(input)?;
46 CrateKorok::parse(&store)?;
47 Ok(quote! {})
48}
49
50#[proc_macro_attribute]
51pub fn codama(attr: TokenStream, input: TokenStream) -> TokenStream {
52 codama_attribute(attr.into(), input.into())
53 .unwrap_or_else(CodamaError::into_compile_error)
54 .into()
55}
56
57fn codama_attribute(attr: TokenStream2, input: TokenStream2) -> CodamaResult<TokenStream2> {
58 let attr: syn::Attribute = syn::parse_quote! { #[codama(#attr)] };
59 let item: syn::Item = syn::parse2(input.clone())?;
60 let ctx: AttributeContext = (&item).into();
61 CodamaAttribute::parse(&attr, &ctx)?;
62 Ok(input)
63}