Helper func to parse ss URL

main.go

@@ -84,18 +84,13 @@ func main() {
 		addr := flags.Client
 		cipher := flags.Cipher
 		password := flags.Password
+		var err error
 
 		if strings.HasPrefix(addr, "ss://") {
-			u, err := url.Parse(addr)
+			addr, cipher, password, err = parseURL(addr)
 			if err != nil {
 				log.Fatal(err)
 			}
-
-			addr = u.Host
-			if u.User != nil {
-				cipher = u.User.Username()
-				password, _ = u.User.Password()
-			}
 		}
 
 		ciph, err := core.PickCipher(cipher, key, password)
@@ -134,18 +129,13 @@ func main() {
 		addr := flags.Server
 		cipher := flags.Cipher
 		password := flags.Password
+		var err error
 
 		if strings.HasPrefix(addr, "ss://") {
-			u, err := url.Parse(addr)
+			addr, cipher, password, err = parseURL(addr)
 			if err != nil {
 				log.Fatal(err)
 			}
-
-			addr = u.Host
-			if u.User != nil {
-				cipher = u.User.Username()
-				password, _ = u.User.Password()
-			}
 		}
 
 		ciph, err := core.PickCipher(cipher, key, password)
@@ -161,3 +151,17 @@ func main() {
 	signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
 	<-sigCh
 }
+
+func parseURL(s string) (addr, cipher, password string, err error) {
+	u, err := url.Parse(s)
+	if err != nil {
+		return
+	}
+
+	addr = u.Host
+	if u.User != nil {
+		cipher = u.User.Username()
+		password, _ = u.User.Password()
+	}
+	return
+}