1use crate::generator::{self, GenerateCode};
2
3pub trait OdraPoet: Sized {
5 type Poet: From<Self> + GenerateCode;
7}
8
9impl<'a> OdraPoet for &'a odra_ir::module::ModuleImpl {
10 type Poet = generator::ModuleImpl<'a>;
11}
12
13impl<'a> OdraPoet for &'a odra_ir::module::ModuleStruct {
14 type Poet = generator::ModuleStruct<'a>;
15}
16
17impl<'a> OdraPoet for &'a odra_ir::EventItem {
18 type Poet = generator::event_item::EventItem<'a>;
19}
20
21impl<'a> OdraPoet for &'a odra_ir::OdraTypeItem {
22 type Poet = generator::odra_type_item::OdraTypeItem<'a>;
23}
24
25impl<'a> OdraPoet for &'a odra_ir::InstanceItem {
26 type Poet = generator::instance_item::InstanceItem<'a>;
27}
28
29impl<'a> OdraPoet for &'a odra_ir::ExternalContractItem {
30 type Poet = generator::external_contract_item::ExternalContractItem<'a>;
31}
32
33impl<'a> OdraPoet for &'a odra_ir::ErrorEnumItem {
34 type Poet = generator::errors::ErrorEnumItem<'a>;
35}
36
37impl<'a> OdraPoet for &'a syn::ItemEnum {
38 type Poet = generator::errors::OdraErrorItem<'a>;
39}
40
41impl<'a> OdraPoet for &'a odra_ir::MapExpr {
42 type Poet = generator::mapping::OdraMapping<'a>;
43}
44
45pub trait OdraPoetUsingImpl: AsRef<odra_ir::module::ModuleImpl> {
46 fn generate_code_using<'a, G>(&'a self) -> proc_macro2::TokenStream
47 where
48 G: GenerateCode + From<&'a odra_ir::module::ModuleImpl>;
49}
50
51impl<T> OdraPoetUsingImpl for T
52where
53 T: AsRef<odra_ir::module::ModuleImpl>
54{
55 fn generate_code_using<'a, G>(&'a self) -> proc_macro2::TokenStream
56 where
57 G: GenerateCode + From<&'a odra_ir::module::ModuleImpl>
58 {
59 <G as GenerateCode>::generate_code(&G::from(
60 <Self as AsRef<odra_ir::module::ModuleImpl>>::as_ref(self)
61 ))
62 }
63}
64
65pub trait OdraPoetUsingStruct: AsRef<odra_ir::module::ModuleStruct> {
66 fn generate_code_using<'a, G>(&'a self) -> proc_macro2::TokenStream
67 where
68 G: GenerateCode + From<&'a odra_ir::module::ModuleStruct>;
69}
70
71impl<T> OdraPoetUsingStruct for T
72where
73 T: AsRef<odra_ir::module::ModuleStruct>
74{
75 fn generate_code_using<'a, G>(&'a self) -> proc_macro2::TokenStream
76 where
77 G: GenerateCode + From<&'a odra_ir::module::ModuleStruct>
78 {
79 <G as GenerateCode>::generate_code(&G::from(<Self as AsRef<
80 odra_ir::module::ModuleStruct
81 >>::as_ref(self)))
82 }
83}