main.py

#!/bin/env python3
from array import array
from struct import unpack
from sys import argv
from typing import TypeVar

from functions import write_functions
from tables import write_tables
from variables import write_variables

T = TypeVar('T')

def read_ksm_container(file: bytes) -> list[bytes]:
    header = list(unpack('4siiiiiiiiii', file[:0x2c]))
    assert header[0] == b'KSMR'
    assert header[1] == 0x10300
    assert header[10] == 0
    
    header[10] = len(file) // 4
    
    # god python can be so beautiful
    sections = [file[start * 4:end * 4] for start, end in zip(header[2:], header[3:])]
    return sections

def print_raw_integers(section: bytes) -> str:
    arr = array('I', section)
    out = ""
    for n in arr:
        if n == 0:
            out += f"  - 0\n"
        else:
            out += f"  - 0x{n:x}\n"
    return out

def main():
    if len(argv) == 1 or argv[1] == '--help' or argv[1] == '-h':
        print("Sticker Star KSM Script Dumper")
        print("Usage: main.py <input file.bin>")
        return
    
    with open(argv[1], 'rb') as f:
        input_file = f.read()
    
    # predefined symbol ids for expressions (incomplete)
    symbol_ids = {
    0x3f: 'next_function',
   #0x40: exit,
    0x41: '(',
    0x42: ')',
    0x43: '||',
    0x44: '&&',
    
    0x4a: '==',
    0x4b: '!=',
    0x4c: '>',
    0x4d: '<',
    0x4e: '>=',
    0x4f: '<=',
    
    0x53: '+',
    0x54: '-',
    0x55: '*',
    0x56: '/',
    }
    sections = read_ksm_container(input_file)
    
    write_variables(sections, symbol_ids)
    write_tables(sections, symbol_ids)
    write_functions(sections, symbol_ids)
    
    # # section 0 (mysterious)
    # out_str = 'section_0:\n'
    # out_str += print_raw_integers(sections[0])
    
    # with open(argv[1] + '.yaml', 'w') as f:
    #     f.write(out_str)

if __name__ ==  '__main__':
    main()