Update middleware.go

server/router/middleware.go

@@ -1,84 +1,66 @@
 package router
 
 import (
-    "encoding/base64"
-    "github.com/0xJacky/Nginx-UI/frontend"
-    "github.com/0xJacky/Nginx-UI/server/model"
-    "github.com/gin-contrib/static"
-    "github.com/gin-gonic/gin"
-    "io/fs"
-    "log"
-    "net/http"
-    "path"
-    "strings"
+	"encoding/base64"
+	"github.com/0xJacky/Nginx-UI/frontend"
+	"github.com/0xJacky/Nginx-UI/server/model"
+	"github.com/gin-contrib/static"
+	"github.com/gin-gonic/gin"
+	"io/fs"
+	"log"
+	"net/http"
+	"path"
 )
 
 func authRequired() gin.HandlerFunc {
-    return func(c *gin.Context) {
-        token := c.GetHeader("Authorization")
-        if token == "" {
-            tmp, _ := base64.StdEncoding.DecodeString(c.Query("token"))
-            token = string(tmp)
-            if token == "" {
-                c.JSON(http.StatusForbidden, gin.H{
-                    "message": "auth fail",
-                })
-                c.Abort()
-                return
-            }
-        }
+	return func(c *gin.Context) {
+		token := c.GetHeader("Authorization")
+		if token == "" {
+			tmp, _ := base64.StdEncoding.DecodeString(c.Query("token"))
+			token = string(tmp)
+			if token == "" {
+				c.JSON(http.StatusForbidden, gin.H{
+					"message": "auth fail",
+				})
+				c.Abort()
+				return
+			}
+		}
 
-        n := model.CheckToken(token)
+		n := model.CheckToken(token)
 
-        if n < 1 {
-            c.JSON(http.StatusForbidden, gin.H{
-                "message": "auth fail",
-            })
-            c.Abort()
-            return
-        }
-        c.Next()
-    }
+		if n < 1 {
+			c.JSON(http.StatusForbidden, gin.H{
+				"message": "auth fail",
+			})
+			c.Abort()
+			return
+		}
+		c.Next()
+	}
 }
 
 type serverFileSystemType struct {
-    http.FileSystem
+	http.FileSystem
 }
 
 func (f serverFileSystemType) Exists(prefix string, _path string) bool {
-    _, err := f.Open(path.Join(prefix, _path))
-    return err == nil
+	_, err := f.Open(path.Join(prefix, _path))
+	return err == nil
 }
 
 func mustFS(dir string) (serverFileSystem static.ServeFileSystem) {
 
-    sub, err := fs.Sub(frontend.DistFS, path.Join("dist", dir))
+	sub, err := fs.Sub(frontend.DistFS, path.Join("dist", dir))
 
-    if err != nil {
-        log.Println(err)
-        return
-    }
+	if err != nil {
+		log.Println(err)
+		return
+	}
 
-    serverFileSystem = serverFileSystemType{
-        http.FS(sub),
-    }
+	serverFileSystem = serverFileSystemType{
+		http.FS(sub),
+	}
 
-    return
-}
-
-// tryStatic Static returns a middleware handler that serves static files in the given directory.
-func tryStatic(urlPrefix string, fs static.ServeFileSystem) gin.HandlerFunc {
-    fileserver := http.FileServer(fs)
-    if urlPrefix != "" {
-        fileserver = http.StripPrefix(urlPrefix, fileserver)
-    }
-    return func(c *gin.Context) {
-        if fs.Exists(urlPrefix, c.Request.URL.Path) {
-            fileserver.ServeHTTP(c.Writer, c.Request)
-            if strings.Contains(c.Request.URL.Path, ".js") {
-                c.Writer.Header().Set("content-type", "application/javascript")
-            }
-            c.Abort()
-        }
-    }
+	return
 }