Function create_script_source

hallucinate/script.py:30–77

(args)

def create_script_source(args):
    scriptsrc = ''
    options = {
        'debug': False,
        'replaceBuffer': args.replaceBuffer,
        'map': {}
    }
    if args.verbose >= 2:
        options['debug'] = True

    if args.mapfile is not None:
        with open(args.mapfile, 'rb') as f:
            options['map'] = json.load(f)

    scriptsrc += 'const options = ' + json.dumps(options) + ';'
    mypath = os.path.realpath(os.path.dirname(__file__))
    with open(mypath + os.path.sep + 'utils.js', 'r') as f:
        scriptsrc += f.read()

    # Avoid scope collisions in scripts
    def wrap_module(modname, source):
        return """
    // from module %s
    (function() {
    %s
    }());
        """ % (modname, source)

    def add_dir(root, predicate):
        for name in os.listdir(root):
            p = os.path.join(root, name)
            if not os.path.isfile(p):
                continue
            if not predicate(name):
                logging.debug('Disabled module %s', name)
                continue

            logging.debug('Adding module %s', name)
            with open(p, 'r') as f:
                yield wrap_module(name, f.read())

    for source in add_dir(mypath + os.path.sep + 'modules/', lambda x: x not in args.disable):
        scriptsrc += source

    for source in add_dir(mypath + os.path.sep + 'modules/optional', lambda x: x in args.enable):
        scriptsrc += source

    return scriptsrc