From e7b49a6bbb47d17e05a2cb6006acdcd727a65658 Mon Sep 17 00:00:00 2001
From: Keivan-sf <keyvan0082@gmail.com>
Date: Tue, 10 Oct 2023 19:00:17 +0330
Subject: [PATCH] feat: remove remark from vless query

---
 src/parser/vless.rs | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/src/parser/vless.rs b/src/parser/vless.rs
index d13254f..fe3370d 100644
--- a/src/parser/vless.rs
+++ b/src/parser/vless.rs
@@ -14,14 +14,16 @@ pub struct VlessQuery {
 
 pub fn get_vless_data(uri: &str) {
     let data = uri.split_once("vless://").unwrap().1;
-    let query = uri.split_once("?").unwrap().1;
-    let query_parsed = querystring::querify(query);
-    for i in query_parsed {
-        println!("{0} : {1}", i.0, i.1);
-    }
+    let query_and_name = uri.split_once("?").unwrap().1;
+    let query = query_and_name
+        .split_once("#")
+        .unwrap_or((query_and_name, ""))
+        .0;
+    let parsed_query = parse_vless_query_data(query);
+    println!("{0}", parsed_query.flow);
 }
 
-pub fn get_vless_query_data(raw_query: &str) -> VlessQuery {
+pub fn parse_vless_query_data(raw_query: &str) -> VlessQuery {
     let query = querystring::querify(raw_query);
     let a = VlessQuery {
         path: query
@@ -87,7 +89,7 @@ mod tests {
     #[test]
     fn parse_vless_query() {
         let query = "security=reality&sni=bench.sh&fp=chrome&pbk=7xhH4b_VkliBxGulljcyPOH-bYUA2dl-XAdZAsfhk04&sid=6ba85179e30d4fc2&type=tcp&flow=xtls-rprx-vision&path=/";
-        let parsed_query = get_vless_query_data(query);
+        let parsed_query = parse_vless_query_data(query);
         assert_eq!(parsed_query.sni, "bench.sh");
         assert_eq!(parsed_query.security, "reality");
         assert_eq!(parsed_query.fp, "chrome");
@@ -103,7 +105,7 @@ mod tests {
     #[test]
     fn parse_vless_query_with_defaults() {
         let query = "";
-        let parsed_query = get_vless_query_data(query);
+        let parsed_query = parse_vless_query_data(query);
         assert_eq!(parsed_query.sni, "");
         assert_eq!(parsed_query.security, "");
         assert_eq!(parsed_query.fp, "");