Merge pull request #63 from pingchieh/main

添加Anytls协议

README.md

@@ -18,6 +18,7 @@
   - Hysteria （Clash.Meta）
   - Hysteria2 （Clash.Meta）
   - Socks5
+  - Anytls （Clash.Meta）
 
 ## 使用
 

api/handler/default.go

@@ -233,6 +233,21 @@ func BuildSub(clashType model.ClashType, query validator.SubValidator, template
 	return temp, nil
 }
 
+func fetchSubscriptionUserInfo(url string, userAgent string) (string, error) {
+	resp, err := common.Head(url, common.WithUserAgent(userAgent))
+	if err != nil {
+		logger.Logger.Debug("创建 HEAD 请求失败", zap.Error(err))
+		return "", err
+	}
+	defer resp.Body.Close()
+	if userInfo := resp.Header.Get("subscription-userinfo"); userInfo != "" {
+		return userInfo, nil
+	}
+
+	logger.Logger.Debug("目标 URL 未返回 subscription-userinfo 头", zap.Error(err))
+	return "", err
+}
+
 func MergeSubAndTemplate(temp *model.Subscription, sub *model.Subscription, igcg bool) {
 
 	var countryGroupNames []string

api/handler/meta.go

@@ -25,6 +25,14 @@ func SubHandler(c *gin.Context) {
 		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

api/handler/short_link.go

@@ -32,16 +32,41 @@ func GenerateLinkHandler(c *gin.Context) {
 		return
 	}
 
-	hash, err := generateUniqueHash()
+	var hash string
+	var password string
+	var err error
+	
+	if params.CustomID != "" {
+		// 检查自定义ID是否已存在
+		exists, err := database.CheckShortLinkHashExists(params.CustomID)
+		if err != nil {
+			respondWithError(c, http.StatusInternalServerError, "数据库错误")
+			return
+		}
+		if exists {
+			respondWithError(c, http.StatusBadRequest, "短链已存在")
+			return
+		}
+		hash = params.CustomID
+		password = params.Password
+	} else {
+		// 自动生成短链ID和密码
+		hash, err = generateUniqueHash()
 		if err != nil {
 			respondWithError(c, http.StatusInternalServerError, "生成短链接失败")
 			return
 		}
+		if params.Password == "" {
+			password = common.RandomString(8) // 生成8位随机密码
+		} else {
+			password = params.Password
+		}
+	}
 
 	shortLink := model.ShortLink{
 		Hash:     hash,
 		Url:      params.Url,
-		Password: params.Password,
+		Password: password,
 	}
 
 	if err := database.SaveShortLink(&shortLink); err != nil {
@@ -49,10 +74,12 @@ func GenerateLinkHandler(c *gin.Context) {
 		return
 	}
 
-	if params.Password != "" {
+	// 返回生成的短链ID和密码
-		hash += "?password=" + params.Password
+	response := map[string]string{
+		"hash": hash,
+		"password": password,
 	}
-	c.String(http.StatusOK, hash)
+	c.JSON(http.StatusOK, response)
 }
 
 func generateUniqueHash() (string, error) {
@@ -74,11 +101,27 @@ func UpdateLinkHandler(c *gin.Context) {
 		respondWithError(c, http.StatusBadRequest, "参数错误: "+err.Error())
 		return
 	}
+
+	// 先获取原有的短链接
+	existingLink, err := database.FindShortLinkByHash(params.Hash)
+	if err != nil {
+		respondWithError(c, http.StatusNotFound, "未找到短链接")
+		return
+	}
+
+	// 验证密码
+	if existingLink.Password != params.Password {
+		respondWithError(c, http.StatusUnauthorized, "密码错误")
+		return
+	}
+
+	// 更新URL，但保持原密码不变
 	shortLink := model.ShortLink{
 		Hash:     params.Hash,
 		Url:      params.Url,
-		Password: params.Password,
+		Password: existingLink.Password, // 保持原密码不变
 	}
+
 	if err := database.SaveShortLink(&shortLink); err != nil {
 		respondWithError(c, http.StatusInternalServerError, "数据库错误")
 		return

api/static/index.html

@@ -79,7 +79,7 @@
         <div class="form-group mb-3">
             <label for="user-agent">ua标识:</label>
             <textarea class="form-control" id="user-agent" name="user-agent"
-                placeholder="用于获取订阅的http请求中的user-agent标识(可选)" rows="1"></textarea>
+                placeholder="用于获取订阅的http请求中的user-agent标识(可选)" rows="3"></textarea>
         </div>
         <!-- Refresh -->
         <div class="form-check mb-3">
@@ -147,24 +147,27 @@
         <div class="form-group mb-5">
             <label for="apiLink">配置链接:</label>
             <div class="input-group mb-2">
-                <input class="form-control" id="apiLink" type="text" placeholder="链接" />
+                <input class="form-control bg-light" id="apiLink" type="text" placeholder="链接" readonly style="cursor: not-allowed;" />
                 <button class="btn btn-primary" onclick="copyToClipboard('apiLink',this)" type="button">
                     复制链接
                 </button>
             </div>
-            <div class="input-group">
+            <div class="input-group mb-2">
-                <input class="form-control" id="apiShortLink" type="text" placeholder="链接" />
+                <input class="form-control" id="customId" type="text" placeholder="短链ID（可选）" />
-                <input class="form-control" id="password" type="text" placeholder="密码" />
+                <input class="form-control" id="password" type="text" placeholder="密码（可选）" />
                 <button class="btn btn-primary" onclick="generateShortLink()" type="button">
                     生成短链
                 </button>
-                <button class="btn btn-primary" onclick="updateShortLink()" type="button">
-                    更新短链
-                </button>
                 <button class="btn btn-primary" onclick="copyToClipboard('apiShortLink',this)" type="button">
                     复制短链
                 </button>
             </div>
+            <div class="input-group">
+                <input class="form-control bg-light" id="apiShortLink" type="text" placeholder="短链接" readonly style="cursor: not-allowed;" />
+                <button class="btn btn-primary" onclick="updateShortLink()" type="button">
+                    更新短链
+                </button>
+            </div>
         </div>
 
         <!-- footer-->

api/static/index.js

@@ -1,3 +1,15 @@
+function setInputReadOnly(input, readonly) {
+  if (readonly) {
+    input.readOnly = true;
+    input.classList.add('bg-light');
+    input.style.cursor = 'not-allowed';
+  } else {
+    input.readOnly = false;
+    input.classList.remove('bg-light');
+    input.style.cursor = 'auto';
+  }
+}
+
 function clearExistingValues() {
   // 清除简单输入框和复选框的值
   document.getElementById("endpoint").value = "clash";
@@ -12,7 +24,23 @@ function clearExistingValues() {
   document.getElementById("remove").value = "";
   document.getElementById("apiLink").value = "";
   document.getElementById("apiShortLink").value = "";
-  document.getElementById("password").value = "";
+  
+  // 恢复短链ID和密码输入框状态
+  const customIdInput = document.getElementById("customId");
+  const passwordInput = document.getElementById("password");
+  const generateButton = document.querySelector('button[onclick="generateShortLink()"]');
+  
+  customIdInput.value = "";
+  setInputReadOnly(customIdInput, false);
+  
+  passwordInput.value = "";
+  setInputReadOnly(passwordInput, false);
+  
+  // 恢复生成短链按钮状态
+  generateButton.disabled = false;
+  generateButton.classList.remove('btn-secondary');
+  generateButton.classList.add('btn-primary');
+  
   document.getElementById("nodeList").checked = false;
 
   // 清除由 createRuleProvider, createReplace, 和 createRule 创建的所有额外输入组
@@ -188,8 +216,32 @@ async function parseInputURL() {
     try {
       const response = await axios.get("./short?" + q.toString());
       url = new URL(window.location.href + response.data);
-      document.querySelector("#apiShortLink").value = inputURL;
+      
-      document.querySelector("#password").value = password;
+      // 回显配置链接
+      const apiLinkInput = document.querySelector("#apiLink");
+      apiLinkInput.value = `${window.location.origin}${window.location.pathname}${response.data}`;
+      setInputReadOnly(apiLinkInput, true);
+      
+      // 回显短链相关信息
+      const apiShortLinkInput = document.querySelector("#apiShortLink");
+      apiShortLinkInput.value = inputURL;
+      setInputReadOnly(apiShortLinkInput, true);
+      
+      // 设置短链ID和密码，并设置为只读
+      const customIdInput = document.querySelector("#customId");
+      const passwordInput = document.querySelector("#password");
+      const generateButton = document.querySelector('button[onclick="generateShortLink()"]');
+      
+      customIdInput.value = hash;
+      setInputReadOnly(customIdInput, true);
+      
+      passwordInput.value = password;
+      setInputReadOnly(passwordInput, true);
+      
+      // 禁用生成短链按钮
+      generateButton.disabled = true;
+      generateButton.classList.add('btn-secondary');
+      generateButton.classList.remove('btn-primary');
     } catch (error) {
       console.log(error);
       alert("获取短链失败，请检查密码！");
@@ -255,6 +307,17 @@ async function parseInputURL() {
     );
   }
 
+  if (params.has("userAgent")) {
+    document.getElementById("user-agent").value = decodeURIComponent(
+      params.get("userAgent")
+    );
+  }
+
+  if (params.has("ignoreCountryGroup")) {
+    document.getElementById("igcg").checked = 
+      params.get("ignoreCountryGroup") === "true";
+  }
+
   if (params.has("replace")) {
     parseAndFillReplaceParams(decodeURIComponent(params.get("replace")));
   }
@@ -428,21 +491,25 @@ function generateURL() {
     return;
   }
   apiLink.value = `${window.location.origin}${window.location.pathname}${uri}`;
+  setInputReadOnly(apiLink, true);
 }
 
 function generateShortLink() {
   const apiShortLink = document.getElementById("apiShortLink");
   const password = document.getElementById("password");
+  const customId = document.getElementById("customId");
   let uri = generateURI();
   if (uri === "") {
     return;
   }
+
   axios
     .post(
       "./short",
       {
         url: uri,
         password: password.value.trim(),
+        customId: customId.value.trim()
       },
       {
         headers: {
@@ -451,11 +518,20 @@ function generateShortLink() {
       }
     )
     .then((response) => {
-      apiShortLink.value = `${window.location.origin}${window.location.pathname}s/${response.data}`;
+      // 设置返回的短链ID和密码
+      customId.value = response.data.hash;
+      password.value = response.data.password;
+      // 生成完整的短链接
+      const shortLink = `${window.location.origin}${window.location.pathname}s/${response.data.hash}?password=${response.data.password}`;
+      apiShortLink.value = shortLink;
     })
     .catch((error) => {
       console.log(error);
+      if (error.response && error.response.data) {
+        alert(error.response.data);
+      } else {
         alert("生成短链失败，请重试！");
+      }
     });
 }
 
@@ -468,7 +544,7 @@ function updateShortLink() {
     hash = u.pathname.substring(u.pathname.lastIndexOf("/s/") + 3);
   }
   if (password.value.trim() === "") {
-    alert("请输入密码！");
+    alert("请输入原密码进行验证！");
     return;
   }
   let uri = generateURI();
@@ -490,11 +566,17 @@ function updateShortLink() {
       }
     )
     .then((response) => {
-      alert("更新短链成功！");
+      alert(`短链 ${hash} 更新成功！`);
     })
     .catch((error) => {
       console.log(error);
+      if (error.response && error.response.status === 401) {
+        alert("密码错误，请输入正确的原密码！");
+      } else if (error.response && error.response.data) {
         alert(error.response.data);
+      } else {
+        alert("更新短链失败，请重试！");
+      }
     });
 }
 

common/get.go

@@ -2,6 +2,7 @@ package common
 
 import (
 	"errors"
+	"fmt"
 	"net/http"
 	"time"
 
@@ -28,10 +29,12 @@ func Get(url string, options ...GetOption) (resp *http.Response, err error) {
 	for _, option := range options {
 		option(&getConfig)
 	}
+	var req *http.Request
+	var get *http.Response
 	for haveTried < retryTimes {
 		client := &http.Client{}
 		//client.Timeout = time.Second * 10
-		req, err := http.NewRequest("GET", url, nil)
+		req, err = http.NewRequest("GET", url, nil)
 		if err != nil {
 			haveTried++
 			time.Sleep(retryDelay)
@@ -40,13 +43,12 @@ func Get(url string, options ...GetOption) (resp *http.Response, err error) {
 		if getConfig.userAgent != "" {
 			req.Header.Set("User-Agent", getConfig.userAgent)
 		}
-		get, err := client.Do(req)
+		get, err = client.Do(req)
 		if err != nil {
 			haveTried++
 			time.Sleep(retryDelay)
 			continue
 		} else {
-
 			if get != nil && get.ContentLength > config.Default.RequestMaxFileSize {
 				return nil, errors.New("文件过大")
 			}
@@ -54,5 +56,47 @@ func Get(url string, options ...GetOption) (resp *http.Response, err error) {
 		}
 
 	}
-	return nil, err
+	return nil, fmt.Errorf("请求失败：%v", err)
+}
+
+func Head(url string, options ...GetOption) (resp *http.Response, err error) {
+	retryTimes := config.Default.RequestRetryTimes
+	haveTried := 0
+	retryDelay := time.Second
+
+	// 解析可选参数（如 User-Agent）
+	getConfig := GetConfig{}
+	for _, option := range options {
+		option(&getConfig)
+	}
+
+	var req *http.Request
+	var headResp *http.Response
+
+	for haveTried < retryTimes {
+		client := &http.Client{}
+		req, err = http.NewRequest("HEAD", url, nil)
+		if err != nil {
+			haveTried++
+			time.Sleep(retryDelay)
+			continue
+		}
+
+		// 设置 User-Agent（如果提供）
+		if getConfig.userAgent != "" {
+			req.Header.Set("User-Agent", getConfig.userAgent)
+		}
+
+		headResp, err = client.Do(req)
+		if err != nil {
+			haveTried++
+			time.Sleep(retryDelay)
+			continue
+		}
+
+		// HEAD 请求不检查 ContentLength，因为没有响应体
+		return headResp, nil
+	}
+
+	return nil, fmt.Errorf("HEAD 请求失败：%v", err)
 }

common/proxy.go

@@ -130,6 +130,9 @@ func ParseProxy(proxies ...string) []model.Proxy {
 			if strings.HasPrefix(proxy, constant.SocksPrefix) {
 				proxyItem, err = parser.ParseSocks(proxy)
 			}
+			if strings.HasPrefix(proxy, constant.AnytlsPrefix) {
+				proxyItem, err = parser.ParseAnytls(proxy)
+			}
 			if err == nil {
 				result = append(result, proxyItem)
 			} else {

constant/prefix.go

@@ -10,4 +10,5 @@ const (
 	VLESSPrefix        string = "vless://"
 	VMessPrefix        string = "vmess://"
 	SocksPrefix        string = "socks"
+	AnytlsPrefix       string = "anytls://"
 )

model/clash.go

@@ -27,6 +27,7 @@ func GetSupportProxyTypes(clashType ClashType) map[string]bool {
 			"hysteria":  true,
 			"hysteria2": true,
 			"socks5":    true,
+			"anytls":    true,
 		}
 	}
 	return nil

model/proxy.go

@@ -69,6 +69,9 @@ type Proxy struct {
 	Peers                    []WireGuardPeerOption `yaml:"peers,omitempty"`
 	RemoteDnsResolve         bool                  `yaml:"remote-dns-resolve,omitempty"`
 	Dns                      []string              `yaml:"dns,omitempty"`
+	IdleSessionCheckInterval int                   `yaml:"idle-session-check-interval,omitempty"`
+	IdleSessionTimeout       int                   `yaml:"idle-session-timeout,omitempty"`
+	MinIdleSession           int                   `yaml:"min-idle-session,omitempty"`
 }
 
 type WireGuardPeerOption struct {
@@ -98,6 +101,8 @@ func (p Proxy) MarshalYAML() (interface{}, error) {
 		return ProxyToHysteria(p), nil
 	case "hysteria2":
 		return ProxyToHysteria2(p), nil
+	case "anytls":
+		return ProxyToAnytls(p), nil
 	default:
 		return _Proxy(p), nil
 	}

model/proxy_anytls.go

@@ -0,0 +1,37 @@
+package model
+
+type Anytls struct {
+	Type                     string   `yaml:"type"`
+	Name                     string   `yaml:"name"`
+	Server                   string   `yaml:"server"`
+	Port                     int      `yaml:"port"`
+	Password                 string   `yaml:"password,omitempty"`
+	Alpn                     []string `yaml:"alpn,omitempty"`
+	SNI                      string   `yaml:"sni,omitempty"`
+	ClientFingerprint        string   `yaml:"client-fingerprint,omitempty"`
+	SkipCertVerify           bool     `yaml:"skip-cert-verify,omitempty"`
+	Fingerprint              string   `yaml:"fingerprint,omitempty"`
+	UDP                      bool     `yaml:"udp,omitempty"`
+	IdleSessionCheckInterval int      `yaml:"idle-session-check-interval,omitempty"`
+	IdleSessionTimeout       int      `yaml:"idle-session-timeout,omitempty"`
+	MinIdleSession           int      `yaml:"min-idle-session,omitempty"`
+}
+
+func ProxyToAnytls(p Proxy) Anytls {
+	return Anytls{
+		Type:                     "anytls",
+		Name:                     p.Name,
+		Server:                   p.Server,
+		Port:                     p.Port,
+		Password:                 p.Password,
+		Alpn:                     p.Alpn,
+		SNI:                      p.Sni,
+		ClientFingerprint:        p.ClientFingerprint,
+		SkipCertVerify:           p.SkipCertVerify,
+		Fingerprint:              p.Fingerprint,
+		UDP:                      p.UDP,
+		IdleSessionCheckInterval: p.IdleSessionCheckInterval,
+		IdleSessionTimeout:       p.IdleSessionTimeout,
+		MinIdleSession:           p.MinIdleSession,
+	}
+}

model/rule_provider.go

@@ -6,6 +6,7 @@ type RuleProvider struct {
 	Url      string `yaml:"url,omitempty"`
 	Path     string `yaml:"path,omitempty"`
 	Interval int    `yaml:"interval,omitempty"`
+	Format string    `yaml:"format,omitempty"`
 }
 
 type Payload struct {

parser/anytls.go

@@ -0,0 +1,74 @@
+package parser
+
+import (
+	"fmt"
+	"net/url"
+	"strings"
+
+	"github.com/nitezs/sub2clash/constant"
+	"github.com/nitezs/sub2clash/model"
+)
+
+func ParseAnytls(proxy string) (model.Proxy, error) {
+	if !strings.HasPrefix(proxy, constant.AnytlsPrefix) {
+		return model.Proxy{}, &ParseError{Type: ErrInvalidPrefix, Raw: proxy}
+	}
+
+	link, err := url.Parse(proxy)
+	if err != nil {
+		return model.Proxy{}, &ParseError{
+			Type:    ErrInvalidStruct,
+			Message: "url parse error",
+			Raw:     proxy,
+		}
+	}
+
+	username := link.User.Username()
+	password, exist := link.User.Password()
+	if !exist {
+		password = username
+	}
+
+	query := link.Query()
+	server := link.Hostname()
+	if server == "" {
+		return model.Proxy{}, &ParseError{
+			Type:    ErrInvalidStruct,
+			Message: "missing server host",
+			Raw:     proxy,
+		}
+	}
+	portStr := link.Port()
+	if portStr == "" {
+		return model.Proxy{}, &ParseError{
+			Type:    ErrInvalidStruct,
+			Message: "missing server port",
+			Raw:     proxy,
+		}
+	}
+	port, err := ParsePort(portStr)
+	if err != nil {
+		return model.Proxy{}, &ParseError{
+			Type: ErrInvalidPort,
+			Raw:  portStr,
+		}
+	}
+	insecure, sni := query.Get("insecure"), query.Get("sni")
+	insecureBool := insecure == "1"
+	remarks := link.Fragment
+	if remarks == "" {
+		remarks = fmt.Sprintf("%s:%s", server, portStr)
+	}
+	remarks = strings.TrimSpace(remarks)
+
+	result := model.Proxy{
+		Type:           "anytls",
+		Name:           remarks,
+		Server:         server,
+		Port:           port,
+		Password:       password,
+		Sni:            sni,
+		SkipCertVerify: insecureBool,
+	}
+	return result, nil
+}

validator/short_link.go

@@ -3,6 +3,7 @@ package validator
 type ShortLinkGenValidator struct {
 	Url      string `form:"url" binding:"required"`
 	Password string `form:"password"`
+	CustomID string `form:"customId"`
 }
 
 type GetUrlValidator struct {