sub2clash/server/handler/convert.go

package handler

import (
	_ "embed"
	"net/http"

	"github.com/bestnite/sub2clash/common"
	"github.com/bestnite/sub2clash/config"
	"github.com/bestnite/sub2clash/model"
	M "github.com/bestnite/sub2clash/model"

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

func ConvertHandler() func(c *gin.Context) {
	return func(c *gin.Context) {
		query, err := M.ParseConvertQuery(c)
		if err != nil {
			c.String(http.StatusBadRequest, err.Error())
			return
		}
		template := ""
		switch query.ClashType {
		case model.Clash:
			template = config.GlobalConfig.ClashTemplate
		case model.ClashMeta:
			template = config.GlobalConfig.MetaTemplate
		}
		sub, err := common.BuildSub(query.ClashType, query, template, config.GlobalConfig.CacheExpire, config.GlobalConfig.RequestRetryTimes)
		if err != nil {
			c.String(http.StatusInternalServerError, err.Error())
			return
		}

		if len(query.Subs) == 1 {
			userInfoHeader, err := common.FetchSubscriptionUserInfo(query.Subs[0], "clash", config.GlobalConfig.RequestRetryTimes)
			if err == nil {
				c.Header("subscription-userinfo", userInfoHeader)
			}
		}

		if query.NodeListMode {
			marshal, err := sub.MarshalNodeListYAML()
			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))
	}
}