Use options

Author: praem90

foundation/console/down_command.go

@@ -16,7 +16,7 @@ type DownCommand struct {
 	app foundation.Application
 }
 
-type DownOptions struct {
+type MaintenanceOptions struct {
 	Reason   string `json:"reason,omitempty"`
 	Redirect string `json:"redirect,omitempty"`
 	Render   string `json:"render,omitempty"`
@@ -82,7 +82,7 @@ func (r *DownCommand) Handle(ctx console.Context) error {
 		return nil
 	}
 
-	options := DownOptions{}
+	options := MaintenanceOptions{}
 
 	options.Status = ctx.OptionInt("status")
 

http/middleware/check_for_maintenance.go

@@ -1,29 +1,51 @@
 package middleware
 
 import (
+	"encoding/json"
+	"os"
+
 	"github.com/goravel/framework/contracts/http"
+	"github.com/goravel/framework/foundation/console"
 	"github.com/goravel/framework/support/file"
 	"github.com/goravel/framework/support/path"
 )
 
 func CheckForMaintenance() http.Middleware {
 	return func(ctx http.Context) {
 		filepath := path.Storage("framework/maintenance")
-		if file.Exists(filepath) {
-			content, err := file.GetContent(filepath)
-
-			if err != nil {
-				ctx.Request().Abort(http.StatusServiceUnavailable)
-				return
-			}
-
-			// Checking err to suppress the linter
-			if err = ctx.Response().String(http.StatusServiceUnavailable, content).Abort(); err != nil {
-				return
-			}
+		if !file.Exists(filepath) {
+			ctx.Request().Next()
+		}
+
+		content, err := os.ReadFile(filepath)
+
+		var maintenanceOptions *console.MaintenanceOptions
+		err = json.Unmarshal(content, &maintenanceOptions)
+
+		if err != nil {
+			ctx.Request().Abort(http.StatusServiceUnavailable)
+			return
+		}
+
+		secret := ctx.Request().Query("secret", "")
+		if secret != "" && maintenanceOptions.Secret != "" && secret == maintenanceOptions.Secret {
+			ctx.Request().Next()
 			return
 		}
 
-		ctx.Request().Next()
+		if maintenanceOptions.Redirect != "" {
+			ctx.Response().Redirect(http.StatusTemporaryRedirect, maintenanceOptions.Redirect)
+			return
+		}
+
+		if maintenanceOptions.Render != "" {
+			ctx.Response().View().Make(maintenanceOptions.Render, nil).Render()
+			return
+		}
+
+		// Checking err to suppress the linter
+		if err = ctx.Response().String(maintenanceOptions.Status, maintenanceOptions.Reason).Abort(); err != nil {
+			return
+		}
 	}
 }