Add --args to ctf challenge deploy for extra docker run arguments

Author: pl4nty

ctfcli/cli/challenges.py

@@ -296,7 +296,7 @@ def lint(self, challenge=None):
 
         lint_challenge(path)
 
-    def deploy(self, challenge, host=None):
+    def deploy(self, challenge, host=None, args=None):
         if challenge is None:
             challenge = os.getcwd()
 
@@ -330,7 +330,7 @@ def deploy(self, challenge, host=None):
             return
 
         status, domain, port = DEPLOY_HANDLERS[url.scheme](
-            challenge=challenge, host=target_host
+            challenge=challenge, host=target_host, args=args
         )
 
         if status:

ctfcli/utils/deploy.py

@@ -6,7 +6,7 @@
 from ctfcli.utils.images import build_image, export_image, get_exposed_ports
 
 
-def ssh(challenge, host):
+def ssh(challenge, host, args):
     # Build image
     image_name = build_image(challenge=challenge)
     print(f"Built {image_name}")
@@ -31,7 +31,7 @@ def ssh(challenge, host):
         [
             "ssh",
             host.netloc,
-            f"docker run -d -p{exposed_port}:{exposed_port} --name {image_name} --restart always {image_name}",
+            f"docker run -d -p{exposed_port}:{exposed_port} --name {image_name} --restart always {image_name}{' '+args if args else ''}",
         ]
     )