chore: resolve rust warning for protocols

This commit is contained in:
Keivan-sf
2023-10-09 18:05:39 +03:30
parent e4db6a92af
commit 3d114db44b
2 changed files with 13 additions and 13 deletions

View File

@@ -4,7 +4,7 @@ mod parser;
fn main() { fn main() {
let protocol = parser::get_uri_protocol("vmess://"); let protocol = parser::get_uri_protocol("vmess://");
match protocol { match protocol {
Some(parser::protocols::Vless) => { Some(parser::Protocols::Vless) => {
println!("The protocol was Vless"); println!("The protocol was Vless");
} }
Some(_) => { Some(_) => {

View File

@@ -1,6 +1,6 @@
use regex::Regex; use regex::Regex;
pub enum protocols { pub enum Protocols {
Vmess, Vmess,
Vless, Vless,
Shadowsocks, Shadowsocks,
@@ -9,28 +9,28 @@ pub enum protocols {
Http, Http,
} }
pub fn get_uri_protocol(uri: &str) -> Option<protocols> { pub fn get_uri_protocol(uri: &str) -> Option<Protocols> {
let uri_regex = Regex::new(r"^[a-z]+:\/\/.+$").unwrap(); let uri_regex = Regex::new(r"^[a-z]+:\/\/.+$").unwrap();
if !uri_regex.is_match(uri) { if !uri_regex.is_match(uri) {
return None; return None;
} }
if uri.starts_with("vmess://") { if uri.starts_with("vmess://") {
return Some(protocols::Vmess); return Some(Protocols::Vmess);
} }
if uri.starts_with("vless://") { if uri.starts_with("vless://") {
return Some(protocols::Vless); return Some(Protocols::Vless);
} }
if uri.starts_with("ss://") { if uri.starts_with("ss://") {
return Some(protocols::Shadowsocks); return Some(Protocols::Shadowsocks);
} }
if uri.starts_with("socks://") { if uri.starts_with("socks://") {
return Some(protocols::Socks); return Some(Protocols::Socks);
} }
if uri.starts_with("http://") { if uri.starts_with("http://") {
return Some(protocols::Http); return Some(Protocols::Http);
} }
if uri.starts_with("trojan://") { if uri.starts_with("trojan://") {
return Some(protocols::Trojan); return Some(Protocols::Trojan);
} }
return None; return None;
} }
@@ -46,21 +46,21 @@ mod tests {
#[test] #[test]
fn recognize_vless_protocol() { 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(); 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] #[test]
fn recognize_vmess_protocol() { fn recognize_vmess_protocol() {
let protocol = get_uri_protocol("vmess://eyJhZGQiOiIxMjcuMC4wLjEiLCJhaWQiOiIwIiwiaG9zdCI6IiIsImlkIjoiOHM2OTdlMmMtZXMxNy00MDNkLTI0ZjMtZHMyYzYwc2I4ZjUiLCJuZXQiOiJ0Y3AiLCJwYXRoIjoiIiwicG9ydCI6IjgwODAiLCJwcyI6InRlc3QiLCJzY3kiOiJhdXRvIiwic25pIjoiIiwidGxzIjoiIiwidHlwZSI6Im5vbmUiLCJ2IjoiMiJ9").unwrap(); let protocol = get_uri_protocol("vmess://eyJhZGQiOiIxMjcuMC4wLjEiLCJhaWQiOiIwIiwiaG9zdCI6IiIsImlkIjoiOHM2OTdlMmMtZXMxNy00MDNkLTI0ZjMtZHMyYzYwc2I4ZjUiLCJuZXQiOiJ0Y3AiLCJwYXRoIjoiIiwicG9ydCI6IjgwODAiLCJwcyI6InRlc3QiLCJzY3kiOiJhdXRvIiwic25pIjoiIiwidGxzIjoiIiwidHlwZSI6Im5vbmUiLCJ2IjoiMiJ9").unwrap();
assert!(matches!(protocol, protocols::Vmess)); assert!(matches!(protocol, Protocols::Vmess));
} }
#[test] #[test]
fn recognize_shadowsocks_protocol() { fn recognize_shadowsocks_protocol() {
let protocol = get_uri_protocol("ss://Y2hhY2hhMjAtaWV0Zi1wb2x5MTMwNTpXNzRYRkFMS0t1dzZtNUlB@www.outline.aasf.cyou:443#test").unwrap(); 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] #[test]
fn recognize_trojan_protocol() { 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(); 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));
} }
} }