3x-ui/sub/subController.go

package sub

import (
	"encoding/base64"
	"strings"

	"github.com/gin-gonic/gin"
)

type SUBController struct {
	subService SubService
}

func NewSUBController(g *gin.RouterGroup) *SUBController {
	a := &SUBController{}
	a.initRouter(g)
	return a
}

func (a *SUBController) initRouter(g *gin.RouterGroup) {
	g = g.Group("/")

	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)))
	}
}