chore: create uri_identifier module in parser

src/main.rs

@@ -1,18 +1,5 @@
-use std::process::exit;
-
 mod parser;
+
 fn main() {
-    let protocol = parser::get_uri_protocol("vmess://");
+    parser::parse("vmess://test");
-    match protocol {
-        Some(parser::Protocols::Vless) => {
-            println!("The protocol was Vless");
-        }
-        Some(_) => {
-            println!("Some recognizable protocol")
-        }
-        None => {
-            println!("The protocol is not supported");
-            exit(0);
-        }
-    }
 }

src/parser/mod.rs

@@ -1,66 +1,18 @@
-use regex::Regex;
+use std::process::exit;
+mod uri_identifier;
 
-pub enum Protocols {
+pub fn parse(uri: &str) {
-    Vmess,
+    let protocol = uri_identifier::get_uri_protocol(uri);
-    Vless,
+    match protocol {
-    Shadowsocks,
+        Some(uri_identifier::Protocols::Vmess) => {
-    Trojan,
+            println!("The protocol was vmess");
-    Socks,
+        }
-    Http,
+        Some(_) => {
-}
+            println!("The protocol was recognized");
-
+        }
-pub fn get_uri_protocol(uri: &str) -> Option<Protocols> {
+        None => {
-    let uri_regex = Regex::new(r"^[a-z]+:\/\/.+$").unwrap();
+            println!("The protcol is not supported");
-    if !uri_regex.is_match(uri) {
+            exit(0);
-        return None;
+        }
-    }
-    if uri.starts_with("vmess://") {
-        return Some(Protocols::Vmess);
-    }
-    if uri.starts_with("vless://") {
-        return Some(Protocols::Vless);
-    }
-    if uri.starts_with("ss://") {
-        return Some(Protocols::Shadowsocks);
-    }
-    if uri.starts_with("socks://") {
-        return Some(Protocols::Socks);
-    }
-    if uri.starts_with("http://") {
-        return Some(Protocols::Http);
-    }
-    if uri.starts_with("trojan://") {
-        return Some(Protocols::Trojan);
-    }
-    return None;
-}
-
-#[cfg(test)]
-mod tests {
-    use super::*;
-    #[test]
-    fn return_none_for_invalid_uri() {
-        let protocol = get_uri_protocol("123-vless://3d1c3f04-729d-59d3-bdb6-3f3f4352e173@root.ii.one:2083?security=reality&sni=www.spamhaus.org&fp=safari&pbk=7xhH4b_VkliBxGulljcyPOH-bYUA2dl-XAdZAsfhk04&sid=6ba85179e30d4fc2&type=tcp&flow=xtls-rprx-vision#Ha-ac");
-        assert!(matches!(protocol, None));
-    }
-    #[test]
-    fn recognize_vless_protocol() {
-        let protocol = get_uri_protocol("vless://3d1c3f04-729d-59d3-bdb6-3f3f4352e173@root.ii.one:2083?security=reality&sni=www.spamhaus.org&fp=safari&pbk=7xhH4b_VkliBxGulljcyPOH-bYUA2dl-XAdZAsfhk04&sid=6ba85179e30d4fc2&type=tcp&flow=xtls-rprx-vision#Ha-ac").unwrap();
-        assert!(matches!(protocol, Protocols::Vless));
-    }
-    #[test]
-    fn recognize_vmess_protocol() {
-        let protocol = get_uri_protocol("vmess://eyJhZGQiOiIxMjcuMC4wLjEiLCJhaWQiOiIwIiwiaG9zdCI6IiIsImlkIjoiOHM2OTdlMmMtZXMxNy00MDNkLTI0ZjMtZHMyYzYwc2I4ZjUiLCJuZXQiOiJ0Y3AiLCJwYXRoIjoiIiwicG9ydCI6IjgwODAiLCJwcyI6InRlc3QiLCJzY3kiOiJhdXRvIiwic25pIjoiIiwidGxzIjoiIiwidHlwZSI6Im5vbmUiLCJ2IjoiMiJ9").unwrap();
-        assert!(matches!(protocol, Protocols::Vmess));
-    }
-    #[test]
-    fn recognize_shadowsocks_protocol() {
-        let protocol = get_uri_protocol("ss://Y2hhY2hhMjAtaWV0Zi1wb2x5MTMwNTpXNzRYRkFMS0t1dzZtNUlB@www.outline.aasf.cyou:443#test").unwrap();
-        assert!(matches!(protocol, Protocols::Shadowsocks));
-    }
-    #[test]
-    fn recognize_trojan_protocol() {
-        let protocol = get_uri_protocol("trojan://test-pw@13.50.100.84:22222?security=tls&sni=trj.rollingnext.co.uk&type=tcp#test").unwrap();
-        assert!(matches!(protocol, Protocols::Trojan));
     }
 }

src/parser/uri_identifier.rs

@@ -0,0 +1,66 @@
+use regex::Regex;
+
+pub enum Protocols {
+    Vmess,
+    Vless,
+    Shadowsocks,
+    Trojan,
+    Socks,
+    Http,
+}
+
+pub fn get_uri_protocol(uri: &str) -> Option<Protocols> {
+    let uri_regex = Regex::new(r"^[a-z]+:\/\/.+$").unwrap();
+    if !uri_regex.is_match(uri) {
+        return None;
+    }
+    if uri.starts_with("vmess://") {
+        return Some(Protocols::Vmess);
+    }
+    if uri.starts_with("vless://") {
+        return Some(Protocols::Vless);
+    }
+    if uri.starts_with("ss://") {
+        return Some(Protocols::Shadowsocks);
+    }
+    if uri.starts_with("socks://") {
+        return Some(Protocols::Socks);
+    }
+    if uri.starts_with("http://") {
+        return Some(Protocols::Http);
+    }
+    if uri.starts_with("trojan://") {
+        return Some(Protocols::Trojan);
+    }
+    return None;
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    #[test]
+    fn return_none_for_invalid_uri() {
+        let protocol = get_uri_protocol("123-vless://3d1c3f04-729d-59d3-bdb6-3f3f4352e173@root.ii.one:2083?security=reality&sni=www.spamhaus.org&fp=safari&pbk=7xhH4b_VkliBxGulljcyPOH-bYUA2dl-XAdZAsfhk04&sid=6ba85179e30d4fc2&type=tcp&flow=xtls-rprx-vision#Ha-ac");
+        assert!(matches!(protocol, None));
+    }
+    #[test]
+    fn recognize_vless_protocol() {
+        let protocol = get_uri_protocol("vless://3d1c3f04-729d-59d3-bdb6-3f3f4352e173@root.ii.one:2083?security=reality&sni=www.spamhaus.org&fp=safari&pbk=7xhH4b_VkliBxGulljcyPOH-bYUA2dl-XAdZAsfhk04&sid=6ba85179e30d4fc2&type=tcp&flow=xtls-rprx-vision#Ha-ac").unwrap();
+        assert!(matches!(protocol, Protocols::Vless));
+    }
+    #[test]
+    fn recognize_vmess_protocol() {
+        let protocol = get_uri_protocol("vmess://eyJhZGQiOiIxMjcuMC4wLjEiLCJhaWQiOiIwIiwiaG9zdCI6IiIsImlkIjoiOHM2OTdlMmMtZXMxNy00MDNkLTI0ZjMtZHMyYzYwc2I4ZjUiLCJuZXQiOiJ0Y3AiLCJwYXRoIjoiIiwicG9ydCI6IjgwODAiLCJwcyI6InRlc3QiLCJzY3kiOiJhdXRvIiwic25pIjoiIiwidGxzIjoiIiwidHlwZSI6Im5vbmUiLCJ2IjoiMiJ9").unwrap();
+        assert!(matches!(protocol, Protocols::Vmess));
+    }
+    #[test]
+    fn recognize_shadowsocks_protocol() {
+        let protocol = get_uri_protocol("ss://Y2hhY2hhMjAtaWV0Zi1wb2x5MTMwNTpXNzRYRkFMS0t1dzZtNUlB@www.outline.aasf.cyou:443#test").unwrap();
+        assert!(matches!(protocol, Protocols::Shadowsocks));
+    }
+    #[test]
+    fn recognize_trojan_protocol() {
+        let protocol = get_uri_protocol("trojan://test-pw@13.50.100.84:22222?security=tls&sni=trj.rollingnext.co.uk&type=tcp#test").unwrap();
+        assert!(matches!(protocol, Protocols::Trojan));
+    }
+}