From 260a0dcc52f05fe00288b3722d4dbdb98c876ec7 Mon Sep 17 00:00:00 2001
From: Enno Tensing <tenno@suij.in>
Date: Mon, 4 Aug 2025 21:33:41 +0200
Subject: [PATCH] generate: container: Improve optional feature handling

Things missing should now cause less problems.

Signed-off-by: Enno Tensing <tenno@suij.in>
---
 generate/container.py | 61 +++++++++++++++++++++++++++----------------
 1 file changed, 39 insertions(+), 22 deletions(-)

diff --git a/generate/container.py b/generate/container.py
index 4bb4ad3..143c313 100644
--- a/generate/container.py
+++ b/generate/container.py
@@ -384,12 +384,17 @@ class Environment:
 
     def create(self) -> str:
         """Create env file."""
-        cmd = f"# Create env-file {self.file}\n"
+        cmd = ""
+        header = f"# Create env-file {self.file}\n"
+
         for var in self.variables:
             escaped_var = var.replace("'", "%b")
             cmd += f"printf '{escaped_var}\\n' \"'\" \"'\" >> '{self.file}'\n"
 
-        return cmd
+        if cmd == "":
+            return ""
+
+        return header + cmd
 
     def remove(self) -> str:
         """Remove env file."""
@@ -418,27 +423,41 @@ class Ports:
             return cls([], [])
         tcp_ports = maybe(val, "tcp")
         udp_ports = maybe(val, "udp")
+        if tcp_ports is None:
+            tcp_ports = []
+        if udp_ports is None:
+            udp_ports = []
+        if not isinstance(tcp_ports, list) and not isinstance(udp_ports, list):
+            logger.log_warning("Port configuration is malformed!")
+            return cls([], [])
         if not isinstance(tcp_ports, list):
-            logger.log_warning("Key tcp_ports is not an array!")
-            return cls([], [])
+            logger.log_warning("tcp_ports configuration is malformed!")
+            return cls([], udp_ports)
         if not isinstance(udp_ports, list):
-            logger.log_warning("Key udp_ports is not an array!")
-            return cls([], [])
+            logger.log_warning("udp_ports configuration is malformed!")
+            return cls(tcp_ports, [])
         return cls(tcp_ports, udp_ports)
 
     def command(self) -> str:
         """Option for podman container create."""
-        ports = ""
         seperator = " \\\n"
-        ports += seperator.join(
+        tcp_ports = seperator.join(
             [f"\t--publish {port}/tcp" for port in self.tcp_ports]
         )
-        ports += seperator
-        ports += seperator.join(
+        udp_ports = seperator.join(
             [f"\t--publish {port}/udp" for port in self.udp_ports]
         )
-        ports += seperator
-        return ports
+
+        if tcp_ports == "" and udp_ports == "":
+            return ""
+
+        if tcp_ports == "":
+            return udp_ports + seperator
+
+        if udp_ports == "":
+            return tcp_ports + seperator
+
+        return tcp_ports + seperator + udp_ports + seperator
 
 
 @dataclass
@@ -469,11 +488,11 @@ class Network:
         """Option for podman container create."""
         if self.mode == "":
             return ""
-        cmd = f"--network={self.mode}"
+        cmd = f"\t--network={self.mode}"
         opts = ",".join(self.options)
         if opts != "":
             cmd += f":{opts}"
-        return cmd
+        return cmd + " \\\n"
 
 
 @dataclass
@@ -513,9 +532,7 @@ class Capability:
         if val is None:
             return []
         if not isinstance(val, dict):
-            logger.log_warning(
-                "Capabilities key is malformed!"
-            )
+            logger.log_warning("Capabilities key is malformed!")
             return []
         add = [cls(value, "add") for value in val["add"]]
         drop = [cls(value, "drop") for value in val["drop"]]
@@ -554,11 +571,11 @@ class Dns:
         if len(self.servers) == 0 and self.search == "":
             return ""
         if len(self.servers) == 0:
-            return f"--dns-search={self.search}"
+            return f"\t--dns-search={self.search} \\\n"
         if self.search == "":
-            return f"--dns={','.join(self.servers)}"
+            return f"\t--dns={','.join(self.servers)} \\\n"
 
-        cmd = f"--dns-search={self.search} \\\n\t--dns="
+        cmd = f"\t--dns-search={self.search} \\\n\t--dns="
         cmd += ",".join(self.servers)
 
         return cmd
@@ -667,8 +684,8 @@ class Container:
         cmd += f"\t--restart={self.restart} \\\n"
         cmd += f"\t--pull={self.pull_policy} \\\n"
         cmd += f"\t--tz={self.timezone} \\\n"
-        cmd += f"\t{self.network.command()} \\\n"
-        cmd += f"\t{self.dns.command()} \\\n"
+        cmd += f"{self.network.command()}"
+        cmd += f"{self.dns.command()}"
         cmd += f"{self.ports.command()}"
         if self.env is not None:
             cmd += f"\t{self.env.command()} \\\n"