@aptre/v86 0.5.0

package/gen/rust_ast.ts

@@ -0,0 +1,107 @@
+import assert from 'node:assert/strict'
+
+export interface SwitchCase {
+    conditions: (string | number)[]
+    body: Statement[]
+}
+
+export interface DefaultCase {
+    varname?: string
+    body: Statement[]
+}
+
+export interface SwitchStatement {
+    type: 'switch'
+    condition: string
+    cases: SwitchCase[]
+    default_case?: DefaultCase
+}
+
+export interface IfBlock {
+    condition: string
+    body: Statement[]
+}
+
+export interface IfElseStatement {
+    type: 'if-else'
+    if_blocks: IfBlock[]
+    else_block?: { body: Statement[] }
+}
+
+export type Statement = string | SwitchStatement | IfElseStatement
+
+function indent(lines: string[], how_much: number): string[] {
+    return lines.map((line) => ' '.repeat(how_much) + line)
+}
+
+export function print_syntax_tree(statements: Statement[]): string[] {
+    const code: string[] = []
+
+    for (const statement of statements) {
+        if (typeof statement === 'string') {
+            code.push(statement)
+        } else if (statement.type === 'switch') {
+            assert(statement.condition)
+
+            const cases: string[] = []
+
+            for (const case_ of statement.cases) {
+                assert(case_.conditions.length >= 1)
+
+                cases.push(case_.conditions.join(' | ') + ' => {')
+                cases.push(...indent(print_syntax_tree(case_.body), 4))
+                cases.push(`},`)
+            }
+
+            if (statement.default_case) {
+                const varname = statement.default_case.varname || '_'
+                cases.push(`${varname} => {`)
+                cases.push(
+                    ...indent(
+                        print_syntax_tree(statement.default_case.body),
+                        4,
+                    ),
+                )
+                cases.push(`}`)
+            }
+
+            code.push(`match ${statement.condition} {`)
+            code.push(...indent(cases, 4))
+            code.push(`}`)
+        } else if (statement.type === 'if-else') {
+            assert(statement.if_blocks.length >= 1)
+
+            const first_if_block = statement.if_blocks[0]
+
+            code.push(`if ${first_if_block.condition} {`)
+            code.push(...indent(print_syntax_tree(first_if_block.body), 4))
+            code.push(`}`)
+
+            for (let i = 1; i < statement.if_blocks.length; i++) {
+                const if_block = statement.if_blocks[i]
+
+                code.push(`else if ${if_block.condition} {`)
+                code.push(...indent(print_syntax_tree(if_block.body), 4))
+                code.push(`}`)
+            }
+
+            if (statement.else_block) {
+                code.push(`else {`)
+                code.push(
+                    ...indent(print_syntax_tree(statement.else_block.body), 4),
+                )
+                code.push(`}`)
+            }
+        } else {
+            assert(
+                false,
+                'Unexpected type: ' +
+                    (statement as { type: string }).type +
+                    ' In: ' +
+                    JSON.stringify(statement),
+            )
+        }
+    }
+
+    return code
+}

package/gen/util.ts

@@ -0,0 +1,35 @@
+import fs from 'node:fs'
+import path from 'node:path'
+import process from 'node:process'
+
+const CYAN_FMT = '\x1b[36m%s\x1b[0m'
+
+export function hex(n: number, pad: number = 0): string {
+    let s = n.toString(16).toUpperCase()
+    while (s.length < pad) s = '0' + s
+    return s
+}
+
+export function get_switch_value(arg_switch: string): string | null {
+    const argv = process.argv
+    const switch_i = argv.indexOf(arg_switch)
+    const val_i = switch_i + 1
+    if (switch_i > -1 && val_i < argv.length) {
+        return argv[switch_i + 1]
+    }
+    return null
+}
+
+export function get_switch_exist(arg_switch: string): boolean {
+    return process.argv.includes(arg_switch)
+}
+
+export function finalize_table_rust(
+    out_dir: string,
+    name: string,
+    contents: string,
+): void {
+    const file_path = path.join(out_dir, name)
+    fs.writeFileSync(file_path, contents)
+    console.log(CYAN_FMT, `[+] Wrote table ${name}.`)
+}