74 lines
2.3 KiB
Rust
74 lines
2.3 KiB
Rust
use std::fmt::{Debug, Write};
|
|
|
|
use super::{
|
|
token::Symbol, Node, NodeParsable, Parsable, ParseResult, ParserError,
|
|
ParserErrors, Statement, TokenCursor,
|
|
};
|
|
use crate::util::Padder;
|
|
|
|
pub struct Body {
|
|
pub statements: Vec<Node<Statement>>,
|
|
}
|
|
|
|
impl Parsable for Body {
|
|
fn parse(cursor: &mut TokenCursor, errors: &mut ParserErrors) -> ParseResult<Self> {
|
|
let mut statements = Vec::new();
|
|
cursor.expect_sym(Symbol::OpenCurly)?;
|
|
if cursor.expect_peek()?.is_symbol(Symbol::CloseCurly) {
|
|
cursor.next();
|
|
return ParseResult::Ok(Self { statements });
|
|
}
|
|
let mut expect_semi = false;
|
|
let mut recover = false;
|
|
loop {
|
|
let Some(next) = cursor.peek() else {
|
|
recover = true;
|
|
errors.add(ParserError::unexpected_end());
|
|
break;
|
|
};
|
|
if next.is_symbol(Symbol::CloseCurly) {
|
|
cursor.next();
|
|
break;
|
|
}
|
|
if next.is_symbol(Symbol::Semicolon) {
|
|
cursor.next();
|
|
expect_semi = false;
|
|
continue;
|
|
} else if expect_semi {
|
|
errors.add(ParserError {
|
|
msg: "expected ';'".to_string(),
|
|
spans: vec![cursor.next_pos().char_span()],
|
|
});
|
|
}
|
|
let res = Statement::parse_node(cursor, errors);
|
|
statements.push(res.node);
|
|
expect_semi = true;
|
|
if res.recover {
|
|
cursor.seek_syms(&[Symbol::Semicolon, Symbol::CloseCurly]);
|
|
if cursor.peek().is_none() {
|
|
recover = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
ParseResult::from_recover(Self { statements }, recover)
|
|
}
|
|
}
|
|
|
|
impl Debug for Body {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
if self.statements.first().is_some() {
|
|
f.write_str("{\n ")?;
|
|
let mut padder = Padder::new(f);
|
|
for s in &self.statements {
|
|
// they don't expose wrap_buf :grief:
|
|
padder.write_str(&format!("{s:?}\n"))?;
|
|
}
|
|
f.write_char('}')?;
|
|
} else {
|
|
f.write_str("{}")?;
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|