Factored out supported URI schemes to a variable

Signed-off-by: jakub-nt <175944085+jakub-nt@users.noreply.github.com>

Author: jakub-nt

cfbs/cfbs_config.py

@@ -13,6 +13,7 @@
     load_bundlenames,
 )
 from cfbs.internal_file_management import (
+    SUPPORTED_URI_SCHEMES,
     clone_url_repo,
     fetch_archive,
     SUPPORTED_ARCHIVES,
@@ -132,7 +133,7 @@ def _add_using_url(
         if url.endswith(SUPPORTED_ARCHIVES):
             config_path, url_commit = fetch_archive(url, checksum)
         else:
-            assert url.startswith(("https://", "git://", "ssh://"))
+            assert url.startswith(SUPPORTED_URI_SCHEMES)
             config_path, url_commit = clone_url_repo(url)
 
         if "@" in url and (url.rindex("@") > url.rindex(".")):
@@ -386,7 +387,7 @@ def add_command(
         before = {m["name"] for m in self.get("build", [])}
 
         if to_add[0].endswith(SUPPORTED_ARCHIVES) or to_add[0].startswith(
-            ("https://", "git://", "ssh://")
+            SUPPORTED_URI_SCHEMES
         ):
             self._add_using_url(
                 url=to_add[0],

cfbs/commands.py

@@ -44,6 +44,7 @@
 from cfbs.cfbs_config import CFBSConfig, CFBSReturnWithoutCommit
 from cfbs.validate import validate_config
 from cfbs.internal_file_management import (
+    SUPPORTED_URI_SCHEMES,
     fetch_archive,
     get_download_path,
     local_module_copy,
@@ -455,7 +456,7 @@ def _get_modules_by_url(name) -> list:
     msg = ""
     files = []
     for name in to_remove:
-        if name.startswith(("https://", "ssh://", "git://")):
+        if name.startswith(SUPPORTED_URI_SCHEMES):
             matches = _get_modules_by_url(name)
             if not matches:
                 user_error("Could not find module with URL '%s'" % name)

cfbs/internal_file_management.py

@@ -28,6 +28,7 @@
 
 _SUPPORTED_TAR_TYPES = (".tar.gz", ".tgz")
 SUPPORTED_ARCHIVES = (".zip",) + _SUPPORTED_TAR_TYPES
+SUPPORTED_URI_SCHEMES = ("https://", "ssh://", "git://")
 
 
 def local_module_name(module_path):
@@ -113,7 +114,7 @@ def local_module_copy(module, counter, max_length):
 
 
 def _get_path_from_url(url):
-    if not url.startswith(("https://", "ssh://", "git://")):
+    if not url.startswith(SUPPORTED_URI_SCHEMES):
         if "://" in url:
             return user_error("Unsupported URL protocol in '%s'" % url)
         else:
@@ -153,7 +154,7 @@ def _clone_and_checkout(url, path, commit):
 
 
 def clone_url_repo(repo_url):
-    assert repo_url.startswith(("https://", "ssh://", "git://"))
+    assert repo_url.startswith(SUPPORTED_URI_SCHEMES)
 
     commit = None
     if "@" in repo_url and (repo_url.rindex("@") > repo_url.rindex(".")):