Run rustfmt on ast module
This commit is contained in:
parent
765a0bec58
commit
5d04a020dc
@ -1,18 +1,22 @@
|
|||||||
#![allow(clippy::upper_case_acronyms)]
|
#![allow(clippy::upper_case_acronyms)]
|
||||||
#![allow(clippy::enum_variant_names)]
|
#![allow(clippy::enum_variant_names)]
|
||||||
|
|
||||||
use std::rc::Rc;
|
use std::{
|
||||||
use std::convert::{AsRef, From};
|
convert::{AsRef, From},
|
||||||
|
rc::Rc,
|
||||||
|
};
|
||||||
|
|
||||||
mod visitor;
|
|
||||||
mod operators;
|
mod operators;
|
||||||
|
mod visitor;
|
||||||
|
|
||||||
pub use operators::{PrefixOp, BinOp};
|
pub use operators::{BinOp, PrefixOp};
|
||||||
pub use visitor::*;
|
pub use visitor::*;
|
||||||
|
|
||||||
use crate::derivative::Derivative;
|
use crate::{
|
||||||
use crate::tokenizing::Location;
|
derivative::Derivative,
|
||||||
use crate::identifier::{Id, define_id_kind};
|
identifier::{define_id_kind, Id},
|
||||||
|
tokenizing::Location,
|
||||||
|
};
|
||||||
|
|
||||||
define_id_kind!(ASTItem);
|
define_id_kind!(ASTItem);
|
||||||
|
|
||||||
@ -54,7 +58,7 @@ pub enum StatementKind {
|
|||||||
|
|
||||||
#[derive(Debug, Clone, PartialEq, Default)]
|
#[derive(Debug, Clone, PartialEq, Default)]
|
||||||
pub struct Block {
|
pub struct Block {
|
||||||
pub statements: Vec<Statement>
|
pub statements: Vec<Statement>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl From<Vec<Statement>> for Block {
|
impl From<Vec<Statement>> for Block {
|
||||||
@ -89,42 +93,20 @@ pub struct QualifiedName {
|
|||||||
pub struct FormalParam {
|
pub struct FormalParam {
|
||||||
pub name: ParamName,
|
pub name: ParamName,
|
||||||
pub default: Option<Expression>,
|
pub default: Option<Expression>,
|
||||||
pub anno: Option<TypeIdentifier>
|
pub anno: Option<TypeIdentifier>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, PartialEq, Clone)]
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
pub enum Declaration {
|
pub enum Declaration {
|
||||||
FuncSig(Signature),
|
FuncSig(Signature),
|
||||||
FuncDecl(Signature, Block),
|
FuncDecl(Signature, Block),
|
||||||
TypeDecl {
|
TypeDecl { name: TypeSingletonName, body: TypeBody, mutable: bool },
|
||||||
name: TypeSingletonName,
|
|
||||||
body: TypeBody,
|
|
||||||
mutable: bool
|
|
||||||
},
|
|
||||||
//TODO this needs to be more sophisticated
|
//TODO this needs to be more sophisticated
|
||||||
TypeAlias {
|
TypeAlias { alias: Rc<String>, original: Rc<String> },
|
||||||
alias: Rc<String>,
|
Binding { name: Rc<String>, constant: bool, type_anno: Option<TypeIdentifier>, expr: Expression },
|
||||||
original: Rc<String>,
|
Impl { type_name: TypeIdentifier, interface_name: Option<TypeSingletonName>, block: Vec<Declaration> },
|
||||||
},
|
Interface { name: Rc<String>, signatures: Vec<Signature> },
|
||||||
Binding {
|
Annotation { name: Rc<String>, arguments: Vec<Expression> },
|
||||||
name: Rc<String>,
|
|
||||||
constant: bool,
|
|
||||||
type_anno: Option<TypeIdentifier>,
|
|
||||||
expr: Expression,
|
|
||||||
},
|
|
||||||
Impl {
|
|
||||||
type_name: TypeIdentifier,
|
|
||||||
interface_name: Option<TypeSingletonName>,
|
|
||||||
block: Vec<Declaration>,
|
|
||||||
},
|
|
||||||
Interface {
|
|
||||||
name: Rc<String>,
|
|
||||||
signatures: Vec<Signature>
|
|
||||||
},
|
|
||||||
Annotation {
|
|
||||||
name: Rc<String>,
|
|
||||||
arguments: Vec<Expression>
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, PartialEq, Clone)]
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
@ -160,7 +142,7 @@ pub struct Expression {
|
|||||||
#[derivative(PartialEq = "ignore")]
|
#[derivative(PartialEq = "ignore")]
|
||||||
pub id: ItemId,
|
pub id: ItemId,
|
||||||
pub kind: ExpressionKind,
|
pub kind: ExpressionKind,
|
||||||
pub type_anno: Option<TypeIdentifier>
|
pub type_anno: Option<TypeIdentifier>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Expression {
|
impl Expression {
|
||||||
@ -177,7 +159,7 @@ impl Expression {
|
|||||||
#[derive(Debug, PartialEq, Clone)]
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
pub enum TypeIdentifier {
|
pub enum TypeIdentifier {
|
||||||
Tuple(Vec<TypeIdentifier>),
|
Tuple(Vec<TypeIdentifier>),
|
||||||
Singleton(TypeSingletonName)
|
Singleton(TypeSingletonName),
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, PartialEq, Clone)]
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
@ -196,60 +178,28 @@ pub enum ExpressionKind {
|
|||||||
PrefixExp(PrefixOp, Box<Expression>),
|
PrefixExp(PrefixOp, Box<Expression>),
|
||||||
TupleLiteral(Vec<Expression>),
|
TupleLiteral(Vec<Expression>),
|
||||||
Value(QualifiedName),
|
Value(QualifiedName),
|
||||||
NamedStruct {
|
NamedStruct { name: QualifiedName, fields: Vec<(Rc<String>, Expression)> },
|
||||||
name: QualifiedName,
|
Call { f: Box<Expression>, arguments: Vec<InvocationArgument> },
|
||||||
fields: Vec<(Rc<String>, Expression)>,
|
Index { indexee: Box<Expression>, indexers: Vec<Expression> },
|
||||||
},
|
IfExpression { discriminator: Option<Box<Expression>>, body: Box<IfExpressionBody> },
|
||||||
Call {
|
WhileExpression { condition: Option<Box<Expression>>, body: Block },
|
||||||
f: Box<Expression>,
|
ForExpression { enumerators: Vec<Enumerator>, body: Box<ForBody> },
|
||||||
arguments: Vec<InvocationArgument>,
|
Lambda { params: Vec<FormalParam>, type_anno: Option<TypeIdentifier>, body: Block },
|
||||||
},
|
|
||||||
Index {
|
|
||||||
indexee: Box<Expression>,
|
|
||||||
indexers: Vec<Expression>,
|
|
||||||
},
|
|
||||||
IfExpression {
|
|
||||||
discriminator: Option<Box<Expression>>,
|
|
||||||
body: Box<IfExpressionBody>,
|
|
||||||
},
|
|
||||||
WhileExpression {
|
|
||||||
condition: Option<Box<Expression>>,
|
|
||||||
body: Block,
|
|
||||||
},
|
|
||||||
ForExpression {
|
|
||||||
enumerators: Vec<Enumerator>,
|
|
||||||
body: Box<ForBody>,
|
|
||||||
},
|
|
||||||
Lambda {
|
|
||||||
params: Vec<FormalParam>,
|
|
||||||
type_anno: Option<TypeIdentifier>,
|
|
||||||
body: Block,
|
|
||||||
},
|
|
||||||
ListLiteral(Vec<Expression>),
|
ListLiteral(Vec<Expression>),
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, PartialEq, Clone)]
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
pub enum InvocationArgument {
|
pub enum InvocationArgument {
|
||||||
Positional(Expression),
|
Positional(Expression),
|
||||||
Keyword {
|
Keyword { name: Rc<String>, expr: Expression },
|
||||||
name: Rc<String>,
|
Ignored,
|
||||||
expr: Expression,
|
|
||||||
},
|
|
||||||
Ignored
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, PartialEq, Clone)]
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
pub enum IfExpressionBody {
|
pub enum IfExpressionBody {
|
||||||
SimpleConditional {
|
SimpleConditional { then_case: Block, else_case: Option<Block> },
|
||||||
then_case: Block,
|
SimplePatternMatch { pattern: Pattern, then_case: Block, else_case: Option<Block> },
|
||||||
else_case: Option<Block>
|
CondList(Vec<ConditionArm>),
|
||||||
},
|
|
||||||
SimplePatternMatch {
|
|
||||||
pattern: Pattern,
|
|
||||||
then_case: Block,
|
|
||||||
else_case: Option<Block>
|
|
||||||
},
|
|
||||||
CondList(Vec<ConditionArm>)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, PartialEq, Clone)]
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
@ -279,10 +229,7 @@ pub enum Pattern {
|
|||||||
|
|
||||||
#[derive(Debug, PartialEq, Clone)]
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
pub enum PatternLiteral {
|
pub enum PatternLiteral {
|
||||||
NumPattern {
|
NumPattern { neg: bool, num: ExpressionKind },
|
||||||
neg: bool,
|
|
||||||
num: ExpressionKind,
|
|
||||||
},
|
|
||||||
StringPattern(Rc<String>),
|
StringPattern(Rc<String>),
|
||||||
BoolPattern(bool),
|
BoolPattern(bool),
|
||||||
}
|
}
|
||||||
@ -305,20 +252,18 @@ pub struct ImportSpecifier {
|
|||||||
#[derivative(PartialEq = "ignore")]
|
#[derivative(PartialEq = "ignore")]
|
||||||
pub id: ItemId,
|
pub id: ItemId,
|
||||||
pub path_components: Vec<Rc<String>>,
|
pub path_components: Vec<Rc<String>>,
|
||||||
pub imported_names: ImportedNames
|
pub imported_names: ImportedNames,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, PartialEq, Clone)]
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
pub enum ImportedNames {
|
pub enum ImportedNames {
|
||||||
All,
|
All,
|
||||||
LastOfPath,
|
LastOfPath,
|
||||||
List(Vec<Rc<String>>)
|
List(Vec<Rc<String>>),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
#[derive(Debug, PartialEq, Clone)]
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
pub struct ModuleSpecifier {
|
pub struct ModuleSpecifier {
|
||||||
pub name: Rc<String>,
|
pub name: Rc<String>,
|
||||||
pub contents: Block,
|
pub contents: Block,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -59,7 +59,7 @@ fn token_kind_to_sigil(tok: & TokenKind) -> Option<&str> {
|
|||||||
LAngleBracket => "<",
|
LAngleBracket => "<",
|
||||||
RAngleBracket => ">",
|
RAngleBracket => ">",
|
||||||
Equals => "=",
|
Equals => "=",
|
||||||
_ => return None
|
_ => return None,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -3,17 +3,27 @@ use crate::ast::*;
|
|||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub enum Recursion {
|
pub enum Recursion {
|
||||||
Continue,
|
Continue,
|
||||||
Stop
|
Stop,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub trait ASTVisitor: Sized {
|
pub trait ASTVisitor: Sized {
|
||||||
fn expression(&mut self, _expression: &Expression) -> Recursion { Recursion::Continue }
|
fn expression(&mut self, _expression: &Expression) -> Recursion {
|
||||||
fn declaration(&mut self, _declaration: &Declaration, _id: &ItemId) -> Recursion { Recursion::Continue }
|
Recursion::Continue
|
||||||
|
}
|
||||||
|
fn declaration(&mut self, _declaration: &Declaration, _id: &ItemId) -> Recursion {
|
||||||
|
Recursion::Continue
|
||||||
|
}
|
||||||
|
|
||||||
fn import(&mut self, _import: &ImportSpecifier) -> Recursion { Recursion::Continue }
|
fn import(&mut self, _import: &ImportSpecifier) -> Recursion {
|
||||||
fn module(&mut self, _module: &ModuleSpecifier) -> Recursion { Recursion::Continue }
|
Recursion::Continue
|
||||||
|
}
|
||||||
|
fn module(&mut self, _module: &ModuleSpecifier) -> Recursion {
|
||||||
|
Recursion::Continue
|
||||||
|
}
|
||||||
|
|
||||||
fn pattern(&mut self, _pat: &Pattern) -> Recursion { Recursion::Continue }
|
fn pattern(&mut self, _pat: &Pattern) -> Recursion {
|
||||||
|
Recursion::Continue
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn walk_ast<V: ASTVisitor>(v: &mut V, ast: &AST) {
|
pub fn walk_ast<V: ASTVisitor>(v: &mut V, ast: &AST) {
|
||||||
@ -32,12 +42,11 @@ pub fn walk_block<V: ASTVisitor>(v: &mut V, block: &Block) {
|
|||||||
}
|
}
|
||||||
Import(ref import_spec) => {
|
Import(ref import_spec) => {
|
||||||
v.import(import_spec);
|
v.import(import_spec);
|
||||||
},
|
}
|
||||||
Module(ref module_spec) => {
|
Module(ref module_spec) =>
|
||||||
if let Recursion::Continue = v.module(module_spec) {
|
if let Recursion::Continue = v.module(module_spec) {
|
||||||
walk_block(v, &module_spec.contents);
|
walk_block(v, &module_spec.contents);
|
||||||
}
|
},
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -50,12 +59,7 @@ pub fn walk_declaration<V: ASTVisitor>(v: &mut V, decl: &Declaration, id: &ItemI
|
|||||||
FuncDecl(_sig, block) => {
|
FuncDecl(_sig, block) => {
|
||||||
walk_block(v, block);
|
walk_block(v, block);
|
||||||
}
|
}
|
||||||
Binding {
|
Binding { name: _, constant: _, type_anno: _, expr } => {
|
||||||
name: _,
|
|
||||||
constant: _,
|
|
||||||
type_anno: _,
|
|
||||||
expr,
|
|
||||||
} => {
|
|
||||||
walk_expression(v, expr);
|
walk_expression(v, expr);
|
||||||
}
|
}
|
||||||
_ => (),
|
_ => (),
|
||||||
@ -76,16 +80,14 @@ pub fn walk_expression<V: ASTVisitor>(v: &mut V, expr: &Expression) {
|
|||||||
PrefixExp(_, arg) => {
|
PrefixExp(_, arg) => {
|
||||||
walk_expression(v, arg);
|
walk_expression(v, arg);
|
||||||
}
|
}
|
||||||
TupleLiteral(exprs) => {
|
TupleLiteral(exprs) =>
|
||||||
for expr in exprs {
|
for expr in exprs {
|
||||||
walk_expression(v, expr);
|
walk_expression(v, expr);
|
||||||
}
|
},
|
||||||
}
|
NamedStruct { name: _, fields } =>
|
||||||
NamedStruct { name: _, fields } => {
|
|
||||||
for (_, expr) in fields.iter() {
|
for (_, expr) in fields.iter() {
|
||||||
walk_expression(v, expr);
|
walk_expression(v, expr);
|
||||||
}
|
},
|
||||||
}
|
|
||||||
Call { f, arguments } => {
|
Call { f, arguments } => {
|
||||||
walk_expression(v, f);
|
walk_expression(v, f);
|
||||||
for arg in arguments.iter() {
|
for arg in arguments.iter() {
|
||||||
@ -102,10 +104,7 @@ pub fn walk_expression<V: ASTVisitor>(v: &mut V, expr: &Expression) {
|
|||||||
walk_expression(v, indexer);
|
walk_expression(v, indexer);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
IfExpression {
|
IfExpression { discriminator, body } => {
|
||||||
discriminator,
|
|
||||||
body,
|
|
||||||
} => {
|
|
||||||
if let Some(d) = discriminator.as_ref() {
|
if let Some(d) = discriminator.as_ref() {
|
||||||
walk_expression(v, d);
|
walk_expression(v, d);
|
||||||
}
|
}
|
||||||
@ -126,18 +125,13 @@ pub fn walk_expression<V: ASTVisitor>(v: &mut V, expr: &Expression) {
|
|||||||
ForBody::StatementBlock(block) => walk_block(v, block),
|
ForBody::StatementBlock(block) => walk_block(v, block),
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
Lambda {
|
Lambda { params: _, type_anno: _, body } => {
|
||||||
params: _,
|
|
||||||
type_anno: _,
|
|
||||||
body,
|
|
||||||
} => {
|
|
||||||
walk_block(v, body);
|
walk_block(v, body);
|
||||||
}
|
}
|
||||||
ListLiteral(exprs) => {
|
ListLiteral(exprs) =>
|
||||||
for expr in exprs {
|
for expr in exprs {
|
||||||
walk_expression(v, expr);
|
walk_expression(v, expr);
|
||||||
}
|
},
|
||||||
}
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -146,27 +140,20 @@ pub fn walk_if_expr_body<V: ASTVisitor>(v: &mut V, body: &IfExpressionBody) {
|
|||||||
use IfExpressionBody::*;
|
use IfExpressionBody::*;
|
||||||
|
|
||||||
match body {
|
match body {
|
||||||
SimpleConditional {
|
SimpleConditional { then_case, else_case } => {
|
||||||
then_case,
|
|
||||||
else_case,
|
|
||||||
} => {
|
|
||||||
walk_block(v, then_case);
|
walk_block(v, then_case);
|
||||||
if let Some(block) = else_case.as_ref() {
|
if let Some(block) = else_case.as_ref() {
|
||||||
walk_block(v, block)
|
walk_block(v, block)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
SimplePatternMatch {
|
SimplePatternMatch { pattern, then_case, else_case } => {
|
||||||
pattern,
|
|
||||||
then_case,
|
|
||||||
else_case,
|
|
||||||
} => {
|
|
||||||
walk_pattern(v, pattern);
|
walk_pattern(v, pattern);
|
||||||
walk_block(v, then_case);
|
walk_block(v, then_case);
|
||||||
if let Some(block) = else_case.as_ref() {
|
if let Some(block) = else_case.as_ref() {
|
||||||
walk_block(v, block)
|
walk_block(v, block)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
CondList(arms) => {
|
CondList(arms) =>
|
||||||
for arm in arms {
|
for arm in arms {
|
||||||
match arm.condition {
|
match arm.condition {
|
||||||
Condition::Pattern(ref pat) => {
|
Condition::Pattern(ref pat) => {
|
||||||
@ -184,8 +171,7 @@ pub fn walk_if_expr_body<V: ASTVisitor>(v: &mut V, body: &IfExpressionBody) {
|
|||||||
walk_expression(v, guard);
|
walk_expression(v, guard);
|
||||||
}
|
}
|
||||||
walk_block(v, &arm.body);
|
walk_block(v, &arm.body);
|
||||||
}
|
},
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -194,21 +180,18 @@ pub fn walk_pattern<V: ASTVisitor>(v: &mut V, pat: &Pattern) {
|
|||||||
|
|
||||||
if let Recursion::Continue = v.pattern(pat) {
|
if let Recursion::Continue = v.pattern(pat) {
|
||||||
match pat {
|
match pat {
|
||||||
TuplePattern(patterns) => {
|
TuplePattern(patterns) =>
|
||||||
for pat in patterns {
|
for pat in patterns {
|
||||||
walk_pattern(v, pat);
|
walk_pattern(v, pat);
|
||||||
}
|
},
|
||||||
}
|
TupleStruct(_, patterns) =>
|
||||||
TupleStruct(_, patterns) => {
|
|
||||||
for pat in patterns {
|
for pat in patterns {
|
||||||
walk_pattern(v, pat);
|
walk_pattern(v, pat);
|
||||||
}
|
},
|
||||||
}
|
Record(_, name_and_patterns) =>
|
||||||
Record(_, name_and_patterns) => {
|
|
||||||
for (_, pat) in name_and_patterns {
|
for (_, pat) in name_and_patterns {
|
||||||
walk_pattern(v, pat);
|
walk_pattern(v, pat);
|
||||||
}
|
},
|
||||||
}
|
|
||||||
_ => (),
|
_ => (),
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user