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
use super::Identifier;

#[derive(Clone, Debug, PartialEq)]
pub enum Expr {
    /// 変数
    Variable(Identifier),

    /// シンボル
    Symbol(Identifier),

    /// 適用
    Apply { lhs: Box<Expr>, rhs: Box<Expr> },

    /// ラムダ抽象
    Lambda { param: Identifier, body: Box<Expr> },
}

impl From<&str> for Expr {
    fn from(s: &str) -> Self {
        match s.chars().nth(0) {
            Some(':') => Expr::Symbol((&s[1..]).into()),
            Some(_) => Expr::Variable(s.into()),
            _ => panic!("invalid identifier"),
        }
    }
}

impl From<String> for Expr {
    fn from(s: String) -> Self {
        s.as_str().into()
    }
}

pub fn v<Id: Into<Identifier>>(id: Id) -> Expr {
    Expr::Variable(id.into())
}

pub fn s<Id: Into<Identifier>>(id: Id) -> Expr {
    Expr::Symbol(id.into())
}

pub fn a<L: Into<Expr>, R: Into<Expr>>(lhs: L, rhs: R) -> Expr {
    Expr::Apply {
        lhs: Box::new(lhs.into()),
        rhs: Box::new(rhs.into()),
    }
}

pub fn l<P: Into<Identifier>, B: Into<Expr>>(param: P, body: B) -> Expr {
    Expr::Lambda {
        param: param.into(),
        body: Box::new(body.into()),
    }
}