♻️ Refactor parsers

parser/hysteria2.go

@@ -1,57 +1,89 @@
 package parser
 
 import (
-	"errors"
 	"net/url"
-	"strconv"
 	"strings"
+	"sub2clash/constant"
 	"sub2clash/model"
 )
 
-// hysteria2://letmein@example.com/?insecure=1&obfs=salamander&obfs-password=gawrgura&pinSHA256=deadbeef&sni=real.example.com#name
-
 func ParseHysteria2(proxy string) (model.Proxy, error) {
-	// 判断是否以 hysteria2:// 开头
-	if !strings.HasPrefix(proxy, "hysteria2://") && !strings.HasPrefix(proxy, "hy2://") {
-		return model.Proxy{}, errors.New("invalid hysteria2 Url")
+	if !strings.HasPrefix(proxy, constant.Hysteria2Prefix1) &&
+		!strings.HasPrefix(proxy, constant.Hysteria2Prefix2) {
+		return model.Proxy{}, &ParseError{Type: ErrInvalidPrefix, Raw: proxy}
 	}
-	// 分割
-	parts := strings.SplitN(strings.TrimPrefix(proxy, "hysteria2://"), "@", 2)
-	// 分割
-	serverInfo := strings.SplitN(parts[1], "/?", 2)
+
+	proxy = strings.TrimPrefix(proxy, constant.Hysteria2Prefix1)
+	proxy = strings.TrimPrefix(proxy, constant.Hysteria2Prefix2)
+	urlParts := strings.SplitN(proxy, "@", 2)
+	if len(urlParts) != 2 {
+		return model.Proxy{}, &ParseError{
+			Type:    ErrInvalidStruct,
+			Message: "missing character '@' in url",
+			Raw:     proxy,
+		}
+	}
+	password := urlParts[0]
+
+	serverInfo := strings.SplitN(urlParts[1], "/?", 2)
+	if len(serverInfo) != 2 {
+		return model.Proxy{}, &ParseError{
+			Type:    ErrInvalidStruct,
+			Message: "missing params in url",
+			Raw:     proxy,
+		}
+	}
+	paramStr := serverInfo[1]
+
 	serverAndPort := strings.SplitN(serverInfo[0], ":", 2)
+	var server string
+	var portStr string
 	if len(serverAndPort) == 1 {
-		serverAndPort = append(serverAndPort, "443")
-	} else if len(serverAndPort) != 2 {
-		return model.Proxy{}, errors.New("invalid hysteria2 Url")
-	}
-	params, err := url.ParseQuery(serverInfo[1])
-	if err != nil {
-		return model.Proxy{}, errors.New("invalid hysteria2 Url")
-	}
-	// 获取端口
-	port, err := strconv.Atoi(serverAndPort[1])
-	if err != nil {
-		return model.Proxy{}, errors.New("invalid hysteria2 Url")
-	}
-	name := ""
-	if strings.Contains(proxy, "#") {
-		splitResult := strings.Split(proxy, "#")
-		name, _ = url.QueryUnescape(splitResult[len(splitResult)-1])
+		portStr = "443"
+	} else if len(serverAndPort) == 2 {
+		server, portStr = serverAndPort[0], serverAndPort[1]
 	} else {
-		name = strings.Join(serverAndPort, ":")
+		return model.Proxy{}, &ParseError{
+			Type:    ErrInvalidStruct,
+			Message: "missing server host or port",
+			Raw:     proxy,
+		}
 	}
-	// 返回结果
+
+	port, err := ParsePort(portStr)
+	if err != nil {
+		return model.Proxy{}, &ParseError{
+			Type:    ErrInvalidPort,
+			Message: err.Error(),
+			Raw:     proxy,
+		}
+	}
+
+	params, err := url.ParseQuery(paramStr)
+	if err != nil {
+		return model.Proxy{}, &ParseError{
+			Type:    ErrCannotParseParams,
+			Raw:     proxy,
+			Message: err.Error(),
+		}
+	}
+
+	remarks, network, obfs, obfsPassword, pinSHA256, insecure, sni := params.Get("name"), params.Get("network"), params.Get("obfs"), params.Get("obfs-password"), params.Get("pinSHA256"), params.Get("insecure"), params.Get("sni")
+	enableTLS := pinSHA256 != ""
+	insecureBool := insecure == "1"
+
 	result := model.Proxy{
 		Type:           "hysteria2",
-		Name:           name,
-		Server:         serverAndPort[0],
+		Name:           remarks,
+		Server:         server,
 		Port:           port,
-		Password:       parts[0],
-		Obfs:           params.Get("obfs"),
-		ObfsParam:      params.Get("obfs-password"),
-		Sni:            params.Get("sni"),
-		SkipCertVerify: params.Get("insecure") == "1",
+		Password:       password,
+		Obfs:           obfs,
+		ObfsParam:      obfsPassword,
+		Sni:            sni,
+		SkipCertVerify: insecureBool,
+		TLS:            enableTLS,
+		Network:        network,
 	}
 	return result, nil
 }