Refactor to use type checking

This commit is contained in:
Jeff 2024-08-11 19:18:13 -04:00
parent 77814c4576
commit c0254e8a94
6 changed files with 52 additions and 195 deletions

View File

@ -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),
}) })
} }

View File

@ -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
}) })
) );
} }
} }

View File

@ -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,
}, },
)] )]
} }

View File

@ -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()),

View File

@ -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),
} }
} }

View File

@ -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))));
} }