doge-compiler 0.3.3

Compiler for the Doge programming language — lexer, parser, semantic checks, and Rust codegen.
Documentation 
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

use super::*;

/// Every nested function defined directly in a body — crossing `if`/`for`/
/// `while`/`pls` blocks but not another function's body — with its name span.
pub(super) fn nested_funcs_with_span(body: &[Stmt]) -> Vec<(&str, Span)> {
    let mut out = Vec::new();
    collect_nested_funcs(body, &mut out);
    out
}

pub(super) fn nested_func_names(body: &[Stmt]) -> HashSet<String> {
    nested_funcs_with_span(body)
        .into_iter()
        .map(|(name, _)| name.to_string())
        .collect()
}

/// The variable-like bindings of one scope — `such`/`so` declarations, `for`
/// loop variables, and `oh no` error names — crossing blocks but not another
/// function's body. Nested-function names are handled separately. Descent into
/// child blocks goes through the shared [`for_each_child_block`] walker, so a new
/// block-carrying statement is covered without editing this function.
pub(super) fn collect_var_bindings(body: &[Stmt], out: &mut HashSet<String>) {
    for stmt in body {
        match stmt {
            Stmt::ConstDecl { name, .. } => {
                out.insert(name.clone());
            }
            Stmt::Decl { names, rest, .. } => {
                out.extend(names.iter().cloned());
                out.extend(rest.iter().cloned());
            }
            Stmt::For { vars, rest, .. } => {
                out.extend(vars.iter().cloned());
                out.extend(rest.iter().cloned());
            }
            Stmt::Try { err_name, .. } => {
                out.insert(err_name.clone());
            }
            _ => {}
        }
        for_each_child_block(stmt, &mut |block| collect_var_bindings(block, out));
    }
}

pub(super) fn collect_nested_funcs<'a>(body: &'a [Stmt], out: &mut Vec<(&'a str, Span)>) {
    for stmt in body {
        if let Stmt::FuncDef { name, span, .. } = stmt {
            out.push((name, *span));
        }
        for_each_child_block(stmt, &mut |block| collect_nested_funcs(block, out));
    }
}