Refactor to use type checking
This commit is contained in:
parent
77814c4576
commit
c0254e8a94
@ -136,7 +136,6 @@ impl Statement {
|
|||||||
let item_type = nodes.first().unwrap().inner.expected_type(context)?;
|
let item_type = nodes.first().unwrap().inner.expected_type(context)?;
|
||||||
|
|
||||||
Some(Type::List {
|
Some(Type::List {
|
||||||
length: nodes.len(),
|
|
||||||
item_type: Box::new(item_type),
|
item_type: Box::new(item_type),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
@ -358,38 +358,14 @@ impl<'a> Analyzer<'a> {
|
|||||||
|
|
||||||
#[derive(Clone, Debug, PartialEq)]
|
#[derive(Clone, Debug, PartialEq)]
|
||||||
pub enum AnalyzerError {
|
pub enum AnalyzerError {
|
||||||
ExpectedBoolean {
|
|
||||||
actual: Node<Statement>,
|
|
||||||
position: Span,
|
|
||||||
},
|
|
||||||
ExpectedFloat {
|
|
||||||
actual: Node<Statement>,
|
|
||||||
position: (usize, usize),
|
|
||||||
},
|
|
||||||
ExpectedFunction {
|
|
||||||
actual: Node<Statement>,
|
|
||||||
position: Span,
|
|
||||||
},
|
|
||||||
ExpectedIdentifier {
|
ExpectedIdentifier {
|
||||||
actual: Node<Statement>,
|
actual: Node<Statement>,
|
||||||
position: Span,
|
position: Span,
|
||||||
},
|
},
|
||||||
ExpectedInteger {
|
ExpectedBoolean {
|
||||||
actual: Node<Statement>,
|
actual: Node<Statement>,
|
||||||
position: Span,
|
position: Span,
|
||||||
},
|
},
|
||||||
ExpectedNumber {
|
|
||||||
actual: Node<Statement>,
|
|
||||||
position: Span,
|
|
||||||
},
|
|
||||||
ExpectedNumberOrString {
|
|
||||||
actual: Node<Statement>,
|
|
||||||
position: Span,
|
|
||||||
},
|
|
||||||
ExpectedString {
|
|
||||||
actual: Node<Statement>,
|
|
||||||
position: (usize, usize),
|
|
||||||
},
|
|
||||||
ExpectedValue {
|
ExpectedValue {
|
||||||
actual: Node<Statement>,
|
actual: Node<Statement>,
|
||||||
},
|
},
|
||||||
@ -418,14 +394,8 @@ impl AnalyzerError {
|
|||||||
pub fn position(&self) -> Span {
|
pub fn position(&self) -> Span {
|
||||||
match self {
|
match self {
|
||||||
AnalyzerError::ExpectedBoolean { position, .. } => *position,
|
AnalyzerError::ExpectedBoolean { position, .. } => *position,
|
||||||
AnalyzerError::ExpectedFloat { position, .. } => *position,
|
|
||||||
AnalyzerError::ExpectedFunction { position, .. } => *position,
|
|
||||||
AnalyzerError::ExpectedIdentifier { position, .. } => *position,
|
AnalyzerError::ExpectedIdentifier { position, .. } => *position,
|
||||||
AnalyzerError::ExpectedValue { actual } => actual.position,
|
AnalyzerError::ExpectedValue { actual } => actual.position,
|
||||||
AnalyzerError::ExpectedInteger { position, .. } => *position,
|
|
||||||
AnalyzerError::ExpectedNumber { position, .. } => *position,
|
|
||||||
AnalyzerError::ExpectedNumberOrString { position, .. } => *position,
|
|
||||||
AnalyzerError::ExpectedString { position, .. } => *position,
|
|
||||||
AnalyzerError::ExpectedValueArgumentCount { position, .. } => *position,
|
AnalyzerError::ExpectedValueArgumentCount { position, .. } => *position,
|
||||||
AnalyzerError::TypeConflict {
|
AnalyzerError::TypeConflict {
|
||||||
actual_statement, ..
|
actual_statement, ..
|
||||||
@ -445,27 +415,9 @@ impl Display for AnalyzerError {
|
|||||||
AnalyzerError::ExpectedBoolean { actual, .. } => {
|
AnalyzerError::ExpectedBoolean { actual, .. } => {
|
||||||
write!(f, "Expected boolean, found {}", actual)
|
write!(f, "Expected boolean, found {}", actual)
|
||||||
}
|
}
|
||||||
AnalyzerError::ExpectedFunction { actual, .. } => {
|
|
||||||
write!(f, "Expected function, found {}", actual)
|
|
||||||
}
|
|
||||||
AnalyzerError::ExpectedFloat { actual, .. } => {
|
|
||||||
write!(f, "Expected float, found {}", actual)
|
|
||||||
}
|
|
||||||
AnalyzerError::ExpectedIdentifier { actual, .. } => {
|
AnalyzerError::ExpectedIdentifier { actual, .. } => {
|
||||||
write!(f, "Expected identifier, found {}", actual)
|
write!(f, "Expected identifier, found {}", actual)
|
||||||
}
|
}
|
||||||
AnalyzerError::ExpectedInteger { actual, .. } => {
|
|
||||||
write!(f, "Expected integer, found {}", actual)
|
|
||||||
}
|
|
||||||
AnalyzerError::ExpectedNumber { actual, .. } => {
|
|
||||||
write!(f, "Expected integer or float, found {}", actual)
|
|
||||||
}
|
|
||||||
AnalyzerError::ExpectedNumberOrString { actual, .. } => {
|
|
||||||
write!(f, "Expected integer, float, or string, found {}", actual)
|
|
||||||
}
|
|
||||||
AnalyzerError::ExpectedString { actual, .. } => {
|
|
||||||
write!(f, "Expected string, found {}", actual)
|
|
||||||
}
|
|
||||||
AnalyzerError::ExpectedValue { actual, .. } => {
|
AnalyzerError::ExpectedValue { actual, .. } => {
|
||||||
write!(f, "Expected value, found {}", actual)
|
write!(f, "Expected value, found {}", actual)
|
||||||
}
|
}
|
||||||
@ -498,26 +450,10 @@ impl Display for AnalyzerError {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use crate::{BuiltInFunction, Identifier, Value};
|
use crate::{Identifier, Value};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn is_even_wrong_type() {
|
|
||||||
let source = "is_even('hello')";
|
|
||||||
|
|
||||||
assert_eq!(
|
|
||||||
analyze(source),
|
|
||||||
Err(DustError::AnalyzerError {
|
|
||||||
analyzer_error: AnalyzerError::ExpectedInteger {
|
|
||||||
actual: Node::new(Statement::Constant(Value::string("hello")), (1, 1)),
|
|
||||||
position: (1, 1)
|
|
||||||
},
|
|
||||||
source
|
|
||||||
})
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn length_no_arguments() {
|
fn length_no_arguments() {
|
||||||
let source = "length()";
|
let source = "length()";
|
||||||
@ -554,107 +490,73 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn integer_plus_boolean() {
|
fn integer_plus_boolean() {
|
||||||
let abstract_tree = AbstractSyntaxTree {
|
let source = "42 + true";
|
||||||
nodes: [Node::new(
|
|
||||||
Statement::BinaryOperation {
|
|
||||||
left: Box::new(Node::new(Statement::Constant(Value::integer(1)), (0, 1))),
|
|
||||||
operator: Node::new(BinaryOperator::Add, (1, 2)),
|
|
||||||
right: Box::new(Node::new(Statement::Constant(Value::boolean(true)), (3, 4))),
|
|
||||||
},
|
|
||||||
(0, 2),
|
|
||||||
)]
|
|
||||||
.into(),
|
|
||||||
};
|
|
||||||
let mut context = Context::new();
|
|
||||||
let mut analyzer = Analyzer::new(&abstract_tree, &mut context);
|
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
analyzer.analyze(),
|
analyze(source),
|
||||||
Err(AnalyzerError::ExpectedInteger {
|
Err(DustError::AnalyzerError {
|
||||||
actual: Node::new(Statement::Constant(Value::boolean(true)), (3, 4)),
|
analyzer_error: AnalyzerError::TypeConflict {
|
||||||
position: (3, 4)
|
actual_statement: Node::new(Statement::Constant(Value::boolean(true)), (5, 9)),
|
||||||
|
actual_type: Type::Boolean,
|
||||||
|
expected: Type::Integer,
|
||||||
|
},
|
||||||
|
source
|
||||||
})
|
})
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn is_even_expects_number() {
|
fn is_even_expects_number() {
|
||||||
let abstract_tree = AbstractSyntaxTree {
|
let source = "is_even('hello')";
|
||||||
nodes: [Node::new(
|
|
||||||
Statement::PropertyAccess(
|
|
||||||
Box::new(Node::new(Statement::Constant(Value::boolean(true)), (0, 1))),
|
|
||||||
Box::new(Node::new(
|
|
||||||
Statement::BuiltInFunctionCall {
|
|
||||||
function: BuiltInFunction::IsEven,
|
|
||||||
type_arguments: None,
|
|
||||||
value_arguments: None,
|
|
||||||
},
|
|
||||||
(1, 2),
|
|
||||||
)),
|
|
||||||
),
|
|
||||||
(0, 2),
|
|
||||||
)]
|
|
||||||
.into(),
|
|
||||||
};
|
|
||||||
let mut context = Context::new();
|
|
||||||
let mut analyzer = Analyzer::new(&abstract_tree, &mut context);
|
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
analyzer.analyze(),
|
analyze(source),
|
||||||
Err(AnalyzerError::ExpectedNumber {
|
Err(DustError::AnalyzerError {
|
||||||
actual: Node::new(Statement::Constant(Value::boolean(true)), (0, 1)),
|
analyzer_error: AnalyzerError::TypeConflict {
|
||||||
position: (0, 1)
|
actual_statement: Node::new(
|
||||||
|
Statement::Constant(Value::string("hello")),
|
||||||
|
(8, 15)
|
||||||
|
),
|
||||||
|
actual_type: Type::String,
|
||||||
|
expected: Type::Number,
|
||||||
|
},
|
||||||
|
source
|
||||||
})
|
})
|
||||||
)
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn is_odd_expects_number() {
|
fn is_odd_expects_number() {
|
||||||
let abstract_tree = AbstractSyntaxTree {
|
let source = "is_odd('hello')";
|
||||||
nodes: [Node::new(
|
|
||||||
Statement::PropertyAccess(
|
|
||||||
Box::new(Node::new(Statement::Constant(Value::boolean(true)), (0, 1))),
|
|
||||||
Box::new(Node::new(
|
|
||||||
Statement::BuiltInFunctionCall {
|
|
||||||
function: BuiltInFunction::IsOdd,
|
|
||||||
type_arguments: None,
|
|
||||||
value_arguments: None,
|
|
||||||
},
|
|
||||||
(1, 2),
|
|
||||||
)),
|
|
||||||
),
|
|
||||||
(0, 2),
|
|
||||||
)]
|
|
||||||
.into(),
|
|
||||||
};
|
|
||||||
let mut context = Context::new();
|
|
||||||
let mut analyzer = Analyzer::new(&abstract_tree, &mut context);
|
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
analyzer.analyze(),
|
analyze(source),
|
||||||
Err(AnalyzerError::ExpectedNumber {
|
Err(DustError::AnalyzerError {
|
||||||
actual: Node::new(Statement::Constant(Value::boolean(true)), (0, 1)),
|
analyzer_error: AnalyzerError::TypeConflict {
|
||||||
position: (0, 1)
|
actual_statement: Node::new(
|
||||||
|
Statement::Constant(Value::string("hello")),
|
||||||
|
(7, 14)
|
||||||
|
),
|
||||||
|
actual_type: Type::String,
|
||||||
|
expected: Type::Number,
|
||||||
|
},
|
||||||
|
source
|
||||||
})
|
})
|
||||||
)
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn undefined_variable() {
|
fn undefined_variable() {
|
||||||
let abstract_tree = AbstractSyntaxTree {
|
let source = "foo";
|
||||||
nodes: [Node::new(
|
|
||||||
Statement::Identifier(Identifier::new("x")),
|
|
||||||
(0, 1),
|
|
||||||
)]
|
|
||||||
.into(),
|
|
||||||
};
|
|
||||||
let mut context = Context::new();
|
|
||||||
let mut analyzer = Analyzer::new(&abstract_tree, &mut context);
|
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
analyzer.analyze(),
|
analyze(source),
|
||||||
Err(AnalyzerError::UndefinedVariable {
|
Err(DustError::AnalyzerError {
|
||||||
identifier: Node::new(Statement::Identifier(Identifier::new("x")), (0, 1))
|
analyzer_error: AnalyzerError::UndefinedVariable {
|
||||||
|
identifier: Node::new(Statement::Identifier(Identifier::new("foo")), (0, 3)),
|
||||||
|
},
|
||||||
|
source
|
||||||
})
|
})
|
||||||
)
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -49,7 +49,6 @@ impl BuiltInFunction {
|
|||||||
"value".into(),
|
"value".into(),
|
||||||
Type::List {
|
Type::List {
|
||||||
item_type: Box::new(Type::Any),
|
item_type: Box::new(Type::Any),
|
||||||
length: 1,
|
|
||||||
},
|
},
|
||||||
)]
|
)]
|
||||||
}
|
}
|
||||||
|
@ -48,7 +48,6 @@ pub enum Type {
|
|||||||
},
|
},
|
||||||
Integer,
|
Integer,
|
||||||
List {
|
List {
|
||||||
length: usize,
|
|
||||||
item_type: Box<Type>,
|
item_type: Box<Type>,
|
||||||
},
|
},
|
||||||
Map(BTreeMap<Identifier, Type>),
|
Map(BTreeMap<Identifier, Type>),
|
||||||
@ -138,15 +137,13 @@ impl Type {
|
|||||||
}
|
}
|
||||||
(
|
(
|
||||||
Type::List {
|
Type::List {
|
||||||
length: left_length,
|
|
||||||
item_type: left_type,
|
item_type: left_type,
|
||||||
},
|
},
|
||||||
Type::List {
|
Type::List {
|
||||||
length: right_length,
|
|
||||||
item_type: right_type,
|
item_type: right_type,
|
||||||
},
|
},
|
||||||
) => {
|
) => {
|
||||||
if left_length != right_length || left_type != right_type {
|
if left_type.check(right_type).is_err() {
|
||||||
return Err(TypeConflict {
|
return Err(TypeConflict {
|
||||||
actual: other.clone(),
|
actual: other.clone(),
|
||||||
expected: self.clone(),
|
expected: self.clone(),
|
||||||
@ -249,7 +246,7 @@ impl Display for Type {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
Type::Integer => write!(f, "int"),
|
Type::Integer => write!(f, "int"),
|
||||||
Type::List { length, item_type } => write!(f, "[{length}; {}]", item_type),
|
Type::List { item_type } => write!(f, "[{item_type}]"),
|
||||||
Type::Map(map) => {
|
Type::Map(map) => {
|
||||||
write!(f, "{{ ")?;
|
write!(f, "{{ ")?;
|
||||||
|
|
||||||
@ -316,11 +313,9 @@ mod tests {
|
|||||||
assert_eq!(Type::Integer.check(&Type::Integer), Ok(()));
|
assert_eq!(Type::Integer.check(&Type::Integer), Ok(()));
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
Type::List {
|
Type::List {
|
||||||
length: 4,
|
|
||||||
item_type: Box::new(Type::Boolean),
|
item_type: Box::new(Type::Boolean),
|
||||||
}
|
}
|
||||||
.check(&Type::List {
|
.check(&Type::List {
|
||||||
length: 4,
|
|
||||||
item_type: Box::new(Type::Boolean),
|
item_type: Box::new(Type::Boolean),
|
||||||
}),
|
}),
|
||||||
Ok(())
|
Ok(())
|
||||||
@ -362,7 +357,6 @@ mod tests {
|
|||||||
Type::Float,
|
Type::Float,
|
||||||
Type::Integer,
|
Type::Integer,
|
||||||
Type::List {
|
Type::List {
|
||||||
length: 10,
|
|
||||||
item_type: Box::new(Type::Integer),
|
item_type: Box::new(Type::Integer),
|
||||||
},
|
},
|
||||||
Type::Map(BTreeMap::new()),
|
Type::Map(BTreeMap::new()),
|
||||||
|
@ -640,7 +640,6 @@ impl ValueInner {
|
|||||||
let item_type = values.first().unwrap().r#type(context);
|
let item_type = values.first().unwrap().r#type(context);
|
||||||
|
|
||||||
Type::List {
|
Type::List {
|
||||||
length: values.len(),
|
|
||||||
item_type: Box::new(item_type),
|
item_type: Box::new(item_type),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -464,42 +464,6 @@ impl Vm {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if let (
|
|
||||||
value,
|
|
||||||
Statement::BuiltInFunctionCall {
|
|
||||||
function,
|
|
||||||
type_arguments: _,
|
|
||||||
value_arguments: value_argument_nodes,
|
|
||||||
},
|
|
||||||
) = (left_value, right.inner)
|
|
||||||
{
|
|
||||||
let mut value_arguments = Vec::new();
|
|
||||||
|
|
||||||
value_arguments.push(value);
|
|
||||||
|
|
||||||
if let Some(value_nodes) = value_argument_nodes {
|
|
||||||
for node in value_nodes {
|
|
||||||
let position = node.position;
|
|
||||||
let value = if let Some(value) = self.run_node(node, context)? {
|
|
||||||
value
|
|
||||||
} else {
|
|
||||||
return Err(VmError::ExpectedValue { position });
|
|
||||||
};
|
|
||||||
|
|
||||||
value_arguments.push(value);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let function_call_return = function.call(None, Some(value_arguments)).map_err(
|
|
||||||
|built_in_function_error| VmError::BuiltInFunctionError {
|
|
||||||
error: built_in_function_error,
|
|
||||||
position: right_span,
|
|
||||||
},
|
|
||||||
)?;
|
|
||||||
|
|
||||||
return Ok(function_call_return);
|
|
||||||
}
|
|
||||||
|
|
||||||
Err(VmError::ExpectedIdentifierOrInteger {
|
Err(VmError::ExpectedIdentifierOrInteger {
|
||||||
position: right_span,
|
position: right_span,
|
||||||
})
|
})
|
||||||
@ -648,7 +612,7 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn to_string() {
|
fn to_string() {
|
||||||
let input = "42.to_string()";
|
let input = "to_string(42)";
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
run(input, &mut Context::new()),
|
run(input, &mut Context::new()),
|
||||||
@ -824,7 +788,7 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn is_even() {
|
fn is_even() {
|
||||||
let input = "42.is_even()";
|
let input = "is_even(42)";
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
run(input, &mut Context::new()),
|
run(input, &mut Context::new()),
|
||||||
@ -834,7 +798,7 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn is_odd() {
|
fn is_odd() {
|
||||||
let input = "42.is_odd()";
|
let input = "is_odd(42)";
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
run(input, &mut Context::new()),
|
run(input, &mut Context::new()),
|
||||||
@ -844,7 +808,7 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn length() {
|
fn length() {
|
||||||
let input = "[1, 2, 3].length()";
|
let input = "length([1, 2, 3])";
|
||||||
|
|
||||||
assert_eq!(run(input, &mut Context::new()), Ok(Some(Value::integer(3))));
|
assert_eq!(run(input, &mut Context::new()), Ok(Some(Value::integer(3))));
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user