sub2clash/api/handler/meta.go

package handler

import (
	_ "embed"
	"net/http"

	"github.com/nitezs/sub2clash/config"
	"github.com/nitezs/sub2clash/model"
	"github.com/nitezs/sub2clash/validator"

	"github.com/gin-gonic/gin"
	"gopkg.in/yaml.v3"
)

func SubHandler(c *gin.Context) {

	query, err := validator.ParseQuery(c)
	if err != nil {
		c.String(http.StatusBadRequest, err.Error())
		return
	}
	sub, err := BuildSub(model.ClashMeta, query, config.Default.MetaTemplate)
	if err != nil {
		c.String(http.StatusInternalServerError, err.Error())
		return
	}

	if len(query.Subs) == 1 {
		userInfoHeader, err := fetchSubscriptionUserInfo(query.Subs[0], "clash")
		if err != nil {
			c.String(http.StatusInternalServerError, err.Error())
		}
		c.Header("subscription-userinfo", userInfoHeader)
	}

	if query.NodeListMode {
		nodelist := model.NodeList{}
		nodelist.Proxies = sub.Proxies
		marshal, err := yaml.Marshal(nodelist)
		if err != nil {
			c.String(http.StatusInternalServerError, "YAML序列化失败: "+err.Error())
			return
		}
		c.String(http.StatusOK, string(marshal))
		return
	}
	marshal, err := yaml.Marshal(sub)
	if err != nil {
		c.String(http.StatusInternalServerError, "YAML序列化失败: "+err.Error())
		return
	}
	c.String(http.StatusOK, string(marshal))
}