118 lines
No EOL
4.4 KiB
Rust
118 lines
No EOL
4.4 KiB
Rust
use twitch_irc::login::StaticLoginCredentials;
|
||
use twitch_irc::ClientConfig;
|
||
use twitch_irc::SecureTCPTransport;
|
||
use twitch_irc::TwitchIRCClient;
|
||
use twitch_irc::message::ServerMessage;
|
||
use std::env;
|
||
use std::time::Instant;
|
||
use rand::Rng;
|
||
|
||
#[tokio::main]
|
||
pub async fn main() {
|
||
let login_name = "modulatingforcebot".to_owned();
|
||
let oauth_token = env::var("access_token").unwrap().to_owned();
|
||
|
||
|
||
/*
|
||
Vector of channels to join
|
||
*/
|
||
|
||
let mut botchannels = Vec::new();
|
||
/* botchannels.push(String::from("modulatingforcebot"));
|
||
botchannels.push(String::from("notohh"));
|
||
botchannels.push(String::from("modulatingforce"));
|
||
botchannels.push(String::from("secondsocksan"));
|
||
*/
|
||
|
||
for chnl in env::var("bot_channels").unwrap().split(',') {
|
||
// println!("(Env Var # {})",chnl);
|
||
botchannels.push(String::from(chnl));
|
||
}
|
||
|
||
|
||
let config = ClientConfig::new_simple(
|
||
StaticLoginCredentials::new(login_name, Some(oauth_token))
|
||
);
|
||
let (mut incoming_messages, client) =
|
||
TwitchIRCClient::<SecureTCPTransport, StaticLoginCredentials>::new(config);
|
||
|
||
|
||
// client.join("modulatingforcebot".to_owned()).unwrap();
|
||
// client.say("modulatingforcebot".to_owned(), "Connected!".to_owned()).await.unwrap();
|
||
|
||
for chnl in botchannels {
|
||
client.join(chnl.to_owned()).unwrap();
|
||
// client.say(chnl.to_owned(), "Connected!".to_owned()).await.unwrap();
|
||
client.say(chnl.to_owned(), "annytfLurk".to_owned()).await.unwrap();
|
||
}
|
||
|
||
// Adding rate limit functionality to be under : https://dev.twitch.tv/docs/irc/#rate-limits
|
||
|
||
let mut ratelimittimer = false.to_owned();
|
||
|
||
let mut ratelimitstart = Instant::now();
|
||
let mut ratelimitcounter = 0;
|
||
|
||
|
||
let join_handle = tokio::spawn(async move {
|
||
while let Some(message) = incoming_messages.recv().await {
|
||
// Below can be used to debug if I want to capture all messages
|
||
// println!("Received message: {:?}", message);
|
||
|
||
match message {
|
||
ServerMessage::Notice(msg) => {
|
||
if let Some(chnl) = msg.channel_login {
|
||
println!("NOTICE : (#{}) {}", chnl, msg.message_text);
|
||
}
|
||
}
|
||
ServerMessage::Privmsg(msg) => {
|
||
println!("(#{}) {}: {}", msg.channel_login, msg.sender.name, msg.message_text);
|
||
if !ratelimittimer {
|
||
println!("(#{}) > {}", msg.channel_login, "started rate limit timer");
|
||
// ratelimitstart = Instant::now();
|
||
// ratelimittimer = true;
|
||
( ratelimittimer , ratelimitstart ) = ( true , Instant::now() );
|
||
} else if ratelimittimer && ratelimitstart.elapsed().as_secs() < 30 && ratelimitcounter >= 20 {
|
||
// skip iteration if rate limit is being reached
|
||
println!("(#{}) > {}", msg.channel_login, "rate limit reached");
|
||
continue;
|
||
} else if ratelimitstart.elapsed().as_secs() >= 30 {
|
||
println!("(#{}) > {}", msg.channel_login, "rate limit timer reset");
|
||
// ratelimittimer = false;
|
||
// ratelimitcounter = 0;
|
||
( ratelimittimer , ratelimitcounter ) = ( false , 0 ) ;
|
||
}
|
||
|
||
let maxblanks = rand::thread_rng().gen_range(1..=5);
|
||
let mut outmsg = "GotTrolled".to_owned();
|
||
|
||
for _i in 1..maxblanks {
|
||
let blankspace: &str = "";
|
||
outmsg.push_str(blankspace);
|
||
}
|
||
|
||
client.say_in_reply_to(&msg,outmsg).await.unwrap();
|
||
println!("(#{}) > {}", msg.channel_login, "rate limit counter increase");
|
||
ratelimitcounter += 1;
|
||
// client.say(msg.channel_login, "GotTrolled".to_owned()).await.unwrap();
|
||
},
|
||
ServerMessage::Whisper(msg) => {
|
||
println!("(w) {}: {}", msg.sender.name, msg.message_text);
|
||
},
|
||
ServerMessage::Join(msg) => {
|
||
println!("JOINED: {}", msg.channel_login);
|
||
},
|
||
ServerMessage::Part(msg) => {
|
||
println!("PARTED: {}", msg.channel_login);
|
||
},
|
||
_ => {}
|
||
}
|
||
}
|
||
});
|
||
|
||
|
||
|
||
|
||
|
||
join_handle.await.unwrap();
|
||
} |