Add extras to {whl,pypi}_repo_name

Author: hartikainen

python/private/pypi/whl_repo_name.bzl

@@ -18,12 +18,13 @@
 load("//python/private:normalize_name.bzl", "normalize_name")
 load(":parse_whl_name.bzl", "parse_whl_name")
 
-def whl_repo_name(filename, sha256):
+def whl_repo_name(filename, sha256, extras):
     """Return a valid whl_library repo name given a distribution filename.
 
     Args:
         filename: {type}`str` the filename of the distribution.
         sha256: {type}`str` the sha256 of the distribution.
+        extras: {type}`list[str]` the extras for the requirement. TODO(hartikainen): Note sure if this is the right place for extras.
 
     Returns:
         a string that can be used in {obj}`whl_library`.
@@ -34,6 +35,7 @@ def whl_repo_name(filename, sha256):
         # Then the filename is basically foo-3.2.1.<ext>
         name, _, tail = filename.rpartition("-")
         parts.append(normalize_name(name))
+        parts.extend(sorted(extras))
         if sha256:
             parts.append("sdist")
             version = ""
@@ -53,6 +55,7 @@ def whl_repo_name(filename, sha256):
         parts.append(python_tag)
         parts.append(abi_tag)
         parts.append(platform_tag)
+        parts.extend(sorted(extras))
 
     if sha256:
         parts.append(sha256[:8])
@@ -61,19 +64,24 @@ def whl_repo_name(filename, sha256):
 
     return "_".join(parts)
 
-def pypi_repo_name(whl_name, *target_platforms):
+def pypi_repo_name(whl_name, extras, *target_platforms):
     """Return a valid whl_library given a requirement line.
 
+    TODO(hartikainen): Note sure if this is the right place for extras.
+
     Args:
         whl_name: {type}`str` the whl_name to use.
+        extras: {type}`list[str]` the extras for the requirement. TODO(hartikainen): Note sure if this is the right place for extras.
         *target_platforms: {type}`list[str]` the target platforms to use in the name.
 
+
     Returns:
         {type}`str` that can be used in {obj}`whl_library`.
     """
     parts = [
         normalize_name(whl_name),
     ]
+    parts.extend(sorted(extras))
     parts.extend([p.partition("_")[-1] for p in target_platforms])
 
     return "_".join(parts)