added wordlist functionality

Signed-off-by: spv <spv0x04@proton.me>

Cargo.toml

@@ -8,7 +8,7 @@ edition = "2021"
 [dependencies]
 anyhow = "1.0.75"
 chrono = { version = "0.4.31", features = ["serde"] }
-clap = "4.5.11"
+clap = { version = "4.5.11", features = ["derive"] }
 console = "0.15.7"
 crossterm = "0.27.0"
 getset = "0.1.2"

src/main.rs

@@ -1,4 +1,6 @@
-use std::{sync::{Arc, RwLock}, ops::Deref, time::Duration};
+use std::{ops::Deref, path::PathBuf, sync::{Arc, RwLock}, time::Duration};
+
+use clap::Parser;
 
 use brute_forcer::Bruter;
 
@@ -7,10 +9,22 @@ pub mod discord_invite;
 pub mod brute_forcer;
 pub mod tui;
 
+#[derive(Parser, Debug, Clone)]
+pub struct Args{
+    wordlist: PathBuf,
+    #[clap(default_value_t = 10)]
+    iterations: usize,
+}
+
 fn main() -> anyhow::Result<()>{
-    let mut bruter = Bruter::new(10);
+    let args = Args::parse();
+    let mut bruter = Bruter::new(args.iterations);
     let b_ref = bruter.clone();
-    bruter.set_codes(vec!["incestcentral".to_string(), "penis".to_string(), "peter".to_string()]);
+    let mut codes = vec![];
+    for w in std::fs::read_to_string(args.wordlist).unwrap().split('\n'){
+        codes.push(w.to_string());
+    }
+    bruter.set_codes(codes);
     Bruter::start(bruter).unwrap();
     Ok(())
 }