typhoon-context-macro 0.1.0-alpha.12

TODO
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

use {
    super::GeneratorResult,
    crate::{context::Context, StagedGenerator},
    quote::quote,
    typhoon_syn::constraints::Constraint,
};

pub struct RentGenerator<'a>(&'a Context);

impl<'a> RentGenerator<'a> {
    pub fn new(context: &'a Context) -> Self {
        Self(context)
    }
}

impl StagedGenerator for RentGenerator<'_> {
    fn append(&mut self, result: &mut GeneratorResult) -> Result<(), syn::Error> {
        if self.0.accounts.iter().any(|acc| {
            acc.constraints
                .0
                .iter()
                .any(|c| matches!(c, Constraint::Init(_) | Constraint::InitIfNeeded(_)))
        }) {
            result
                .inside
                .extend(quote!(let rent = <Rent as Sysvar>::get()?;));
        }

        Ok(())
    }
}