From 5fa33bcad524bc4e90f858d710b41eeb25414418 Mon Sep 17 00:00:00 2001
From: tobi <tobi.smethurst@protonmail.com>
Date: Wed, 5 Feb 2025 12:19:39 +0100
Subject: [PATCH] use switch for RobotsHeaders

---
 internal/middleware/robots.go | 22 ++++++++++++----------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/internal/middleware/robots.go b/internal/middleware/robots.go
index 4981b891e..fefd93be0 100644
--- a/internal/middleware/robots.go
+++ b/internal/middleware/robots.go
@@ -41,25 +41,27 @@ func RobotsHeaders(mode string) gin.HandlerFunc {
 		noai = "noai, noimageai"
 	)
 
-	if mode == "aiOnly" {
-		// Just set ai headers and
-		// leave the other headers be.
+	switch mode {
+
+	// Just set ai headers and
+	// leave the other headers be.
+	case "aiOnly":
 		return func(c *gin.Context) {
 			c.Writer.Header().Set(key, noai)
 		}
-	}
 
-	if mode == "allowSome" {
-		// Allow some limited indexing.
+	// Allow some limited indexing.
+	case "allowSome":
 		return func(c *gin.Context) {
 			c.Writer.Header().Set(key, apiutil.RobotsDirectivesAllowSome)
 			c.Writer.Header().Add(key, noai)
 		}
-	}
 
 	// Disallow indexing via noindex, nofollow.
-	return func(c *gin.Context) {
-		c.Writer.Header().Set(key, apiutil.RobotsDirectivesDisallow)
-		c.Writer.Header().Add(key, noai)
+	default:
+		return func(c *gin.Context) {
+			c.Writer.Header().Set(key, apiutil.RobotsDirectivesDisallow)
+			c.Writer.Header().Add(key, noai)
+		}
 	}
 }