aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJJ2023-11-08 21:18:08 +0000
committerJJ2023-11-08 21:18:08 +0000
commit3851012eeb1420bef0db7cd3e9a76affdb6145b9 (patch)
treeab2fbf7b80ea76ac971f85271bbc5a05337c4550
parentb7b2f41976eb0315130e002e0a40631b30f28add (diff)
compiler: progress. parse most of the hard things.
-rw-r--r--src/frontend/ast.rs35
-rw-r--r--src/frontend/lex.rs20
-rw-r--r--src/frontend/parse.rs441
3 files changed, 380 insertions, 116 deletions
diff --git a/src/frontend/ast.rs b/src/frontend/ast.rs
index ed7a01a..ed44712 100644
--- a/src/frontend/ast.rs
+++ b/src/frontend/ast.rs
@@ -2,6 +2,7 @@
pub type Id = String;
/// Puck's fundamental types.
+#[derive(Clone, PartialEq)]
pub enum Type {
Void, Never,
Integer, Float, String, // char et al are defined later
@@ -12,10 +13,7 @@ pub enum Type {
Struct(Vec<(Id, Box<Type>)>),
Tuple(Vec<(Option<Id>, Box<Type>)>),
Union(Vec<(Id, Box<Type>)>),
- Interface {
- funcs: Vec<Sig>,
- for_type: Option<Box<Type>>,
- },
+ Interface(Vec<Sig>),
Array{size: usize, kind: Box<Type>},
List(Box<Type>),
Slice(Box<Type>), // todo: plus ownership
@@ -31,12 +29,13 @@ pub enum Type {
}
/// Function signatures.
+#[derive(Clone, PartialEq)]
pub struct Sig {
- effect: Option<Id>,
- id: Id,
- generics: Vec<(Id, Option<Type>)>,
- params: Vec<Type>,
- result: Option<Type>
+ pub effect: Option<Id>,
+ pub id: Id,
+ pub generics: Vec<(Id, Option<Type>)>,
+ pub parameters: Vec<Type>,
+ pub kind: Option<Type>
}
/// Patterns are recognizable given zero context.
@@ -69,16 +68,24 @@ pub enum Binding {
kind: Option<Type>,
value: Box<Expr>
},
- FuncDecl {
+ Func {
public: bool,
effect: Option<Id>,
id: Id,
generics: Vec<(Id, Option<Type>)>, // id, kind
- params: Vec<(Id, Type)>, // id, kind
+ parameters: Vec<(Id, Type)>, // id, kind
kind: Type,
body: Vec<Expr>
},
- TypeDecl { id: Id, generics: Vec<Id>, alias: Type },
+ Macro {
+ public: bool,
+ id: Id,
+ generics: Vec<(Id, Option<Type>)>, // id, kind
+ parameters: Vec<(Id, Option<Type>)>, // id, kind
+ kind: Option<Type>,
+ body: Vec<Expr>
+ },
+ TypeDecl { id: Id, generics: Vec<(Id, Option<Type>)>, alias: Type },
Import { from: Option<Id>, imports: Vec<Id>, alias: Option<Id> },
Module { id: Id, body: Vec<Expr> },
}
@@ -107,8 +114,8 @@ pub enum Control {
}
pub struct CondBranch { pub cond: Expr, pub body: Vec<Expr> }
-pub struct CatchBranch { pub exceptions: Vec<Id>, pub binding: Option<Id>, pub body: Vec<Expr> }
-pub struct MatchBranch { pub pattern: Pattern, pub guard: Option<Expr>, pub body: Vec<Expr> }
+pub struct CatchBranch { pub exceptions: Vec<(Id, Option<Id>)>, pub body: Vec<Expr> }
+pub struct MatchBranch { pub patterns: Vec<Pattern>, pub guard: Option<Expr>, pub body: Vec<Expr> }
/// Expressions are either Patterns, Bindings, or Control flow constructs.
pub enum Expr {
diff --git a/src/frontend/lex.rs b/src/frontend/lex.rs
index a51b21d..67c4ae3 100644
--- a/src/frontend/lex.rs
+++ b/src/frontend/lex.rs
@@ -54,15 +54,15 @@ pub enum Literal {
pub enum Keyword {
Pub, Let, Var, Const,
Func, Macro, Type,
- Mod, From, Import,
+ Mod, Use, As,
For, While, Loop,
Block, Static,
- If, When, Elif, Else, Match,
+ If, When, Elif, Else, Match, Where,
Try, Catch, Finally,
Struct, Tuple, Enum, Union, Interface,
Distinct, Ref, Ptr, Mut,
Break, Continue, Return,
- In, Is, Of, As,
+ In, Is, Of,
}
/// All punctuation recognized by the lexer.
@@ -272,9 +272,9 @@ pub fn tokenize(input: &str) -> Result<TokenStream> {
"func" => res.push(Key(Func)),
"macro" => res.push(Key(Macro)),
"type" => res.push(Key(Type)),
- "mod" => res.push(Key(Mod)),
- "from" => res.push(Key(From)),
- "import" => res.push(Key(Import)),
+ "mod" => res.push(Key(Mod)),
+ "use" => res.push(Key(Use)),
+ "as" => res.push(Key(As)),
"for" => res.push(Key(For)),
"while" => res.push(Key(While)),
"loop" => res.push(Key(Loop)),
@@ -285,6 +285,7 @@ pub fn tokenize(input: &str) -> Result<TokenStream> {
"elif" => res.push(Key(Elif)),
"else" => res.push(Key(Else)),
"match" => res.push(Key(Match)),
+ "where" => res.push(Key(Where)),
"try" => res.push(Key(Try)),
"catch" => res.push(Key(Catch)),
"finally" => res.push(Key(Finally)),
@@ -303,7 +304,6 @@ pub fn tokenize(input: &str) -> Result<TokenStream> {
"in" => res.push(Key(In)),
"is" => res.push(Key(Is)),
"of" => res.push(Key(Of)),
- "as" => res.push(Key(As)),
_ => res.push(Word(String::from(&buf)))
}
match x { // () and [] denote both parameters/generics and tuples/arrays
@@ -470,8 +470,8 @@ impl std::fmt::Display for Keyword {
Macro => write!(f, "macro"),
Type => write!(f, "type"),
Mod => write!(f, "mod"),
- From => write!(f, "from"),
- Import => write!(f, "import"),
+ Use => write!(f, "use"),
+ As => write!(f, "as"),
For => write!(f, "for"),
While => write!(f, "while"),
Loop => write!(f, "loop"),
@@ -482,6 +482,7 @@ impl std::fmt::Display for Keyword {
Elif => write!(f, "elif"),
Else => write!(f, "else"),
Match => write!(f, "match"),
+ Where => write!(f, "where"),
Try => write!(f, "try"),
Catch => write!(f, "catch"),
Finally => write!(f, "finally"),
@@ -500,7 +501,6 @@ impl std::fmt::Display for Keyword {
In => write!(f, "in"),
Is => write!(f, "is"),
Of => write!(f, "of"),
- As => write!(f, "as"),
}
}
}
diff --git a/src/frontend/parse.rs b/src/frontend/parse.rs
index dba94ec..eee3911 100644
--- a/src/frontend/parse.rs
+++ b/src/frontend/parse.rs
@@ -1,13 +1,14 @@
+use multipeek::multipeek;
+
use crate::frontend::lex::*;
use crate::frontend::ast::*;
use crate::frontend::ast::Binding::*;
use crate::frontend::ast::Control::*;
-use crate::frontend::ast::Pattern::*;
use Token::*;
use Literal::*;
use Punctuation::*;
-struct Input(std::iter::Peekable<std::vec::IntoIter<Token>>);
+struct Input(multipeek::MultiPeek<std::vec::IntoIter<Token>>);
impl Input {
/// Map input.next() to return Results for use with the propagation operator
@@ -20,6 +21,10 @@ impl Input {
self.0.peek().ok_or("end of input".into())
}
+ fn peek_nth(&mut self, n: usize) -> Result<&Token> {
+ self.0.peek_nth(n).ok_or("end of input".into())
+ }
+
/// Asserts the next character to be a known token
fn then(&mut self, expected: Token) -> Result<()> {
match self.next()? {
@@ -31,7 +36,7 @@ impl Input {
/// Convert a basic TokenStream into an AbstractSyntaxTree
pub fn astify(input: TokenStream, name: &str) -> Result<Expr> {
- let mut input = Input(input.into_iter().peekable());
+ let mut input = Input(multipeek(input));
let body = parse_body(&mut input)?;
Ok(Expr::Binding(Module{ id: name.to_string(), body }))
}
@@ -44,14 +49,14 @@ fn parse_body(input: &mut Input) -> Result<Vec<Expr>> {
res.push(parse_expr(input)?);
return Ok(res);
}
- input.then(Sep(ScopeLeftBrace))?;
+ input.next()?;
while input.peek()? != &Sep(ScopeRightBrace) {
res.push(parse_expr(input)?);
if input.peek()? == &Sep(Semicolon) {
input.next()?;
}
}
- input.then(Sep(ScopeRightBrace))?;
+ input.next()?;
Ok(res)
}
@@ -65,7 +70,8 @@ fn parse_expr(input: &mut Input) -> Result<Expr> {
match input.next()? {
Key(word) => match word {
Const => parse_const(input, true),
- Func => parse_funcdecl(input, true),
+ Func => parse_func(input, true),
+ Macro => parse_macro(input, true),
Type => parse_typedecl(input, true),
Mod => parse_mod(input, true),
_ => return Err("unrecognized keyword following pub".into()),
@@ -76,11 +82,11 @@ fn parse_expr(input: &mut Input) -> Result<Expr> {
Let => parse_let(input),
Var => parse_var(input),
Const => parse_const(input, false),
- Func => parse_funcdecl(input, false),
+ Func => parse_func(input, false),
+ Macro => parse_macro(input, false),
Type => parse_typedecl(input, false),
Mod => parse_mod(input, false),
- From => parse_import(input, true), // todo: probably rework imports
- Import => parse_import(input, false),
+ Use => parse_use(input),
Block => parse_block(input),
Static => parse_static(input),
For => parse_for(input),
@@ -90,13 +96,13 @@ fn parse_expr(input: &mut Input) -> Result<Expr> {
When => parse_when(input),
Try => parse_try(input),
Match => parse_match(input),
- _ => return Err("invalid keyword starting expression".into()),
+ word => return Err(format!("attempting to parse {} out of context", word).into()),
},
_ => todo!(), // what can i do with this?? match line here
}
}
-/// Let ::= 'let' Pattern Annotation? '=' Expr
+/// `Let ::= 'let' Pattern Annotation? '=' Expr`
fn parse_let(input: &mut Input) -> Result<Expr> {
let id = parse_pattern(input)?;
let kind = parse_annotation(input)?;
@@ -105,7 +111,7 @@ fn parse_let(input: &mut Input) -> Result<Expr> {
Ok(Expr::Binding(Let { id, kind, value }))
}
-/// Var ::= 'var' Pattern Annotation? ('=' Expr)?
+/// `Var ::= 'var' Pattern Annotation? ('=' Expr)?`
fn parse_var(input: &mut Input) -> Result<Expr> {
let id = parse_pattern(input)?;
let kind = parse_annotation(input)?;
@@ -116,7 +122,7 @@ fn parse_var(input: &mut Input) -> Result<Expr> {
Ok(Expr::Binding(Var { id, kind, value }))
}
-/// Const ::= 'pub'? 'const' Pattern Annotation? '=' Expr
+/// `Const ::= 'pub'? 'const' Pattern Annotation? '=' Expr`
fn parse_const(input: &mut Input, public: bool) -> Result<Expr> {
let id = parse_pattern(input)?;
let kind = parse_annotation(input)?;
@@ -135,16 +141,119 @@ fn parse_annotation(input: &mut Input) -> Result<Option<Type>> {
Ok(kind)
}
-/// Func ::= 'pub'? 'func' Ident Generics? Parameters? (':' TypeDesc) '=' Body
-fn parse_funcdecl(input: &mut Input, public: bool) -> Result<Expr> { todo!() }
+/// `Func ::= 'pub'? 'func' Ident ('[' Parameters ']')? ('(' Parameters ')')? (':' TypeDesc) '=' Body`
+fn parse_func(input: &mut Input, public: bool) -> Result<Expr> {
+ let effect = None;
+ let id = parse_ident(input)?;
+ let mut generics = Vec::new();
+ if input.peek()? == &Sep(GenericLeftBracket) {
+ input.next()?;
+ generics = parse_parameters(input)?;
+ input.then(Sep(GenericRightBracket))?;
+ }
+ let mut parameters = Vec::new();
+ if input.peek()? == &Sep(FuncLeftParen) {
+ input.next()?; // todo: rewrite to map over an input
+ // let temp_parameters = parse_parameters(input)?;
+ // if temp_parameters.last().is_none() {
+ // return Err("expected a type annotation on the last function parameter".into());
+ // }
+ // parameters = parse_parameters(input)?.iter().map(|x| x).collect();
+ let mut stack = Vec::new();
+ let (id, kind) = parse_parameter(input)?;
+ if kind.is_some() {
+ parameters.push((id, kind.unwrap()));
+ } else {
+ stack.push(id);
+ }
+ while input.peek()? == &Sep(Comma) {
+ input.next()?;
+ let (id, kind) = parse_parameter(input)?;
+ stack.push(id);
+ if kind.is_some() {
+ for id in &stack {
+ parameters.push((id.clone(), kind.clone().unwrap()));
+ }
+ stack.clear();
+ }
+ }
+ if stack.len() != 0 {
+ return Err("expected a type annotation on the last function parameter".into());
+ }
+ input.then(Sep(FuncRightParen))?;
+ }
+ let mut kind = Type::Void;
+ if input.peek()? == &Sep(Colon) {
+ input.next()?;
+ kind = parse_type(input)?;
+ }
+ input.then(Sep(Equals))?;
+ let body = parse_body(input)?;
+ Ok(Expr::Binding(Func { public, effect, id, generics, parameters, kind, body }))
+}
-/// TypeDecl ::= 'pub'? 'type' Pattern Generics? '=' 'distinct'? 'ref'? TypeDesc
+/// `Macro ::= 'pub'? 'macro' Ident ('[' Paremeters ']')? ('(' Paremeters ')')? (':' TypeDesc) '=' Body`
+fn parse_macro(input: &mut Input, public: bool) -> Result<Expr> {
+ let id = parse_ident(input)?;
+ let mut generics = Vec::new();
+ if input.peek()? == &Sep(GenericLeftBracket) {
+ input.next()?;
+ generics = parse_parameters(input)?;
+ input.then(Sep(GenericRightBracket))?;
+ }
+ let mut parameters = Vec::new();
+ if input.peek()? == &Sep(FuncLeftParen) {
+ input.next()?;
+ parameters = parse_parameters(input)?;
+ input.then(Sep(FuncRightParen))?;
+ }
+ let mut kind = None;
+ if input.peek()? == &Sep(Colon) {
+ input.next()?;
+ kind = Some(parse_type(input)?);
+ }
+ input.then(Sep(Equals))?;
+ let body = parse_body(input)?;
+ Ok(Expr::Binding(Macro { public, id, generics, parameters, kind, body }))
+}
+
+/// `TypeDecl ::= 'pub'? 'type' Ident ('[' Parameters ']')? '=' Type
fn parse_typedecl(input: &mut Input, public: bool) -> Result<Expr> {
- let pattern = parse_pattern(input)?;
- todo!()
+ let id = parse_ident(input)?;
+ let mut generics = Vec::new();
+ if input.peek()? == &Sep(GenericLeftBracket) {
+ input.next()?;
+ generics = parse_parameters(input)?;
+ input.then(Sep(GenericRightBracket))?;
+ }
+ input.then(Sep(Equals))?;
+ let alias = parse_type(input)?;
+ Ok(Expr::Binding(TypeDecl { id, generics, alias }))
+}
+
+/// `Parameter ::= Ident (':' Type)?`
+fn parse_parameter(input: &mut Input) -> Result<(Id, Option<Type>)> {
+ let id = parse_ident(input)?;
+ let mut kind = None;
+ if input.peek()? == &Sep(Colon) {
+ input.next()?;
+ kind = Some(parse_type(input)?);
+ }
+ Ok((id, kind))
+}
+
+/// `Parameters ::= Parameter (',' Parameter)*
+fn parse_parameters(input: &mut Input) -> Result<Vec<(Id, Option<Type>)>> {
+ let mut res = Vec::new();
+ res.push(parse_parameter(input)?);
+ while input.peek()? == &Sep(Comma) {
+ input.next()?;
+ res.push(parse_parameter(input)?);
+ }
+ Ok(res)
}
-/// Mod ::= 'pub'? 'mod' Ident ':' Body
+/// `Mod ::= 'pub'? 'mod' Ident ':' Body`
fn parse_mod(input: &mut Input, public: bool) -> Result<Expr> {
match input.next()? {
Word(id) => {
@@ -160,21 +269,11 @@ fn parse_mod(input: &mut Input, public: bool) -> Result<Expr> {
}
}
-/// Import ::= ('from' Ident)? 'import' Ident (',' Ident)* ('as' Ident)?
-fn parse_import(input: &mut Input, from_scope: bool) -> Result<Expr> {
- let mut from = None;
- if from_scope {
- match input.next()? {
- Word(id) => from = Some(id),
- _ => return Err("identifier not following from keyword".into())
- }
- input.then(Key(Keyword::Import))?;
- }
- todo!()
-}
+/// `Use ::= 'use' Ident ('/' Ident)* ('/' (('[' Ident (',' Ident)* ']') | '*'))?`
+fn parse_use(input: &mut Input) -> Result<Expr> { todo!() }
-/// Block ::= 'block' Ident? ':' Body
-fn parse_block(input: &mut Input) -> Result<Expr> { // todo: body + offset
+/// `Block ::= 'block' Ident? ':' Body`
+fn parse_block(input: &mut Input) -> Result<Expr> {
match input.next()? {
Sep(Colon) => {
let id = None;
@@ -182,27 +281,23 @@ fn parse_block(input: &mut Input) -> Result<Expr> { // todo: body + offset
Ok(Expr::Control(Block { id, body }))
},
Word(label) => {
- match input.next()? {
- Sep(Colon) => {
- let id = Some(label);
- let body = parse_body(input)?;
- Ok(Expr::Control(Block { id, body }))
- },
- _ => return Err("unexpected token following block label".into()),
- }
+ input.then(Sep(Colon))?;
+ let id = Some(label);
+ let body = parse_body(input)?;
+ Ok(Expr::Control(Block { id, body }))
},
_ => return Err("unexpected thing following block keyword".into()),
}
}
-/// Static ::= 'static' ':' Body
+/// `Static ::= 'static' ':' Body`
fn parse_static(input: &mut Input) -> Result<Expr> {
input.then(Sep(Colon))?;
let body = parse_body(input)?;
Ok(Expr::Control(Static { body }))
}
-/// For ::= 'for' Pattern 'in' Expr ':' Body
+/// `For ::= 'for' Pattern 'in' Expr ':' Body`
fn parse_for(input: &mut Input) -> Result<Expr> {
let binding = parse_pattern(input)?;
input.then(Key(Keyword::In))?;
@@ -212,7 +307,7 @@ fn parse_for(input: &mut Input) -> Result<Expr> {
Ok(Expr::Control(For { binding, range, body }))
}
-/// While ::= 'while' Expr ':' Body
+/// `While ::= 'while' Expr ':' Body`
fn parse_while(input: &mut Input) -> Result<Expr> {
let cond = Box::new(parse_expr(input)?);
input.then(Sep(Colon))?;
@@ -220,14 +315,14 @@ fn parse_while(input: &mut Input) -> Result<Expr> {
Ok(Expr::Control(While { cond, body }))
}
-/// Loop ::= 'loop' ':' Body
+/// `Loop ::= 'loop' ':' Body`
fn parse_loop(input: &mut Input) -> Result<Expr> {
input.then(Sep(Colon))?;
let body = parse_body(input)?;
Ok(Expr::Control(Loop { body }))
}
-/// If ::= 'if' CondBranch ('elif' CondBranch)* ('else' ':' Body)?
+/// `If ::= 'if' CondBranch ('elif' CondBranch)* ('else' ':' Body)?`
fn parse_if(input: &mut Input) -> Result<Expr> {
let mut branches = Vec::new();
branches.push(parse_cond_branch(input)?);
@@ -243,7 +338,7 @@ fn parse_if(input: &mut Input) -> Result<Expr> {
Ok(Expr::Control(If { branches, else_body }))
}
-// When ::= 'when' CondBranch ('elif' CondBranch)* ('else' ':' Body)?
+/// `When ::= 'when' CondBranch ('elif' CondBranch)* ('else' ':' Body)?`
fn parse_when(input: &mut Input) -> Result<Expr> {
let mut branches = Vec::new();
branches.push(parse_cond_branch(input)?);
@@ -262,7 +357,7 @@ fn parse_when(input: &mut Input) -> Result<Expr> {
Ok(Expr::Control(Static { body }))
}
-/// CondBranch ::= Expr ':' Body
+/// `CondBranch ::= Expr ':' Body`
fn parse_cond_branch(input: &mut Input) -> Result<CondBranch> {
let cond = parse_expr(input)?;
input.then(Sep(Colon))?;
@@ -270,14 +365,22 @@ fn parse_cond_branch(input: &mut Input) -> Result<CondBranch> {
Ok(CondBranch { cond, body })
}
-/// Try ::= 'try' ':' Body ('except' Ident (',' Ident)* ':' Body) ('finally' ':' Body)?
+/// `Try ::= 'try' ':' Body ('except' Exception (',' Exception)* ':' Body) ('finally' ':' Body)?`
fn parse_try(input: &mut Input) -> Result<Expr> {
input.then(Sep(Colon))?;
let body = parse_body(input)?;
let mut catches = Vec::new();
while input.peek()? == &Key(Keyword::Catch) {
input.next()?;
- todo!();
+ let mut exceptions = Vec::new();
+ exceptions.push(parse_catch_exception(input)?);
+ while input.peek()? == &Sep(Comma) {
+ input.next()?;
+ exceptions.push(parse_catch_exception(input)?);
+ }
+ input.then(Sep(Colon))?;
+ let body = parse_body(input)?;
+ catches.push(CatchBranch { exceptions, body });
}
let mut finally = None;
if input.peek()? == &Key(Keyword::Finally) {
@@ -288,49 +391,72 @@ fn parse_try(input: &mut Input) -> Result<Expr> {
Ok(Expr::Control(Try { body, catches, finally }))
}
-/// Match ::= 'match' Expr ('of' Pattern (',' Pattern)* ('where' Expr)? ':' Body)+
+/// `Exception ::= Ident ('as' Ident)?`
+fn parse_catch_exception(input: &mut Input) -> Result<(Id, Option<Id>)> {
+ let id = parse_ident(input)?;
+ let mut alias = None;
+ if input.peek()? == &Key(Keyword::As) {
+ input.next()?;
+ alias = Some(parse_ident(input)?);
+ }
+ Ok((id, alias))
+}
+
+/// `Match ::= 'match' Expr ('of' Pattern (',' Pattern)* ('where' Expr)? ':' Body)+`
fn parse_match(input: &mut Input) -> Result<Expr> {
let item = parse_pattern(input)?; // fixme
let mut branches = Vec::new();
while input.peek()? == &Key(Keyword::Of) {
input.next()?;
- todo!();
+ let mut patterns = Vec::new();
+ patterns.push(parse_pattern(input)?);
+ while input.peek()? == &Sep(Comma) {
+ input.next()?;
+ patterns.push(parse_pattern(input)?);
+ }
+ let mut guard = None;
+ if input.peek()? == &Key(Keyword::Where) {
+ input.next()?;
+ guard = Some(parse_expr(input)?)
+ }
+ input.then(Sep(Colon))?;
+ let body = parse_body(input)?;
+ branches.push(MatchBranch { patterns, guard, body })
}
Ok(Expr::Control(Match { item, branches }))
}
-/// Type ::=
-/// ('ref' | 'ptr' | 'mut' | 'static' | 'struct' | 'tuple' | 'enum' | 'union' | 'interface' | 'concept') |
-/// ('ref' WrappedType) | ('ptr' WrappedType) | ('mut' WrappedType) | ('static' WrappedType) | ('distinct' WrappedType) |
-/// StructType | TupleType | EnumType | UnionType | InterfaceType
-/// The input stream must be normalized before attempting to parse types, because otherwise it's just a little bit hellish.
-/// In particular: ref, ptr, mut, static, distinct must wrap their parameters in '[' ']' and all type declarations must be on one line.
+/// `Type ::= (('distinct' | 'ref' | 'ptr' | 'mut' | 'static') BracketType?) |
+/// StructType | TupleType | EnumType | UnionType | InterfaceType`
+///
+/// The input stream must be normalized (i.e. all type declarations must be on one line)
+/// before attempting to parse types, because otherwise it's just a little bit hellish.
fn parse_type(input: &mut Input) -> Result<Type> {
use Type::*;
match input.next()? {
- Key(word) => {
- match input.peek()? { // todo: check if the type is a special typeclass
- Sep(GenericLeftBracket) => (),
- _ => todo!() // ref, ptr, mut, static, struct, tuple, enum, union, interface, concept
- }
- match word {
- Keyword::Distinct => Ok(Distinct(Box::new(parse_wrapped_type(input)?))),
- Keyword::Ref => Ok(Reference(Box::new(parse_wrapped_type(input)?))),
- Keyword::Ptr => Ok(Pointer(Box::new(parse_wrapped_type(input)?))),
- Keyword::Var => Ok(Mutable(Box::new(parse_wrapped_type(input)?))),
- Keyword::Const => Ok(Static(Box::new(parse_wrapped_type(input)?))),
- Keyword::Struct => parse_struct_type(input),
- Keyword::Tuple => parse_tuple_type(input),
- Keyword::Enum => parse_enum_type(input),
- Keyword::Union => parse_union_type(input),
- Keyword::Interface => parse_interface(input),
- _ => return Err("invalid keyword present in type!".into())
- }
+ Key(word) => match word {
+ Keyword::Distinct => Ok(Distinct(Box::new(parse_wrapped_type(input)?))),
+ Keyword::Ref => Ok(Reference(Box::new(parse_wrapped_type(input)?))),
+ Keyword::Ptr => Ok(Pointer(Box::new(parse_wrapped_type(input)?))),
+ Keyword::Mut => Ok(Mutable(Box::new(parse_wrapped_type(input)?))),
+ Keyword::Static => Ok(Static(Box::new(parse_wrapped_type(input)?))),
+ Keyword::Struct => parse_struct_type(input),
+ Keyword::Tuple => parse_tuple_type(input),
+ Keyword::Enum => parse_enum_type(input),
+ Keyword::Union => parse_union_type(input),
+ Keyword::Interface => parse_interface(input),
+ _ => return Err("invalid keyword present in type!".into())
},
Word(id) => {
let mut generics = Vec::new();
if input.peek()? == &Sep(GenericLeftBracket) {
- generics = parse_generics(input)?;
+ input.next()?;
+ generics.push(parse_type(input)?);
+ while input.peek()? == &Sep(Comma) {
+ input.next()?;
+ generics.push(parse_type(input)?);
+ }
+ input.then(Sep(GenericRightBracket))?;
}
Ok(Alias { id, generics })
},
@@ -338,18 +464,145 @@ fn parse_type(input: &mut Input) -> Result<Type> {
}
}
-/// `StructType ::= ('struct' '[' Ident ':' Type (',' Ident ':' Type)* ']'`
-fn parse_struct_type(input: &mut Input) -> Result<Type> { todo!() }
-/// `TupleType ::= 'tuple' '[' (Ident ':')? Type (',' (Ident ':')? Type)* ']'`
-fn parse_tuple_type(input: &mut Input) -> Result<Type> { todo!() }
-/// `EnumType ::= 'enum' '[' Ident ('=' Pattern)? (Ident ('=' Pattern)?)* ']'`
-fn parse_enum_type(input: &mut Input) -> Result<Type> { todo!() }
-/// `UnionType ::= 'union' '[' Ident (':' Type)? (',' Ident (':' Type)?)* ']'`
-fn parse_union_type(input: &mut Input) -> Result<Type> { todo!() }
-/// `Interface ::= 'interface' '[' Signature (',' Signature)* ']'`
-fn parse_interface(input: &mut Input) -> Result<Type> { todo!() }
-/// `Signature ::= Ident ('[' Ident (':' Type)? (',' Ident (':' Type)?)* ']')? ('(' Type (',' Type)* ')')? (':' Type)?`
-fn parse_signature(input: &mut Input) -> Result<Sig> { todo!() }
+/// `StructType ::= 'struct' ('[' Ident ':' Type (',' Ident ':' Type)* ']')?`
+fn parse_struct_type(input: &mut Input) -> Result<Type> {
+ let mut res = Vec::new();
+ if input.peek()? == &Sep(GenericLeftBracket) {
+ input.next()?;
+ res.push(parse_struct_field(input)?);
+ while input.peek()? == &Sep(Comma) {
+ input.next()?;
+ res.push(parse_struct_field(input)?);
+ }
+ input.then(Sep(GenericRightBracket))?;
+ }
+ Ok(Type::Struct(res))
+}
+
+fn parse_struct_field(input: &mut Input) -> Result<(Id, Box<Type>)> {
+ let id = parse_ident(input)?;
+ input.then(Sep(Colon))?;
+ let kind = Box::new(parse_type(input)?);
+ Ok((id, kind))
+}
+
+/// `TupleType ::= 'tuple' ('[' (Ident ':')? Type (',' (Ident ':')? Type)* ']')?`
+fn parse_tuple_type(input: &mut Input) -> Result<Type> {
+ let mut res = Vec::new();
+ if input.peek()? == &Sep(GenericLeftBracket) {
+ input.next()?;
+ res.push(parse_tuple_field(input)?);
+ while input.peek()? == &Sep(Comma) {
+ input.next()?;
+ res.push(parse_tuple_field(input)?);
+ }
+ input.then(Sep(GenericRightBracket))?;
+ }
+ Ok(Type::Tuple(res))
+}
+
+fn parse_tuple_field(input: &mut Input) -> Result<(Option<Id>, Box<Type>)> {
+ match input.peek()?.clone() { // huh???
+ Word(id) if input.peek_nth(1)? == &Sep(Colon) => {
+ input.next()?;
+ Ok((Some(id.to_string()), Box::new(parse_type(input)?)))
+ },
+ _ => Ok((None, Box::new(parse_type(input)?)))
+ }
+}
+
+/// `EnumType ::= 'enum' ('[' Ident ('=' Pattern)? (Ident ('=' Pattern)?)* ']')?`
+fn parse_enum_type(input: &mut Input) -> Result<Type> {
+ let mut res = Vec::new();
+ if input.peek()? == &Sep(GenericLeftBracket) {
+ input.next()?;
+ res.push(parse_enum_variant(input)?);
+ while input.peek()? == &Sep(Comma) {
+ input.next()?;
+ res.push(parse_enum_variant(input)?);
+ }
+ input.then(Sep(GenericRightBracket))?;
+ }
+ todo!()
+}
+
+fn parse_enum_variant(input: &mut Input) -> Result<(Id, Option<Pattern>)> {
+ let id = parse_ident(input)?;
+ let mut kind = None;
+ if input.peek()? == &Sep(Equals) {
+ input.next()?;
+ kind = Some(parse_pattern(input)?);
+ }
+ Ok((id, kind))
+}
+
+/// `UnionType ::= 'union' ('[' Ident (':' Type)? (',' Ident (':' Type)?)* ']')?`
+fn parse_union_type(input: &mut Input) -> Result<Type> {
+ let mut res = Vec::new();
+ if input.peek()? == &Sep(GenericLeftBracket) {
+ input.next()?;
+ res.push(parse_union_variant(input)?);
+ while input.peek()? == &Sep(Comma) {
+ input.next()?;
+ res.push(parse_union_variant(input)?);
+ }
+ input.then(Sep(GenericRightBracket))?;
+ }
+ Ok(Type::Union(res))
+}
+
+fn parse_union_variant(input: &mut Input) -> Result<(Id, Box<Type>)> {
+ let id = parse_ident(input)?;
+ let mut kind = Box::new(Type::Alias { id: "unit".to_string(), generics: Vec::new() });
+ if input.peek()? == &Sep(Colon) {
+ input.next()?;
+ kind = Box::new(parse_type(input)?);
+ }
+ Ok((id, kind))
+}
+
+/// `Interface ::= 'interface' ('[' Signature (',' Signature)* ']')?`
+fn parse_interface(input: &mut Input) -> Result<Type> {
+ let mut res = Vec::new();
+ if input.peek()? == &Sep(GenericLeftBracket) {
+ input.next()?;
+ res.push(parse_signature(input)?);
+ while input.peek()? == &Sep(Comma) {
+ input.next()?;
+ res.push(parse_signature(input)?);
+ }
+ input.then(Sep(GenericRightBracket))?;
+ }
+ Ok(Type::Interface(res))
+}
+
+/// `Signature ::= Ident ('[' Parameters ']')? ('(' Type (',' Type)* ')')? (':' Type)?`
+fn parse_signature(input: &mut Input) -> Result<Sig> {
+ let effect = None;
+ let id = parse_ident(input)?;
+ let mut generics = Vec::new();
+ if input.peek()? == &Sep(GenericLeftBracket) {
+ input.next()?;
+ generics = parse_parameters(input)?;
+ input.then(Sep(GenericRightBracket))?;
+ }
+ let mut parameters = Vec::new();
+ if input.peek()? == &Sep(FuncLeftParen) {
+ input.next()?;
+ parameters.push(parse_type(input)?);
+ if input.peek()? == &Sep(Comma) {
+ input.next()?;
+ parameters.push(parse_type(input)?);
+ }
+ input.then(Sep(FuncRightParen))?;
+ }
+ let mut kind = None;
+ if input.peek()? == &Sep(Colon) {
+ input.next()?;
+ kind = Some(parse_type(input)?);
+ }
+ Ok(Sig { effect, id, generics, parameters, kind })
+}
/// `WrappedType ::= Type | ('[' Type ']')`
fn parse_wrapped_type(input: &mut Input) -> Result<Type> {
@@ -362,11 +615,15 @@ fn parse_wrapped_type(input: &mut Input) -> Result<Type> {
}
}
-/// `GenericType ::= '[' Type (',' Type)* ']'`
-fn parse_generics(input: &mut Input) -> Result<Vec<Type>> { todo!() }
-
/// Pattern ::= Literal | Ident | '(' Pattern (',' Pattern)* ')' | Ident '(' Pattern (',' Pattern)* ')'
fn parse_pattern(input: &mut Input) -> Result<Pattern> { todo!() }
/// Literal ::= Char | String | Number | Float
fn parse_literal(input: &mut Input) -> Result<Pattern> { todo!() }
+
+fn parse_ident(input: &mut Input) -> Result<Id> {
+ match input.next()? {
+ Word(id) => Ok(id),
+ token => Err(format!("expected identifier but found token {}", token).into())
+ }
+}