#!/usr/bin/env python3 import os import re import shlex import shutil import subprocess from sys import stderr cmds = {} warnings = [] class CheckError(Exception): pass def checking(thing): stderr.write("checking %s... " % thing) stderr.flush() def warn(string): warnings.append(string) stderr.write("WARNING: %s\n" % string) def check(name, why, flags=[], optional=False, var=None, run=True): global warn checking("for %s" % name) var = var or name.upper() varvalue = os.getenv(var) if varvalue == '': stderr.write("skipping\n") if not optional: warn("%s is required to %s, the code will not fall back gracefully without it!" % (name, why)) return elif varvalue is None: varvalue = name split = shlex.split(varvalue) exe = shutil.which(split[0]) if not exe: stderr.write("not found\n") if not optional: raise CheckError() else: warn("%s is recommended to %s." % (name, why)) return stderr.write("%s\n" % exe) cmd = [exe] + split[1:] + flags + shlex.split(os.getenv(var + "FLAGS") or "") if run: checking("%s usability" % exe) try: subprocess.check_call(cmd + ["-h"], stdout=subprocess.DEVNULL, stderr=subprocess.STDOUT) stderr.write("yes\n") except subprocess.CalledProcessError as e: raise CheckError() from e cmds[var] = cmd return cmd def run_ffmpeg(arg): return subprocess.check_output(ffmpeg + [arg], stderr=subprocess.STDOUT).decode('utf-8') def ffmpeg_check(name, regex, checks, output): for check in checks: checking("if %s supports %s %s" % (ffmpeg[0], name, check)) if re.search("\n %s %s " % (regex, check), output): stderr.write("yes\n") else: stderr.write("no\n") raise CheckError() if __name__ == "__main__": # no, this line isn't bugged. really. examine check() and Makefile if not check("zopfli", "compress gzip files better than gzip -9", optional=True, var="GZIP"): check("gzip", "pre-compress the JSON files", ["-9"]) check("apngasm", "assemble the click-to-continue image", run=False) check("convert", "perform miscellaneous actions on the images") check("cwebp", "convert all images to WebP for webkit browsers", ["-quiet", "-alpha_cleanup", "-m", "6"]) ffmpeg = check("ffmpeg", "convert audio and video to HTML5 formats", ["-v", "warning", "-y"], run=False) if ffmpeg: ffmpeg_formats = run_ffmpeg("-formats") ffmpeg_check("demuxing", "D.", ["matroska,webm", "ogg", "wav", "yuv4mpegpipe"], ffmpeg_formats) ffmpeg_check("muxing", ".E", ["ipod", "mp4", "ogg", "wav", "webm", "yuv4mpegpipe"], ffmpeg_formats) ffmpeg_check("decoding", ".{6}", ["mpeg4", "rawvideo", "pcm_s16le", "vorbis"], run_ffmpeg("-decoders")) ffmpeg_check("encoding", ".{6}", ["libx264", "rawvideo", "libtheora", "libvpx", "libvpx-vp9", "libfdk_aac", "libopus", "pcm_s16le"], run_ffmpeg("-encoders")) check("npm", "install uglifyjs", ["--quiet"]) check("webpmux", "assemble the WebP click-to-continue image") check("defluff", "decrease the size of DEFLATE files including gzip and PNG files", optional=True, run=False) check("pngquant", "quantize PNG images, improving compatibility and dramatically decreasing size", optional=True) check("zopflipng", "recompress DEFLATE files including gzip and PNG files", optional=True) stderr.write("creating Makefile.inc\n") with open("Makefile.inc", "w") as f: f.write(''.join('%s := %s\n' % (k, subprocess.list2cmdline(cmds[k])) for k in cmds)) if len(warnings): stderr.write("Warnings during configuration:\n") stderr.write('\n'.join(warnings) + '\n') stderr.write("Run `make' now to build HTML5KS.\n")