Client/server
5
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat(middleware): enhance user agent handling by querying client list in PanDomainMiddleware

Browse Source
This commit is contained in:
Chang lue Tsen 2025-09-05 04:28:05 -04:00
parent f3c73a5224
commit b9ef8f4572
2 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
internal/handler/subscribe.go
View File

@ -8,6 +8,7 @@ import (
"github.com/perfect-panel/server/internal/logic/subscribe" "github.com/perfect-panel/server/internal/logic/subscribe"
"github.com/perfect-panel/server/internal/svc" "github.com/perfect-panel/server/internal/svc"
"github.com/perfect-panel/server/internal/types" "github.com/perfect-panel/server/internal/types"
"github.com/perfect-panel/server/pkg/logger"
"github.com/perfect-panel/server/pkg/tool" "github.com/perfect-panel/server/pkg/tool"
) )
@ -30,6 +31,18 @@ func SubscribeHandler(svcCtx *svc.ServiceContext) func(c *gin.Context) {
return return
} }
clientUserAgents := tool.RemoveDuplicateElements(strings.Split(svcCtx.Config.Subscribe.UserAgentList, "\n")...) clientUserAgents := tool.RemoveDuplicateElements(strings.Split(svcCtx.Config.Subscribe.UserAgentList, "\n")...)
// query client list
clients, err := svcCtx.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, " ")
clientUserAgents = append(clientUserAgents, u)
}
var allow = false var allow = false
for _, keyword := range clientUserAgents { for _, keyword := range clientUserAgents {
keyword = strings.Trim(keyword, " ") keyword = strings.Trim(keyword, " ")

13
internal/middleware/panDomainMiddleware.go
View File

@ -8,6 +8,7 @@ import (
"github.com/perfect-panel/server/internal/logic/subscribe" "github.com/perfect-panel/server/internal/logic/subscribe"
"github.com/perfect-panel/server/internal/svc" "github.com/perfect-panel/server/internal/svc"
"github.com/perfect-panel/server/internal/types" "github.com/perfect-panel/server/internal/types"
"github.com/perfect-panel/server/pkg/logger"
"github.com/perfect-panel/server/pkg/tool" "github.com/perfect-panel/server/pkg/tool"
) )
@ -26,6 +27,18 @@ func PanDomainMiddleware(svc *svc.ServiceContext) func(c *gin.Context) {
} }
browserKeywords := tool.RemoveDuplicateElements(strings.Split(svc.Config.Subscribe.UserAgentList, "\n")...) browserKeywords := tool.RemoveDuplicateElements(strings.Split(svc.Config.Subscribe.UserAgentList, "\n")...)
var allow = false 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 { for _, keyword := range browserKeywords {
keyword = strings.ToLower(strings.Trim(keyword, " ")) keyword = strings.ToLower(strings.Trim(keyword, " "))
if keyword == "" { if keyword == "" {