aboutsummaryrefslogtreecommitdiff
path: root/argparse/src/num.rs
blob: 9f34de7b198df7d5d3d0d764db3334355be0ecea (plain)
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
use std::cell::RefCell;
use std::rc::Rc;
use std::ops::{Add, Sub};

use super::{IncrBy, DecrBy};
use super::action::{TypedAction, Action, ParseResult};
use super::action::ParseResult::Parsed;
use super::action::IFlagAction;
use super::action::Action::Flag;

pub struct IncrByAction<'a, T: 'a> {
    delta: T,
    cell: Rc<RefCell<&'a mut T>>,
}

pub struct DecrByAction<'a, T: 'a> {
    delta: T,
    cell: Rc<RefCell<&'a mut T>>,
}

impl<T: 'static + Add<Output = T> + Clone> TypedAction<T> for IncrBy<T> {
    fn bind<'x>(&self, cell: Rc<RefCell<&'x mut T>>) -> Action<'x> {
        let IncrBy(ref delta) = *self;
        return Flag(Box::new(IncrByAction { cell: cell, delta: delta.clone() }));
    }
}

impl<T: 'static + Sub<Output = T> + Clone> TypedAction<T> for DecrBy<T> {
    fn bind<'x>(&self, cell: Rc<RefCell<&'x mut T>>) -> Action<'x> {
        let DecrBy(ref delta) = *self;
        return Flag(Box::new(DecrByAction { cell: cell, delta: delta.clone() }));
    }
}

impl<'a, T: Add<Output = T> + Clone> IFlagAction for IncrByAction<'a, T> {
    fn parse_flag(&self) -> ParseResult {
        let oldval = {
            let targ = self.cell.borrow();
            targ.clone()
        };
        let mut targ = self.cell.borrow_mut();
        **targ = oldval + self.delta.clone();
        return Parsed;
    }
}

impl<'a, T: Sub<Output = T> + Clone> IFlagAction for DecrByAction<'a, T> {
    fn parse_flag(&self) -> ParseResult {
        let oldval = {
            let targ = self.cell.borrow();
            targ.clone()
        };
        let mut targ = self.cell.borrow_mut();
        **targ = oldval - self.delta.clone();
        return Parsed;
    }
}