Add subcommand to get url for a media file

Author: ColdHeat

ctfcli/cli/media.py

@@ -1,5 +1,7 @@
 import os
 
+import click
+
 from ctfcli.core.api import API
 from ctfcli.core.config import Config
 
@@ -53,3 +55,20 @@ def rm(self, path):
                 del config["media"][path]
                 with open(config.config_path, "w+") as f:
                     config.write(f)
+
+    def url(self, path):
+        """Get server URL for a file key"""
+        config = Config()
+
+        if config.config.has_section("media") is False:
+            config.config.add_section("media")
+
+        try:
+            location = config["media"][path]
+        except KeyError:
+            click.secho(f"Could not locate media '{path}'", fg="red")
+            return 1
+
+        base_url = config["config"]["url"]
+        base_url = base_url.rstrip("/")
+        return f"{base_url}{location}"