peace_code_gen 0.0.14

Code generation macros for the peace automation framework.
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

use quote::quote;

use crate::cmd::{CmdCtxBuilderTypeBuilder, FlowCount, ImplHeaderBuilder, ScopeStruct};

/// Generates functions for the command context builder that are not constrained
/// by type parameters.
pub fn impl_common_fns(scope_struct: &ScopeStruct) -> proc_macro2::TokenStream {
    let scope = scope_struct.scope();
    let scope_builder_name = &scope_struct.item_struct().ident;

    let return_type = CmdCtxBuilderTypeBuilder::new(scope_builder_name.clone()).build();

    let mut common_fns = quote! {
        /// Sets the interrupt receiver and strategy so `CmdExecution`s can be interrupted.
        pub fn with_interruptibility(
            mut self,
            interruptibility: interruptible::Interruptibility<'static>,
        ) -> #return_type {
            let crate::ctx::CmdCtxBuilder {
                output,
                interruptibility: _,
                workspace,
                resources,
                scope_builder,
            } = self;

            crate::ctx::CmdCtxBuilder {
                output,
                interruptibility,
                workspace,
                resources,
                scope_builder,
            }
        }

        /// Sets the interrupt receiver and strategy so `CmdExecution`s can be interrupted.
        pub fn with_resource<R>(
            mut self,
            resource: R,
        ) -> Self
        where R: peace_resource_rt::Resource
        {
            self.resources.insert(resource);
            self
        }
    };

    if scope.flow_count() == FlowCount::One {
        common_fns.extend(quote! {
            /// Sets an item's parameters.
            ///
            /// Note: this **must** be called for each item in the flow.
            pub fn with_item_params<I>(
                mut self,
                item_id: peace_cfg::ItemId,
                params_spec: <I::Params<'_> as peace_params::Params>::Spec,
            ) -> Self
            where
                I: peace_cfg::Item,
                AppError: From<I::Error>,
            {
                self.scope_builder.params_specs_provided.insert(item_id, params_spec);
                self
            }
        });
    };

    let builder_type = CmdCtxBuilderTypeBuilder::new(scope_builder_name.clone()).build();
    let impl_header = ImplHeaderBuilder::new(builder_type).build();

    quote! {
        #impl_header
        {
            #common_fns
        }
    }
}