157 lines
		
	
	
		
			4.8 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
			
		
		
	
	
			157 lines
		
	
	
		
			4.8 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
#!/usr/bin/env python3
 | 
						|
 | 
						|
# Build the Debian packages using Docker images.
 | 
						|
#
 | 
						|
# This script builds the Docker images and then executes them sequentially, each
 | 
						|
# one building a Debian package for the targeted operating system. It is
 | 
						|
# designed to be a "single command" to produce all the images.
 | 
						|
#
 | 
						|
# By default, builds for all known distributions, but a list of distributions
 | 
						|
# can be passed on the commandline for debugging.
 | 
						|
 | 
						|
import argparse
 | 
						|
import os
 | 
						|
import signal
 | 
						|
import subprocess
 | 
						|
import sys
 | 
						|
import threading
 | 
						|
from concurrent.futures import ThreadPoolExecutor
 | 
						|
 | 
						|
DISTS = (
 | 
						|
    "debian:stretch",
 | 
						|
    "debian:buster",
 | 
						|
    "debian:bullseye",
 | 
						|
    "debian:sid",
 | 
						|
    "ubuntu:xenial",
 | 
						|
    "ubuntu:bionic",
 | 
						|
    "ubuntu:eoan",
 | 
						|
    "ubuntu:focal",
 | 
						|
)
 | 
						|
 | 
						|
DESC = '''\
 | 
						|
Builds .debs for synapse, using a Docker image for the build environment.
 | 
						|
 | 
						|
By default, builds for all known distributions, but a list of distributions
 | 
						|
can be passed on the commandline for debugging.
 | 
						|
'''
 | 
						|
 | 
						|
 | 
						|
class Builder(object):
 | 
						|
    def __init__(self, redirect_stdout=False):
 | 
						|
        self.redirect_stdout = redirect_stdout
 | 
						|
        self.active_containers = set()
 | 
						|
        self._lock = threading.Lock()
 | 
						|
        self._failed = False
 | 
						|
 | 
						|
    def run_build(self, dist):
 | 
						|
        """Build deb for a single distribution"""
 | 
						|
 | 
						|
        if self._failed:
 | 
						|
            print("not building %s due to earlier failure" % (dist, ))
 | 
						|
            raise Exception("failed")
 | 
						|
 | 
						|
        try:
 | 
						|
            self._inner_build(dist)
 | 
						|
        except Exception as e:
 | 
						|
            print("build of %s failed: %s" % (dist, e), file=sys.stderr)
 | 
						|
            self._failed = True
 | 
						|
            raise
 | 
						|
 | 
						|
    def _inner_build(self, dist):
 | 
						|
        projdir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
 | 
						|
        os.chdir(projdir)
 | 
						|
 | 
						|
        tag = dist.split(":", 1)[1]
 | 
						|
 | 
						|
        # Make the dir where the debs will live.
 | 
						|
        #
 | 
						|
        # Note that we deliberately put this outside the source tree, otherwise
 | 
						|
        # we tend to get source packages which are full of debs. (We could hack
 | 
						|
        # around that with more magic in the build_debian.sh script, but that
 | 
						|
        # doesn't solve the problem for natively-run dpkg-buildpakage).
 | 
						|
        debsdir = os.path.join(projdir, '../debs')
 | 
						|
        os.makedirs(debsdir, exist_ok=True)
 | 
						|
 | 
						|
        if self.redirect_stdout:
 | 
						|
            logfile = os.path.join(debsdir, "%s.buildlog" % (tag, ))
 | 
						|
            print("building %s: directing output to %s" % (dist, logfile))
 | 
						|
            stdout = open(logfile, "w")
 | 
						|
        else:
 | 
						|
            stdout = None
 | 
						|
 | 
						|
        # first build a docker image for the build environment
 | 
						|
        subprocess.check_call([
 | 
						|
            "docker", "build",
 | 
						|
            "--tag", "dh-venv-builder:" + tag,
 | 
						|
            "--build-arg", "distro=" + dist,
 | 
						|
            "-f", "docker/Dockerfile-dhvirtualenv",
 | 
						|
            "docker",
 | 
						|
        ], stdout=stdout, stderr=subprocess.STDOUT)
 | 
						|
 | 
						|
        container_name = "synapse_build_" + tag
 | 
						|
        with self._lock:
 | 
						|
            self.active_containers.add(container_name)
 | 
						|
 | 
						|
        # then run the build itself
 | 
						|
        subprocess.check_call([
 | 
						|
            "docker", "run",
 | 
						|
            "--rm",
 | 
						|
            "--name", container_name,
 | 
						|
            "--volume=" + projdir + ":/synapse/source:ro",
 | 
						|
            "--volume=" + debsdir + ":/debs",
 | 
						|
            "-e", "TARGET_USERID=%i" % (os.getuid(), ),
 | 
						|
            "-e", "TARGET_GROUPID=%i" % (os.getgid(), ),
 | 
						|
            "dh-venv-builder:" + tag,
 | 
						|
        ], stdout=stdout, stderr=subprocess.STDOUT)
 | 
						|
 | 
						|
        with self._lock:
 | 
						|
            self.active_containers.remove(container_name)
 | 
						|
 | 
						|
        if stdout is not None:
 | 
						|
            stdout.close()
 | 
						|
            print("Completed build of %s" % (dist, ))
 | 
						|
 | 
						|
    def kill_containers(self):
 | 
						|
        with self._lock:
 | 
						|
            active = list(self.active_containers)
 | 
						|
 | 
						|
        for c in active:
 | 
						|
            print("killing container %s" % (c,))
 | 
						|
            subprocess.run([
 | 
						|
                "docker", "kill", c,
 | 
						|
            ], stdout=subprocess.DEVNULL)
 | 
						|
            with self._lock:
 | 
						|
                self.active_containers.remove(c)
 | 
						|
 | 
						|
 | 
						|
def run_builds(dists, jobs=1):
 | 
						|
    builder = Builder(redirect_stdout=(jobs > 1))
 | 
						|
 | 
						|
    def sig(signum, _frame):
 | 
						|
        print("Caught SIGINT")
 | 
						|
        builder.kill_containers()
 | 
						|
    signal.signal(signal.SIGINT, sig)
 | 
						|
 | 
						|
    with ThreadPoolExecutor(max_workers=jobs) as e:
 | 
						|
        res = e.map(builder.run_build, dists)
 | 
						|
 | 
						|
    # make sure we consume the iterable so that exceptions are raised.
 | 
						|
    for r in res:
 | 
						|
        pass
 | 
						|
 | 
						|
 | 
						|
if __name__ == '__main__':
 | 
						|
    parser = argparse.ArgumentParser(
 | 
						|
        description=DESC,
 | 
						|
    )
 | 
						|
    parser.add_argument(
 | 
						|
        '-j', '--jobs', type=int, default=1,
 | 
						|
        help='specify the number of builds to run in parallel',
 | 
						|
    )
 | 
						|
    parser.add_argument(
 | 
						|
        'dist', nargs='*', default=DISTS,
 | 
						|
        help='a list of distributions to build for. Default: %(default)s',
 | 
						|
    )
 | 
						|
    args = parser.parse_args()
 | 
						|
    run_builds(dists=args.dist, jobs=args.jobs)
 |