j00lz/pinbot-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Make any errors not crash the entire application
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse source
This commit is contained in:
Julius 2022-06-24 22:43:08 +02:00
parent 1ab7e4bffe
commit fca4872a73
Signed by: j00lz
GPG key ID: AF241B0AA237BBA2
1 changed files with 38 additions and 35 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

73
src/main.rs
View file

@ -2,7 +2,8 @@ use serenity::{
client::{Context, EventHandler}, client::{Context, EventHandler},
model::{ model::{
channel::Reaction, channel::Reaction,
gateway::{Activity, Ready}, id::{ChannelId, MessageId}, gateway::{Activity, Ready},
id::{ChannelId, MessageId},
}, },
prelude::GatewayIntents, prelude::GatewayIntents,
Client, Client,
@ -20,48 +21,50 @@ impl EventHandler for Handler {
} }
async fn reaction_add(&self, c: Context, r: Reaction) { async fn reaction_add(&self, c: Context, r: Reaction) {
let emoji = r.emoji.as_data(); if let Err(e) = handle_react(c, r).await {
if emoji == "📌" { println!("Error: {:?}", e);
let msg = r.message(c.http.clone()).await.expect("Could not find message");
for r in &msg.reactions {
let emoji = r.reaction_type.as_data();
if emoji == "📌" {
let count = r.count;
if count >= 5 {
msg.pin(c.http.clone()).await.expect("Could not pin message");
}
}
}
} }
} }
async fn reaction_remove(&self, c: Context, r: Reaction) { async fn reaction_remove(&self, c: Context, r: Reaction) {
let emoji = r.emoji.as_data(); if let Err(e) = handle_react(c, r).await {
if emoji == "📌" { println!("Error: {:?}", e);
let msg = r.message(&c.http).await.expect("Could not find message"); }
for r in &msg.reactions { }
let emoji = r.reaction_type.as_data();
if emoji == "📌" { async fn reaction_remove_all(&self, c: Context, c_id: ChannelId, m_id: MessageId) {
let count = r.count; let msg = match c.http.get_message(c_id.0, m_id.0).await {
if count < 3 { Ok(m) => m,
msg.unpin(&c.http).await.expect("Could not unpin message"); Err(e) => {
} println!("Error: {:?}", e);
return;
}
};
if msg.pinned {
if let Err(e) = msg.unpin(&c.http).await {
println!("Error: {:?}", e);
}
}
}
}
async fn handle_react(c: Context, r: Reaction) -> Result<(), serenity::Error> {
let emoji = r.emoji.as_data();
if emoji == "📌" {
let msg = r.message(c.http.clone()).await?;
for r in &msg.reactions {
let emoji = r.reaction_type.as_data();
if emoji == "📌" {
let count = r.count;
if count >= 5 {
msg.pin(c.http.clone()).await?;
} else if count < 3 {
msg.unpin(&c.http).await?;
} }
} }
} }
} }
Ok(())
async fn reaction_remove_all(
&self,
c: Context,
c_id: ChannelId,
m_id: MessageId,
) {
let msg = c.http.get_message(c_id.0, m_id.0).await.expect("Could not find message");
if msg.pinned {
msg.unpin(&c.http).await.expect("Could not unpin message");
}
}
} }
#[tokio::main] #[tokio::main]