tiny-basic/parser_util.py at main · dilshan/tiny-basic · GitHub

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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
# SPDX-License-Identifier: BSD-3-Clause
# Copyright (c) 2026 Dilshan R Jayakody. All rights reserved.

import os
import subprocess
import shutil

Import("env")


def run_parser_generators(env):
    print("\n[Bison/Flex] Generating parser and lexer files...")

    proj_dir = env.get("PROJECT_DIR")
    src_dir = os.path.join(proj_dir, "src")
    include_dir = os.path.join(proj_dir, "include")
    parser_dir = os.path.join(proj_dir, "parser")

    bison_file = os.path.join(proj_dir, "parser", "tinybasic.y")
    flex_file = os.path.join(proj_dir, "parser", "tinybasic.l")

    os.makedirs(src_dir, exist_ok=True)
    os.makedirs(include_dir, exist_ok=True)

    for f in [
        os.path.join(parser_dir, "tinybasic.tab.c"),
        os.path.join(src_dir, "tinybasic.tab.c"),
        os.path.join(src_dir, "lex.yy.c"),
        os.path.join(include_dir, "tinybasic.tab.h"),
    ]:
        if os.path.exists(f):
            os.remove(f)

    print("[Bison] Processing tinybasic.y...")

    subprocess.check_call([
        "bison",
        "-d",
        "-o",
        os.path.join(src_dir, "tinybasic.tab.c"),
        bison_file
    ])

    generated_header = os.path.join(src_dir, "tinybasic.tab.h")

    print("[Flex] Processing tinybasic.l...")

    subprocess.check_call([
        "flex",
        "-o",
        os.path.join(src_dir, "lex.yy.c"),
        flex_file
    ])

    if os.path.exists(generated_header):
        shutil.move(
            generated_header,
            os.path.join(include_dir, "tinybasic.tab.h")
        )

    print("[Bison/Flex] Target files successfully generated\n")


def clean_generated_files(env):
    print("\n[Clean] Removing generated parser and lexer files...")

    proj_dir = env.get("PROJECT_DIR")
    src_dir = os.path.join(proj_dir, "src")
    include_dir = os.path.join(proj_dir, "include")
    parser_dir = os.path.join(proj_dir, "parser")

    for f in [
        os.path.join(parser_dir, "tinybasic.tab.c"),
        os.path.join(src_dir, "tinybasic.tab.c"),
        os.path.join(src_dir, "lex.yy.c"),
        os.path.join(include_dir, "tinybasic.tab.h"),
    ]:
        if os.path.exists(f):
            os.remove(f)

    print("[Clean] Generated parser and lexer files removed\n")


if env.IsCleanTarget():
    clean_generated_files(env)
else:
    run_parser_generators(env)