From 999875d0e45c7b1414b78b0d0939bf0631abf6d6 Mon Sep 17 00:00:00 2001
From: Erik Montnemery <erik@montnemery.com>
Date: Wed, 29 Nov 2023 11:26:50 +0100
Subject: [PATCH] Autogenerate Dockerfile (#104669)

---
 Dockerfile                  |  3 ++
 script/hassfest/__main__.py |  2 +
 script/hassfest/docker.py   | 89 +++++++++++++++++++++++++++++++++++++
 3 files changed, 94 insertions(+)
 create mode 100644 script/hassfest/docker.py

diff --git a/Dockerfile b/Dockerfile
index b61e1461c52..97eeb5b0dfa 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,3 +1,6 @@
+# Automatically generated by hassfest.
+#
+# To update, run python3 -m script.hassfest -p docker
 ARG BUILD_FROM
 FROM ${BUILD_FROM}
 
diff --git a/script/hassfest/__main__.py b/script/hassfest/__main__.py
index 32803731ecd..c454c69d141 100644
--- a/script/hassfest/__main__.py
+++ b/script/hassfest/__main__.py
@@ -16,6 +16,7 @@ from . import (
     coverage,
     dependencies,
     dhcp,
+    docker,
     json,
     manifest,
     metadata,
@@ -50,6 +51,7 @@ INTEGRATION_PLUGINS = [
 ]
 HASS_PLUGINS = [
     coverage,
+    docker,
     mypy_config,
     metadata,
 ]
diff --git a/script/hassfest/docker.py b/script/hassfest/docker.py
new file mode 100644
index 00000000000..1849e8e7ec8
--- /dev/null
+++ b/script/hassfest/docker.py
@@ -0,0 +1,89 @@
+"""Generate and validate the dockerfile."""
+from homeassistant import core
+from homeassistant.util import executor, thread
+
+from .model import Config, Integration
+
+DOCKERFILE_TEMPLATE = """# Automatically generated by hassfest.
+#
+# To update, run python3 -m script.hassfest -p docker
+ARG BUILD_FROM
+FROM ${{BUILD_FROM}}
+
+# Synchronize with homeassistant/core.py:async_stop
+ENV \\
+    S6_SERVICES_GRACETIME={timeout}
+
+ARG QEMU_CPU
+
+WORKDIR /usr/src
+
+## Setup Home Assistant Core dependencies
+COPY requirements.txt homeassistant/
+COPY homeassistant/package_constraints.txt homeassistant/homeassistant/
+RUN \\
+    pip3 install \\
+        --only-binary=:all: \\
+        -r homeassistant/requirements.txt
+
+COPY requirements_all.txt home_assistant_frontend-* home_assistant_intents-* homeassistant/
+RUN \\
+    if ls homeassistant/home_assistant_frontend*.whl 1> /dev/null 2>&1; then \\
+        pip3 install homeassistant/home_assistant_frontend-*.whl; \\
+    fi \\
+    && if ls homeassistant/home_assistant_intents*.whl 1> /dev/null 2>&1; then \\
+        pip3 install homeassistant/home_assistant_intents-*.whl; \\
+    fi \\
+    && \\
+        LD_PRELOAD="/usr/local/lib/libjemalloc.so.2" \\
+        MALLOC_CONF="background_thread:true,metadata_thp:auto,dirty_decay_ms:20000,muzzy_decay_ms:20000" \\
+        pip3 install \\
+            --only-binary=:all: \\
+            -r homeassistant/requirements_all.txt
+
+## Setup Home Assistant Core
+COPY . homeassistant/
+RUN \\
+    pip3 install \\
+        --only-binary=:all: \\
+        -e ./homeassistant \\
+    && python3 -m compileall \\
+        homeassistant/homeassistant
+
+# Home Assistant S6-Overlay
+COPY rootfs /
+
+WORKDIR /config
+"""
+
+
+def _generate_dockerfile() -> str:
+    timeout = (
+        core.STAGE_1_SHUTDOWN_TIMEOUT
+        + core.STAGE_2_SHUTDOWN_TIMEOUT
+        + core.STAGE_3_SHUTDOWN_TIMEOUT
+        + executor.EXECUTOR_SHUTDOWN_TIMEOUT
+        + thread.THREADING_SHUTDOWN_TIMEOUT
+        + 10
+    )
+    return DOCKERFILE_TEMPLATE.format(timeout=timeout * 1000)
+
+
+def validate(integrations: dict[str, Integration], config: Config) -> None:
+    """Validate dockerfile."""
+    dockerfile_content = _generate_dockerfile()
+    config.cache["dockerfile"] = dockerfile_content
+
+    dockerfile_path = config.root / "Dockerfile"
+    if dockerfile_path.read_text() != dockerfile_content:
+        config.add_error(
+            "docker",
+            "File Dockerfile is not up to date. Run python3 -m script.hassfest",
+            fixable=True,
+        )
+
+
+def generate(integrations: dict[str, Integration], config: Config) -> None:
+    """Generate dockerfile."""
+    dockerfile_path = config.root / "Dockerfile"
+    dockerfile_path.write_text(config.cache["dockerfile"])