From dacfd413dcfc5ead5d019efbe45059f2f6922696 Mon Sep 17 00:00:00 2001
From: tobi <31960611+tsmethurst@users.noreply.github.com>
Date: Tue, 5 Dec 2023 12:43:07 +0100
Subject: [chore/frontend] Refactor status templates slightly, put polls behind
 CWs if present (#2419)

* [chore/frontend] Reorder templates, allow polls to be inside content warnings

* show when multiple-choice
---
 internal/api/model/attachment.go         | 6 ++++++
 internal/typeutils/internaltofrontend.go | 6 ++++++
 2 files changed, 12 insertions(+)

(limited to 'internal')

diff --git a/internal/api/model/attachment.go b/internal/api/model/attachment.go
index 1911fc9c3..7e81759f2 100644
--- a/internal/api/model/attachment.go
+++ b/internal/api/model/attachment.go
@@ -90,6 +90,12 @@ type Attachment struct {
 	// A hash computed by the BlurHash algorithm, for generating colorful preview thumbnails when media has not been downloaded yet.
 	// See https://github.com/woltapp/blurhash
 	Blurhash *string `json:"blurhash"`
+
+	// Additional fields not exposed via JSON
+	// (used only internally for templating etc).
+
+	// Parent status of this media is sensitive.
+	Sensitive bool `json:"-"`
 }
 
 // MediaMeta models media metadata.
diff --git a/internal/typeutils/internaltofrontend.go b/internal/typeutils/internaltofrontend.go
index fe5c8ec8f..b3d263963 100644
--- a/internal/typeutils/internaltofrontend.go
+++ b/internal/typeutils/internaltofrontend.go
@@ -721,6 +721,12 @@ func (c *Converter) StatusToWebStatus(
 		webStatus.WebPollOptions = webPollOptions
 	}
 
+	// Set additional templating
+	// variables on media attachments.
+	for _, a := range webStatus.MediaAttachments {
+		a.Sensitive = webStatus.Sensitive
+	}
+
 	return webStatus, nil
 }
 
-- 
cgit v1.2.3