Get direct/gallery from header too

Wikidepia · Nov 1, 2024 · f0dd30e · f0dd30e
1 parent 0ae9f64
commit f0dd30e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/handlers/embed.go b/handlers/embed.go
@@ -55,6 +55,14 @@ func Embed(w http.ResponseWriter, r *http.Request) {
 	isDirect, _ := strconv.ParseBool(urlQuery.Get("direct"))
 	isGallery, _ := strconv.ParseBool(urlQuery.Get("gallery"))
 
+	// Get direct/gallery from header too, nginx query params is pain in the ass
+	embedType := r.Header.Get("X-Embed-Type")
+	if embedType == "direct" {
+		isDirect = true
+	} else if embedType == "gallery" {
+		isGallery = true
+	}
+
 	// Stories use mediaID (int) instead of postID
 	if strings.Contains(r.URL.Path, "/stories/") {
 		mediaID, err := strconv.Atoi(postID)