1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

use super::pattern::ParsePatternRules;
use super::ParseCtx;
use super::Parser;
use crate::ast::Node;
use crate::ast::Syntax;
use crate::error::SyntaxResult;
use crate::session::Session;
use crate::symbol::Scope;
use crate::symbol::ScopeType;
use crate::symbol::SymbolGenerator;
use crate::token::TokenType;
use core::str::FromStr;

#[derive(Clone, Copy, PartialEq, Eq, Debug)]
pub enum TopLevelMode {
  Global,
  Module,
}

impl FromStr for TopLevelMode {
  type Err = &'static str;

  fn from_str(s: &str) -> Result<Self, Self::Err> {
    match s {
      "global" => Ok(TopLevelMode::Global),
      "module" => Ok(TopLevelMode::Module),
      _ => Err("invalid top-level mode"),
    }
  }
}

impl<'a> Parser<'a> {
  pub fn parse_top_level(
    &mut self,
    session: &'a Session,
    symbol_generator: SymbolGenerator,
    top_level_mode: TopLevelMode,
  ) -> SyntaxResult<'a, Node<'a>> {
    let ctx = ParseCtx {
      scope: Scope::new(session, symbol_generator, None, match top_level_mode {
        TopLevelMode::Global => ScopeType::Global,
        TopLevelMode::Module => ScopeType::Module,
      }),
      session,
      rules: ParsePatternRules {
        await_allowed: true,
        yield_allowed: true,
      },
    };
    let mut body = ctx.session.new_vec();
    while !self.consume_if(TokenType::EOF)?.is_match() {
      body.push(self.parse_stmt(ctx)?);
    }
    let top_level_node = ctx.create_node(self.source_range(), Syntax::TopLevel { body });
    Ok(top_level_node)
  }
}