diff options
Diffstat (limited to 'support')
-rw-r--r-- | support/lib/util.py | 20 | ||||
-rw-r--r-- | support/openscad/__init__.py | 0 | ||||
-rw-r--r-- | support/openscad/__main__.py | 37 |
3 files changed, 56 insertions, 1 deletions
diff --git a/support/lib/util.py b/support/lib/util.py index a3bc3dc..bede240 100644 --- a/support/lib/util.py +++ b/support/lib/util.py @@ -1,4 +1,4 @@ -import contextlib, subprocess, tempfile, shutil +import contextlib, subprocess, tempfile, shutil, re, os @contextlib.contextmanager @@ -16,3 +16,21 @@ def command(args): process.wait() assert not process.returncode + + +def bash_escape_string(string): + return "'{}'".format(re.sub("'", "'\"'\"'", string)) + + +def write_file(path, data): + temp_path = path + '~' + + with open(temp_path, 'wb') as file: + file.write(data) + + os.rename(temp_path, path) + + +def read_file(path): + with open(path, 'rb') as file: + return file.read() diff --git a/support/openscad/__init__.py b/support/openscad/__init__.py new file mode 100644 index 0000000..e69de29 --- /dev/null +++ b/support/openscad/__init__.py diff --git a/support/openscad/__main__.py b/support/openscad/__main__.py new file mode 100644 index 0000000..08b2ab6 --- /dev/null +++ b/support/openscad/__main__.py @@ -0,0 +1,37 @@ +import os, sys +from lib import util + + +def _openscad(in_path, out_path, deps_path): + util.command([os.environ['OPENSCAD'], '-o', out_path, '-d', deps_path, in_path]) + + +def _write_dependencies(path, target, dependencies): + util.write_file(path, '{}: {}\n'.format(target, ' '.join(dependencies)).encode()) + + +def main(in_path, out_path, deps_path): + cwd = os.getcwd() + + def relpath(path): + return os.path.relpath(path, cwd) + + with util.TemporaryDirectory() as temp_dir: + temp_deps_path = os.path.join(temp_dir, 'deps') + temp_mk_path = os.path.join(temp_dir, 'mk') + temp_files_path = os.path.join(temp_dir, 'files') + + _openscad(in_path, out_path, temp_deps_path) + + mk_content = '%:; echo "$@" >> {}'.format(util.bash_escape_string(temp_files_path)) + + util.write_file(temp_mk_path, mk_content.encode()) + util.command(['make', '-s', '-B', '-f', temp_mk_path, '-f', temp_deps_path]) + + deps = set(map(relpath, util.read_file(temp_files_path).decode().splitlines())) + ignored_files = set(map(relpath, [temp_deps_path, temp_mk_path, in_path, out_path])) + + _write_dependencies(deps_path, relpath(out_path), deps - ignored_files) + + +main(*sys.argv[1:]) |