1use proc_macro::TokenStream;
 2use quote::quote;
 3use syn::{parse_macro_input, parse_quote, DeriveInput};
 4
 5pub fn derive_component(input: TokenStream) -> TokenStream {
 6    let ast = parse_macro_input!(input as DeriveInput);
 7    let name = &ast.ident;
 8
 9    let mut trait_generics = ast.generics.clone();
10    let view_type = if let Some(view_type) = specified_view_type(&ast) {
11        quote! { #view_type }
12    } else {
13        if let Some(first_type_param) = ast.generics.params.iter().find_map(|param| {
14            if let syn::GenericParam::Type(type_param) = param {
15                Some(type_param.ident.clone())
16            } else {
17                None
18            }
19        }) {
20            quote! { #first_type_param }
21        } else {
22            trait_generics.params.push(parse_quote! { V: 'static });
23            quote! { V }
24        }
25    };
26
27    let (impl_generics, _, where_clause) = trait_generics.split_for_impl();
28    let (_, ty_generics, _) = ast.generics.split_for_impl();
29
30    let expanded = quote! {
31        impl #impl_generics gpui2::Component<#view_type> for #name #ty_generics #where_clause {
32            fn render(self) -> gpui2::AnyElement<#view_type> {
33                (move |view_state: &mut #view_type, cx: &mut gpui2::ViewContext<'_, '_, #view_type>| self.render(view_state, cx))
34                    .render()
35            }
36        }
37    };
38
39    if name == "CollabPanel" {
40        println!("{}", expanded)
41    }
42
43    TokenStream::from(expanded)
44}
45
46fn specified_view_type(ast: &DeriveInput) -> Option<proc_macro2::Ident> {
47    let component_attr = ast
48        .attrs
49        .iter()
50        .find(|attr| attr.path.is_ident("component"))?;
51
52    if let Ok(syn::Meta::List(meta_list)) = component_attr.parse_meta() {
53        meta_list.nested.iter().find_map(|nested| {
54            if let syn::NestedMeta::Meta(syn::Meta::NameValue(nv)) = nested {
55                if nv.path.is_ident("view_type") {
56                    if let syn::Lit::Str(lit_str) = &nv.lit {
57                        return Some(
58                            lit_str
59                                .parse::<syn::Ident>()
60                                .expect("Failed to parse view_type"),
61                        );
62                    }
63                }
64            }
65            None
66        })
67    } else {
68        None
69    }
70}