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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
use nom::character::complete::{multispace0, multispace1};
use nom::{alt, delimited, do_parse, map, named, opt, separated_list0, tag, terminated, tuple};
use super::ident;
use crate::ast::{FunctionType, Type};
named!(pub function_type(&str) -> FunctionType, do_parse!(
tag!("fn")
>> multispace1
>> args: map!(opt!(terminated!(separated_list0!(
tuple!(
multispace0,
tag!(","),
multispace0
),
type_
), multispace1)), |args| args.unwrap_or_default())
>> tag!("->")
>> multispace1
>> ret: type_
>> (FunctionType {
args,
ret: Box::new(ret)
})
));
named!(pub type_(&str) -> Type, alt!(
tag!("int") => { |_| Type::Int } |
tag!("float") => { |_| Type::Float } |
tag!("bool") => { |_| Type::Bool } |
tag!("cstring") => { |_| Type::CString } |
function_type => { |ft| Type::Function(ft) }|
ident => { |id| Type::Var(id) } |
delimited!(
tuple!(tag!("("), multispace0),
type_,
tuple!(tag!(")"), multispace0)
)
));
#[cfg(test)]
mod tests {
use std::convert::TryFrom;
use super::*;
use crate::ast::Ident;
#[test]
fn simple_types() {
assert_eq!(test_parse!(type_, "int"), Type::Int);
assert_eq!(test_parse!(type_, "float"), Type::Float);
assert_eq!(test_parse!(type_, "bool"), Type::Bool);
assert_eq!(test_parse!(type_, "cstring"), Type::CString);
}
#[test]
fn no_arg_fn_type() {
assert_eq!(
test_parse!(type_, "fn -> int"),
Type::Function(FunctionType {
args: vec![],
ret: Box::new(Type::Int)
})
);
}
#[test]
fn fn_type_with_args() {
assert_eq!(
test_parse!(type_, "fn int, bool -> int"),
Type::Function(FunctionType {
args: vec![Type::Int, Type::Bool],
ret: Box::new(Type::Int)
})
);
}
#[test]
fn fn_taking_fn() {
assert_eq!(
test_parse!(type_, "fn fn int, bool -> bool, float -> float"),
Type::Function(FunctionType {
args: vec![
Type::Function(FunctionType {
args: vec![Type::Int, Type::Bool],
ret: Box::new(Type::Bool)
}),
Type::Float
],
ret: Box::new(Type::Float)
})
)
}
#[test]
fn parenthesized() {
assert_eq!(
test_parse!(type_, "fn (fn int, bool -> bool), float -> float"),
Type::Function(FunctionType {
args: vec![
Type::Function(FunctionType {
args: vec![Type::Int, Type::Bool],
ret: Box::new(Type::Bool)
}),
Type::Float
],
ret: Box::new(Type::Float)
})
)
}
#[test]
fn type_vars() {
assert_eq!(
test_parse!(type_, "fn x, y -> x"),
Type::Function(FunctionType {
args: vec![
Type::Var(Ident::try_from("x").unwrap()),
Type::Var(Ident::try_from("y").unwrap()),
],
ret: Box::new(Type::Var(Ident::try_from("x").unwrap())),
})
)
}
}
|