use auto_delegate_impl::{delegate, Delegate};

#[delegate]
trait Increment {
    fn increment(&mut self) -> usize;
}


#[delegate]
trait Calc {
    fn calc(&mut self, x1: usize, x2: usize) -> &usize;
}

struct CalcAdd {
    num: usize,
}

impl Calc for CalcAdd {
    fn calc(&mut self, x1: usize, x2: usize) -> &usize {
        self.num = x1 + x2;
        &self.num
    }
}

impl Increment for CalcAdd {
    fn increment(&mut self) -> usize {
        self.num += 1;
        self.num
    }
}

struct CalcSub {
    num: usize,
}

impl Calc for CalcSub {
    fn calc(&mut self, x1: usize, x2: usize) -> &usize {
        self.num = x1 - x2;
        &self.num
    }
}

impl Increment for CalcSub {
    fn increment(&mut self) -> usize {
        self.num -= 1;
        self.num
    }
}

#[derive(Delegate)]
#[to(Calc, Increment)]
enum EnumCalc {
    Add(CalcAdd),
    Sub(CalcSub),
}

fn main() {
    let mut c = EnumCalc::Add(CalcAdd { num: 0 });
    assert_eq!(*c.calc(3, 5), 8);
    assert_eq!(c.increment(), 9);

    let mut c = EnumCalc::Sub(CalcSub { num: 1 });
    assert_eq!(*c.calc(3, 2), 1);
    assert_eq!(c.increment(), 0);
}