narrative-macros 0.12.2

Procedural macros for the narrative crate
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

// The generated trait is used as a super trait of the story trait, and provides `Self::*` items
// for coupled data types.
// Sealing the trait is not necessary because we have blanket impls of this for all types.

use proc_macro2::TokenStream;
use quote::{format_ident, quote};

use crate::{
    item_story::{story_const::StoryConst, ItemStory},
    Asyncness,
};

pub fn generate(input: &ItemStory, asyncness: Asyncness) -> TokenStream {
    let trait_ident = match asyncness {
        Asyncness::Sync => quote!(BaseTrait),
        Asyncness::Async => quote!(AsyncBaseTrait),
    };
    let story_trait_ident = match asyncness {
        Asyncness::Sync => input.ident.clone(),
        Asyncness::Async => format_ident!("Async{}", input.ident),
    };
    let consts = input.consts().map(|StoryConst { raw, .. }| {
        let ident = &raw.ident;
        let ty = &raw.ty;
        Some(quote! {
            const #ident: #ty;
        })
    });
    let consts_assigns = input.consts().map(|StoryConst { raw, .. }| raw);
    quote! {
        pub trait #trait_ident {
            #(#consts)*
            type Context: narrative::story::StoryContext;
            const CONTEXT: Self::Context;
        }
        impl<T: #story_trait_ident> #trait_ident for T {
            #(#consts_assigns)*
            type Context = StoryContext;
            const CONTEXT: StoryContext = StoryContext;
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use pretty_assertions::assert_eq;

    #[test]
    fn test_empty() {
        let story_syntax = syn::parse_quote! {
            trait User {
                #[step("Step 1")]
                fn step1();
                #[step("Step 2")]
                fn step2();
            }
        };
        let actual = generate(&story_syntax, Asyncness::Sync);
        let expected = quote! {
            pub trait BaseTrait {
                type Context: narrative::story::StoryContext;
                const CONTEXT: Self::Context;
            }
            impl<T: User> BaseTrait for T {
                type Context = StoryContext;
                const CONTEXT: StoryContext = StoryContext;
            }
        };
        assert_eq!(actual.to_string(), expected.to_string());
    }


    #[test]
    fn test_async() {
        let story_syntax = syn::parse_quote! {
            trait User {
                #[step("Step 1")]
                async fn step1();
                #[step("Step 2")]
                async fn step2();
            }
        };
        let actual = generate(&story_syntax, Asyncness::Async);
        let expected = quote! {
            pub trait AsyncBaseTrait {
                type Context: narrative::story::StoryContext;
                const CONTEXT: Self::Context;
            }
            impl<T: AsyncUser> AsyncBaseTrait for T {
                type Context = StoryContext;
                const CONTEXT: StoryContext = StoryContext;
            }
        };
        assert_eq!(actual.to_string(), expected.to_string());
    }
}