102871c91SPatrick Williams#!/usr/bin/env python3
202871c91SPatrick Williams#
302871c91SPatrick Williams# Build the required docker image to run package unit tests
402871c91SPatrick Williams#
502871c91SPatrick Williams# Script Variables:
602871c91SPatrick Williams#   DOCKER_IMG_NAME:  <optional, the name of the docker image to generate>
702871c91SPatrick Williams#                     default is openbmc/ubuntu-unit-test
802871c91SPatrick Williams#   DISTRO:           <optional, the distro to build a docker image against>
973395159SPatrick Williams#                     default is ubuntu:hirsute
1050837436SPatrick Williams#   FORCE_DOCKER_BUILD: <optional, a non-zero value with force all Docker
1150837436SPatrick Williams#                     images to be rebuilt rather than reusing caches.>
1250837436SPatrick Williams#   BUILD_URL:        <optional, used to detect running under CI context
1350837436SPatrick Williams#                     (ex. Jenkins)>
1402871c91SPatrick Williams#   BRANCH:           <optional, branch to build from each of the openbmc/
1502871c91SPatrick Williams#                     repositories>
1602871c91SPatrick Williams#                     default is master, which will be used if input branch not
1702871c91SPatrick Williams#                     provided or not found
1802871c91SPatrick Williams#   UBUNTU_MIRROR:    <optional, the URL of a mirror of Ubuntu to override the
1902871c91SPatrick Williams#                     default ones in /etc/apt/sources.list>
2002871c91SPatrick Williams#                     default is empty, and no mirror is used.
2102871c91SPatrick Williams#   http_proxy        The HTTP address of the proxy server to connect to.
2202871c91SPatrick Williams#                     Default: "", proxy is not setup if this is not set
2302871c91SPatrick Williams
2402871c91SPatrick Williamsimport os
2502871c91SPatrick Williamsimport sys
26b16f3e20SPatrick Williamsimport threading
27a18d9c57SPatrick Williamsfrom datetime import date
28a18d9c57SPatrick Williamsfrom hashlib import sha256
29ee3c9eebSPatrick Williamsfrom sh import docker, git, nproc, uname  # type: ignore
30ee3c9eebSPatrick Williamsfrom typing import Any, Callable, Dict, Iterable, Optional
3102871c91SPatrick Williams
32ee3c9eebSPatrick Williamstry:
33ee3c9eebSPatrick Williams    # Python before 3.8 doesn't have TypedDict, so reroute to standard 'dict'.
34ee3c9eebSPatrick Williams    from typing import TypedDict
35ee3c9eebSPatrick Williamsexcept:
36ee3c9eebSPatrick Williams
37ee3c9eebSPatrick Williams    class TypedDict(dict):  # type: ignore
38ee3c9eebSPatrick Williams        # We need to do this to eat the 'total' argument.
39ee3c9eebSPatrick Williams        def __init_subclass__(cls, **kwargs):
40ee3c9eebSPatrick Williams            super().__init_subclass__()
41ee3c9eebSPatrick Williams
42ee3c9eebSPatrick Williams
43ee3c9eebSPatrick Williams# Declare some variables used in package definitions.
44aae36d18SPatrick Williamsprefix = "/usr/local"
4502871c91SPatrick Williamsproc_count = nproc().strip()
4602871c91SPatrick Williams
47ee3c9eebSPatrick Williams
48ee3c9eebSPatrick Williamsclass PackageDef(TypedDict, total=False):
49ee3c9eebSPatrick Williams    """ Package Definition for packages dictionary. """
50ee3c9eebSPatrick Williams
51ee3c9eebSPatrick Williams    # rev [optional]: Revision of package to use.
52ee3c9eebSPatrick Williams    rev: str
53ee3c9eebSPatrick Williams    # url [optional]: lambda function to create URL: (package, rev) -> url.
54ee3c9eebSPatrick Williams    url: Callable[[str, str], str]
55ee3c9eebSPatrick Williams    # depends [optional]: List of package dependencies.
56ee3c9eebSPatrick Williams    depends: Iterable[str]
57ee3c9eebSPatrick Williams    # build_type [required]: Build type used for package.
58ee3c9eebSPatrick Williams    #   Currently supported: autoconf, cmake, custom, make, meson
59ee3c9eebSPatrick Williams    build_type: str
60ee3c9eebSPatrick Williams    # build_steps [optional]: Steps to run for 'custom' build_type.
61ee3c9eebSPatrick Williams    build_steps: Iterable[str]
62ee3c9eebSPatrick Williams    # config_flags [optional]: List of options to pass configuration tool.
63ee3c9eebSPatrick Williams    config_flags: Iterable[str]
64ee3c9eebSPatrick Williams    # config_env [optional]: List of environment variables to set for config.
65ee3c9eebSPatrick Williams    config_env: Iterable[str]
66ee3c9eebSPatrick Williams    # custom_post_dl [optional]: List of steps to run after download, but
67ee3c9eebSPatrick Williams    #   before config / build / install.
68ee3c9eebSPatrick Williams    custom_post_dl: Iterable[str]
696bce2ca1SPatrick Williams    # custom_post_install [optional]: List of steps to run after install.
706bce2ca1SPatrick Williams    custom_post_install: Iterable[str]
71ee3c9eebSPatrick Williams
72ee3c9eebSPatrick Williams    # __tag [private]: Generated Docker tag name for package stage.
73ee3c9eebSPatrick Williams    __tag: str
74ee3c9eebSPatrick Williams    # __package [private]: Package object associated with this package.
75ee3c9eebSPatrick Williams    __package: Any  # Type is Package, but not defined yet.
76ee3c9eebSPatrick Williams
7702871c91SPatrick Williams
787204324cSPatrick Williams# Packages to include in image.
797204324cSPatrick Williamspackages = {
80ee3c9eebSPatrick Williams    "boost": PackageDef(
81ffd6b733SWilliam A. Kennington III        rev="1.76.0",
82ee3c9eebSPatrick Williams        url=(
835f2549eaSPatrick Williams            lambda pkg, rev: f"https://downloads.yoctoproject.org/mirror/sources/{pkg}_{rev.replace('.', '_')}.tar.bz2"
842abc4a48SPatrick Williams        ),
85ee3c9eebSPatrick Williams        build_type="custom",
86ee3c9eebSPatrick Williams        build_steps=[
87aae36d18SPatrick Williams            f"./bootstrap.sh --prefix={prefix} --with-libraries=context,coroutine",
88aae36d18SPatrick Williams            "./b2",
89aae36d18SPatrick Williams            f"./b2 install --prefix={prefix}",
90aae36d18SPatrick Williams        ],
91ee3c9eebSPatrick Williams    ),
92ee3c9eebSPatrick Williams    "USCiLab/cereal": PackageDef(
933f8b5294SPatrick Williams        rev="3e4d1b84cab4891368d2179a61a7ba06a5693e7f",
94ee3c9eebSPatrick Williams        build_type="custom",
95ee3c9eebSPatrick Williams        build_steps=[f"cp -a include/cereal/ {prefix}/include/"],
96ee3c9eebSPatrick Williams    ),
97ee3c9eebSPatrick Williams    "catchorg/Catch2": PackageDef(
9896b43455SWilliam A. Kennington III        rev="v2.13.6",
99ee3c9eebSPatrick Williams        build_type="cmake",
100ee3c9eebSPatrick Williams        config_flags=["-DBUILD_TESTING=OFF", "-DCATCH_INSTALL_DOCS=OFF"],
101ee3c9eebSPatrick Williams    ),
102ee3c9eebSPatrick Williams    "CLIUtils/CLI11": PackageDef(
103ee3c9eebSPatrick Williams        rev="v1.9.1",
104ee3c9eebSPatrick Williams        build_type="cmake",
105ee3c9eebSPatrick Williams        config_flags=[
106aae36d18SPatrick Williams            "-DBUILD_TESTING=OFF",
107aae36d18SPatrick Williams            "-DCLI11_BUILD_DOCS=OFF",
108aae36d18SPatrick Williams            "-DCLI11_BUILD_EXAMPLES=OFF",
109aae36d18SPatrick Williams        ],
110ee3c9eebSPatrick Williams    ),
111ee3c9eebSPatrick Williams    "fmtlib/fmt": PackageDef(
112ee3c9eebSPatrick Williams        rev="7.1.3",
113ee3c9eebSPatrick Williams        build_type="cmake",
114ee3c9eebSPatrick Williams        config_flags=[
115aae36d18SPatrick Williams            "-DFMT_DOC=OFF",
116aae36d18SPatrick Williams            "-DFMT_TEST=OFF",
117aae36d18SPatrick Williams        ],
118ee3c9eebSPatrick Williams    ),
119ee3c9eebSPatrick Williams    "Naios/function2": PackageDef(
12096b43455SWilliam A. Kennington III        rev="4.1.0",
121ee3c9eebSPatrick Williams        build_type="custom",
122ee3c9eebSPatrick Williams        build_steps=[
123aae36d18SPatrick Williams            f"mkdir {prefix}/include/function2",
124aae36d18SPatrick Williams            f"cp include/function2/function2.hpp {prefix}/include/function2/",
125aae36d18SPatrick Williams        ],
126ee3c9eebSPatrick Williams    ),
1274dd32c02SWilliam A. Kennington III    # Snapshot from 2021-05-13
128ee3c9eebSPatrick Williams    "google/googletest": PackageDef(
1294dd32c02SWilliam A. Kennington III        rev="662fe38e44900c007eccb65a5d2ea19df7bd520e",
130ee3c9eebSPatrick Williams        build_type="cmake",
1314dd32c02SWilliam A. Kennington III        config_env=["CXXFLAGS=-std=c++20"],
132ee3c9eebSPatrick Williams        config_flags=["-DTHREADS_PREFER_PTHREAD_FLAG=ON"],
133ee3c9eebSPatrick Williams    ),
13402871c91SPatrick Williams    # Release 2020-08-06
135ee3c9eebSPatrick Williams    "nlohmann/json": PackageDef(
136ee3c9eebSPatrick Williams        rev="v3.9.1",
1376bce2ca1SPatrick Williams        build_type="cmake",
1386bce2ca1SPatrick Williams        config_flags=["-DJSON_BuildTests=OFF"],
1396bce2ca1SPatrick Williams        custom_post_install=[
140aae36d18SPatrick Williams            f"ln -s {prefix}/include/nlohmann/json.hpp {prefix}/include/json.hpp",
141aae36d18SPatrick Williams        ],
142ee3c9eebSPatrick Williams    ),
14302871c91SPatrick Williams    # Snapshot from 2019-05-24
144ee3c9eebSPatrick Williams    "linux-test-project/lcov": PackageDef(
145ee3c9eebSPatrick Williams        rev="v1.15",
146ee3c9eebSPatrick Williams        build_type="make",
147ee3c9eebSPatrick Williams    ),
1480eedeedaSPatrick Williams    # dev-5.8 2021-01-11
149ee3c9eebSPatrick Williams    "openbmc/linux": PackageDef(
150ee3c9eebSPatrick Williams        rev="3cc95ae40716e56f81b69615781f54c78079042d",
151ee3c9eebSPatrick Williams        build_type="custom",
152ee3c9eebSPatrick Williams        build_steps=[
153aae36d18SPatrick Williams            f"make -j{proc_count} defconfig",
154aae36d18SPatrick Williams            f"make INSTALL_HDR_PATH={prefix} headers_install",
155aae36d18SPatrick Williams        ],
156ee3c9eebSPatrick Williams    ),
1570eedeedaSPatrick Williams    # Snapshot from 2020-06-13
158ee3c9eebSPatrick Williams    "LibVNC/libvncserver": PackageDef(
159ee3c9eebSPatrick Williams        rev="LibVNCServer-0.9.13",
160ee3c9eebSPatrick Williams        build_type="cmake",
161ee3c9eebSPatrick Williams    ),
162ee3c9eebSPatrick Williams    "martinmoene/span-lite": PackageDef(
16396b43455SWilliam A. Kennington III        rev="v0.9.2",
164ee3c9eebSPatrick Williams        build_type="cmake",
165ee3c9eebSPatrick Williams        config_flags=[
166aae36d18SPatrick Williams            "-DSPAN_LITE_OPT_BUILD_TESTS=OFF",
167aae36d18SPatrick Williams        ],
168ee3c9eebSPatrick Williams    ),
1690eedeedaSPatrick Williams    # version from meta-openembedded/meta-oe/recipes-support/libtinyxml2/libtinyxml2_8.0.0.bb
170ee3c9eebSPatrick Williams    "leethomason/tinyxml2": PackageDef(
171ee3c9eebSPatrick Williams        rev="8.0.0",
172ee3c9eebSPatrick Williams        build_type="cmake",
173ee3c9eebSPatrick Williams    ),
17402871c91SPatrick Williams    # version from /meta-openembedded/meta-oe/recipes-devtools/boost-url/boost-url_git.bb
175ee3c9eebSPatrick Williams    "CPPAlliance/url": PackageDef(
176ee3c9eebSPatrick Williams        rev="a56ae0df6d3078319755fbaa67822b4fa7fd352b",
177ee3c9eebSPatrick Williams        build_type="cmake",
178ee3c9eebSPatrick Williams        config_flags=[
179aae36d18SPatrick Williams            "-DBOOST_URL_BUILD_EXAMPLES=OFF",
180aae36d18SPatrick Williams            "-DBOOST_URL_BUILD_TESTS=OFF",
181aae36d18SPatrick Williams            "-DBOOST_URL_STANDALONE=ON",
182aae36d18SPatrick Williams        ],
183ee3c9eebSPatrick Williams    ),
184f84f7965SAndrew Jeffery    # valijson v0.4, which contains the nlohmann/json.hpp include fix:
185f84f7965SAndrew Jeffery    # 66262bafb82c ("Include nlohmann/json.hpp instead of json.hpp")
186ee3c9eebSPatrick Williams    "tristanpenman/valijson": PackageDef(
187f84f7965SAndrew Jeffery        rev="v0.4",
188ee3c9eebSPatrick Williams        build_type="cmake",
189ee3c9eebSPatrick Williams        config_flags=[
1900eedeedaSPatrick Williams            "-Dvalijson_BUILD_TESTS=0",
1910eedeedaSPatrick Williams            "-Dvalijson_INSTALL_HEADERS=1",
192aae36d18SPatrick Williams        ],
193ee3c9eebSPatrick Williams    ),
19402871c91SPatrick Williams    # version from meta-openembedded/meta-oe/recipes-devtools/nlohmann-fifo/nlohmann-fifo_git.bb
195ee3c9eebSPatrick Williams    "nlohmann/fifo_map": PackageDef(
196ee3c9eebSPatrick Williams        rev="0dfbf5dacbb15a32c43f912a7e66a54aae39d0f9",
197ee3c9eebSPatrick Williams        build_type="custom",
198ee3c9eebSPatrick Williams        build_steps=[f"cp src/fifo_map.hpp {prefix}/include/"],
199ee3c9eebSPatrick Williams    ),
200ee3c9eebSPatrick Williams    "open-power/pdbg": PackageDef(build_type="autoconf"),
201ee3c9eebSPatrick Williams    "openbmc/gpioplus": PackageDef(
202ee3c9eebSPatrick Williams        depends=["openbmc/stdplus"],
203ee3c9eebSPatrick Williams        build_type="meson",
204ee3c9eebSPatrick Williams        config_flags=[
205aae36d18SPatrick Williams            "-Dexamples=false",
206aae36d18SPatrick Williams            "-Dtests=disabled",
207aae36d18SPatrick Williams        ],
208ee3c9eebSPatrick Williams    ),
209ee3c9eebSPatrick Williams    "openbmc/phosphor-dbus-interfaces": PackageDef(
210ee3c9eebSPatrick Williams        depends=["openbmc/sdbusplus"],
211ee3c9eebSPatrick Williams        build_type="meson",
212ee3c9eebSPatrick Williams        config_flags=[
213aae36d18SPatrick Williams            "-Ddata_com_ibm=true",
214aae36d18SPatrick Williams            "-Ddata_org_open_power=true",
215aae36d18SPatrick Williams        ],
216ee3c9eebSPatrick Williams    ),
217ee3c9eebSPatrick Williams    "openbmc/phosphor-logging": PackageDef(
218ee3c9eebSPatrick Williams        depends=[
21983394610SPatrick Williams            "USCiLab/cereal",
22083394610SPatrick Williams            "nlohmann/fifo_map",
22183394610SPatrick Williams            "openbmc/phosphor-dbus-interfaces",
22283394610SPatrick Williams            "openbmc/sdbusplus",
22383394610SPatrick Williams            "openbmc/sdeventplus",
224aae36d18SPatrick Williams        ],
225f79ce4c4SPatrick Williams        build_type="meson",
226ee3c9eebSPatrick Williams        config_flags=[
227f79ce4c4SPatrick Williams            f"-Dyaml_dir={prefix}/share/phosphor-dbus-yaml/yaml",
228aae36d18SPatrick Williams        ],
229ee3c9eebSPatrick Williams    ),
230ee3c9eebSPatrick Williams    "openbmc/phosphor-objmgr": PackageDef(
231ee3c9eebSPatrick Williams        depends=[
23283394610SPatrick Williams            "boost",
23383394610SPatrick Williams            "leethomason/tinyxml2",
23483394610SPatrick Williams            "openbmc/phosphor-logging",
23583394610SPatrick Williams            "openbmc/sdbusplus",
236aae36d18SPatrick Williams        ],
2371197e359SBrad Bishop        build_type="meson",
2381197e359SBrad Bishop        config_flags=[
2391197e359SBrad Bishop            "-Dtests=disabled",
2401197e359SBrad Bishop        ],
241ee3c9eebSPatrick Williams    ),
242ee3c9eebSPatrick Williams    "openbmc/pldm": PackageDef(
243ee3c9eebSPatrick Williams        depends=[
24483394610SPatrick Williams            "CLIUtils/CLI11",
24583394610SPatrick Williams            "boost",
24683394610SPatrick Williams            "nlohmann/json",
24783394610SPatrick Williams            "openbmc/phosphor-dbus-interfaces",
24883394610SPatrick Williams            "openbmc/phosphor-logging",
24983394610SPatrick Williams            "openbmc/sdbusplus",
25083394610SPatrick Williams            "openbmc/sdeventplus",
251aae36d18SPatrick Williams        ],
252ee3c9eebSPatrick Williams        build_type="meson",
253ee3c9eebSPatrick Williams        config_flags=[
254aae36d18SPatrick Williams            "-Dlibpldm-only=enabled",
255aae36d18SPatrick Williams            "-Doem-ibm=enabled",
256aae36d18SPatrick Williams            "-Dtests=disabled",
257aae36d18SPatrick Williams        ],
258ee3c9eebSPatrick Williams    ),
259ee3c9eebSPatrick Williams    "openbmc/sdbusplus": PackageDef(
260ee3c9eebSPatrick Williams        build_type="meson",
261ee3c9eebSPatrick Williams        custom_post_dl=[
262aae36d18SPatrick Williams            "cd tools",
263aae36d18SPatrick Williams            f"./setup.py install --root=/ --prefix={prefix}",
264aae36d18SPatrick Williams            "cd ..",
265aae36d18SPatrick Williams        ],
266ee3c9eebSPatrick Williams        config_flags=[
267aae36d18SPatrick Williams            "-Dexamples=disabled",
268aae36d18SPatrick Williams            "-Dtests=disabled",
269aae36d18SPatrick Williams        ],
270b16f3e20SPatrick Williams    ),
271ee3c9eebSPatrick Williams    "openbmc/sdeventplus": PackageDef(
272ee3c9eebSPatrick Williams        depends=["Naios/function2", "openbmc/stdplus"],
273ee3c9eebSPatrick Williams        build_type="meson",
274ee3c9eebSPatrick Williams        config_flags=[
275ee3c9eebSPatrick Williams            "-Dexamples=false",
276ee3c9eebSPatrick Williams            "-Dtests=disabled",
277ee3c9eebSPatrick Williams        ],
278ee3c9eebSPatrick Williams    ),
279ee3c9eebSPatrick Williams    "openbmc/stdplus": PackageDef(
280ee3c9eebSPatrick Williams        depends=["fmtlib/fmt", "martinmoene/span-lite"],
281ee3c9eebSPatrick Williams        build_type="meson",
282ee3c9eebSPatrick Williams        config_flags=[
283ee3c9eebSPatrick Williams            "-Dexamples=false",
284ee3c9eebSPatrick Williams            "-Dtests=disabled",
285ee3c9eebSPatrick Williams        ],
286ee3c9eebSPatrick Williams    ),
287ee3c9eebSPatrick Williams}  # type: Dict[str, PackageDef]
28802871c91SPatrick Williams
28902871c91SPatrick Williams# Define common flags used for builds
29002871c91SPatrick Williamsconfigure_flags = " ".join(
29102871c91SPatrick Williams    [
29202871c91SPatrick Williams        f"--prefix={prefix}",
29302871c91SPatrick Williams    ]
29402871c91SPatrick Williams)
29502871c91SPatrick Williamscmake_flags = " ".join(
29602871c91SPatrick Williams    [
29702871c91SPatrick Williams        "-DBUILD_SHARED_LIBS=ON",
2980f2086b3SPatrick Williams        "-DCMAKE_BUILD_TYPE=RelWithDebInfo",
29902871c91SPatrick Williams        f"-DCMAKE_INSTALL_PREFIX:PATH={prefix}",
3000f2086b3SPatrick Williams        "-GNinja",
3010f2086b3SPatrick Williams        "-DCMAKE_MAKE_PROGRAM=ninja",
30202871c91SPatrick Williams    ]
30302871c91SPatrick Williams)
30402871c91SPatrick Williamsmeson_flags = " ".join(
30502871c91SPatrick Williams    [
30602871c91SPatrick Williams        "--wrap-mode=nodownload",
30702871c91SPatrick Williams        f"-Dprefix={prefix}",
30802871c91SPatrick Williams    ]
30902871c91SPatrick Williams)
31002871c91SPatrick Williams
311ee3c9eebSPatrick Williams
312ee3c9eebSPatrick Williamsclass Package(threading.Thread):
313ee3c9eebSPatrick Williams    """Class used to build the Docker stages for each package.
314ee3c9eebSPatrick Williams
315ee3c9eebSPatrick Williams    Generally, this class should not be instantiated directly but through
316ee3c9eebSPatrick Williams    Package.generate_all().
317ee3c9eebSPatrick Williams    """
318ee3c9eebSPatrick Williams
319ee3c9eebSPatrick Williams    # Copy the packages dictionary.
320ee3c9eebSPatrick Williams    packages = packages.copy()
321ee3c9eebSPatrick Williams
322ee3c9eebSPatrick Williams    # Lock used for thread-safety.
323ee3c9eebSPatrick Williams    lock = threading.Lock()
324ee3c9eebSPatrick Williams
325ee3c9eebSPatrick Williams    def __init__(self, pkg: str):
326ee3c9eebSPatrick Williams        """ pkg - The name of this package (ex. foo/bar ) """
327ee3c9eebSPatrick Williams        super(Package, self).__init__()
328ee3c9eebSPatrick Williams
329ee3c9eebSPatrick Williams        self.package = pkg
330ee3c9eebSPatrick Williams        self.exception = None  # type: Optional[Exception]
331ee3c9eebSPatrick Williams
332ee3c9eebSPatrick Williams        # Reference to this package's
333ee3c9eebSPatrick Williams        self.pkg_def = Package.packages[pkg]
334ee3c9eebSPatrick Williams        self.pkg_def["__package"] = self
335ee3c9eebSPatrick Williams
336ee3c9eebSPatrick Williams    def run(self) -> None:
337ee3c9eebSPatrick Williams        """ Thread 'run' function.  Builds the Docker stage. """
338ee3c9eebSPatrick Williams
339ee3c9eebSPatrick Williams        # In case this package has no rev, fetch it from Github.
340ee3c9eebSPatrick Williams        self._update_rev()
341ee3c9eebSPatrick Williams
342ee3c9eebSPatrick Williams        # Find all the Package objects that this package depends on.
343ee3c9eebSPatrick Williams        #   This section is locked because we are looking into another
344ee3c9eebSPatrick Williams        #   package's PackageDef dict, which could be being modified.
345ee3c9eebSPatrick Williams        Package.lock.acquire()
346ee3c9eebSPatrick Williams        deps: Iterable[Package] = [
347ee3c9eebSPatrick Williams            Package.packages[deppkg]["__package"]
348ee3c9eebSPatrick Williams            for deppkg in self.pkg_def.get("depends", [])
349ee3c9eebSPatrick Williams        ]
350ee3c9eebSPatrick Williams        Package.lock.release()
351ee3c9eebSPatrick Williams
352ee3c9eebSPatrick Williams        # Wait until all the depends finish building.  We need them complete
353ee3c9eebSPatrick Williams        # for the "COPY" commands.
354ee3c9eebSPatrick Williams        for deppkg in deps:
355ee3c9eebSPatrick Williams            deppkg.join()
356ee3c9eebSPatrick Williams
357ee3c9eebSPatrick Williams        # Generate this package's Dockerfile.
358ee3c9eebSPatrick Williams        dockerfile = f"""
359ee3c9eebSPatrick WilliamsFROM {docker_base_img_name}
360ee3c9eebSPatrick Williams{self._df_copycmds()}
361ee3c9eebSPatrick Williams{self._df_build()}
362ee3c9eebSPatrick Williams"""
363ee3c9eebSPatrick Williams
364ee3c9eebSPatrick Williams        # Generate the resulting tag name and save it to the PackageDef.
365ee3c9eebSPatrick Williams        #   This section is locked because we are modifying the PackageDef,
366ee3c9eebSPatrick Williams        #   which can be accessed by other threads.
367ee3c9eebSPatrick Williams        Package.lock.acquire()
368ee3c9eebSPatrick Williams        tag = Docker.tagname(self._stagename(), dockerfile)
369ee3c9eebSPatrick Williams        self.pkg_def["__tag"] = tag
370ee3c9eebSPatrick Williams        Package.lock.release()
371ee3c9eebSPatrick Williams
372ee3c9eebSPatrick Williams        # Do the build / save any exceptions.
373ee3c9eebSPatrick Williams        try:
374ee3c9eebSPatrick Williams            Docker.build(self.package, tag, dockerfile)
375ee3c9eebSPatrick Williams        except Exception as e:
376ee3c9eebSPatrick Williams            self.exception = e
377ee3c9eebSPatrick Williams
378ee3c9eebSPatrick Williams    @classmethod
379ee3c9eebSPatrick Williams    def generate_all(cls) -> None:
380ee3c9eebSPatrick Williams        """Ensure a Docker stage is created for all defined packages.
381ee3c9eebSPatrick Williams
382ee3c9eebSPatrick Williams        These are done in parallel but with appropriate blocking per
383ee3c9eebSPatrick Williams        package 'depends' specifications.
384ee3c9eebSPatrick Williams        """
385ee3c9eebSPatrick Williams
386ee3c9eebSPatrick Williams        # Create a Package for each defined package.
387ee3c9eebSPatrick Williams        pkg_threads = [Package(p) for p in cls.packages.keys()]
388ee3c9eebSPatrick Williams
389ee3c9eebSPatrick Williams        # Start building them all.
3906dbd7807SPatrick Williams        #   This section is locked because threads depend on each other,
3916dbd7807SPatrick Williams        #   based on the packages, and they cannot 'join' on a thread
3926dbd7807SPatrick Williams        #   which is not yet started.  Adding a lock here allows all the
3936dbd7807SPatrick Williams        #   threads to start before they 'join' their dependencies.
3946dbd7807SPatrick Williams        Package.lock.acquire()
395ee3c9eebSPatrick Williams        for t in pkg_threads:
396ee3c9eebSPatrick Williams            t.start()
3976dbd7807SPatrick Williams        Package.lock.release()
398ee3c9eebSPatrick Williams
399ee3c9eebSPatrick Williams        # Wait for completion.
400ee3c9eebSPatrick Williams        for t in pkg_threads:
401ee3c9eebSPatrick Williams            t.join()
402ee3c9eebSPatrick Williams            # Check if the thread saved off its own exception.
403ee3c9eebSPatrick Williams            if t.exception:
404ee3c9eebSPatrick Williams                print(f"Package {t.package} failed!", file=sys.stderr)
405ee3c9eebSPatrick Williams                raise t.exception
406ee3c9eebSPatrick Williams
407ee3c9eebSPatrick Williams    @staticmethod
408ee3c9eebSPatrick Williams    def df_all_copycmds() -> str:
409ee3c9eebSPatrick Williams        """Formulate the Dockerfile snippet necessary to copy all packages
410ee3c9eebSPatrick Williams        into the final image.
411ee3c9eebSPatrick Williams        """
412ee3c9eebSPatrick Williams        return Package.df_copycmds_set(Package.packages.keys())
413ee3c9eebSPatrick Williams
414ee3c9eebSPatrick Williams    @classmethod
415ee3c9eebSPatrick Williams    def depcache(cls) -> str:
416ee3c9eebSPatrick Williams        """Create the contents of the '/tmp/depcache'.
417ee3c9eebSPatrick Williams        This file is a comma-separated list of "<pkg>:<rev>".
418ee3c9eebSPatrick Williams        """
419ee3c9eebSPatrick Williams
420ee3c9eebSPatrick Williams        # This needs to be sorted for consistency.
421ee3c9eebSPatrick Williams        depcache = ""
422ee3c9eebSPatrick Williams        for pkg in sorted(cls.packages.keys()):
423ee3c9eebSPatrick Williams            depcache += "%s:%s," % (pkg, cls.packages[pkg]["rev"])
424ee3c9eebSPatrick Williams        return depcache
425ee3c9eebSPatrick Williams
426ee3c9eebSPatrick Williams    def _update_rev(self) -> None:
427ee3c9eebSPatrick Williams        """ Look up the HEAD for missing a static rev. """
428ee3c9eebSPatrick Williams
429ee3c9eebSPatrick Williams        if "rev" in self.pkg_def:
430ee3c9eebSPatrick Williams            return
431ee3c9eebSPatrick Williams
43265b21fb9SPatrick Williams        # Check if Jenkins/Gerrit gave us a revision and use it.
43365b21fb9SPatrick Williams        if gerrit_project == self.package and gerrit_rev:
43465b21fb9SPatrick Williams            print(
43565b21fb9SPatrick Williams                f"Found Gerrit revision for {self.package}: {gerrit_rev}",
43665b21fb9SPatrick Williams                file=sys.stderr,
43765b21fb9SPatrick Williams            )
43865b21fb9SPatrick Williams            self.pkg_def["rev"] = gerrit_rev
43965b21fb9SPatrick Williams            return
44065b21fb9SPatrick Williams
441ee3c9eebSPatrick Williams        # Ask Github for all the branches.
442ee3c9eebSPatrick Williams        lookup = git("ls-remote", "--heads", f"https://github.com/{self.package}")
443ee3c9eebSPatrick Williams
444ee3c9eebSPatrick Williams        # Find the branch matching {branch} (or fallback to master).
445ee3c9eebSPatrick Williams        #   This section is locked because we are modifying the PackageDef.
446ee3c9eebSPatrick Williams        Package.lock.acquire()
447ee3c9eebSPatrick Williams        for line in lookup.split("\n"):
448ee3c9eebSPatrick Williams            if f"refs/heads/{branch}" in line:
449ee3c9eebSPatrick Williams                self.pkg_def["rev"] = line.split()[0]
450ee3c9eebSPatrick Williams            elif f"refs/heads/master" in line and "rev" not in self.pkg_def:
451ee3c9eebSPatrick Williams                self.pkg_def["rev"] = line.split()[0]
452ee3c9eebSPatrick Williams        Package.lock.release()
453ee3c9eebSPatrick Williams
454ee3c9eebSPatrick Williams    def _stagename(self) -> str:
455ee3c9eebSPatrick Williams        """ Create a name for the Docker stage associated with this pkg. """
456ee3c9eebSPatrick Williams        return self.package.replace("/", "-").lower()
457ee3c9eebSPatrick Williams
458ee3c9eebSPatrick Williams    def _url(self) -> str:
459ee3c9eebSPatrick Williams        """ Get the URL for this package. """
460ee3c9eebSPatrick Williams        rev = self.pkg_def["rev"]
461ee3c9eebSPatrick Williams
462ee3c9eebSPatrick Williams        # If the lambda exists, call it.
463ee3c9eebSPatrick Williams        if "url" in self.pkg_def:
464ee3c9eebSPatrick Williams            return self.pkg_def["url"](self.package, rev)
465ee3c9eebSPatrick Williams
466ee3c9eebSPatrick Williams        # Default to the github archive URL.
467ee3c9eebSPatrick Williams        return f"https://github.com/{self.package}/archive/{rev}.tar.gz"
468ee3c9eebSPatrick Williams
469ee3c9eebSPatrick Williams    def _cmd_download(self) -> str:
470ee3c9eebSPatrick Williams        """Formulate the command necessary to download and unpack to source."""
471ee3c9eebSPatrick Williams
472ee3c9eebSPatrick Williams        url = self._url()
473ee3c9eebSPatrick Williams        if ".tar." not in url:
474ee3c9eebSPatrick Williams            raise NotImplementedError(
475ee3c9eebSPatrick Williams                f"Unhandled download type for {self.package}: {url}"
476ee3c9eebSPatrick Williams            )
477ee3c9eebSPatrick Williams
478ee3c9eebSPatrick Williams        cmd = f"curl -L {url} | tar -x"
479ee3c9eebSPatrick Williams
480ee3c9eebSPatrick Williams        if url.endswith(".bz2"):
481ee3c9eebSPatrick Williams            cmd += "j"
482ee3c9eebSPatrick Williams        elif url.endswith(".gz"):
483ee3c9eebSPatrick Williams            cmd += "z"
484ee3c9eebSPatrick Williams        else:
485ee3c9eebSPatrick Williams            raise NotImplementedError(
486ee3c9eebSPatrick Williams                f"Unknown tar flags needed for {self.package}: {url}"
487ee3c9eebSPatrick Williams            )
488ee3c9eebSPatrick Williams
489ee3c9eebSPatrick Williams        return cmd
490ee3c9eebSPatrick Williams
491ee3c9eebSPatrick Williams    def _cmd_cd_srcdir(self) -> str:
492ee3c9eebSPatrick Williams        """ Formulate the command necessary to 'cd' into the source dir. """
493ee3c9eebSPatrick Williams        return f"cd {self.package.split('/')[-1]}*"
494ee3c9eebSPatrick Williams
495ee3c9eebSPatrick Williams    def _df_copycmds(self) -> str:
496ee3c9eebSPatrick Williams        """ Formulate the dockerfile snippet necessary to COPY all depends. """
497ee3c9eebSPatrick Williams
498ee3c9eebSPatrick Williams        if "depends" not in self.pkg_def:
499ee3c9eebSPatrick Williams            return ""
500ee3c9eebSPatrick Williams        return Package.df_copycmds_set(self.pkg_def["depends"])
501ee3c9eebSPatrick Williams
502ee3c9eebSPatrick Williams    @staticmethod
503ee3c9eebSPatrick Williams    def df_copycmds_set(pkgs: Iterable[str]) -> str:
504ee3c9eebSPatrick Williams        """Formulate the Dockerfile snippet necessary to COPY a set of
505ee3c9eebSPatrick Williams        packages into a Docker stage.
506ee3c9eebSPatrick Williams        """
507ee3c9eebSPatrick Williams
508ee3c9eebSPatrick Williams        copy_cmds = ""
509ee3c9eebSPatrick Williams
510ee3c9eebSPatrick Williams        # Sort the packages for consistency.
511ee3c9eebSPatrick Williams        for p in sorted(pkgs):
512ee3c9eebSPatrick Williams            tag = Package.packages[p]["__tag"]
513ee3c9eebSPatrick Williams            copy_cmds += f"COPY --from={tag} {prefix} {prefix}\n"
514ee3c9eebSPatrick Williams            # Workaround for upstream docker bug and multiple COPY cmds
515ee3c9eebSPatrick Williams            # https://github.com/moby/moby/issues/37965
516ee3c9eebSPatrick Williams            copy_cmds += "RUN true\n"
517ee3c9eebSPatrick Williams
518ee3c9eebSPatrick Williams        return copy_cmds
519ee3c9eebSPatrick Williams
520ee3c9eebSPatrick Williams    def _df_build(self) -> str:
521ee3c9eebSPatrick Williams        """Formulate the Dockerfile snippet necessary to download, build, and
522ee3c9eebSPatrick Williams        install a package into a Docker stage.
523ee3c9eebSPatrick Williams        """
524ee3c9eebSPatrick Williams
525ee3c9eebSPatrick Williams        # Download and extract source.
526ee3c9eebSPatrick Williams        result = f"RUN {self._cmd_download()} && {self._cmd_cd_srcdir()} && "
527ee3c9eebSPatrick Williams
528ee3c9eebSPatrick Williams        # Handle 'custom_post_dl' commands.
529ee3c9eebSPatrick Williams        custom_post_dl = self.pkg_def.get("custom_post_dl")
530ee3c9eebSPatrick Williams        if custom_post_dl:
531ee3c9eebSPatrick Williams            result += " && ".join(custom_post_dl) + " && "
532ee3c9eebSPatrick Williams
533ee3c9eebSPatrick Williams        # Build and install package based on 'build_type'.
534ee3c9eebSPatrick Williams        build_type = self.pkg_def["build_type"]
535ee3c9eebSPatrick Williams        if build_type == "autoconf":
536ee3c9eebSPatrick Williams            result += self._cmd_build_autoconf()
537ee3c9eebSPatrick Williams        elif build_type == "cmake":
538ee3c9eebSPatrick Williams            result += self._cmd_build_cmake()
539ee3c9eebSPatrick Williams        elif build_type == "custom":
540ee3c9eebSPatrick Williams            result += self._cmd_build_custom()
541ee3c9eebSPatrick Williams        elif build_type == "make":
542ee3c9eebSPatrick Williams            result += self._cmd_build_make()
543ee3c9eebSPatrick Williams        elif build_type == "meson":
544ee3c9eebSPatrick Williams            result += self._cmd_build_meson()
545ee3c9eebSPatrick Williams        else:
546ee3c9eebSPatrick Williams            raise NotImplementedError(
547ee3c9eebSPatrick Williams                f"Unhandled build type for {self.package}: {build_type}"
548ee3c9eebSPatrick Williams            )
549ee3c9eebSPatrick Williams
5506bce2ca1SPatrick Williams        # Handle 'custom_post_install' commands.
5516bce2ca1SPatrick Williams        custom_post_install = self.pkg_def.get("custom_post_install")
5526bce2ca1SPatrick Williams        if custom_post_install:
5536bce2ca1SPatrick Williams            result += " && " + " && ".join(custom_post_install)
5546bce2ca1SPatrick Williams
555ee3c9eebSPatrick Williams        return result
556ee3c9eebSPatrick Williams
557ee3c9eebSPatrick Williams    def _cmd_build_autoconf(self) -> str:
558ee3c9eebSPatrick Williams        options = " ".join(self.pkg_def.get("config_flags", []))
559ee3c9eebSPatrick Williams        env = " ".join(self.pkg_def.get("config_env", []))
560ee3c9eebSPatrick Williams        result = "./bootstrap.sh && "
561ee3c9eebSPatrick Williams        result += f"{env} ./configure {configure_flags} {options} && "
562ee3c9eebSPatrick Williams        result += f"make -j{proc_count} && make install"
563ee3c9eebSPatrick Williams        return result
564ee3c9eebSPatrick Williams
565ee3c9eebSPatrick Williams    def _cmd_build_cmake(self) -> str:
566ee3c9eebSPatrick Williams        options = " ".join(self.pkg_def.get("config_flags", []))
567ee3c9eebSPatrick Williams        env = " ".join(self.pkg_def.get("config_env", []))
568ee3c9eebSPatrick Williams        result = "mkdir builddir && cd builddir && "
569ee3c9eebSPatrick Williams        result += f"{env} cmake {cmake_flags} {options} .. && "
570ee3c9eebSPatrick Williams        result += "cmake --build . --target all && "
571ee3c9eebSPatrick Williams        result += "cmake --build . --target install && "
572ee3c9eebSPatrick Williams        result += "cd .."
573ee3c9eebSPatrick Williams        return result
574ee3c9eebSPatrick Williams
575ee3c9eebSPatrick Williams    def _cmd_build_custom(self) -> str:
576ee3c9eebSPatrick Williams        return " && ".join(self.pkg_def.get("build_steps", []))
577ee3c9eebSPatrick Williams
578ee3c9eebSPatrick Williams    def _cmd_build_make(self) -> str:
579ee3c9eebSPatrick Williams        return f"make -j{proc_count} && make install"
580ee3c9eebSPatrick Williams
581ee3c9eebSPatrick Williams    def _cmd_build_meson(self) -> str:
582ee3c9eebSPatrick Williams        options = " ".join(self.pkg_def.get("config_flags", []))
583ee3c9eebSPatrick Williams        env = " ".join(self.pkg_def.get("config_env", []))
584ee3c9eebSPatrick Williams        result = f"{env} meson builddir {meson_flags} {options} && "
585ee3c9eebSPatrick Williams        result += "ninja -C builddir && ninja -C builddir install"
586ee3c9eebSPatrick Williams        return result
587ee3c9eebSPatrick Williams
588ee3c9eebSPatrick Williams
589ee3c9eebSPatrick Williamsclass Docker:
590ee3c9eebSPatrick Williams    """Class to assist with Docker interactions.  All methods are static."""
591ee3c9eebSPatrick Williams
592ee3c9eebSPatrick Williams    @staticmethod
593ee3c9eebSPatrick Williams    def timestamp() -> str:
594ee3c9eebSPatrick Williams        """ Generate a timestamp for today using the ISO week. """
595ee3c9eebSPatrick Williams        today = date.today().isocalendar()
596ee3c9eebSPatrick Williams        return f"{today[0]}-W{today[1]:02}"
597ee3c9eebSPatrick Williams
598ee3c9eebSPatrick Williams    @staticmethod
599ee3c9eebSPatrick Williams    def tagname(pkgname: str, dockerfile: str) -> str:
600ee3c9eebSPatrick Williams        """ Generate a tag name for a package using a hash of the Dockerfile. """
601ee3c9eebSPatrick Williams        result = docker_image_name
602ee3c9eebSPatrick Williams        if pkgname:
603ee3c9eebSPatrick Williams            result += "-" + pkgname
604ee3c9eebSPatrick Williams
605ee3c9eebSPatrick Williams        result += ":" + Docker.timestamp()
606ee3c9eebSPatrick Williams        result += "-" + sha256(dockerfile.encode()).hexdigest()[0:16]
607ee3c9eebSPatrick Williams
608ee3c9eebSPatrick Williams        return result
609ee3c9eebSPatrick Williams
610ee3c9eebSPatrick Williams    @staticmethod
611ee3c9eebSPatrick Williams    def build(pkg: str, tag: str, dockerfile: str) -> None:
612ee3c9eebSPatrick Williams        """Build a docker image using the Dockerfile and tagging it with 'tag'."""
613ee3c9eebSPatrick Williams
614ee3c9eebSPatrick Williams        # If we're not forcing builds, check if it already exists and skip.
615ee3c9eebSPatrick Williams        if not force_build:
616ee3c9eebSPatrick Williams            if docker.image.ls(tag, "--format", '"{{.Repository}}:{{.Tag}}"'):
617ee3c9eebSPatrick Williams                print(f"Image {tag} already exists.  Skipping.", file=sys.stderr)
618ee3c9eebSPatrick Williams                return
619ee3c9eebSPatrick Williams
620ee3c9eebSPatrick Williams        # Build it.
621ee3c9eebSPatrick Williams        #   Capture the output of the 'docker build' command and send it to
622ee3c9eebSPatrick Williams        #   stderr (prefixed with the package name).  This allows us to see
623ee3c9eebSPatrick Williams        #   progress but not polute stdout.  Later on we output the final
624ee3c9eebSPatrick Williams        #   docker tag to stdout and we want to keep that pristine.
625ee3c9eebSPatrick Williams        #
626ee3c9eebSPatrick Williams        #   Other unusual flags:
627ee3c9eebSPatrick Williams        #       --no-cache: Bypass the Docker cache if 'force_build'.
628ee3c9eebSPatrick Williams        #       --force-rm: Clean up Docker processes if they fail.
629ee3c9eebSPatrick Williams        docker.build(
630ee3c9eebSPatrick Williams            proxy_args,
631ee3c9eebSPatrick Williams            "--network=host",
632ee3c9eebSPatrick Williams            "--force-rm",
633ee3c9eebSPatrick Williams            "--no-cache=true" if force_build else "--no-cache=false",
634ee3c9eebSPatrick Williams            "-t",
635ee3c9eebSPatrick Williams            tag,
636ee3c9eebSPatrick Williams            "-",
637ee3c9eebSPatrick Williams            _in=dockerfile,
638ee3c9eebSPatrick Williams            _out=(
639ee3c9eebSPatrick Williams                lambda line: print(
640ee3c9eebSPatrick Williams                    pkg + ":", line, end="", file=sys.stderr, flush=True
641ee3c9eebSPatrick Williams                )
642ee3c9eebSPatrick Williams            ),
643ee3c9eebSPatrick Williams        )
644ee3c9eebSPatrick Williams
645ee3c9eebSPatrick Williams
646ee3c9eebSPatrick Williams# Read a bunch of environment variables.
647ee3c9eebSPatrick Williamsdocker_image_name = os.environ.get("DOCKER_IMAGE_NAME", "openbmc/ubuntu-unit-test")
648ee3c9eebSPatrick Williamsforce_build = os.environ.get("FORCE_DOCKER_BUILD")
649ee3c9eebSPatrick Williamsis_automated_ci_build = os.environ.get("BUILD_URL", False)
65073395159SPatrick Williamsdistro = os.environ.get("DISTRO", "ubuntu:hirsute")
651ee3c9eebSPatrick Williamsbranch = os.environ.get("BRANCH", "master")
652ee3c9eebSPatrick Williamsubuntu_mirror = os.environ.get("UBUNTU_MIRROR")
653ee3c9eebSPatrick Williamshttp_proxy = os.environ.get("http_proxy")
654ee3c9eebSPatrick Williams
65565b21fb9SPatrick Williamsgerrit_project = os.environ.get("GERRIT_PROJECT")
65665b21fb9SPatrick Williamsgerrit_rev = os.environ.get("GERRIT_PATCHSET_REVISION")
65765b21fb9SPatrick Williams
658ee3c9eebSPatrick Williams# Set up some common variables.
659ee3c9eebSPatrick Williamsusername = os.environ.get("USER", "root")
660ee3c9eebSPatrick Williamshomedir = os.environ.get("HOME", "/root")
661ee3c9eebSPatrick Williamsgid = os.getgid()
662ee3c9eebSPatrick Williamsuid = os.getuid()
663ee3c9eebSPatrick Williams
664ee3c9eebSPatrick Williams# Determine the architecture for Docker.
665ee3c9eebSPatrick Williamsarch = uname("-m").strip()
666ee3c9eebSPatrick Williamsif arch == "ppc64le":
667ee3c9eebSPatrick Williams    docker_base = "ppc64le/"
668ee3c9eebSPatrick Williamselif arch == "x86_64":
669ee3c9eebSPatrick Williams    docker_base = ""
670ee3c9eebSPatrick Williamselse:
671ee3c9eebSPatrick Williams    print(
672ee3c9eebSPatrick Williams        f"Unsupported system architecture({arch}) found for docker image",
673ee3c9eebSPatrick Williams        file=sys.stderr,
674ee3c9eebSPatrick Williams    )
675ee3c9eebSPatrick Williams    sys.exit(1)
676ee3c9eebSPatrick Williams
67702871c91SPatrick Williams# Special flags if setting up a deb mirror.
67802871c91SPatrick Williamsmirror = ""
67902871c91SPatrick Williamsif "ubuntu" in distro and ubuntu_mirror:
68002871c91SPatrick Williams    mirror = f"""
68102871c91SPatrick WilliamsRUN echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME) main restricted universe multiverse" > /etc/apt/sources.list && \\
68202871c91SPatrick Williams    echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME)-updates main restricted universe multiverse" >> /etc/apt/sources.list && \\
68302871c91SPatrick Williams    echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME)-security main restricted universe multiverse" >> /etc/apt/sources.list && \\
68402871c91SPatrick Williams    echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME)-proposed main restricted universe multiverse" >> /etc/apt/sources.list && \\
68502871c91SPatrick Williams    echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME)-backports main restricted universe multiverse" >> /etc/apt/sources.list
68602871c91SPatrick Williams"""
68702871c91SPatrick Williams
68802871c91SPatrick Williams# Special flags for proxying.
68902871c91SPatrick Williamsproxy_cmd = ""
69034ec77e8SAdrian Ambrożewiczproxy_keyserver = ""
69102871c91SPatrick Williamsproxy_args = []
69202871c91SPatrick Williamsif http_proxy:
69302871c91SPatrick Williams    proxy_cmd = f"""
69402871c91SPatrick WilliamsRUN echo "[http]" >> {homedir}/.gitconfig && \
69502871c91SPatrick Williams    echo "proxy = {http_proxy}" >> {homedir}/.gitconfig
69602871c91SPatrick Williams"""
69734ec77e8SAdrian Ambrożewicz    proxy_keyserver = f"--keyserver-options http-proxy={http_proxy}"
69834ec77e8SAdrian Ambrożewicz
69902871c91SPatrick Williams    proxy_args.extend(
70002871c91SPatrick Williams        [
70102871c91SPatrick Williams            "--build-arg",
70202871c91SPatrick Williams            f"http_proxy={http_proxy}",
70302871c91SPatrick Williams            "--build-arg",
704d461cd6aSLei YU            f"https_proxy={http_proxy}",
70502871c91SPatrick Williams        ]
70602871c91SPatrick Williams    )
70702871c91SPatrick Williams
708ee3c9eebSPatrick Williams# Create base Dockerfile.
709a18d9c57SPatrick Williamsdockerfile_base = f"""
710a18d9c57SPatrick WilliamsFROM {docker_base}{distro}
71102871c91SPatrick Williams
71202871c91SPatrick Williams{mirror}
71302871c91SPatrick Williams
71402871c91SPatrick WilliamsENV DEBIAN_FRONTEND noninteractive
71502871c91SPatrick Williams
71602871c91SPatrick WilliamsENV PYTHONPATH "/usr/local/lib/python3.8/site-packages/"
71702871c91SPatrick Williams
718bb16ac14SPatrick Williams# Sometimes the ubuntu key expires and we need a way to force an execution
719bb16ac14SPatrick Williams# of the apt-get commands for the dbgsym-keyring.  When this happens we see
720bb16ac14SPatrick Williams# an error like: "Release: The following signatures were invalid:"
721bb16ac14SPatrick Williams# Insert a bogus echo that we can change here when we get this error to force
722bb16ac14SPatrick Williams# the update.
723bb16ac14SPatrick WilliamsRUN echo "ubuntu keyserver rev as of 2021-04-21"
724bb16ac14SPatrick Williams
72502871c91SPatrick Williams# We need the keys to be imported for dbgsym repos
72602871c91SPatrick Williams# New releases have a package, older ones fall back to manual fetching
72702871c91SPatrick Williams# https://wiki.ubuntu.com/Debug%20Symbol%20Packages
72850837436SPatrick WilliamsRUN apt-get update && apt-get dist-upgrade -yy && \
729f79ce4c4SPatrick Williams    ( apt-get install gpgv ubuntu-dbgsym-keyring || \
73050837436SPatrick Williams        ( apt-get install -yy dirmngr && \
73150837436SPatrick Williams          apt-key adv --keyserver keyserver.ubuntu.com \
73234ec77e8SAdrian Ambrożewicz                      {proxy_keyserver} \
73350837436SPatrick Williams                      --recv-keys F2EDC64DC5AEE1F6B9C621F0C8CAB6595FDFF622 ) )
73402871c91SPatrick Williams
73502871c91SPatrick Williams# Parse the current repo list into a debug repo list
73602871c91SPatrick WilliamsRUN sed -n '/^deb /s,^deb [^ ]* ,deb http://ddebs.ubuntu.com ,p' /etc/apt/sources.list >/etc/apt/sources.list.d/debug.list
73702871c91SPatrick Williams
73802871c91SPatrick Williams# Remove non-existent debug repos
73902871c91SPatrick WilliamsRUN sed -i '/-\(backports\|security\) /d' /etc/apt/sources.list.d/debug.list
74002871c91SPatrick Williams
74102871c91SPatrick WilliamsRUN cat /etc/apt/sources.list.d/debug.list
74202871c91SPatrick Williams
74302871c91SPatrick WilliamsRUN apt-get update && apt-get dist-upgrade -yy && apt-get install -yy \
744b84d59dcSWilliam A. Kennington III    gcc-11 \
745b84d59dcSWilliam A. Kennington III    g++-11 \
74602871c91SPatrick Williams    libc6-dbg \
74702871c91SPatrick Williams    libc6-dev \
74802871c91SPatrick Williams    libtool \
74902871c91SPatrick Williams    bison \
75002871c91SPatrick Williams    libdbus-1-dev \
75102871c91SPatrick Williams    flex \
75202871c91SPatrick Williams    cmake \
75302871c91SPatrick Williams    python3 \
75402871c91SPatrick Williams    python3-dev\
75502871c91SPatrick Williams    python3-yaml \
75602871c91SPatrick Williams    python3-mako \
75702871c91SPatrick Williams    python3-pip \
75802871c91SPatrick Williams    python3-setuptools \
75902871c91SPatrick Williams    python3-git \
76002871c91SPatrick Williams    python3-socks \
76102871c91SPatrick Williams    pkg-config \
76202871c91SPatrick Williams    autoconf \
76302871c91SPatrick Williams    autoconf-archive \
76402871c91SPatrick Williams    libsystemd-dev \
76502871c91SPatrick Williams    systemd \
76602871c91SPatrick Williams    libssl-dev \
76702871c91SPatrick Williams    libevdev-dev \
76802871c91SPatrick Williams    libjpeg-dev \
76902871c91SPatrick Williams    libpng-dev \
77002871c91SPatrick Williams    ninja-build \
77102871c91SPatrick Williams    sudo \
77202871c91SPatrick Williams    curl \
77302871c91SPatrick Williams    git \
77402871c91SPatrick Williams    dbus \
77502871c91SPatrick Williams    iputils-ping \
7764569bf49SPatrick Williams    clang-12 \
7774569bf49SPatrick Williams    clang-format-12 \
7784569bf49SPatrick Williams    clang-tidy-12 \
7794569bf49SPatrick Williams    clang-tools-12 \
78002871c91SPatrick Williams    shellcheck \
78102871c91SPatrick Williams    npm \
78202871c91SPatrick Williams    iproute2 \
78302871c91SPatrick Williams    libnl-3-dev \
78402871c91SPatrick Williams    libnl-genl-3-dev \
78502871c91SPatrick Williams    libconfig++-dev \
78602871c91SPatrick Williams    libsnmp-dev \
78702871c91SPatrick Williams    valgrind \
78802871c91SPatrick Williams    valgrind-dbg \
78902871c91SPatrick Williams    libpam0g-dev \
79002871c91SPatrick Williams    xxd \
79102871c91SPatrick Williams    libi2c-dev \
79202871c91SPatrick Williams    wget \
79302871c91SPatrick Williams    libldap2-dev \
79402871c91SPatrick Williams    libprotobuf-dev \
795dafe7529SWilliam A. Kennington III    liburing-dev \
796dafe7529SWilliam A. Kennington III    liburing1-dbgsym \
79702871c91SPatrick Williams    libperlio-gzip-perl \
79802871c91SPatrick Williams    libjson-perl \
79902871c91SPatrick Williams    protobuf-compiler \
80002871c91SPatrick Williams    libgpiod-dev \
80102871c91SPatrick Williams    device-tree-compiler \
80202871c91SPatrick Williams    cppcheck \
80302871c91SPatrick Williams    libpciaccess-dev \
80402871c91SPatrick Williams    libmimetic-dev \
80502871c91SPatrick Williams    libxml2-utils \
8060eedeedaSPatrick Williams    libxml-simple-perl \
8070eedeedaSPatrick Williams    rsync
80802871c91SPatrick Williams
809b84d59dcSWilliam A. Kennington IIIRUN update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-11 11 \
810b84d59dcSWilliam A. Kennington III  --slave /usr/bin/g++ g++ /usr/bin/g++-11 \
811b84d59dcSWilliam A. Kennington III  --slave /usr/bin/gcov gcov /usr/bin/gcov-11 \
812b84d59dcSWilliam A. Kennington III  --slave /usr/bin/gcov-dump gcov-dump /usr/bin/gcov-dump-11 \
813b84d59dcSWilliam A. Kennington III  --slave /usr/bin/gcov-tool gcov-tool /usr/bin/gcov-tool-11
81402871c91SPatrick Williams
8154569bf49SPatrick WilliamsRUN update-alternatives --install /usr/bin/clang clang /usr/bin/clang-12 1000 \
8164569bf49SPatrick Williams  --slave /usr/bin/clang++ clang++ /usr/bin/clang++-12 \
8174569bf49SPatrick Williams  --slave /usr/bin/clang-tidy clang-tidy /usr/bin/clang-tidy-12 \
8184569bf49SPatrick Williams  --slave /usr/bin/clang-format clang-format /usr/bin/clang-format-12 \
8194569bf49SPatrick Williams  --slave /usr/bin/run-clang-tidy.py run-clang-tidy.py /usr/bin/run-clang-tidy-12.py \
8204569bf49SPatrick Williams  --slave /usr/bin/scan-build scan-build /usr/bin/scan-build-12
82102871c91SPatrick Williams
82250837436SPatrick Williams"""
82350837436SPatrick Williams
82450837436SPatrick Williamsif is_automated_ci_build:
82550837436SPatrick Williams    dockerfile_base += f"""
82650837436SPatrick Williams# Run an arbitrary command to polute the docker cache regularly force us
82750837436SPatrick Williams# to re-run `apt-get update` daily.
828ee3c9eebSPatrick WilliamsRUN echo {Docker.timestamp()}
82950837436SPatrick WilliamsRUN apt-get update && apt-get dist-upgrade -yy
83050837436SPatrick Williams
83150837436SPatrick Williams"""
83250837436SPatrick Williams
83350837436SPatrick Williamsdockerfile_base += f"""
83402871c91SPatrick WilliamsRUN pip3 install inflection
83502871c91SPatrick WilliamsRUN pip3 install pycodestyle
83602871c91SPatrick WilliamsRUN pip3 install jsonschema
837*b3e88fb6SPatrick WilliamsRUN pip3 install meson==0.58.1
83802871c91SPatrick WilliamsRUN pip3 install protobuf
839e6f120aaSManojkiran EdaRUN pip3 install codespell
840a18d9c57SPatrick Williams"""
84102871c91SPatrick Williams
842ee3c9eebSPatrick Williams# Build the base and stage docker images.
843ee3c9eebSPatrick Williamsdocker_base_img_name = Docker.tagname("base", dockerfile_base)
844ee3c9eebSPatrick WilliamsDocker.build("base", docker_base_img_name, dockerfile_base)
845ee3c9eebSPatrick WilliamsPackage.generate_all()
84602871c91SPatrick Williams
847ee3c9eebSPatrick Williams# Create the final Dockerfile.
848a18d9c57SPatrick Williamsdockerfile = f"""
84902871c91SPatrick Williams# Build the final output image
850a18d9c57SPatrick WilliamsFROM {docker_base_img_name}
851ee3c9eebSPatrick Williams{Package.df_all_copycmds()}
85202871c91SPatrick Williams
85302871c91SPatrick Williams# Some of our infrastructure still relies on the presence of this file
85402871c91SPatrick Williams# even though it is no longer needed to rebuild the docker environment
85502871c91SPatrick Williams# NOTE: The file is sorted to ensure the ordering is stable.
856ee3c9eebSPatrick WilliamsRUN echo '{Package.depcache()}' > /tmp/depcache
85702871c91SPatrick Williams
85802871c91SPatrick Williams# Final configuration for the workspace
85902871c91SPatrick WilliamsRUN grep -q {gid} /etc/group || groupadd -g {gid} {username}
86002871c91SPatrick WilliamsRUN mkdir -p "{os.path.dirname(homedir)}"
86102871c91SPatrick WilliamsRUN grep -q {uid} /etc/passwd || useradd -d {homedir} -m -u {uid} -g {gid} {username}
86202871c91SPatrick WilliamsRUN sed -i '1iDefaults umask=000' /etc/sudoers
86302871c91SPatrick WilliamsRUN echo "{username} ALL=(ALL) NOPASSWD: ALL" >>/etc/sudoers
86402871c91SPatrick Williams
865305a9a5dSAndrew Geissler# Ensure user has ability to write to /usr/local for different tool
866305a9a5dSAndrew Geissler# and data installs
8677bb00b13SAndrew GeisslerRUN chown -R {username}:{username} /usr/local/share
868305a9a5dSAndrew Geissler
86902871c91SPatrick Williams{proxy_cmd}
87002871c91SPatrick Williams
87102871c91SPatrick WilliamsRUN /bin/bash
87202871c91SPatrick Williams"""
87302871c91SPatrick Williams
874a18d9c57SPatrick Williams# Do the final docker build
875ee3c9eebSPatrick Williamsdocker_final_img_name = Docker.tagname(None, dockerfile)
876ee3c9eebSPatrick WilliamsDocker.build("final", docker_final_img_name, dockerfile)
877ee3c9eebSPatrick Williams
87800536fbeSPatrick Williams# Print the tag of the final image.
87900536fbeSPatrick Williamsprint(docker_final_img_name)
880