moverox-codegen 0.0.12

Generate Rust code from Move IR parsed by move-syn
Documentation 
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

use std::collections::HashSet;

use quote::quote;
use unsynn::{Ident, ToTokens as _, TokenStream};

use crate::Result;

pub(super) trait GenericsExt {
    fn to_rust(&self) -> TokenStream;

    /// Generics declaration for the enum/struct definition.
    fn type_generics(
        &self,
        thecrate: &TokenStream,
        otw_types: HashSet<Ident>,
    ) -> Result<TokenStream>;

    fn phantoms(&self) -> impl Iterator<Item = &Ident>;
}

impl GenericsExt for move_syn::Generics {
    fn to_rust(&self) -> TokenStream {
        let idents = self.generics().map(|g| &g.ident);
        quote! {
            <#(#idents),*>
        }
    }

    fn type_generics(
        &self,
        thecrate: &TokenStream,
        mut otw_types: HashSet<Ident>,
    ) -> Result<TokenStream> {
        let idents: Vec<_> = self
            .generics()
            .map(|g| &g.ident)
            .map(|ident| {
                if otw_types.remove(ident) {
                    quote!(#ident = #thecrate::Otw)
                } else {
                    ident.to_token_stream()
                }
            })
            .collect();

        if !otw_types.is_empty() {
            let excess = otw_types
                .into_iter()
                .map(|ident| ident.to_string())
                .reduce(|a, b| a + ", " + &b)
                .unwrap_or_default();
            return Err(format!("Not a type parameter of this datatype: {excess}").into());
        }

        Ok(quote! {
            <#(#idents),*>
        })
    }

    fn phantoms(&self) -> impl Iterator<Item = &Ident> {
        self.generics()
            .filter(|d| d.phantom.is_some())
            .map(|d| &d.ident)
    }
}