rush_build.py

#!/bin/python3
import os
import sys
import subprocess
from pathlib import Path
from enum import Enum


class Backend(Enum):
    llvm = 0
    riscv = 1
    c = 2

JOBS = [
    # Not automatically built with the document
    {
        'in': './listings/fib.rush',
        'out': './listings/fib.ll',
        'backend': Backend.llvm,
        'auto': False,
    },
    {
        'in': './listings/simple.rush',
        'out': './listings/simple.ll',
        'backend': Backend.llvm,
        'auto': False,
    },
    # Automatically built with the document
    {
        'in': './listings/simple_scope.rush',
        'out': './listings/generated/rush_simple_scope.c',
        'backend': Backend.c,
        'auto': True,
    },
    {
        'in': './listings/riscv_simple.rush',
        'out': './listings/generated/riscv_simple.s',
        'backend': Backend.riscv,
        'auto': True,
    },
    {
        'in': './listings/rush_simple_pointer.rush',
        'out': './listings/generated/riscv_rush_simple_pointer.s',
        'backend': Backend.riscv,
        'auto': True,
    },
    {
        'in': './listings/simple_add.rush',
        'out': './listings/generated/rush_simple_add.s',
        'backend': Backend.riscv,
        'auto': True,
    },
    {
        'in': './listings/while_loop.rush',
        'out': './listings/generated/rush_while_loop.s',
        'backend': Backend.riscv,
        'auto': True,
    },
]


def llvm_gen_ir(source: str, output: str):
    input_path = os.path.realpath(source)
    output_path = os.path.realpath(output)

    start_dir = os.path.realpath(os.curdir)
    os.chdir('./deps/rush/crates/rush-compiler-llvm/')

    subprocess.run(
        f'cargo r {input_path}',
        shell=True,
    ).check_returncode()

    os.rename('output.ll', output_path)
    os.chdir(start_dir)

    print(f'generated {output_path}')


def riscv_asm(source: str, output: str, line_width: int):
    input_path = os.path.realpath(source)
    output_path = os.path.realpath(output)

    start_dir = os.path.realpath(os.curdir)
    os.chdir('./deps/rush/crates/rush-compiler-risc-v/')

    subprocess.run(
        f'cargo r {input_path} {line_width}',
        shell=True,
    ).check_returncode()

    os.rename('output.s', output_path)
    os.chdir(start_dir)

    print(f'generated {output_path}')


def transpile(source: str, output: str):
    input_path = os.path.realpath(source)
    output_path = os.path.realpath(output)

    start_dir = os.path.realpath(os.curdir)
    os.chdir('./deps/rush/crates/rush-transpiler-c/')

    subprocess.run(
        f'cargo r {input_path}',
        shell=True,
    ).check_returncode()

    os.rename('output.c', output_path)
    os.chdir(start_dir)

    print(f'generated {output_path}')


def lint_all():
    rush_files = list(Path('.').rglob('*.rush'))
    for rush_path in rush_files:
        if str(rush_path).startswith('deps/rush'):
            continue
        input_path = os.path.realpath(rush_path)

        start_dir = os.path.realpath(os.curdir)
        os.chdir('./deps/rush/crates/rush-analyzer/')

        res = subprocess.run(
            f'cargo r {input_path}',
            shell=True,
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
        )

        if res.returncode:
            print('\x1b[1;31m=== CHECK FAILURE ===\x1b[1;m')
            sys.stdout.buffer.write(res.stderr)
            sys.stdout.buffer.write(res.stdout)
            exit(1)

        os.chdir(start_dir)
        print(f'ok: {input_path}')
    print('\x1b[1;32m=== CHECK SUCCESS ===\x1b[1;m')


def assure_used_listings():
    unused = list()
    files = list(Path('.').rglob('listings/**/*'))
    for f_path in files:
        if not str(f_path).startswith('.git'):
            res = subprocess.run(
                f'rg -q {f_path} -t tex',
                shell=True,
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
            )
            if res.returncode:
                unused.append(str(f_path))
            print(f'ok (used): {f_path}')
    if unused:
        print('\x1b[1;31m=== CHECK FAILURE: NOT ALL USED ===\x1b[1;m')
        for f_path in unused:
            print(f'UNUSED: {f_path}')
        exit(1)
    print('\x1b[1;32m=== CHECK SUCCESS ===\x1b[1;m')


if __name__ == '__main__':
    if len(sys.argv) != 2:
        print(f'Expected exactly one argument, got {len(sys.argv) - 1}')
        exit(1)

    if sys.argv[1] == 'check':
        lint_all()
    elif sys.argv[1] == 'used':
        assure_used_listings()
    elif sys.argv[1] == 'build' or sys.argv[1] == 'regen':
        for job in JOBS:
            if sys.argv[1] == 'build' and job['auto'] == False:
                continue
            if job['backend'] == Backend.riscv:
                riscv_asm(job['in'], job['out'], 17)
            elif job['backend'] == Backend.llvm:
                llvm_gen_ir(job['in'], job['out'])
            elif job['backend'] == Backend.c:
                transpile(job['in'], job['out'])
            else:
                raise Exception('Invalid enum in job')
    else:
        print(f'Invalid command-line argument: {sys.argv[1]}')
        exit(1)