summaryrefslogtreecommitdiff
path: root/src/util.rs
blob: 9ed38d17ab43c82689a8db2e794c8a1f1134a562 (plain) (blame)
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
59
60
#![allow(non_snake_case)]

use crate::ast::*;

// intentionally small: i want to run into errors
/// assumption: the count is instantiated to zero
pub fn unique_ident(count: &mut u8) -> String {
    *count += 1;
    if *count == 0 {
        panic!("we've overflowed!");
    } else {
        return String::from(format!("{:X}", count));
    }
}

pub fn Term(val: Value, kind: Type) -> Term {
    return Term {val, kind};
}

pub fn Ann(expr: Expression, kind: Type) -> Expression {
    return Expression::Annotation {
        expr: Box::new(expr),
        kind: kind
    };
}

pub fn Const(val: Value, kind: Type) -> Expression {
    return Expression::Constant {
        term: Term {val, kind}
    };
}

pub fn Var(id: &str) -> Expression {
    return Expression::Variable {
        id: String::from(id)
    };
}

pub fn Abs(param: &str, func: Expression) -> Expression {
    return Expression::Abstraction {
        param: String::from(param),
        func: Box::new(func),
    };
}

pub fn App(func: Expression, arg: Expression) -> Expression {
    return Expression::Application {
        func: Box::new(func),
        arg: Box::new(arg)
    };
}

pub fn Cond(if_cond: Expression, if_then: Expression, if_else: Expression) -> Expression {
    return Expression::Conditional {
        if_cond: Box::new(if_cond),
        if_then: Box::new(if_then),
        if_else: Box::new(if_else)
    };
}