Merge pull request #4 from Lurk/evil-mode

Evil mode

Author: Serhiy Barhamon

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "clink"
-version = "0.2.1"
+version = "0.3.0"
 authors = ["Sergey Bargamon <sergey@bargamon.ru>"]
 edition = "2018"
 
@@ -10,4 +10,5 @@ edition = "2018"
 clipboard = "0.5.0"
 linkify = "0.4.0"
 url = "2.2.0"
-rustop = "1.1.1"
+rustop = "1.1.1"
+rand = "0.8.3"

Cargo.toml

@@ -20,6 +20,11 @@ sets values of fbclid, utm_source, utm_campaign and utm_medium GET params to "yo
 
 inspired by this [tweet](https://twitter.com/ftrain/status/1359138516681314311?s=21)
 
+### Evil mode
+```
+clink -m evil
+```
+swap two random chars in values of fbclid, utm_source, utm_campaign and utm_medium GET params in links that are in clipboard (Diabolical Laughter)
 
 ## Build
 

readme.md

@@ -31,13 +31,13 @@ fn main() {
                 if previous_clipboard != current_clipboard {
                     let cleaned = find_and_replace(&current_clipboard, &args.mode);
                     if cleaned != current_clipboard {
-                        ctx.set_contents(cleaned).unwrap();
+                        ctx.set_contents(cleaned.clone()).unwrap();
                     }
-                    previous_clipboard = current_clipboard;
+                    previous_clipboard = cleaned;
                 }
             }
             Err(_e) => {}
         }
-        thread::sleep(Duration::from_millis(100))
+        thread::sleep(Duration::from_millis(150))
     }
 }

src/main.rs

@@ -6,6 +6,7 @@ use std::str::FromStr;
 pub enum Mode {
     Remove,
     YourMom,
+    Evil,
 }
 
 #[derive(Debug)]
@@ -39,7 +40,10 @@ impl FromStr for Mode {
         match day {
             "remove" => Ok(Mode::Remove),
             "your_mom" => Ok(Mode::YourMom),
-            _ => Err(ModeError::new("Mode can be \"remove\" or \"your_mom\"")),
+            "evil" => Ok(Mode::Evil),
+            _ => Err(ModeError::new(
+                "Mode can be \"remove\", \"your_mom\" or \"evil\"",
+            )),
         }
     }
 }
@@ -49,6 +53,7 @@ impl fmt::Display for Mode {
         match *self {
             Mode::Remove => write!(f, "Remove"),
             Mode::YourMom => write!(f, "YourMom"),
+            Mode::Evil => write!(f, "Evil"),
         }
     }
 }

src/mode.rs

@@ -1,5 +1,6 @@
 use crate::mode::Mode;
 use linkify::{LinkFinder, LinkKind};
+use rand::Rng;
 use url::Url;
 
 #[cfg(test)]
