From d3e86624f22e10dc30f6109c34f7221b40a63fa1 Mon Sep 17 00:00:00 2001
From: liuzhihang1 <1625167628@qq.com>
Date: Wed, 19 Feb 2025 16:34:10 +0800
Subject: [PATCH] add demo model

---
 internal/app/middle/demo.go | 23 +++++++++++++++++++++++
 internal/app/route/route.go |  1 +
 2 files changed, 24 insertions(+)
 create mode 100644 internal/app/middle/demo.go

diff --git a/internal/app/middle/demo.go b/internal/app/middle/demo.go
new file mode 100644
index 0000000..af01a12
--- /dev/null
+++ b/internal/app/middle/demo.go
@@ -0,0 +1,23 @@
+package middle
+
+import (
+	"net/http"
+	"slices"
+
+	"github.com/gin-gonic/gin"
+)
+
+// 演示模式
+func DemoMiddle() func(c *gin.Context) {
+	return func(ctx *gin.Context) {
+		whiteListUri := []string{
+			"/api/user/login",
+			"/api/log",
+		}
+		if ctx.Request.Method == http.MethodGet || slices.Contains(whiteListUri, ctx.Request.URL.String()) {
+			ctx.Next()
+		} else {
+			ctx.Abort()
+		}
+	}
+}
diff --git a/internal/app/route/route.go b/internal/app/route/route.go
index 188066b..ace1c67 100644
--- a/internal/app/route/route.go
+++ b/internal/app/route/route.go
@@ -54,6 +54,7 @@ func routePathInit(r *gin.Engine) {
 	apiGroup := r.Group("/api")
 	apiGroup.Use(middle.CheckToken())
 	apiGroup.Use(middle.PanicMiddle())
+	// apiGroup.Use(middle.DemoMiddle())
 	{
 		apiGroup.GET("/ws", middle.OprPermission(constants.OPERATION_TERMINAL), api.WsApi.WebsocketHandle)