Parse OBJ endpoint domain dynamically (#851)

Author: ezilber-akamai

linodecli/plugins/obj/config.py

@@ -2,13 +2,24 @@
 The config of the object storage plugin.
 """
 
+import os
+import re
 import shutil
 
 ENV_ACCESS_KEY_NAME = "LINODE_CLI_OBJ_ACCESS_KEY"
 ENV_SECRET_KEY_NAME = "LINODE_CLI_OBJ_SECRET_KEY"
-# replace {} with the cluster name
-BASE_URL_TEMPLATE = "https://{}.linodeobjects.com"
-BASE_WEBSITE_TEMPLATE = "{bucket}.website-{cluster}.linodeobjects.com"
+
+API_HOST = os.getenv("LINODE_CLI_API_HOST", "")
+
+OBJ_DOMAIN = "linodeobjects.com"
+
+if API_HOST:
+    match = re.match(r"api\.([^.]+)\.linode\.com", API_HOST)
+    if match:
+        OBJ_DOMAIN = f"{match.group(1)}.linodeobjects.com"
+
+BASE_URL_TEMPLATE = f"https://{{}}.{OBJ_DOMAIN}"
+BASE_WEBSITE_TEMPLATE = f"{{bucket}}.website-{{cluster}}.{OBJ_DOMAIN}"
 
 # for all date output
 DATE_FORMAT = "%Y-%m-%d %H:%M"