typhoon-context-macro 0.1.0-alpha.0

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
32
33
34
35

use {
    super::{ConstraintGenerator, GeneratorResult},
    crate::{constraints::ConstraintInit, visitor::ContextVisitor},
    quote::quote,
};

pub struct RentGenerator(bool);

impl RentGenerator {
    pub fn new() -> Self {
        RentGenerator(false)
    }
}

impl ConstraintGenerator for RentGenerator {
    fn generate(&self) -> Result<GeneratorResult, syn::Error> {
        if self.0 {
            Ok(GeneratorResult {
                at_init: quote! {
                    let rent = <Rent as Sysvar>::get()?;
                },
                ..Default::default()
            })
        } else {
            Ok(GeneratorResult::default())
        }
    }
}

impl ContextVisitor for RentGenerator {
    fn visit_init(&mut self, _contraint: &ConstraintInit) -> Result<(), syn::Error> {
        self.0 = true; // TODO disable when creating ATA
        Ok(())
    }
}