From 16005e14ef46c6721f4bf4e01a54e6c1bf0ecdf7 Mon Sep 17 00:00:00 2001
From: Andrew Godwin <andrew@aeracode.org>
Date: Wed, 14 Dec 2022 21:21:18 -0700
Subject: [PATCH] Make GCS backend handle webp right, and use gs://

Fixes #164
---
 core/uploads.py    | 20 ++++++++++++++++++--
 takahe/settings.py | 13 ++++++-------
 2 files changed, 24 insertions(+), 9 deletions(-)

diff --git a/core/uploads.py b/core/uploads.py
index c83888c..41b6e94 100644
--- a/core/uploads.py
+++ b/core/uploads.py
@@ -2,6 +2,7 @@ import os
 import secrets
 
 from django.utils import timezone
+from storages.backends.gcloud import GoogleCloudStorage
 from storages.backends.s3boto3 import S3Boto3Storage
 
 
@@ -16,11 +17,26 @@ def upload_namer(prefix, instance, filename):
 
 
 class TakaheS3Storage(S3Boto3Storage):
+    """
+    Custom override backend that makes webp files store correctly
+    """
+
     def get_object_parameters(self, name: str):
         params = self.object_parameters.copy()
-
         if name.endswith(".webp"):
             params["ContentDisposition"] = "inline"
             params["ContentType"] = "image/webp"
-
+        return params
+
+
+class TakaheGoogleCloudStorage(GoogleCloudStorage):
+    """
+    Custom override backend that makes webp files store correctly
+    """
+
+    def get_object_parameters(self, name: str):
+        params = self.object_parameters.copy()
+        if name.endswith(".webp"):
+            params["content_disposition"] = "inline"
+            params["content_type"] = "image/webp"
         return params
diff --git a/takahe/settings.py b/takahe/settings.py
index 552dc76..ee3854c 100644
--- a/takahe/settings.py
+++ b/takahe/settings.py
@@ -338,15 +338,14 @@ if SETUP.EMAIL_SERVER:
 if SETUP.MEDIA_BACKEND:
     parsed = urllib.parse.urlparse(SETUP.MEDIA_BACKEND)
     query = urllib.parse.parse_qs(parsed.query)
-    if parsed.scheme == "gcs":
-        DEFAULT_FILE_STORAGE = "storages.backends.gcloud.GoogleCloudStorage"
-        if parsed.path.lstrip("/"):
-            GS_BUCKET_NAME = parsed.path.lstrip("/")
-        else:
-            GS_BUCKET_NAME = parsed.hostname
+    if parsed.scheme == "gs":
+        DEFAULT_FILE_STORAGE = "core.uploads.TakaheGoogleCloudStorage"
+        GS_BUCKET_NAME = parsed.path.lstrip("/")
         GS_QUERYSTRING_AUTH = False
+        if parsed.hostname is not None:
+            port = parsed.port or 443
+            GS_CUSTOM_ENDPOINT = f"https://{parsed.hostname}:{port}"
     elif parsed.scheme == "s3":
-        # DEFAULT_FILE_STORAGE = "storages.backends.s3boto3.S3Boto3Storage"
         DEFAULT_FILE_STORAGE = "core.uploads.TakaheS3Storage"
         AWS_STORAGE_BUCKET_NAME = parsed.path.lstrip("/")
         AWS_QUERYSTRING_AUTH = False