From 18f4689d21febed0a9478280fe15796c89c6a59a Mon Sep 17 00:00:00 2001
From: ssrlive <30760636+ssrlive@users.noreply.github.com>
Date: Mon, 8 Apr 2024 20:49:44 +0800
Subject: [PATCH] refine code

---
 src/args.rs | 15 ++++++---------
 1 file changed, 6 insertions(+), 9 deletions(-)

diff --git a/src/args.rs b/src/args.rs
index 24c0e41..d0cc99f 100644
--- a/src/args.rs
+++ b/src/args.rs
@@ -12,7 +12,7 @@ pub struct Args {
     /// Proxy URL in the form proto://[username[:password]@]host:port,
     /// where proto is one of socks4, socks5, http.
     /// Username and password are encoded in percent encoding. For example:
-    /// socks5://myname:password@127.0.0.1:1080
+    /// socks5://myname:pass%40word@127.0.0.1:1080
     #[arg(short, long, value_parser = ArgProxy::from_url, value_name = "URL")]
     pub proxy: ArgProxy,
 
@@ -333,15 +333,12 @@ impl ArgProxy {
             Some(UserKey::new(username, password))
         };
 
-        let scheme = url.scheme();
-
         let proxy_type = match url.scheme().to_ascii_lowercase().as_str() {
-            "socks4" => Some(ProxyType::Socks4),
-            "socks5" => Some(ProxyType::Socks5),
-            "http" => Some(ProxyType::Http),
-            _ => None,
-        }
-        .ok_or(Error::from(&format!("`{scheme}` is an invalid proxy type")))?;
+            "socks4" => Ok(ProxyType::Socks4),
+            "socks5" => Ok(ProxyType::Socks5),
+            "http" => Ok(ProxyType::Http),
+            scheme => Err(Error::from(&format!("`{scheme}` is an invalid proxy type"))),
+        }?;
 
         Ok(ArgProxy {
             proxy_type,