updated rust ast through structs to work
This commit is contained in:
165
src/ast.rs
165
src/ast.rs
@@ -1,86 +1,73 @@
|
||||
use std::collections::HashMap;
|
||||
use crate::types;
|
||||
|
||||
use std::cell::RefCell;
|
||||
|
||||
pub struct IdGenerator {
|
||||
counter: i64,
|
||||
counter: RefCell<i64>,
|
||||
}
|
||||
|
||||
impl IdGenerator {
|
||||
pub fn new() -> Self {
|
||||
IdGenerator{counter: 0}
|
||||
IdGenerator{counter: RefCell::new(0)}
|
||||
}
|
||||
|
||||
pub fn next(&mut self) -> String {
|
||||
self.counter += 1;
|
||||
("S" + self.counter.to_string()).to_string()
|
||||
pub fn next(&self) -> String {
|
||||
*self.counter.borrow_mut() += 1;
|
||||
("S".to_owned() + &self.counter.borrow().to_string()).to_string()
|
||||
}
|
||||
}
|
||||
|
||||
pub fn new_named(name: String) -> {
|
||||
ast::TypeUsage::Named(ast::NamedTypeUsage{
|
||||
name: ast::Identifier{
|
||||
name: ast::Spanned{
|
||||
span: ast::Span{left: 0, right: 0}, //todo: figure out a sane value for these
|
||||
value: name,
|
||||
}
|
||||
}
|
||||
)
|
||||
}
|
||||
|
||||
pub fn new_unit() -> {
|
||||
ast::TypeUsage::Named(ast::NamedTypeUsage{
|
||||
name: ast::Identifier{
|
||||
name: ast::Spanned{
|
||||
span: ast::Span{left: 0, right: 0}, //todo: figure out a sane value for these
|
||||
pub fn new_unit() -> TypeUsage {
|
||||
TypeUsage::Named(NamedTypeUsage{
|
||||
name: Identifier{
|
||||
name: Spanned{
|
||||
span: Span{left: 0, right: 0}, //todo: figure out a sane value for these
|
||||
value: "()".to_string(),
|
||||
}
|
||||
}
|
||||
)
|
||||
})
|
||||
}
|
||||
|
||||
pub fn new_never() -> {
|
||||
ast::TypeUsage::Named(ast::NamedTypeUsage{
|
||||
name: ast::Identifier{
|
||||
name: ast::Spanned{
|
||||
span: ast::Span{left: 0, right: 0}, //todo: figure out a sane value for these
|
||||
pub fn new_never() -> TypeUsage {
|
||||
TypeUsage::Named(NamedTypeUsage{
|
||||
name: Identifier{
|
||||
name: Spanned{
|
||||
span: Span{left: 0, right: 0}, //todo: figure out a sane value for these
|
||||
value: "!".to_string(),
|
||||
}
|
||||
}
|
||||
)
|
||||
})
|
||||
}
|
||||
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct Span {
|
||||
pub left: usize,
|
||||
pub right: usize
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct Spanned<T> {
|
||||
pub span: Span,
|
||||
pub value: T,
|
||||
}
|
||||
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct FunctionTypeUsage {
|
||||
pub arguments: Vec<TypeUsage>,
|
||||
pub return_type: Box<TypeUsage>,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct NamedTypeUsage {
|
||||
name: Identifier,
|
||||
pub name: Identifier,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct UnknownTypeUsage {
|
||||
name: String,
|
||||
pub name: String,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub enum TypeUsage {
|
||||
Function(FunctionTypeUsage),
|
||||
Named(NamedTypeUsage),
|
||||
@@ -88,38 +75,38 @@ pub enum TypeUsage {
|
||||
}
|
||||
|
||||
impl TypeUsage {
|
||||
pub fn new_unknown(id_gen: &mut IdGenerator) -> TypeUsage {
|
||||
pub fn new_unknown(id_gen: &IdGenerator) -> TypeUsage {
|
||||
return TypeUsage::Unknown(UnknownTypeUsage{
|
||||
name: id_gen.next(),
|
||||
});
|
||||
}
|
||||
|
||||
pub fn new_named(identifier: &Identifier) -> TypeUsage {
|
||||
pub fn new_named(identifier: Identifier) -> TypeUsage {
|
||||
return TypeUsage::Named(NamedTypeUsage{
|
||||
name: identifier.clone(),
|
||||
});
|
||||
}
|
||||
|
||||
pub fn new_builtin(name: String) -> TypeUsage {
|
||||
ast::TypeUsage::Named(ast::NamedTypeUsage{
|
||||
name: ast::Identifier{
|
||||
name: ast::Spanned{
|
||||
span: ast::Span{left: 0, right: 0}, //todo: figure out a sane value for these
|
||||
TypeUsage::Named(NamedTypeUsage{
|
||||
name: Identifier{
|
||||
name: Spanned{
|
||||
span: Span{left: 0, right: 0}, //todo: figure out a sane value for these
|
||||
value: name,
|
||||
}
|
||||
}
|
||||
)
|
||||
})
|
||||
}
|
||||
|
||||
pub fn new_function(arg_count: usize, id_gen: &mut IdGenerator) -> TypeUsage {
|
||||
pub fn new_function(arg_count: usize, id_gen: &IdGenerator) -> TypeUsage {
|
||||
return TypeUsage::Function(FunctionTypeUsage{
|
||||
arguments: 0..arg_count.map(|_| => TypeUsage.new_unknown(&mut id_gen)).collect(),
|
||||
return_type: Box::new(TypeUsage.new_unknown(&mut id_gen)),
|
||||
arguments: (0..arg_count).map(|_| TypeUsage::new_unknown(&id_gen)).collect(),
|
||||
return_type: Box::new(TypeUsage::new_unknown(&id_gen)),
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub enum Operator {
|
||||
Mul,
|
||||
Div,
|
||||
@@ -127,98 +114,100 @@ pub enum Operator {
|
||||
Minus,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct LiteralInt {
|
||||
pub value: Spanned<i64>,
|
||||
pub value: Spanned<String>,
|
||||
pub type_: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct LiteralFloat {
|
||||
pub value: Spanned<f64>,
|
||||
pub value: Spanned<String>,
|
||||
pub type_: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct LiteralStruct {
|
||||
pub name: Identifier,
|
||||
pub fields: HashMap<Identifier, Expression>,
|
||||
pub fields: Vec<(Identifier, Expression)>,
|
||||
pub type_: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct Identifier {
|
||||
pub name: Spanned<String>,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct FunctionCall {
|
||||
pub source: Expression,
|
||||
pub arguments: Vec<Expression>,
|
||||
pub type_: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct StructGetter {
|
||||
pub source: Expression,
|
||||
pub attribute: Identifier,
|
||||
pub type_: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct Operation {
|
||||
pub left: Expression,
|
||||
pub op: Operator,
|
||||
pub right: Expression,
|
||||
pub type_: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct VariableUsage {
|
||||
pub name: Identifier,
|
||||
pub type_: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub enum Subexpression {
|
||||
LiteralInt(LiteralInt),
|
||||
LiteralFloat(LiteralFloat),
|
||||
LiteralStruct(LiteralStruct),
|
||||
FunctionCall(FunctionCall),
|
||||
Identifier(Identifier),
|
||||
VariableUsage(VariableUsage),
|
||||
StructGetter(StructGetter),
|
||||
Block(Block),
|
||||
Op(Operation),
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct Expression {
|
||||
pub subexpression: Spanned<Box<Subexpression>>,
|
||||
pub subexpression: Box<Subexpression>,
|
||||
pub type_: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct ReturnStatement {
|
||||
pub source: Expression,
|
||||
};
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
pub struct LetStatement {
|
||||
variable_name: Identifier,
|
||||
expression: Expression,
|
||||
type_: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct LetStatement {
|
||||
pub variable_name: Identifier,
|
||||
pub expression: Expression,
|
||||
pub type_: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub enum AssignmentTarget {
|
||||
Variable(VariableUsage),
|
||||
StructAttr(StructGetter),
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct AssignmentStatement {
|
||||
pub source: AssignmentTarget,
|
||||
pub expression: Expression,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub enum Statement {
|
||||
Return(ReturnStatement),
|
||||
Let(LetStatement),
|
||||
@@ -226,75 +215,75 @@ pub enum Statement {
|
||||
Expression(Expression),
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct Block {
|
||||
pub statements: Vec<Statement>,
|
||||
pub type_: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct VariableDeclaration {
|
||||
pub name: Identifier,
|
||||
pub type_: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct FunctionDeclaration {
|
||||
pub name: Identifier,
|
||||
pub arguments: Vec<VariableDeclaration>,
|
||||
pub return_type: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct Function {
|
||||
pub declaration: FunctionDeclaration,
|
||||
pub block: Block,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct PrimitiveTypeDeclaration {
|
||||
pub name: String, // cannot be identifier as it's not declared anywhere specific, it's builtins
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct StructField {
|
||||
pub name: Identifier,
|
||||
pub type_: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct StructTypeDeclaration {
|
||||
pub name: Identifier,
|
||||
pub fields: Vec<StructField>,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct AliasTypeDeclaration {
|
||||
pub name: Identifier,
|
||||
pub replaces: TypeUsage,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub enum TypeDeclaration {
|
||||
Struct(StructTypeDeclaration),
|
||||
Primitive(PrimitiveTypeDeclaration),
|
||||
Alias(AliasTypeDeclaration),
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct Impl {
|
||||
pub struct_name: Identifier,
|
||||
pub functions: Vec<Function>,
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub enum ModuleItem {
|
||||
Function(Function),
|
||||
TypeDeclaration(TypeDeclaration),
|
||||
Impl(Impl),
|
||||
}
|
||||
|
||||
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
|
||||
pub struct Module {
|
||||
pub items: Vec<ModuleItem>,
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user