jude/soundfx-bot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 3 Wiki Activity

check for patreon role

Browse Source
This commit is contained in:
jude-lafitteIII 2020-04-30 01:21:34 +01:00
parent 9f2317eb6a
commit 5a0b208d81
1 changed files with 52 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
src/main.rs
View File

@ -16,6 +16,10 @@ use serenity::{
} }
}, },
model::{ model::{
id::{
GuildId,
RoleId,
},
channel::Message, channel::Message,
guild::Guild, guild::Guild,
}, },
@ -70,6 +74,16 @@ lazy_static! {
dotenv().unwrap(); dotenv().unwrap();
env::var("MAX_SOUNDS").unwrap().parse::<u32>().unwrap() env::var("MAX_SOUNDS").unwrap().parse::<u32>().unwrap()
}; };
static ref PATREON_GUILD: u64 = {
dotenv().unwrap();
env::var("PATREON_GUILD").unwrap().parse::<u64>().unwrap()
};
static ref PATREON_ROLE: u64 = {
dotenv().unwrap();
env::var("PATREON_ROLE").unwrap().parse::<u64>().unwrap()
};
} }
#[group] #[group]
@ -591,45 +605,57 @@ async fn upload_new_sound(ctx: &mut Context, msg: &Message, mut args: Args) -> C
// need to check how many sounds user currently has // need to check how many sounds user currently has
let count = Sound::count_user_sounds(*msg.author.id.as_u64(), pool.clone()).await?; let count = Sound::count_user_sounds(*msg.author.id.as_u64(), pool.clone()).await?;
let mut permit_upload = true;
// need to check if user is patreon or nah // need to check if user is patreon or nah
if count >= *MAX_SOUNDS { if count >= *MAX_SOUNDS {
let patreon_guild_member = GuildId(*PATREON_GUILD).member(&ctx, msg.author.id).await?;
if patreon_guild_member.roles.contains(&RoleId(*PATREON_ROLE)) {
permit_upload = true;
}
else {
permit_upload = false;
}
} }
msg.channel_id.say(&ctx, "Please now upload an audio file under 1MB in size:").await?; if permit_upload {
msg.channel_id.say(&ctx, "Please now upload an audio file under 1MB in size:").await?;
let reply = msg.channel_id.await_reply(&ctx) let reply = msg.channel_id.await_reply(&ctx)
.author_id(msg.author.id) .author_id(msg.author.id)
.timeout(Duration::from_secs(30)) .timeout(Duration::from_secs(30))
.await; .await;
match reply { match reply {
Some(reply_msg) => { Some(reply_msg) => {
if reply_msg.attachments.len() == 1 { if reply_msg.attachments.len() == 1 {
match Sound::create_anon( match Sound::create_anon(
new_name, new_name,
&reply_msg.attachments[0].url, &reply_msg.attachments[0].url,
*msg.guild_id.unwrap().as_u64(), *msg.guild_id.unwrap().as_u64(),
*msg.author.id.as_u64(), *msg.author.id.as_u64(),
pool).await { pool).await {
Ok(_) => { Ok(_) => {
msg.channel_id.say(&ctx, "Sound has been uploaded").await?; msg.channel_id.say(&ctx, "Sound has been uploaded").await?;
} }
Err(_) => { Err(_) => {
msg.channel_id.say(&ctx, "Sound failed to upload. Size may be too large").await?; msg.channel_id.say(&ctx, "Sound failed to upload. Size may be too large").await?;
}
} }
} else {
msg.channel_id.say(&ctx, "Please upload 1 attachment following your upload command. Aborted").await?;
} }
} }
else {
msg.channel_id.say(&ctx, "Please upload 1 attachment following your upload command. Aborted").await?; None => {
msg.channel_id.say(&ctx, "Upload timed out. Please redo the command").await?;
} }
} }
}
None => { else {
msg.channel_id.say(&ctx, "Upload timed out. Please redo the command").await?; msg.channel_id.say(&ctx, "You have reached the maximum number of sounds ({}). Either delete some with `{}delete` or join our Patreon for unlimited uploads at **https://patreon.com/jellywx**").await?;
}
} }
} }
else { else {