From 2024aa0ad58aa8f3aa8fb42546fd3e15c1a0f903 Mon Sep 17 00:00:00 2001
From: Jake Howard <git@theorangeone.net>
Date: Wed, 14 Jun 2023 09:13:49 +0100
Subject: [PATCH] Support more images for hero sizes

---
 website/common/models.py                   | 20 +++++++++++++++-----
 website/common/templates/wagtail_base.html |  2 +-
 website/contrib/unsplash/models.py         |  3 +++
 3 files changed, 19 insertions(+), 6 deletions(-)

diff --git a/website/common/models.py b/website/common/models.py
index 74ffe65..1ffcd62 100644
--- a/website/common/models.py
+++ b/website/common/models.py
@@ -24,6 +24,7 @@ from wagtail.snippets.blocks import SnippetChooserBlock
 from wagtail.snippets.models import register_snippet
 from wagtailmetadata.models import MetadataMixin
 
+from website.contrib.unsplash.models import SIZES as UNSPLASH_SIZES
 from website.contrib.unsplash.widgets import UnsplashPhotoChooser
 
 from .serializers import PaginationSerializer
@@ -143,26 +144,35 @@ class BaseContentPage(BasePage, MetadataMixin):
     def plain_text(self) -> str:
         return extract_text(self.content_html)
 
-    def hero_url(self, unsplash_size: str, wagtail_image_spec: str) -> Optional[str]:
+    def hero_url(
+        self, image_size: str, wagtail_image_spec_extra: Optional[str] = None
+    ) -> Optional[str]:
         if self.hero_unsplash_photo_id is not None:
-            return self.hero_unsplash_photo.get_image_urls()[unsplash_size]
+            return self.hero_unsplash_photo.get_image_urls()[image_size]
         elif self.hero_image_id is not None:
+            image_width = UNSPLASH_SIZES[image_size]
+            wagtail_image_spec = f"width-{image_width}"
+            if wagtail_image_spec_extra:
+                wagtail_image_spec += wagtail_image_spec_extra
             return generate_image_url(self.hero_image, wagtail_image_spec)
         return None
 
     @cached_property
+    def hero_image_urls(self) -> dict:
+        return {width: self.hero_url(size) for size, width in UNSPLASH_SIZES.items()}
+
     def hero_image_url(self) -> Optional[str]:
-        return self.hero_url("full", "width-2000")
+        return self.hero_url("full")
 
     @cached_property
     def list_image_url(self) -> Optional[str]:
-        return self.hero_url("small", "width-400")
+        return self.hero_url("small")
 
     def get_meta_url(self) -> str:
         return self.full_url
 
     def get_meta_image_url(self, request: HttpRequest) -> Optional[str]:
-        return self.hero_url("regular", "width-1000|format-png")
+        return self.hero_url("regular", "|format-png")
 
     def get_meta_title(self) -> str:
         return self.html_title
diff --git a/website/common/templates/wagtail_base.html b/website/common/templates/wagtail_base.html
index 04113e7..0b5b709 100644
--- a/website/common/templates/wagtail_base.html
+++ b/website/common/templates/wagtail_base.html
@@ -13,7 +13,7 @@
 {% block main_content %}
   {% if page.hero_image_url %}
     <picture>
-      {% if page.list_image_url %}<source srcset="{{ page.list_image_url }}" media="(max-width: 800px)" />{% endif %}
+      {% for width, image_url in page.hero_image_urls.items %}<source srcset="{{ image_url }}" media="(max-width: {{ width }}px)" />{% endfor %}
 
       <img class="hero" src="{{ page.hero_image_url }}" decoding="async" alt="" />
     </picture>
diff --git a/website/contrib/unsplash/models.py b/website/contrib/unsplash/models.py
index b802ccc..c7e6f57 100644
--- a/website/contrib/unsplash/models.py
+++ b/website/contrib/unsplash/models.py
@@ -13,6 +13,9 @@ class ImageURLs(TypedDict):
     thumb: str
 
 
+SIZES = {"full": 2000, "regular": 1080, "small": 400, "thumb": 200}
+
+
 class UnsplashPhoto(index.Indexed, models.Model):
     unsplash_id = models.CharField(unique=True, max_length=11, db_index=True)
     data = models.JSONField()