shanshanzhong
c582087c0f
Some checks failed
Build docker and publish / build (20.15.1) (push) Failing after 6m27s
feat: 添加版本和构建时间变量 fix: 修正短信队列类型注释错误 style: 清理未使用的代码和测试文件 docs: 更新安装文档中的下载链接 chore: 迁移数据库脚本添加日志和订阅配置
79 lines
2.0 KiB
Go
79 lines
2.0 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"
|
|
"github.com/perfect-panel/server/pkg/logger"
|
|
"github.com/perfect-panel/server/pkg/tool"
|
|
)
|
|
|
|
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 svc.Config.Subscribe.UserAgentLimit {
|
|
if ua == "" {
|
|
c.String(http.StatusForbidden, "Access denied")
|
|
c.Abort()
|
|
return
|
|
}
|
|
browserKeywords := tool.RemoveDuplicateElements(strings.Split(svc.Config.Subscribe.UserAgentList, "\n")...)
|
|
var allow = false
|
|
|
|
// query client list
|
|
clients, err := svc.ClientModel.List(c.Request.Context())
|
|
if err != nil {
|
|
logger.Errorw("[PanDomainMiddleware] Query client list failed", logger.Field("error", err.Error()))
|
|
}
|
|
for _, item := range clients {
|
|
u := strings.ToLower(item.UserAgent)
|
|
u = strings.Trim(u, " ")
|
|
browserKeywords = append(browserKeywords, u)
|
|
}
|
|
|
|
for _, keyword := range browserKeywords {
|
|
keyword = strings.ToLower(strings.Trim(keyword, " "))
|
|
if keyword == "" {
|
|
continue
|
|
}
|
|
if strings.Contains(strings.ToLower(ua), keyword) {
|
|
allow = true
|
|
}
|
|
}
|
|
if !allow {
|
|
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()
|
|
}
|
|
}
|