1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
use crate::utils::{add_where_clauses_for_new_ident, SingleFieldData, State};
use proc_macro2::{Span, TokenStream};
use quote::quote;
use syn::{parse::Result, DeriveInput, Ident};
/// Provides the hook to expand `#[derive(Index)]` into an implementation of `Index`
pub fn expand(input: &DeriveInput, trait_name: &'static str) -> Result<TokenStream> {
let index_type = &Ident::new("__IdxT", Span::call_site());
let mut state = State::with_field_ignore(
input,
trait_name,
quote!(::core::ops),
trait_name.to_lowercase(),
)?;
state.add_trait_path_type_param(quote!(#index_type));
let SingleFieldData {
field,
field_type,
input_type,
trait_path_with_params,
casted_trait,
member,
..
} = state.assert_single_enabled_field();
let type_where_clauses = quote! {
where #field_type: #trait_path_with_params
};
let new_generics = add_where_clauses_for_new_ident(
&input.generics,
&[field],
index_type,
type_where_clauses,
true,
);
let (impl_generics, _, where_clause) = new_generics.split_for_impl();
let (_, ty_generics, _) = input.generics.split_for_impl();
Ok(quote! {
impl#impl_generics #trait_path_with_params for #input_type#ty_generics #where_clause
{
type Output = #casted_trait::Output;
#[inline]
fn index(&self, idx: #index_type) -> &Self::Output {
#casted_trait::index(&#member, idx)
}
}
})
}
|
