micropython/mip/mip/__init__.py: Add an alias for codeberg repos.

This commit introduces an alias to access codeberg repos from within the
mip embedded package manager.

Right now packages hosted on codeberg could only be referenced by their
full URL, unlike other packages hosted on either GitHub or GitLab.  To
make access those packages easier, now they can be referenced as
"codeberg:org/repo@branch".

Signed-off-by: Alessandro Gatti <a.gatti@frob.it>

Author: agatti

micropython/mip/mip/__init__.py

@@ -10,7 +10,19 @@
 _PACKAGE_INDEX = const("https://micropython.org/pi/v2")
 _CHUNK_SIZE = const(128)
 
-allowed_mip_url_prefixes = ("http://", "https://", "github:", "gitlab:")
+# URL components must appear in the same order to avoid having to embed
+# component names in the format string themselves.  Since all URLs are
+# accessed via HTTPS, the URL scheme has to be omitted here.
+_PROVIDERS = {
+    # https://codeberg.org/{org}/{repo}/raw/branch/{branch}/{path}
+    "codeberg:": "codeberg.org/{}/{}/raw/branch/{}/{}",
+    # https://raw.githubusercontent.com/{org}/{repo}/{branch}/{path}
+    "github:": "raw.githubusercontent.com/{}/{}/{}/{}",
+    # https://gitlab.com/{org}/{repo}/-/raw/{branch}/{path}
+    "gitlab:": "gitlab.com/{}/{}/-/raw/{}/{}",
+}
+
+allowed_mip_url_prefixes = tuple(["http://", "https://"] + list(_PROVIDERS.keys()))
 
 
 # This implements os.makedirs(os.dirname(path))
@@ -64,29 +76,13 @@ def _check_exists(path, short_hash):
 def _rewrite_url(url, branch=None):
     if not branch:
         branch = "HEAD"
-    if url.startswith("github:"):
-        url = url[7:].split("/")
-        url = (
-            "https://raw.githubusercontent.com/"
-            + url[0]
-            + "/"
-            + url[1]
-            + "/"
-            + branch
-            + "/"
-            + "/".join(url[2:])
-        )
-    elif url.startswith("gitlab:"):
-        url = url[7:].split("/")
-        url = (
-            "https://gitlab.com/"
-            + url[0]
-            + "/"
-            + url[1]
-            + "/-/raw/"
-            + branch
-            + "/"
-            + "/".join(url[2:])
+    for provider, url_format in _PROVIDERS.items():
+        if not url.startswith(provider):
+            continue
+        components = url[len(provider) :].split("/")
+        # Add https:// prefix to final URL.
+        return allowed_mip_url_prefixes[1] + url_format.format(
+            components[0], components[1], branch, "/".join(components[2:])
         )
     return url