From e3d403f49fd1de5b4492a0966086b6668c03319f Mon Sep 17 00:00:00 2001 From: legitnull Date: Sun, 19 Feb 2023 14:08:48 -0700 Subject: [PATCH] added some color --- Cargo.toml | 3 ++- src/main.rs | 15 +++++++++++---- src/modules/ai.rs | 10 ++++++---- src/modules/invade.rs | 31 +++++++++++++++++++++++-------- src/modules/test.rs | 0 5 files changed, 42 insertions(+), 17 deletions(-) create mode 100644 src/modules/test.rs diff --git a/Cargo.toml b/Cargo.toml index a17c21e..60c996f 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -14,4 +14,5 @@ regex = "1.7.1" toml = "0.7.2" serde = "1.0.152" tokio-openssl = "0.6.3" -colored = "2" \ No newline at end of file +colored = "2" +socks = "0.3.4" \ No newline at end of file diff --git a/src/main.rs b/src/main.rs index a096a20..ba62049 100644 --- a/src/main.rs +++ b/src/main.rs @@ -13,11 +13,13 @@ mod modules { pub mod kill; pub mod ai; pub mod invade; + pub mod test; } use modules::ai::Ai; // FIX THIS BS use modules::ping::PingCommand; use modules::invade::InvadeCommand; +//use modules::test::TestCommand; use modules::kill::KillCommand; // ... use crate::modules::Command; @@ -70,12 +72,12 @@ fn main() { let message = received.trim(); //debug chat - println!("{}", received); // ADD COLORS + println!("{} {}","[%] DEBUG:".bold().green(), received.purple()); // RESPOND TO PINGS if message.starts_with("PING") { - println!("[%] PONG {}", config.nick.blue()); + println!("{} {}","[%] PONG:".bold().green(), config.nick.blue()); // DEBUG ssl_stream.write_all("PONG ircd.chat\r\n".as_bytes()).unwrap(); continue; // skip processing the PING message further } @@ -84,6 +86,7 @@ fn main() { let ping_command = PingCommand; let kill_command = KillCommand; let invade_command = InvadeCommand; + //let test_command = TestCommand; let ai = Ai; // ADMIN MODULES @@ -91,7 +94,7 @@ fn main() { let parts: Vec<&str> = message.splitn(2, ' ').collect(); // Check if user is admin_user let username = parts[0].trim_start_matches(':').split("!").next().unwrap(); if !admin_users.contains(&username.to_string()) { - println!("[!] UNAUTHORIZED: {}", username.red()); + println!("{} {}","[!] UNAUTHORIZED:".bold().clear().on_red(), username.red().bold()); continue; // ... } if message.contains(":%ping") { @@ -106,7 +109,11 @@ fn main() { for response in invade_command.handle(message) { ssl_stream.write_all(response.as_bytes()).unwrap(); } - } + } //else if message.contains(":%test") { + // for response in test_command.handle(message) { + // ssl_stream.write_all(response.as_bytes()).unwrap(); + // } + //} } // Check if the message is user and respond via ai diff --git a/src/modules/ai.rs b/src/modules/ai.rs index 22ef96a..d78af3e 100644 --- a/src/modules/ai.rs +++ b/src/modules/ai.rs @@ -4,6 +4,8 @@ use regex::Regex; use crate::modules::Command; use toml::Value; use serde::Deserialize; +use colored::*; + #[derive(Deserialize)] struct Config { nick: String, @@ -45,7 +47,7 @@ async fn ai(user_message: &str, username: &str, channel: &str) -> Vec { let api_key = config.openai; // set this from config let client = Client::new().with_api_key(api_key); - println!("[?] PROMPT: {}: {}", username, user_message); + println!("{} {} {}: {}", "[?]".on_green().bold(), "PROMPT:".green().bold(), username, user_message); let chat_request = CreateCompletionRequestArgs::default() .prompt(user_message) .max_tokens(40_u16) @@ -57,10 +59,10 @@ async fn ai(user_message: &str, username: &str, channel: &str) -> Vec { .create(chat_request) .await .unwrap(); - println!("[+] RESPONSE: {}", chat_response.choices.first().unwrap().text); + println!("{} {} {}","[+]".on_green().bold(), "RESPONSE:".green().bold(), chat_response.choices.first().unwrap().text); //modify regex for varible username ie G1R g1r GIR gir but as handle nick for bots let response_text = &chat_response.choices.first().unwrap().text; - let regex = Regex::new(r#""|[gG][1iI][rR]:\s*|[mM][eE]:?\s"#).unwrap(); // THIS IS FUCKING UP EVERYTHING + let regex = Regex::new(r#""|[gG][1iI][rR]:\s"#).unwrap(); // THIS IS FUCKING UP EVERYTHING //let nick = &config.nick; //let regex_str = format!( // r#""|[{}{}{}]|\b[gG][1iI][rR]:\s*|\b[mM][eE]:?\s"#, @@ -80,7 +82,7 @@ async fn ai(user_message: &str, username: &str, channel: &str) -> Vec { let response_lines = response_text.split("\n").filter(|line| !line.trim().is_empty()); let mut responses = Vec::new(); for line in response_lines { - responses.push(format!("PRIVMSG {} :{}: {}\r\n", channel, username, line)); + responses.push(format!("PRIVMSG {} :\x0313{}\x0f: {}\r\n", channel, username, line)); } responses diff --git a/src/modules/invade.rs b/src/modules/invade.rs index 6ae5b39..192b7be 100644 --- a/src/modules/invade.rs +++ b/src/modules/invade.rs @@ -1,11 +1,13 @@ use crate::modules::Command; -use std::io::{Write}; +use std::io::{BufRead, BufReader, Write}; use std::net::TcpStream; - use openssl::ssl::{SslConnector, SslMethod}; use serde::Deserialize; use toml::Value; +use colored::*; + + #[derive(Clone, Deserialize)] struct Config { invaders: Vec, @@ -29,6 +31,7 @@ impl Command for InvadeCommand { let config_value = config_str.parse::().unwrap(); let config: Config = config_value.try_into().unwrap(); + for invader in &config.invaders[0..num_invaders] { let thread_channel = invadechannel.to_string(); let thread_invader = invader.to_string(); @@ -49,19 +52,31 @@ impl Command for InvadeCommand { ssl_stream.write_all(msg.as_bytes()).unwrap(); loop { - let mut buffer = [0; 512]; - match ssl_stream.ssl_read(&mut buffer) { + + + + let mut buf = [0; 512]; + match ssl_stream.ssl_read(&mut buf) { Ok(0) => break, Ok(n) => { - let message = String::from_utf8_lossy(&buffer[..n]); + let received = String::from_utf8_lossy(&buf[0..n]); + let message = received.trim(); + + //debug chat + println!("{} {} {}","[%] DEBUG:".bold().green(), thread_invader.green(), received.blue()); + if message.starts_with("PING") { let response = message.replace("PING", "PONG"); - println!("[%] PONG {}", thread_invader); + println!("{} {}","[%] PONG:".bold().green(), thread_invader.blue()); + ssl_stream.write_all(response.as_bytes()).unwrap(); + } + if message.contains("PRIVMSG") && message.contains(":%%fuck") { + let response = format!("PRIVMSG {} :FUCKFUCKFUCK\r\n", thread_channel); ssl_stream.write_all(response.as_bytes()).unwrap(); } } Err(e) => { - eprintln!("Error reading from server: {}", e); + eprintln!("{} {}","[!] ERROR FROM SERVER:".on_red(), e); break; } } @@ -69,7 +84,7 @@ impl Command for InvadeCommand { }); } - response.push(format!("PRIVMSG {} :INVADING WITH {} INVADERS...\r\n", channel, num_invaders)); + response.push(format!("PRIVMSG {} :\x0304,01[!] INVADING {} WITH {} INVADERS...\x0f\r\n", channel, invadechannel, num_invaders)); } response diff --git a/src/modules/test.rs b/src/modules/test.rs new file mode 100644 index 0000000..e69de29