🚧 remove customize head and body for manage

DerekTso · Jan 8, 2022 · 925f386 · 925f386
1 parent 0fbbd54
commit 925f386
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/conf/var.go b/conf/var.go
@@ -83,6 +83,7 @@ var (
 
 var (
 	RawIndexHtml string
+	ManageHtml   string
 	IndexHtml    string
 	Token        string
 

diff --git a/model/setting.go b/model/setting.go
@@ -97,15 +97,15 @@ func LoadSettings() {
 	favicon, err := GetSettingByKey("favicon")
 	if err == nil {
 		//conf.Favicon = favicon.Value
-		conf.IndexHtml = strings.Replace(conf.RawIndexHtml, "https://store.heytapimage.com/cdo-portal/feedback/202110/30/d43c41c5d257c9bc36366e310374fb19.png", favicon.Value, 1)
+		conf.ManageHtml = strings.Replace(conf.RawIndexHtml, "https://store.heytapimage.com/cdo-portal/feedback/202110/30/d43c41c5d257c9bc36366e310374fb19.png", favicon.Value, 1)
 	}
 	title, err := GetSettingByKey("title")
 	if err == nil {
-		conf.IndexHtml = strings.Replace(conf.IndexHtml, "Loading...", title.Value, 1)
+		conf.ManageHtml = strings.Replace(conf.ManageHtml, "Loading...", title.Value, 1)
 	}
 	customizeHead, err := GetSettingByKey("customize head")
 	if err == nil {
-		conf.IndexHtml = strings.Replace(conf.IndexHtml, "<!-- customize head -->", customizeHead.Value, 1)
+		conf.IndexHtml = strings.Replace(conf.ManageHtml, "<!-- customize head -->", customizeHead.Value, 1)
 	}
 	customizeBody, err := GetSettingByKey("customize body")
 	if err == nil {

diff --git a/server/static.go b/server/static.go
@@ -8,6 +8,7 @@ import (
 	"io/fs"
 	"io/ioutil"
 	"net/http"
+	"strings"
 )
 
 func InitIndex() {
@@ -41,7 +42,11 @@ func Static(r *gin.Engine) {
 	r.NoRoute(func(c *gin.Context) {
 		c.Status(200)
 		c.Header("Content-Type", "text/html")
-		_, _ = c.Writer.WriteString(conf.IndexHtml)
+		if strings.HasPrefix(c.Request.URL.Path, "/@manage") {
+			_, _ = c.Writer.WriteString(conf.ManageHtml)
+		} else {
+			_, _ = c.Writer.WriteString(conf.IndexHtml)
+		}
 		c.Writer.Flush()
 		c.Writer.WriteHeaderNow()
 	})