56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package middleware
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/perfect-panel/server/internal/logic/subscribe"
|
|
"github.com/perfect-panel/server/internal/svc"
|
|
"github.com/perfect-panel/server/internal/types"
|
|
)
|
|
|
|
func PanDomainMiddleware(svc *svc.ServiceContext) func(c *gin.Context) {
|
|
return func(c *gin.Context) {
|
|
|
|
if svc.Config.Subscribe.PanDomain && c.Request.URL.Path == "/" {
|
|
|
|
// intercept browser
|
|
ua := c.GetHeader("User-Agent")
|
|
if ua == "" {
|
|
c.String(http.StatusForbidden, "Access denied")
|
|
c.Abort()
|
|
return
|
|
}
|
|
browserKeywords := []string{"chrome", "firefox", "safari", "edge", "opera", "micromessenger"}
|
|
for _, keyword := range browserKeywords {
|
|
lcUA := strings.ToLower(ua)
|
|
if strings.Contains(lcUA, keyword) {
|
|
c.String(http.StatusForbidden, "Access denied")
|
|
c.Abort()
|
|
return
|
|
}
|
|
}
|
|
|
|
domain := c.Request.Host
|
|
domainArr := strings.Split(domain, ".")
|
|
domainFirst := domainArr[0]
|
|
request := types.SubscribeRequest{
|
|
Token: domainFirst,
|
|
Flag: domainArr[1],
|
|
UA: c.Request.Header.Get("User-Agent"),
|
|
}
|
|
l := subscribe.NewSubscribeLogic(c, svc)
|
|
resp, err := l.Handler(&request)
|
|
if err != nil {
|
|
return
|
|
}
|
|
c.Header("subscription-userinfo", resp.Header)
|
|
c.String(200, "%s", string(resp.Config))
|
|
c.Abort()
|
|
return
|
|
}
|
|
c.Next()
|
|
}
|
|
}
|