diff --git a/src/main.rs b/src/main.rs index f5e3312..6e9b734 100644 --- a/src/main.rs +++ b/src/main.rs @@ -4,7 +4,7 @@ mod parser; fn main() { let protocol = parser::get_uri_protocol("vmess://"); match protocol { - Some(parser::protocols::Vless) => { + Some(parser::Protocols::Vless) => { println!("The protocol was Vless"); } Some(_) => { diff --git a/src/parser/mod.rs b/src/parser/mod.rs index 51523e9..5dad642 100644 --- a/src/parser/mod.rs +++ b/src/parser/mod.rs @@ -1,6 +1,6 @@ use regex::Regex; -pub enum protocols { +pub enum Protocols { Vmess, Vless, Shadowsocks, @@ -9,28 +9,28 @@ pub enum protocols { Http, } -pub fn get_uri_protocol(uri: &str) -> Option { +pub fn get_uri_protocol(uri: &str) -> Option { 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); + return Some(Protocols::Vmess); } if uri.starts_with("vless://") { - return Some(protocols::Vless); + return Some(Protocols::Vless); } if uri.starts_with("ss://") { - return Some(protocols::Shadowsocks); + return Some(Protocols::Shadowsocks); } if uri.starts_with("socks://") { - return Some(protocols::Socks); + return Some(Protocols::Socks); } if uri.starts_with("http://") { - return Some(protocols::Http); + return Some(Protocols::Http); } if uri.starts_with("trojan://") { - return Some(protocols::Trojan); + return Some(Protocols::Trojan); } return None; } @@ -46,21 +46,21 @@ mod tests { #[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)); + assert!(matches!(protocol, Protocols::Vless)); } #[test] fn recognize_vmess_protocol() { let protocol = get_uri_protocol("vmess://eyJhZGQiOiIxMjcuMC4wLjEiLCJhaWQiOiIwIiwiaG9zdCI6IiIsImlkIjoiOHM2OTdlMmMtZXMxNy00MDNkLTI0ZjMtZHMyYzYwc2I4ZjUiLCJuZXQiOiJ0Y3AiLCJwYXRoIjoiIiwicG9ydCI6IjgwODAiLCJwcyI6InRlc3QiLCJzY3kiOiJhdXRvIiwic25pIjoiIiwidGxzIjoiIiwidHlwZSI6Im5vbmUiLCJ2IjoiMiJ9").unwrap(); - assert!(matches!(protocol, protocols::Vmess)); + 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)); + 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)); + assert!(matches!(protocol, Protocols::Trojan)); } }