odra_codegen/
poet.rs

1use crate::generator::{self, GenerateCode};
2
3/// Types which can generate code.
4pub trait OdraPoet: Sized {
5    /// The underlying generator generating the code.
6    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}