From ef10fab47d555d28c79058ab4b69fa8c854a5991 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Wed, 31 Jan 2024 00:36:04 +0800
Subject: [PATCH] :sparkles: Force HTTPS

---
 pkg/hypertext/server.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/pkg/hypertext/server.go b/pkg/hypertext/server.go
index 5c89463..d153994 100644
--- a/pkg/hypertext/server.go
+++ b/pkg/hypertext/server.go
@@ -26,6 +26,19 @@ func InitServer() *fiber.App {
 		BodyLimit:             viper.GetInt("hypertext.limitation.max_body_size"),
 	})
 
+	if viper.GetBool("hypertext.force_https") {
+		app.Use(func(c *fiber.Ctx) error {
+			if !c.Secure() {
+				return c.Redirect(
+					strings.Replace(c.Request().URI().String(), "http", "https", 1),
+					fiber.StatusMovedPermanently,
+				)
+			}
+
+			return c.Next()
+		})
+	}
+
 	if viper.GetBool("telemetry.request_logging") {
 		app.Use(logger.New(logger.Config{
 			Output: log.Logger,