mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2025-03-01 01:20:49 +03:00
49 lines
999 B
Go
49 lines
999 B
Go
package controller
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"strings"
|
|
"x-ui/web/service"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type SUBController struct {
|
|
BaseController
|
|
|
|
subService service.SubService
|
|
}
|
|
|
|
func NewSUBController(g *gin.RouterGroup) *SUBController {
|
|
a := &SUBController{}
|
|
a.initRouter(g)
|
|
return a
|
|
}
|
|
|
|
func (a *SUBController) initRouter(g *gin.RouterGroup) {
|
|
g = g.Group("/sub")
|
|
|
|
g.GET("/:subid", a.subs)
|
|
}
|
|
|
|
func (a *SUBController) subs(c *gin.Context) {
|
|
subId := c.Param("subid")
|
|
host := strings.Split(c.Request.Host, ":")[0]
|
|
subs, headers, err := a.subService.GetSubs(subId, host)
|
|
if err != nil || len(subs) == 0 {
|
|
c.String(400, "Error!")
|
|
} else {
|
|
result := ""
|
|
for _, sub := range subs {
|
|
result += sub + "\n"
|
|
}
|
|
|
|
// Add headers
|
|
c.Writer.Header().Set("Subscription-Userinfo", headers[0])
|
|
c.Writer.Header().Set("Profile-Update-Interval", headers[1])
|
|
c.Writer.Header().Set("Profile-Title", headers[2])
|
|
|
|
c.String(200, base64.StdEncoding.EncodeToString([]byte(result)))
|
|
}
|
|
}
|