Support ipv6 proxy address

cmd/params/params.go

@@ -42,11 +42,13 @@ var (
 	// nolint
 	matchNoneRegex = regexp.MustCompile("a^")
 	// nolint
+	ntlmProxyRegex = regexp.MustCompile(`^.*\\.+$`)
+	// nolint
 	pluginRegex = regexp.MustCompile(`(?i)([a-z\/0-9.]+\s)?(?P<editor>[a-z-]+)\-wakatime\/[0-9.]+`)
 	// nolint
-	proxyRegex = regexp.MustCompile(`^((https?|socks5)://)?([^:@]+(:([^:@])+)?@)?[^:]+(:\d+)?$`)
+	proxyIPv6Regex = regexp.MustCompile(`^((https?|socks5)://)?([^:@]+(:([^:@])+)?@)?(([0-9a-fA-F]{1,4}:){7,7}[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,7}:|([0-9a-fA-F]{1,4}:){1,6}:[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,5}(:[0-9a-fA-F]{1,4}){1,2}|([0-9a-fA-F]{1,4}:){1,4}(:[0-9a-fA-F]{1,4}){1,3}|([0-9a-fA-F]{1,4}:){1,3}(:[0-9a-fA-F]{1,4}){1,4}|([0-9a-fA-F]{1,4}:){1,2}(:[0-9a-fA-F]{1,4}){1,5}|[0-9a-fA-F]{1,4}:((:[0-9a-fA-F]{1,4}){1,6})|:((:[0-9a-fA-F]{1,4}){1,7}|:)|fe80:(:[0-9a-fA-F]{0,4}){0,4}%[0-9a-zA-Z]{1,}|::(ffff(:0{1,4}){0,1}:){0,1}((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])|([0-9a-fA-F]{1,4}:){1,4}:((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9]))(:\d+)?$`)
 	// nolint
-	ntlmProxyRegex = regexp.MustCompile(`^.*\\.+$`)
+	proxyRegex = regexp.MustCompile(`^((https?|socks5)://)?([^:@]+(:([^:@])+)?@)?[^:]+(:\d+)?$`)
 )
 
 type (
@@ -273,7 +275,9 @@ func LoadAPIParams(v *viper.Viper) (API, error) {
 	}
 
 	if proxyURL != "" && !rgx.MatchString(proxyURL) {
-		return API{}, api.ErrAuth{Err: fmt.Errorf(errMsgTemplate, proxyURL)}
+		if !proxyIPv6Regex.MatchString(proxyURL) {
+			return API{}, api.ErrAuth{Err: fmt.Errorf(errMsgTemplate, proxyURL)}
+		}
 	}
 
 	proxyEnv := httpproxy.FromEnvironment()

cmd/params/params_test.go

@@ -2011,8 +2011,9 @@ func TestLoad_API_ProxyURL(t *testing.T) {
 	tests := map[string]string{
 		"https":  "https://john:[email protected]:8888",
 		"http":   "http://john:[email protected]:8888",
-		"socks5": "socks5://john:secret@example.org:8888",
+		"ipv6":   "socks5://john:secret@2001:0db8:85a3:0000:0000:8a2e:0370:7334:8888",
 		"ntlm":   `domain\\john:123456`,
+		"socks5": "socks5://john:[email protected]:8888",
 	}
 
 	for name, proxyURL := range tests {