Introduce distribution installer finalize function

mkosi/__init__.py

@@ -295,6 +295,9 @@ def install_distribution(context: Context) -> None:
                     context, context.config.packages
                 )
 
+    with complete_step(f"Finalizing {context.config.distribution.pretty_name()}"):
+        context.config.distribution.finalize(context)
+
     for f in (
         "var/lib/systemd/random-seed",
         "var/lib/systemd/credential.secret",

mkosi/distributions/__init__.py

@@ -42,6 +42,10 @@ def setup(cls, context: "Context") -> None:
     def install(cls, context: "Context") -> None:
         raise NotImplementedError
 
+    @classmethod
+    def finalize(cls, context: "Context") -> None:
+        raise NotImplementedError
+
     @classmethod
     def filesystem(cls) -> str:
         return "ext4"
@@ -127,6 +131,9 @@ def setup(self, context: "Context") -> None:
     def install(self, context: "Context") -> None:
         return self.installer().install(context)
 
+    def finalize(self, context: "Context") -> None:
+        return self.installer().finalize(context)
+
     def filesystem(self) -> str:
         return self.installer().filesystem()
 

mkosi/distributions/arch.py

@@ -63,6 +63,10 @@ def setup(cls, context: Context) -> None:
     def install(cls, context: Context) -> None:
         Pacman.install(context, ["filesystem"], apivfs=False)
 
+    @classmethod
+    def finalize(cls, context: Context) -> None:
+        pass
+
     @classmethod
     def repositories(cls, context: Context) -> Iterable[PacmanRepository]:
         if context.config.local_mirror:

mkosi/distributions/centos.py

@@ -79,6 +79,10 @@ def setup(cls, context: Context) -> None:
     def install(cls, context: Context) -> None:
         Dnf.install(context, ["basesystem"], apivfs=False)
 
+    @classmethod
+    def finalize(cls, context: Context) -> None:
+        pass
+
     @classmethod
     def architecture(cls, arch: Architecture) -> str:
         a = {

mkosi/distributions/custom.py

@@ -27,3 +27,7 @@ def setup(cls, context: Context) -> None:
     @classmethod
     def install(cls, context: Context) -> None:
         pass
+
+    @classmethod
+    def finalize(cls, context: Context) -> None:
+        pass

mkosi/distributions/debian.py

@@ -189,6 +189,8 @@ def install(cls, context: Context) -> None:
 
         fixup_os_release(context)
 
+    @classmethod
+    def finalize(cls, context: Context) -> None:
         if (
             "apt" in itertools.chain(context.config.packages, context.config.volatile_packages)
             or (context.root / "usr/bin/apt").exists()

mkosi/distributions/fedora.py

@@ -130,6 +130,10 @@ def setup(cls, context: Context) -> None:
     def install(cls, context: Context) -> None:
         Dnf.install(context, ["basesystem"], apivfs=False)
 
+    @classmethod
+    def finalize(cls, context: Context) -> None:
+        pass
+
     @classmethod
     def repositories(cls, context: Context) -> Iterable[RpmRepository]:
         gpgurls = find_fedora_rpm_gpgkeys(context)

mkosi/distributions/opensuse.py

@@ -55,6 +55,10 @@ def setup(cls, context: Context) -> None:
     def install(cls, context: Context) -> None:
         cls.package_manager(context.config).install(context, ["filesystem"], apivfs=False)
 
+    @classmethod
+    def finalize(cls, context: Context) -> None:
+        pass
+
     @classmethod
     def repositories(cls, context: Context) -> Iterable[RpmRepository]:
         if context.config.local_mirror: