Adding GCP support

jschneier/django-storages#1193 Required PR: pulp/pulpcore#3367 [noissue]
fao89 · Oct 28, 2022 · 7d91c5e · 7d91c5e
1 parent 3a98395
commit 7d91c5e
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/pulp_container/app/redirects.py b/pulp_container/app/redirects.py
@@ -132,3 +132,20 @@ def redirect_to_object_storage(self, artifact, return_media_type):
         }
         content_url = artifact.file.storage.url(artifact.file.name, parameters=parameters)
         return redirect(content_url)
+
+class GCloudStorageRedirects(S3StorageRedirects):
+    """
+    A class that implements methods for the direct retrieval of manifest objects.
+    """
+
+    def redirect_to_object_storage(self, artifact, return_media_type):
+        """
+        Redirect to the passed artifact's file stored in the Azure storage.
+        """
+        filename = f"sha256:{artifact.sha256}"
+        parameters = {
+            "content_type": return_media_type,
+            "response_disposition": f"attachment;filename={filename}",
+        }
+        content_url = artifact.file.storage.url(artifact.file.name, parameters=parameters)
+        return redirect(content_url)
diff --git a/pulp_container/app/registry_api.py b/pulp_container/app/registry_api.py
@@ -65,6 +65,7 @@
     FileStorageRedirects,
     S3StorageRedirects,
     AzureStorageRedirects,
+    GCloudStorageRedirects,
 )
 from pulp_container.app.token_verification import (
     RegistryAuthentication,
@@ -794,6 +795,8 @@ def __init__(self, *args, **kwargs):
             self.redirects_class = S3StorageRedirects
         elif settings.DEFAULT_FILE_STORAGE == "storages.backends.azure_storage.AzureStorage":
             self.redirects_class = AzureStorageRedirects
+        elif settings.DEFAULT_FILE_STORAGE == "storages.backends.gcloud.GoogleCloudStorage":
+            self.redirects_class = GCloudStorageRedirects
         else:
             raise NotImplementedError()