114 lines
5.6 KiB
Rust
114 lines
5.6 KiB
Rust
use crate::modules::Command;
|
|
|
|
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<String>,
|
|
server: String,
|
|
port: u16,
|
|
|
|
}
|
|
|
|
pub struct InvadeCommand;
|
|
|
|
impl Command for InvadeCommand {
|
|
fn handle(&self, message: &str) -> Vec<String> {
|
|
let mut response = vec![];
|
|
|
|
if message.contains("PRIVMSG") && message.contains(":%invade") {
|
|
let parts: Vec<&str> = message.split_whitespace().collect();
|
|
let num_invaders = parts[4].parse::<u32>().unwrap_or(1) as usize;
|
|
let channel = parts[2];
|
|
let invadechannel = parts[5];
|
|
let scream = if parts.len() > 6 { parts[6] } else { "" };
|
|
let config_str = std::fs::read_to_string("config.toml").unwrap();
|
|
let config_value = config_str.parse::<Value>().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();
|
|
let config_clone = config.clone();
|
|
let screaming = scream.to_string();
|
|
let command_channel = channel.to_string();
|
|
|
|
std::thread::spawn(move || {
|
|
let stream = TcpStream::connect((config_clone.server.as_str(), config_clone.port)).unwrap();
|
|
let connector = SslConnector::builder(SslMethod::tls()).unwrap().build();
|
|
let mut ssl_stream = connector.connect(config_clone.server.as_str(), stream).unwrap();
|
|
let nick_command = format!("NICK {}\r\n", thread_invader);
|
|
let user_command = format!("USER {} 0 * :{}\r\n", thread_invader, thread_invader);
|
|
ssl_stream.write_all(nick_command.as_bytes()).unwrap();
|
|
ssl_stream.write_all(user_command.as_bytes()).unwrap();
|
|
let join_command = format!("JOIN {} \r\n", thread_channel);
|
|
let commander = format!("JOIN {} \r\n", command_channel);
|
|
ssl_stream.write_all(commander.as_bytes()).unwrap();
|
|
ssl_stream.write_all(join_command.as_bytes()).unwrap();
|
|
let msg = format!("PRIVMSG {} :{}\r\n", thread_channel, screaming);
|
|
ssl_stream.write_all(msg.as_bytes()).unwrap();
|
|
|
|
loop {
|
|
|
|
|
|
|
|
let mut buf = [0; 512];
|
|
match ssl_stream.ssl_read(&mut buf) {
|
|
Ok(0) => break,
|
|
Ok(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:".bold().green(), thread_invader.blue());
|
|
ssl_stream.write_all(response.as_bytes()).unwrap();
|
|
}
|
|
// setup so these will only run from the commander
|
|
if message.contains("PRIVMSG") && message.contains(":%%join") {
|
|
let parts: Vec<&str> = message.splitn(3, ":%%join ").collect();
|
|
let invade_channel = parts[1];
|
|
let response = format!("JOIN {} \r\n", invade_channel);
|
|
ssl_stream.write_all(response.as_bytes()).unwrap();
|
|
}
|
|
if message.contains("PRIVMSG") && message.contains(":%%leave") {
|
|
let parts: Vec<&str> = message.splitn(3, ":%%leave ").collect();
|
|
let invade_channel = parts[1];
|
|
let response = format!("PART {} \r\n", invade_channel);
|
|
ssl_stream.write_all(response.as_bytes()).unwrap();
|
|
}
|
|
if message.contains("PRIVMSG") && message.contains(":%%scream") {
|
|
let parts: Vec<&str> = message.splitn(3, ":%%scream ").collect();
|
|
let invade_channel = parts[1];
|
|
if parts.len() == 2 {
|
|
let scream = parts[1];
|
|
let response = format!("PRIVMSG {} :{}\r\n", invade_channel, scream);
|
|
ssl_stream.write_all(response.as_bytes()).unwrap();
|
|
}
|
|
}
|
|
}
|
|
Err(e) => {
|
|
eprintln!("{} {}","[!] ERROR FROM SERVER:".on_red(), e);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
response.push(format!("PRIVMSG {} :\x0304,01[!] INVADING {} WITH {} INVADERS...\x0f\r\n", channel, invadechannel, num_invaders));
|
|
}
|
|
|
|
response
|
|
}
|
|
} |