import json
import signal
import sys

from docutils.core import publish_string


OVERRIDES = {'embed_stylesheet': False,
             'smart_quotes': True,
             'syntax_highlight': 'short',
             'trim_footnote_reference_space': True,
             'initial_header_level': 2}


def signal_handler(_sig, _frame):
    sys.exit(0)


def assoc(d, **kvs):
    return({**d, **kvs})


def main():
    command = ''
    while command != 'quit':
        line = sys.stdin.readline()
        data = json.loads(line)
        command = data['command']
        if command == 'translate':
            overrides = {}
            if 'slug' in data and data['slug']:
                id_prefix = data['slug'] + '_%'
                overrides = assoc(OVERRIDES, auto_id_prefix=id_prefix)

            output = publish_string(data['string'],
                                    writer_name='html',
                                    settings_overrides=overrides)
            sys.stdout.write(json.dumps({'output': output.decode()}))
            sys.stdout.write('\n')
            sys.stdout.flush()


if __name__ == '__main__':
    signal.signal(signal.SIGINT, signal_handler)
    main()