#!/usr/bin/env python3

# BGI script dumper

import glob
import os
import struct
import sys
import io

import bgi_common
import bgi_setup

def dump_text(fo, marker, id, text, comment):
    fo.write(u'//%s\n' % comment)
    fo.write(u'<%s%s%04d>%s\n' % (bgi_setup.slang,marker,id,text))
    for lang in bgi_setup.dlang:
        if bgi_setup.dcopy:
            fo.write(u'<%s%s%04d>%s\n' % (lang,marker,id,text))
        else:
            fo.write(u'<%s%s%04d>\n' % (lang,marker,id))
    fo.write(u'\n')

def dump_unique(fo, code_section, imarker):
    text_set = set()
    for addr in sorted(code_section):
        text, id, marker, comment = code_section[addr]
        if marker == imarker and text not in text_set:
            dump_text(fo, marker, id, bgi_common.escape(text), comment)
            text_set.add(text)

def dump_sequential(fo, code_section, imarker):
    for addr in sorted(code_section):
        text, id, marker, comment = code_section[addr]
        if marker == imarker:
            dump_text(fo, marker, id, bgi_common.escape(text), comment)

def dump_script(script):
    data = open(script, 'rb').read()
    hdr_bytes, code_bytes, text_bytes, config = bgi_common.split_data(data)
    text_section = bgi_common.get_text_section(text_bytes)
    code_section = bgi_common.get_code_section(code_bytes, text_section, config)
    fo = io.open(script+bgi_setup.dext, 'w', encoding=bgi_setup.denc)
    dump_unique(fo, code_section, 'N')        # names
    fo.write(u'///'+u'='*80+u'\n\n')
    dump_sequential(fo, code_section, 'T')    # text
    fo.write(u'///'+u'='*80+u'\n\n')
    dump_unique(fo, code_section, 'Z')        # other
    fo.close()

if __name__ == '__main__':
    if len(sys.argv) < 2:
        print('Usage: bgi_dump.py <file(s)>')
        print('(only extension-less files amongst <file(s)> will be processed)')
        sys.exit(1)
    for arg in sys.argv[1:]:
        for script in glob.glob(arg):
            base, ext = os.path.splitext(script)
            if not ext and os.path.isfile(script):
                print('Dumping %s...' % script)
                dump_script(script)