1.4.5.2

EdgeUser/internal/web/actions/actionutils/portal_action.go

@@ -0,0 +1,88 @@
+// Copyright 2024 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package actionutils
+
+import (
+	"bytes"
+	"github.com/TeaOSLab/EdgeUser/internal/configloaders"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/maps"
+	"github.com/iwind/TeaGo/types"
+	"net/http"
+	"regexp"
+)
+
+var faviconReg = regexp.MustCompile(`<link\s+rel="icon"[^>]+>`)
+
+type PortalAction struct {
+	ParentAction
+}
+
+func (this *PortalAction) IsFrontRequest() bool {
+	return len(this.Request.URL.Query().Get("X_FROM_FRONT")) > 0
+}
+
+func (this *PortalAction) Success() {
+	if this.IsFrontRequest() {
+		this.prepareCORSHeader()
+	}
+	this.ActionObject.Success()
+}
+
+func (this *PortalAction) Show() {
+	if this.IsFrontRequest() {
+		this.prepareCORSHeader()
+		this.ActionObject.Success()
+		return
+	}
+
+	// add header
+	this.ViewFunc("X_VIEW_DATA", func() string {
+		if this.Data == nil {
+			return ""
+		}
+		return string(this.Data.AsJSON())
+	})
+	this.ViewFunc("X_VIEW_FAVICON", func() string {
+		uiConfig, err := configloaders.LoadUIConfig()
+		if err != nil || uiConfig == nil {
+			return ""
+		}
+		if uiConfig.FaviconFileId > 0 {
+			return `<link rel="icon" href="/ui/image/` + types.String(uiConfig.FaviconFileId) + `"/>`
+		}
+		return ""
+	})
+
+	this.TemplateFilter(func(body []byte) []byte {
+		// head
+		body = bytes.ReplaceAll(body, []byte("</head>"), []byte(`
+<script>
+window.X_FROM_SERVER = true;
+window.X_VIEW_DATA = {$ X_VIEW_DATA };
+</script>
+</head>`))
+
+		// favicon
+		body = faviconReg.ReplaceAll(body, []byte(`{$ X_VIEW_FAVICON }`))
+
+		return body
+	})
+
+	this.ViewDir(Tea.Root)
+	this.ActionObject.Show()
+}
+
+func (this *PortalAction) Redirect(url string, status int) {
+	if this.IsFrontRequest() {
+		this.Data["X_FRONT_REDIRECT"] = maps.Map{"url": url}
+		this.Success()
+		return
+	}
+	http.Redirect(this.ResponseWriter, this.Request, url, status)
+}
+
+func (this *PortalAction) prepareCORSHeader() {
+	this.ResponseWriter.Header().Set("Access-Control-Allow-Origin", this.Request.Header.Get("Origin"))
+	this.ResponseWriter.Header().Set("Access-Control-Allow-Credentials", "true")
+}