aboutsummaryrefslogtreecommitdiff
path: root/test/script_loader.py
blob: 22196c32230c68434a3bd23bdaecd612423d4507 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
# SPDX-License-Identifier: GPL-3.0-or-later

"""
Loading of parts of Haketilo source for testing in browser
"""

# This file is part of Haketilo.
#
# Copyright (C) 2021 Wojtek Kosior <koszko@koszko.org>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.
#
# I, Wojtek Kosior, thereby promise not to sue for violation of this file's
# license. Although I request that you do not make use this code in a
# proprietary program, I am not going to enforce this in court.

from pathlib import Path
import subprocess, re

from .misc_constants import *

script_root = here.parent
awk_script = script_root / 'compute_scripts.awk'

def make_relative_path(path):
    path = Path(path)

    if path.is_absolute():
        path = path.relative_to(script_root)

    return path

"""Used to ignore hidden files and emacs auto-save files."""
script_name_regex = re.compile(r'^[^.#].*\.js$')

def available_scripts(directory):
    for script in directory.rglob('*.js'):
        if script_name_regex.match(script.name):
            yield script

def get_wrapped_script(script_path):
    if script_path == 'exports_init.js':
        with open(script_root / 'MOZILLA_exports_init.js') as script:
            return script.read()

    awk = subprocess.run(['awk', '-f', str(awk_script), 'wrapped_code',
                          str(script_path)],
                         stdout=subprocess.PIPE, cwd=script_root, check=True)

    return awk.stdout.decode()

def load_script(path, import_dirs):
    """
    `path` and `import_dirs` are .js file path and a list of directory paths,
    respectively. They may be absolute or specified relative to Haketilo's
    project directory.

    Return a string containing script from `path` together with all other
    scripts it depends on, wrapped in the same way Haketilo's build system wraps
    them, with imports properly satisfied.
    """
    path = make_relative_path(path)

    import_dirs = [make_relative_path(dir) for dir in import_dirs]
    available = [s for dir in import_dirs for s in available_scripts(dir)]

    awk = subprocess.run(['awk', '-f', str(awk_script), 'script_dependencies',
                          str(path), *[str(s) for s in available]],
                         stdout=subprocess.PIPE, cwd=script_root, check=True)

    output = awk.stdout.decode()

    return '\n'.join([get_wrapped_script(path) for path in output.split()])