From 486a38440f78ea06c2dc3629300bd7bdeac127e6 Mon Sep 17 00:00:00 2001
From: Enno Tensing <tenno@suij.in>
Date: Fri, 8 Aug 2025 15:43:39 +0200
Subject: [PATCH] generate: container: Add command section to image
 configuration

Signed-off-by: Enno Tensing <tenno@suij.in>
---
 generate/container.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/generate/container.py b/generate/container.py
index c8e19f9..a1890aa 100644
--- a/generate/container.py
+++ b/generate/container.py
@@ -506,6 +506,7 @@ class Image:
     registry: str
     image: str
     tag: str
+    cmd: str
 
     @classmethod
     def from_json(cls, val: ConfigValue, logger: Log) -> Self | None:
@@ -516,10 +517,13 @@ class Image:
         registry = maybe_or(val, "registry", "")
         image = maybe_or(val, "image", "")
         tag = maybe_or(val, "tag", "")
-        return cls(str(registry), str(image), str(tag))
+        cmd = maybe_or(val, "command", "")
+        return cls(str(registry), str(image), str(tag), cmd)
 
     def command(self) -> str:
         """Option for podman container create."""
+        if self.cmd != "":
+            return f"{self.registry}/{self.image}:{self.tag} {self.cmd}"
         return f"{self.registry}/{self.image}:{self.tag}"