1
0
dust/src/parser.rs

854 lines
29 KiB
Rust
Raw Normal View History

use std::{cell::RefCell, collections::HashMap};
2024-02-25 18:49:26 +00:00
use chumsky::{input::SpannedInput, pratt::*, prelude::*};
2024-03-07 10:37:26 +00:00
use crate::{
abstract_tree::*,
error::Error,
2024-03-07 11:57:33 +00:00
lexer::{Control, Operator, Token},
2024-03-07 10:37:26 +00:00
};
2024-02-25 18:49:26 +00:00
2024-03-06 20:36:58 +00:00
pub type DustParser<'src> = Boxed<
'src,
'src,
ParserInput<'src>,
2024-03-08 21:14:47 +00:00
Vec<(Statement, SimpleSpan)>,
2024-03-06 20:36:58 +00:00
extra::Err<Rich<'src, Token<'src>, SimpleSpan>>,
>;
pub type ParserInput<'src> =
SpannedInput<Token<'src>, SimpleSpan, &'src [(Token<'src>, SimpleSpan)]>;
2024-02-25 18:49:26 +00:00
2024-03-06 20:36:58 +00:00
pub fn parse<'src>(
tokens: &'src [(Token<'src>, SimpleSpan)],
2024-03-08 21:14:47 +00:00
) -> Result<Vec<(Statement, SimpleSpan)>, Vec<Error>> {
2024-03-02 01:17:55 +00:00
parser()
2024-03-06 20:36:58 +00:00
.parse(tokens.spanned((tokens.len()..tokens.len()).into()))
2024-03-02 01:17:55 +00:00
.into_result()
2024-03-06 20:36:58 +00:00
.map_err(|errors| errors.into_iter().map(|error| error.into()).collect())
2024-03-02 01:17:55 +00:00
}
2024-03-06 20:36:58 +00:00
pub fn parser<'src>() -> DustParser<'src> {
let identifiers: RefCell<HashMap<&str, Identifier>> = RefCell::new(HashMap::new());
2024-02-26 21:27:01 +00:00
let identifier = select! {
Token::Identifier(text) => {
let mut identifiers = identifiers.borrow_mut();
if let Some(identifier) = identifiers.get(&text) {
identifier.clone()
} else {
let new = Identifier::new(text);
identifiers.insert(text, new.clone());
new
}
}
2024-02-26 21:27:01 +00:00
};
2024-02-25 18:49:26 +00:00
2024-03-07 21:13:15 +00:00
let basic_type = choice((
just(Token::Keyword("bool")).to(Type::Boolean),
just(Token::Keyword("float")).to(Type::Float),
just(Token::Keyword("int")).to(Type::Integer),
just(Token::Keyword("range")).to(Type::Range),
just(Token::Keyword("str")).to(Type::String),
just(Token::Keyword("list")).to(Type::List),
));
let type_arguments = basic_type.clone().delimited_by(
just(Token::Control(Control::ParenOpen)),
just(Token::Control(Control::ParenClose)),
);
let type_specification = just(Token::Control(Control::Colon)).ignore_then(choice((
basic_type
.clone()
.separated_by(just(Token::Control(Control::Comma)))
.collect()
.delimited_by(
just(Token::Control(Control::SquareOpen)),
just(Token::Control(Control::SquareClose)),
)
.map(|types| Type::ListExact(types)),
just(Token::Keyword("list"))
.then(type_arguments)
.map(|(_, item_type)| Type::ListOf(Box::new(item_type))),
basic_type.clone(),
identifier
.clone()
.map(|identifier| Type::Custom(identifier)),
)));
2024-02-26 21:27:01 +00:00
let expression = recursive(|expression| {
2024-02-25 18:49:26 +00:00
let basic_value = select! {
2024-02-26 21:27:01 +00:00
Token::Boolean(boolean) => ValueNode::Boolean(boolean),
Token::Integer(integer) => ValueNode::Integer(integer),
Token::Float(float) => ValueNode::Float(float),
2024-03-08 21:14:47 +00:00
Token::String(string) => ValueNode::String(string.to_string()),
}
.map(|value| Expression::Value(value))
.boxed();
2024-02-25 18:49:26 +00:00
2024-02-26 21:27:01 +00:00
let identifier_expression = identifier
.clone()
2024-02-26 21:27:01 +00:00
.map(|identifier| Expression::Identifier(identifier))
.boxed();
2024-03-09 01:30:26 +00:00
let range = {
let raw_integer = select! {
Token::Integer(integer) => integer
};
raw_integer
.clone()
.then_ignore(just(Token::Control(Control::DoubleDot)))
.then(raw_integer)
.map(|(start, end)| Expression::Value(ValueNode::Range(start..end)))
};
2024-02-26 21:27:01 +00:00
let list = expression
2024-02-25 18:49:26 +00:00
.clone()
2024-03-07 11:57:33 +00:00
.separated_by(just(Token::Control(Control::Comma)))
2024-02-25 18:49:26 +00:00
.allow_trailing()
.collect()
2024-03-07 11:57:33 +00:00
.delimited_by(
just(Token::Control(Control::SquareOpen)),
just(Token::Control(Control::SquareClose)),
)
.map(|list| Expression::Value(ValueNode::List(list)))
.boxed();
2024-02-25 18:49:26 +00:00
2024-03-07 21:13:15 +00:00
let map_assignment = identifier
.clone()
.then(type_specification.clone().or_not())
.then_ignore(just(Token::Operator(Operator::Assign)))
.then(expression.clone())
.map(|((identifier, r#type), expression)| (identifier, r#type, expression));
let map = map_assignment
.separated_by(just(Token::Control(Control::Comma)).or_not())
.allow_trailing()
.collect()
.delimited_by(
just(Token::Control(Control::CurlyOpen)),
just(Token::Control(Control::CurlyClose)),
)
.map(|map_assigment_list| Expression::Value(ValueNode::Map(map_assigment_list)));
let r#enum = identifier
.clone()
2024-03-07 11:57:33 +00:00
.then_ignore(just(Token::Control(Control::DoubleColon)))
.then(identifier.clone())
.map(|(name, variant)| Expression::Value(ValueNode::Enum(name, variant)))
.boxed();
2024-02-26 21:27:01 +00:00
let atom = choice((
identifier_expression.clone(),
basic_value.clone(),
list.clone(),
r#enum.clone(),
2024-03-07 11:57:33 +00:00
expression.clone().delimited_by(
just(Token::Control(Control::ParenOpen)),
just(Token::Control(Control::ParenClose)),
),
2024-02-25 18:49:26 +00:00
));
2024-03-07 10:37:26 +00:00
use Operator::*;
2024-03-07 21:13:15 +00:00
let logic_math_and_index = atom
2024-02-26 21:27:01 +00:00
.pratt((
2024-03-07 10:37:26 +00:00
prefix(2, just(Token::Operator(Not)), |expression| {
2024-02-26 21:27:01 +00:00
Expression::Logic(Box::new(Logic::Not(expression)))
}),
2024-03-07 10:37:26 +00:00
infix(left(1), just(Token::Operator(Equal)), |left, right| {
2024-02-26 21:27:01 +00:00
Expression::Logic(Box::new(Logic::Equal(left, right)))
}),
2024-03-07 10:37:26 +00:00
infix(left(1), just(Token::Operator(NotEqual)), |left, right| {
2024-02-26 21:27:01 +00:00
Expression::Logic(Box::new(Logic::NotEqual(left, right)))
}),
2024-03-07 10:37:26 +00:00
infix(left(1), just(Token::Operator(Greater)), |left, right| {
2024-02-26 21:27:01 +00:00
Expression::Logic(Box::new(Logic::Greater(left, right)))
}),
2024-03-07 10:37:26 +00:00
infix(left(1), just(Token::Operator(Less)), |left, right| {
2024-02-26 21:27:01 +00:00
Expression::Logic(Box::new(Logic::Less(left, right)))
}),
2024-03-07 10:37:26 +00:00
infix(
left(1),
just(Token::Operator(GreaterOrEqual)),
|left, right| Expression::Logic(Box::new(Logic::GreaterOrEqual(left, right))),
),
infix(
left(1),
just(Token::Operator(LessOrEqual)),
|left, right| Expression::Logic(Box::new(Logic::LessOrEqual(left, right))),
),
infix(left(1), just(Token::Operator(And)), |left, right| {
2024-02-26 21:27:01 +00:00
Expression::Logic(Box::new(Logic::And(left, right)))
}),
2024-03-07 10:37:26 +00:00
infix(left(1), just(Token::Operator(Or)), |left, right| {
2024-02-26 21:27:01 +00:00
Expression::Logic(Box::new(Logic::Or(left, right)))
}),
2024-03-07 11:33:54 +00:00
infix(left(1), just(Token::Operator(Add)), |left, right| {
Expression::Math(Box::new(Math::Add(left, right)))
}),
infix(left(1), just(Token::Operator(Subtract)), |left, right| {
Expression::Math(Box::new(Math::Subtract(left, right)))
}),
infix(left(2), just(Token::Operator(Multiply)), |left, right| {
Expression::Math(Box::new(Math::Multiply(left, right)))
}),
infix(left(2), just(Token::Operator(Divide)), |left, right| {
Expression::Math(Box::new(Math::Divide(left, right)))
}),
infix(left(1), just(Token::Operator(Modulo)), |left, right| {
Expression::Math(Box::new(Math::Modulo(left, right)))
}),
2024-03-07 17:29:07 +00:00
infix(
left(3),
just(Token::Control(Control::Dot)),
|left, right| Expression::Index(Box::new(Index::new(left, right))),
),
2024-02-26 21:27:01 +00:00
))
.boxed();
2024-03-07 11:33:54 +00:00
choice((
2024-03-09 01:30:26 +00:00
range,
2024-03-07 11:33:54 +00:00
r#enum,
2024-03-07 21:13:15 +00:00
logic_math_and_index,
2024-03-07 11:33:54 +00:00
identifier_expression,
list,
2024-03-07 21:13:15 +00:00
map,
2024-03-07 11:33:54 +00:00
basic_value,
))
2024-02-26 21:27:01 +00:00
});
let statement = recursive(|statement| {
let expression_statement = expression
2024-03-08 17:24:11 +00:00
.clone()
2024-02-26 21:27:01 +00:00
.map(|expression| Statement::Expression(expression))
.boxed();
2024-03-08 17:24:11 +00:00
let r#break = just(Token::Keyword("break"))
2024-03-08 19:01:05 +00:00
.ignore_then(expression.clone())
2024-03-08 17:24:11 +00:00
.map(|expression| Statement::Break(expression));
2024-02-25 18:49:26 +00:00
let assignment = identifier
2024-03-08 21:14:47 +00:00
.clone()
.then(type_specification.clone().or_not())
.then(choice((
just(Token::Operator(Operator::Assign)).to(AssignmentOperator::Assign),
just(Token::Operator(Operator::AddAssign)).to(AssignmentOperator::AddAssign),
just(Token::Operator(Operator::SubAssign)).to(AssignmentOperator::SubAssign),
)))
2024-02-25 18:49:26 +00:00
.then(statement.clone())
.map(|(((identifier, r#type), operator), statement)| {
Statement::Assignment(Assignment::new(identifier, r#type, operator, statement))
2024-02-26 21:27:01 +00:00
})
.boxed();
2024-02-25 18:49:26 +00:00
2024-02-28 23:16:25 +00:00
let block = statement
.clone()
2024-03-07 00:45:41 +00:00
.repeated()
2024-02-28 23:16:25 +00:00
.collect()
2024-03-07 11:57:33 +00:00
.delimited_by(
just(Token::Control(Control::CurlyOpen)),
just(Token::Control(Control::CurlyClose)),
)
2024-02-28 23:16:25 +00:00
.map(|statements| Statement::Block(Block::new(statements)))
.boxed();
2024-03-02 01:17:55 +00:00
let r#loop = statement
.clone()
2024-03-07 00:45:41 +00:00
.repeated()
2024-03-08 17:24:11 +00:00
.at_least(1)
2024-03-02 01:17:55 +00:00
.collect()
.delimited_by(
2024-03-07 11:57:33 +00:00
just(Token::Keyword("loop")).then(just(Token::Control(Control::CurlyOpen))),
just(Token::Control(Control::CurlyClose)),
2024-03-02 01:17:55 +00:00
)
.map(|statements| Statement::Loop(Loop::new(statements)))
.boxed();
2024-03-08 19:01:05 +00:00
let if_else = just(Token::Keyword("if"))
.ignore_then(expression.clone())
.then(statement.clone())
2024-03-08 19:29:53 +00:00
.then(
just(Token::Keyword("else"))
.ignore_then(statement.clone())
.or_not(),
)
.map(|((if_expression, if_statement), else_statement)| {
Statement::IfElse(IfElse::new(if_expression, if_statement, else_statement))
2024-03-08 19:01:05 +00:00
})
.boxed();
2024-03-08 21:14:47 +00:00
let function = identifier
.clone()
.then(type_specification.clone())
.separated_by(just(Token::Control(Control::Comma)))
.collect::<Vec<(Identifier, Type)>>()
.delimited_by(
just(Token::Control(Control::ParenOpen)),
just(Token::Control(Control::ParenClose)),
)
.then(type_specification)
.then(statement.clone())
.map(|((parameters, return_type), body)| {
Statement::Expression(Expression::Value(ValueNode::Function {
parameters,
return_type,
body: Box::new(body),
}))
});
2024-03-08 19:01:05 +00:00
choice((
assignment,
expression_statement,
r#break,
block,
r#loop,
if_else,
2024-03-08 21:14:47 +00:00
function,
2024-03-08 19:01:05 +00:00
))
.then_ignore(just(Token::Control(Control::Semicolon)).or_not())
2024-02-26 21:27:01 +00:00
});
2024-02-25 18:49:26 +00:00
2024-02-26 21:27:01 +00:00
statement
.map_with(|item, state| (item, state.span()))
.repeated()
.collect()
2024-03-06 20:36:58 +00:00
.boxed()
2024-02-25 18:49:26 +00:00
}
#[cfg(test)]
mod tests {
2024-02-26 21:27:01 +00:00
use crate::{abstract_tree::Logic, lexer::lex};
2024-02-25 18:49:26 +00:00
use super::*;
2024-03-09 01:30:26 +00:00
#[test]
fn range() {
assert_eq!(
parse(&lex("1..10").unwrap()).unwrap()[0].0,
Statement::Expression(Expression::Value(ValueNode::Range(1..10)))
)
}
2024-03-08 21:14:47 +00:00
#[test]
fn function() {
assert_eq!(
parse(&lex("(x: int): int x ").unwrap()).unwrap()[0].0,
Statement::Expression(Expression::Value(ValueNode::Function {
parameters: vec![(Identifier::new("x"), Type::Integer)],
return_type: Type::Integer,
body: Box::new(Statement::Expression(Expression::Identifier(
Identifier::new("x")
)))
}))
)
}
2024-03-08 19:01:05 +00:00
#[test]
2024-03-08 19:29:53 +00:00
fn r#if() {
assert_eq!(
parse(&lex("if true 'foo'").unwrap()).unwrap()[0].0,
Statement::IfElse(IfElse::new(
Expression::Value(ValueNode::Boolean(true)),
2024-03-08 21:14:47 +00:00
Statement::Expression(Expression::Value(ValueNode::String("foo".to_string()))),
2024-03-08 19:29:53 +00:00
None
))
)
}
#[test]
2024-03-08 19:01:05 +00:00
fn if_else() {
assert_eq!(
parse(&lex("if true 'foo' else 'bar'").unwrap()).unwrap()[0].0,
Statement::IfElse(IfElse::new(
Expression::Value(ValueNode::Boolean(true)),
2024-03-08 21:14:47 +00:00
Statement::Expression(Expression::Value(ValueNode::String("foo".to_string()))),
2024-03-08 19:01:05 +00:00
Some(Statement::Expression(Expression::Value(ValueNode::String(
2024-03-08 21:14:47 +00:00
"bar".to_string()
2024-03-08 19:01:05 +00:00
))))
))
)
}
2024-03-07 21:13:15 +00:00
#[test]
fn map() {
assert_eq!(
parse(&lex("{ foo = 'bar' }").unwrap()).unwrap()[0].0,
Statement::Expression(Expression::Value(ValueNode::Map(vec![(
Identifier::new("foo"),
None,
2024-03-08 21:14:47 +00:00
Expression::Value(ValueNode::String("bar".to_string()))
2024-03-07 21:13:15 +00:00
)])))
);
assert_eq!(
parse(&lex("{ x = 1, y = 2, }").unwrap()).unwrap()[0].0,
Statement::Expression(Expression::Value(ValueNode::Map(vec![
(
Identifier::new("x"),
None,
Expression::Value(ValueNode::Integer(1))
),
(
Identifier::new("y"),
None,
Expression::Value(ValueNode::Integer(2))
),
])))
);
assert_eq!(
parse(&lex("{ x = 1 y = 2 }").unwrap()).unwrap()[0].0,
Statement::Expression(Expression::Value(ValueNode::Map(vec![
(
Identifier::new("x"),
None,
Expression::Value(ValueNode::Integer(1))
),
(
Identifier::new("y"),
None,
Expression::Value(ValueNode::Integer(2))
),
])))
);
}
2024-03-07 11:33:54 +00:00
#[test]
fn math() {
assert_eq!(
parse(&lex("1 + 1").unwrap()).unwrap()[0].0,
Statement::Expression(Expression::Math(Box::new(Math::Add(
Expression::Value(ValueNode::Integer(1)),
Expression::Value(ValueNode::Integer(1))
))))
);
}
2024-03-02 01:17:55 +00:00
#[test]
fn r#loop() {
assert_eq!(
2024-03-08 17:24:11 +00:00
parse(&lex("loop { 42 }").unwrap()).unwrap()[0].0,
Statement::Loop(Loop::new(vec![Statement::Expression(Expression::Value(
ValueNode::Integer(42)
))]))
2024-03-02 01:17:55 +00:00
);
}
2024-02-28 23:16:25 +00:00
#[test]
fn block() {
assert_eq!(
parse(&lex("{ x }").unwrap()).unwrap()[0].0,
Statement::Block(Block::new(vec![Statement::Expression(
Expression::Identifier(Identifier::new("x"))
),]))
);
assert_eq!(
parse(
&lex("
{
x;
y;
z
}
")
.unwrap()
)
.unwrap()[0]
.0,
Statement::Block(Block::new(vec![
Statement::Expression(Expression::Identifier(Identifier::new("x"))),
Statement::Expression(Expression::Identifier(Identifier::new("y"))),
Statement::Expression(Expression::Identifier(Identifier::new("z"))),
]))
);
assert_eq!(
parse(
&lex("
{
1 == 1
z
}
")
.unwrap()
)
.unwrap()[0]
.0,
Statement::Block(Block::new(vec![
Statement::Expression(Expression::Logic(Box::new(Logic::Equal(
Expression::Value(ValueNode::Integer(1)),
Expression::Value(ValueNode::Integer(1))
)))),
Statement::Expression(Expression::Identifier(Identifier::new("z"))),
]))
);
}
2024-02-25 18:49:26 +00:00
#[test]
fn identifier() {
assert_eq!(
parse(&lex("x").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Identifier(Identifier::new("x")))
2024-02-25 18:49:26 +00:00
);
assert_eq!(
parse(&lex("foobar").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Identifier(Identifier::new("foobar")))
2024-02-25 18:49:26 +00:00
);
assert_eq!(
parse(&lex("HELLO").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Identifier(Identifier::new("HELLO")))
2024-02-25 18:49:26 +00:00
);
}
#[test]
fn assignment() {
assert_eq!(
parse(&lex("foobar = 1").unwrap()).unwrap()[0].0,
Statement::Assignment(Assignment::new(
Identifier::new("foobar"),
None,
AssignmentOperator::Assign,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Integer(1)))
2024-02-25 18:49:26 +00:00
)),
);
}
#[test]
2024-02-29 02:34:14 +00:00
fn assignment_with_basic_type() {
assert_eq!(
parse(&lex("foobar: int = 1").unwrap()).unwrap()[0].0,
Statement::Assignment(Assignment::new(
Identifier::new("foobar"),
Some(Type::Integer),
AssignmentOperator::Assign,
Statement::Expression(Expression::Value(ValueNode::Integer(1)))
)),
);
2024-02-29 02:34:14 +00:00
}
2024-02-29 02:34:14 +00:00
#[test]
fn assignment_with_custom_type() {
assert_eq!(
parse(&lex("foobar: Foo = Foo::Bar").unwrap()).unwrap()[0].0,
Statement::Assignment(Assignment::new(
Identifier::new("foobar"),
Some(Type::Custom(Identifier::new("Foo"))),
AssignmentOperator::Assign,
Statement::Expression(Expression::Value(ValueNode::Enum(
Identifier::new("Foo"),
Identifier::new("Bar")
)))
)),
);
}
2024-02-29 02:34:14 +00:00
#[test]
fn assignment_with_list_types() {
assert_eq!(
parse(&lex("foobar: list = []").unwrap()).unwrap()[0].0,
Statement::Assignment(Assignment::new(
Identifier::new("foobar"),
Some(Type::List),
AssignmentOperator::Assign,
2024-02-29 02:34:14 +00:00
Statement::Expression(Expression::Value(ValueNode::List(vec![])))
)),
);
assert_eq!(
parse(&lex("foobar: list(int) = []").unwrap()).unwrap()[0].0,
Statement::Assignment(Assignment::new(
Identifier::new("foobar"),
Some(Type::ListOf(Box::new(Type::Integer))),
AssignmentOperator::Assign,
2024-02-29 02:34:14 +00:00
Statement::Expression(Expression::Value(ValueNode::List(vec![])))
)),
);
assert_eq!(
parse(&lex("foobar: [int, str] = [ 42, 'foo' ]").unwrap()).unwrap()[0].0,
Statement::Assignment(Assignment::new(
Identifier::new("foobar"),
Some(Type::ListExact(vec![Type::Integer, Type::String])),
AssignmentOperator::Assign,
2024-02-29 02:34:14 +00:00
Statement::Expression(Expression::Value(ValueNode::List(vec![
Expression::Value(ValueNode::Integer(42)),
2024-03-08 21:14:47 +00:00
Expression::Value(ValueNode::String("foo".to_string()))
2024-02-29 02:34:14 +00:00
])))
)),
);
}
2024-02-25 18:49:26 +00:00
#[test]
fn logic() {
assert_eq!(
parse(&lex("x == 1").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Logic(Box::new(Logic::Equal(
Expression::Identifier(Identifier::new("x")),
Expression::Value(ValueNode::Integer(1))
))))
2024-02-25 18:49:26 +00:00
);
2024-02-28 22:02:30 +00:00
assert_eq!(
parse(&lex("(x == 1) && (y == 2)").unwrap()).unwrap()[0].0,
Statement::Expression(Expression::Logic(Box::new(Logic::And(
Expression::Logic(Box::new(Logic::Equal(
Expression::Identifier(Identifier::new("x")),
Expression::Value(ValueNode::Integer(1))
))),
Expression::Logic(Box::new(Logic::Equal(
Expression::Identifier(Identifier::new("y")),
Expression::Value(ValueNode::Integer(2))
))),
))))
);
assert_eq!(
parse(&lex("(x == 1) && (y == 2) && true").unwrap()).unwrap()[0].0,
Statement::Expression(Expression::Logic(Box::new(Logic::And(
Expression::Logic(Box::new(Logic::And(
Expression::Logic(Box::new(Logic::Equal(
Expression::Identifier(Identifier::new("x")),
Expression::Value(ValueNode::Integer(1))
))),
Expression::Logic(Box::new(Logic::Equal(
Expression::Identifier(Identifier::new("y")),
Expression::Value(ValueNode::Integer(2))
))),
))),
Expression::Value(ValueNode::Boolean(true))
))))
);
2024-02-25 18:49:26 +00:00
}
#[test]
fn r#enum() {
assert_eq!(
parse(&lex("Option::None").unwrap()).unwrap()[0].0,
Statement::Expression(Expression::Value(ValueNode::Enum(
Identifier::new("Option"),
Identifier::new("None")
)))
);
}
2024-02-25 18:49:26 +00:00
#[test]
fn list() {
assert_eq!(
parse(&lex("[]").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::List(Vec::with_capacity(0))))
2024-02-25 18:49:26 +00:00
);
assert_eq!(
parse(&lex("[42]").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::List(vec![Expression::Value(
ValueNode::Integer(42)
)])))
2024-02-25 18:49:26 +00:00
);
assert_eq!(
parse(&lex("[42, 'foo', 'bar', [1, 2, 3,]]").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::List(vec![
Expression::Value(ValueNode::Integer(42)),
2024-03-08 21:14:47 +00:00
Expression::Value(ValueNode::String("foo".to_string())),
Expression::Value(ValueNode::String("bar".to_string())),
2024-02-26 21:27:01 +00:00
Expression::Value(ValueNode::List(vec![
Expression::Value(ValueNode::Integer(1)),
Expression::Value(ValueNode::Integer(2)),
Expression::Value(ValueNode::Integer(3)),
2024-02-25 18:49:26 +00:00
]))
2024-02-26 21:27:01 +00:00
])),)
2024-02-25 18:49:26 +00:00
);
}
#[test]
fn r#true() {
assert_eq!(
parse(&lex("true").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Boolean(true)))
2024-02-25 18:49:26 +00:00
);
}
#[test]
fn r#false() {
assert_eq!(
parse(&lex("false").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Boolean(false)))
2024-02-25 18:49:26 +00:00
);
}
#[test]
fn positive_float() {
assert_eq!(
parse(&lex("0.0").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Float(0.0)))
2024-02-25 18:49:26 +00:00
);
assert_eq!(
parse(&lex("42.0").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Float(42.0)))
2024-02-25 18:49:26 +00:00
);
let max_float = f64::MAX.to_string() + ".0";
assert_eq!(
parse(&lex(&max_float).unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Float(f64::MAX)))
2024-02-25 18:49:26 +00:00
);
let min_positive_float = f64::MIN_POSITIVE.to_string();
assert_eq!(
parse(&lex(&min_positive_float).unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Float(f64::MIN_POSITIVE)))
2024-02-25 18:49:26 +00:00
);
}
#[test]
fn negative_float() {
assert_eq!(
parse(&lex("-0.0").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Float(-0.0)))
2024-02-25 18:49:26 +00:00
);
assert_eq!(
parse(&lex("-42.0").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Float(-42.0)))
2024-02-25 18:49:26 +00:00
);
let min_float = f64::MIN.to_string() + ".0";
assert_eq!(
parse(&lex(&min_float).unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Float(f64::MIN)))
2024-02-25 18:49:26 +00:00
);
let max_negative_float = format!("-{}", f64::MIN_POSITIVE);
assert_eq!(
parse(&lex(&max_negative_float).unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Float(-f64::MIN_POSITIVE)))
2024-02-25 18:49:26 +00:00
);
}
#[test]
fn other_float() {
assert_eq!(
parse(&lex("Infinity").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Float(f64::INFINITY)))
2024-02-25 18:49:26 +00:00
);
assert_eq!(
parse(&lex("-Infinity").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Float(f64::NEG_INFINITY)))
2024-02-25 18:49:26 +00:00
);
2024-02-26 21:27:01 +00:00
if let Statement::Expression(Expression::Value(ValueNode::Float(float))) =
&parse(&lex("NaN").unwrap()).unwrap()[0].0
{
assert!(float.is_nan());
} else {
panic!("Expected a float.");
2024-02-25 18:49:26 +00:00
}
}
#[test]
fn positive_integer() {
for i in 0..10 {
let source = i.to_string();
2024-03-06 20:36:58 +00:00
let tokens = lex(&source).unwrap();
let statements = parse(&tokens).unwrap();
2024-02-25 18:49:26 +00:00
2024-02-26 21:27:01 +00:00
assert_eq!(
statements[0].0,
Statement::Expression(Expression::Value(ValueNode::Integer(i)))
)
2024-02-25 18:49:26 +00:00
}
assert_eq!(
parse(&lex("42").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Integer(42)))
2024-02-25 18:49:26 +00:00
);
let maximum_integer = i64::MAX.to_string();
assert_eq!(
parse(&lex(&maximum_integer).unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Integer(i64::MAX)))
2024-02-25 18:49:26 +00:00
);
}
#[test]
fn negative_integer() {
for i in -9..1 {
let source = i.to_string();
2024-03-06 20:36:58 +00:00
let tokens = lex(&source).unwrap();
let statements = parse(&tokens).unwrap();
2024-02-25 18:49:26 +00:00
2024-02-26 21:27:01 +00:00
assert_eq!(
statements[0].0,
Statement::Expression(Expression::Value(ValueNode::Integer(i)))
)
2024-02-25 18:49:26 +00:00
}
assert_eq!(
parse(&lex("-42").unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Integer(-42)))
2024-02-25 18:49:26 +00:00
);
let minimum_integer = i64::MIN.to_string();
assert_eq!(
parse(&lex(&minimum_integer).unwrap()).unwrap()[0].0,
2024-02-26 21:27:01 +00:00
Statement::Expression(Expression::Value(ValueNode::Integer(i64::MIN)))
2024-02-25 18:49:26 +00:00
);
}
#[test]
fn double_quoted_string() {
assert_eq!(
parse(&lex("\"\"").unwrap()).unwrap()[0].0,
2024-03-08 21:14:47 +00:00
Statement::Expression(Expression::Value(ValueNode::String("".to_string())))
2024-02-25 18:49:26 +00:00
);
assert_eq!(
parse(&lex("\"42\"").unwrap()).unwrap()[0].0,
2024-03-08 21:14:47 +00:00
Statement::Expression(Expression::Value(ValueNode::String("42".to_string())))
2024-02-25 18:49:26 +00:00
);
assert_eq!(
parse(&lex("\"foobar\"").unwrap()).unwrap()[0].0,
2024-03-08 21:14:47 +00:00
Statement::Expression(Expression::Value(ValueNode::String("foobar".to_string())))
2024-02-25 18:49:26 +00:00
);
}
#[test]
fn single_quoted_string() {
assert_eq!(
parse(&lex("''").unwrap()).unwrap()[0].0,
2024-03-08 21:14:47 +00:00
Statement::Expression(Expression::Value(ValueNode::String("".to_string())))
2024-02-25 18:49:26 +00:00
);
assert_eq!(
parse(&lex("'42'").unwrap()).unwrap()[0].0,
2024-03-08 21:14:47 +00:00
Statement::Expression(Expression::Value(ValueNode::String("42".to_string())))
2024-02-25 18:49:26 +00:00
);
assert_eq!(
parse(&lex("'foobar'").unwrap()).unwrap()[0].0,
2024-03-08 21:14:47 +00:00
Statement::Expression(Expression::Value(ValueNode::String("foobar".to_string())))
2024-02-25 18:49:26 +00:00
);
}
#[test]
fn grave_quoted_string() {
assert_eq!(
parse(&lex("``").unwrap()).unwrap()[0].0,
2024-03-08 21:14:47 +00:00
Statement::Expression(Expression::Value(ValueNode::String("".to_string())))
2024-02-25 18:49:26 +00:00
);
assert_eq!(
parse(&lex("`42`").unwrap()).unwrap()[0].0,
2024-03-08 21:14:47 +00:00
Statement::Expression(Expression::Value(ValueNode::String("42".to_string())))
2024-02-25 18:49:26 +00:00
);
assert_eq!(
parse(&lex("`foobar`").unwrap()).unwrap()[0].0,
2024-03-08 21:14:47 +00:00
Statement::Expression(Expression::Value(ValueNode::String("foobar".to_string())))
2024-02-25 18:49:26 +00:00
);
}
}