radix_engine_interface/
macros.rs

1// TODO: Move this logic into procedural macro.
2#[macro_export]
3macro_rules! access_and_or {
4    (|| $tt:tt) => {{
5        let next = $crate::composite_requirement!($tt);
6        move |e: $crate::blueprints::resource::CompositeRequirement| e.or(next)
7    }};
8    (|| $right1:ident $right2:tt) => {{
9        let next = $crate::composite_requirement!($right1 $right2);
10        move |e: $crate::blueprints::resource::CompositeRequirement| e.or(next)
11    }};
12    (|| $right:tt && $($rest:tt)+) => {{
13        let f = $crate::access_and_or!(&& $($rest)+);
14        let next = $crate::composite_requirement!($right);
15        move |e: $crate::blueprints::resource::CompositeRequirement| e.or(f(next))
16    }};
17    (|| $right:tt || $($rest:tt)+) => {{
18        let f = $crate::access_and_or!(|| $($rest)+);
19        let next = $crate::composite_requirement!($right);
20        move |e: $crate::blueprints::resource::CompositeRequirement| f(e.or(next))
21    }};
22    (|| $right1:ident $right2:tt && $($rest:tt)+) => {{
23        let f = $crate::access_and_or!(&& $($rest)+);
24        let next = $crate::composite_requirement!($right1 $right2);
25        move |e: $crate::blueprints::resource::CompositeRequirement| e.or(f(next))
26    }};
27    (|| $right1:ident $right2:tt || $($rest:tt)+) => {{
28        let f = $crate::access_and_or!(|| $($rest)+);
29        let next = $crate::composite_requirement!($right1 $right2);
30        move |e: $crate::blueprints::resource::CompositeRequirement| f(e.or(next))
31    }};
32
33    (&& $tt:tt) => {{
34        let next = $crate::composite_requirement!($tt);
35        move |e: $crate::blueprints::resource::CompositeRequirement| e.and(next)
36    }};
37    (&& $right1:ident $right2:tt) => {{
38        let next = $crate::composite_requirement!($right1 $right2);
39        move |e: $crate::blueprints::resource::CompositeRequirement| e.and(next)
40    }};
41    (&& $right:tt && $($rest:tt)+) => {{
42        let f = $crate::access_and_or!(&& $($rest)+);
43        let next = $crate::composite_requirement!($right);
44        move |e: $crate::blueprints::resource::CompositeRequirement| f(e.and(next))
45    }};
46    (&& $right:tt || $($rest:tt)+) => {{
47        let f = $crate::access_and_or!(|| $($rest)+);
48        let next = $crate::composite_requirement!($right);
49        move |e: $crate::blueprints::resource::CompositeRequirement| f(e.and(next))
50    }};
51    (&& $right1:ident $right2:tt && $($rest:tt)+) => {{
52        let f = $crate::access_and_or!(&& $($rest)+);
53        let next = $crate::composite_requirement!($right1 $right2);
54        move |e: $crate::blueprints::resource::CompositeRequirement| f(e.and(next))
55    }};
56    (&& $right1:ident $right2:tt || $($rest:tt)+) => {{
57        let f = $crate::access_and_or!(|| $($rest)+);
58        let next = $crate::composite_requirement!($right1 $right2);
59        move |e: $crate::blueprints::resource::CompositeRequirement| f(e.and(next))
60    }};
61}
62
63#[macro_export]
64macro_rules! composite_requirement {
65    // Handle leaves
66    ($rule:ident $args:tt) => {{
67        $rule $args
68    }};
69
70    // Handle group
71    (($($tt:tt)+)) => {{ $crate::composite_requirement!($($tt)+) }};
72
73    // Handle and/or logic
74    ($left1:ident $left2:tt $($right:tt)+) => {{
75        let f = $crate::access_and_or!($($right)+);
76        f($crate::composite_requirement!($left1 $left2))
77    }};
78    ($left:tt $($right:tt)+) => {{
79        let f = $crate::access_and_or!($($right)+);
80        f($crate::composite_requirement!($left))
81    }};
82}
83
84#[macro_export]
85macro_rules! rule {
86    (allow_all) => {{
87        $crate::blueprints::resource::AccessRule::AllowAll
88    }};
89    (deny_all) => {{
90        $crate::blueprints::resource::AccessRule::DenyAll
91    }};
92    ($($tt:tt)+) => {{
93        $crate::blueprints::resource::AccessRule::Protected($crate::composite_requirement!($($tt)+))
94    }};
95}
96
97#[macro_export]
98macro_rules! role_entry {
99    ($roles: expr, $role: expr, $rule:expr) => {{
100        $roles.define_role($role, $rule);
101    }};
102}
103
104#[macro_export]
105macro_rules! roles2 {
106    ( ) => ({
107        $crate::blueprints::resource::RoleAssignmentInit::new()
108    });
109    ( $($role:expr => $rule:expr $(, $updatable:ident)?;)* ) => ({
110        let mut roles = $crate::blueprints::resource::RoleAssignmentInit::new();
111        $(
112            role_entry!(roles, $role, $rule);
113        )*
114        roles
115    })
116}
117
118extern crate radix_common_derive;
119pub use radix_common_derive::dec;
120pub use radix_common_derive::pdec;