@@ -10,60 +11,60 @@ mod find_and_replace {
     fn naive() {
         assert_eq!(
             find_and_replace(
-                &"https://test.test/?fbclid=dsadsa&utm_source=fafa&utm_campaign=fafas&utm_medium=adsa".to_string(),
+                "https://test.test/?fbclid=dsadsa&utm_source=fafa&utm_campaign=fafas&utm_medium=adsa",
                 &Mode::Remove
             ),
             "https://test.test/"
         );
         assert_eq!(
             find_and_replace(
-                &"https://test.test/?fbclid=dsadsa&utm_source=fafa&utm_campaign=fafas&utm_medium=adsa".to_string(),
+                "https://test.test/?fbclid=dsadsa&utm_source=fafa&utm_campaign=fafas&utm_medium=adsa",
                 &Mode::YourMom
             ),
             "https://test.test/?fbclid=your_mom&utm_source=your_mom&utm_campaign=your_mom&utm_medium=your_mom"
         );
+        assert_ne!(
+            find_and_replace(
+                "https://test.test/?fbclid=IwAR3l6qn8TzOT254dIa7jBAM1dG3OHn3f8ZoRGsADTmqG1Zfmmko-oRhE8Qs&utm_source=IwAR3l6qn8TzOT254dIa7jBAM1dG3OHn3f8ZoRGsADTmqG1Zfmmko-oRhE8Qs&utm_campaign=IwAR3l6qn8TzOT254dIa7jBAM1dG3OHn3f8ZoRGsADTmqG1Zfmmko-oRhE8Qs&utm_medium=IwAR3l6qn8TzOT254dIa7jBAM1dG3OHn3f8ZoRGsADTmqG1Zfmmko-oRhE8Qs",
+                &Mode::Evil
+            ),
+            "https://test.test/?fbclid=IwAR3l6qn8TzOT254dIa7jBAM1dG3OHn3f8ZoRGsADTmqG1Zfmmko-oRhE8Qs&utm_source=IwAR3l6qn8TzOT254dIa7jBAM1dG3OHn3f8ZoRGsADTmqG1Zfmmko-oRhE8Qs&utm_campaign=IwAR3l6qn8TzOT254dIa7jBAM1dG3OHn3f8ZoRGsADTmqG1Zfmmko-oRhE8Qs&utm_medium=IwAR3l6qn8TzOT254dIa7jBAM1dG3OHn3f8ZoRGsADTmqG1Zfmmko-oRhE8Qs"
+        );
     }
     #[test]
     fn should_preserve_query() {
         assert_eq!(
-            find_and_replace(&"https://test.test/?abc=abc".to_string(), &Mode::Remove),
+            find_and_replace("https://test.test/?abc=abc", &Mode::Remove),
             "https://test.test/?abc=abc"
         );
         assert_eq!(
-            find_and_replace(&"https://test.test/?abc=abc".to_string(), &Mode::YourMom),
+            find_and_replace("https://test.test/?abc=abc", &Mode::YourMom),
             "https://test.test/?abc=abc"
         );
     }
     #[test]
     fn multiple_params() {
         assert_eq!(
-            find_and_replace(
-                &"https://test.test/?abc=abc&fbclid=flksj".to_string(),
-                &Mode::Remove
-            ),
+            find_and_replace("https://test.test/?abc=abc&fbclid=flksj", &Mode::Remove),
             "https://test.test/?abc=abc"
         );
         assert_eq!(
-            find_and_replace(
-                &"https://test.test/?abc=abc&fbclid=flksj".to_string(),
-                &Mode::YourMom
-            ),
+            find_and_replace("https://test.test/?abc=abc&fbclid=flksj", &Mode::YourMom),
             "https://test.test/?abc=abc&fbclid=your_mom"
         );
     }
     #[test]
     fn multiple_links() {
         assert_eq!(
             find_and_replace(
-                &"https://test.test/?abc=abc&fbclid=flksj\nhttps://test.test/?abc=abc&fbclid=flksj"
-                    .to_string(),
+                "https://test.test/?abc=abc&fbclid=flksj\nhttps://test.test/?abc=abc&fbclid=flksj",
                 &Mode::Remove
             ),
             "https://test.test/?abc=abc\nhttps://test.test/?abc=abc"
         );
         assert_eq!(
             find_and_replace(
-                &"https://test.test/?abc=abc&fbclid=flksj\nhttps://test.test/?abc=abc&fbclid=flksj".to_string(),
+                "https://test.test/?abc=abc&fbclid=flksj\nhttps://test.test/?abc=abc&fbclid=flksj",
                 &Mode::YourMom
             ),
             "https://test.test/?abc=abc&fbclid=your_mom\nhttps://test.test/?abc=abc&fbclid=your_mom"
@@ -73,14 +74,14 @@ mod find_and_replace {
     fn multiple_links_and_text() {
         assert_eq!(
             find_and_replace(
-                &"some text here https://test.test/?abc=abc&fbclid=flksj here \nand herehttps://test.test/?abc=abc&fbclid=flksj".to_string(),
+                "some text here https://test.test/?abc=abc&fbclid=flksj here \nand herehttps://test.test/?abc=abc&fbclid=flksj",
                 &Mode::Remove
             ),
             "some text here https://test.test/?abc=abc here \nand herehttps://test.test/?abc=abc"
         );
         assert_eq!(
             find_and_replace(
-                &"some text here https://test.test/?abc=abc&fbclid=flksj here \nand herehttps://test.test/?abc=abc&fbclid=flksj".to_string(),
+                "some text here https://test.test/?abc=abc&fbclid=flksj here \nand herehttps://test.test/?abc=abc&fbclid=flksj",
                 &Mode::YourMom
             ),
             "some text here https://test.test/?abc=abc&fbclid=your_mom here \nand herehttps://test.test/?abc=abc&fbclid=your_mom"
@@ -126,6 +127,25 @@ fn process_query(query: url::form_urlencoded::Parse<'_>, mode: &Mode) -> Vec<(St
                 }
             })
             .collect(),
+        Mode::Evil => {
+            let mut rng = rand::thread_rng();
+            query
+                .map(|p| {
+                    if is_hit(&p.0) {
+                        (
+                            p.0.to_string(),
+                            swap_two_chars(
+                                &p.1,
+                                rng.gen_range(0..p.1.to_string().len()),
+                                rng.gen_range(0..p.1.to_string().len()),
+                            ),
+                        )
+                    } else {
+                        (p.0.to_string(), p.1.to_string())
+                    }
+                })
+                .collect()
+        }
     }
 }
 
@@ -144,3 +164,19 @@ mod is_hit {
 fn is_hit(p: &str) -> bool {
     p == "fbclid" || p == "utm_source" || p == "utm_campaign" || p == "utm_medium"
 }
+
+#[cfg(test)]
+mod swap {
+    use super::*;
+
+    #[test]
+    fn test_all() {
+        assert_eq!(swap_two_chars("0123456789", 2, 7), "0173456289");
+    }
+}
+
+fn swap_two_chars(s: &str, a: usize, b: usize) -> String {
+    let mut char_vector: Vec<char> = s.chars().collect();
+    char_vector.swap(a, b);
+    char_vector.iter().collect()
+}