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

moved create_from_guild to the prefix handler.

Browse Source
This commit is contained in:
jude-lafitteIII 2020-05-29 16:13:20 +01:00
parent d6d825e86d
commit b4db99934d
2 changed files with 27 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/guilddata.rs
View File

@ -36,19 +36,19 @@ SELECT id, name, prefix, volume, allow_greets
pub async fn create_from_guild(guild: Guild, db_pool: MySqlPool) -> Result<GuildData, Box<dyn std::error::Error>> { pub async fn create_from_guild(guild: Guild, db_pool: MySqlPool) -> Result<GuildData, Box<dyn std::error::Error>> {
sqlx::query!( sqlx::query!(
" "
INSERT INTO roles (guild_id, role) INSERT INTO servers (id, name)
VALUES (?, ?) VALUES (?, ?)
", ", guild.id.as_u64(), guild.name
guild.id.as_u64(), guild.id.as_u64()
) )
.execute(&db_pool) .execute(&db_pool)
.await?; .await?;
sqlx::query!( sqlx::query!(
" "
INSERT INTO servers (id, name) INSERT INTO roles (guild_id, role)
VALUES (?, ?) VALUES (?, ?)
", guild.id.as_u64(), guild.name ",
guild.id.as_u64(), guild.id.as_u64()
) )
.execute(&db_pool) .execute(&db_pool)
.await?; .await?;

34
src/main.rs
View File

@ -339,10 +339,29 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
let pool = ctx.data.read().await let pool = ctx.data.read().await
.get::<SQLPool>().cloned().expect("Could not get SQLPool from data"); .get::<SQLPool>().cloned().expect("Could not get SQLPool from data");
match GuildData::get_from_id(*msg.guild_id.unwrap().as_u64(), pool).await { let guild = match msg.guild(&ctx.cache).await {
Some(guild) => Some(guild.prefix), Some(guild) => guild,
None => Some(String::from("?")) None => {
return Some(String::from("?"));
}
};
match GuildData::get_from_id(*msg.guild_id.unwrap().as_u64(), pool.clone()).await {
Some(mut guild_data) => {
let name = Some(guild.name);
if guild_data.name != name {
guild_data.name = name;
guild_data.commit(pool).await.unwrap();
}
Some(guild_data.prefix)
},
None => {
GuildData::create_from_guild(guild, pool).await.unwrap();
Some(String::from("?"))
}
} }
})) }))
.allow_dm(false) .allow_dm(false)
@ -539,11 +558,6 @@ async fn change_volume(ctx: &Context, msg: &Message, mut args: Args) -> CommandR
.get::<SQLPool>().cloned().expect("Could not get SQLPool from data"); .get::<SQLPool>().cloned().expect("Could not get SQLPool from data");
let mut guild_data_opt = GuildData::get_from_id(*guild.id.as_u64(), pool.clone()).await; let mut guild_data_opt = GuildData::get_from_id(*guild.id.as_u64(), pool.clone()).await;
if guild_data_opt.is_none() {
guild_data_opt = Some(GuildData::create_from_guild(guild, pool.clone()).await.unwrap())
}
let mut guild_data = guild_data_opt.unwrap(); let mut guild_data = guild_data_opt.unwrap();
if args.len() == 1 { if args.len() == 1 {
@ -590,10 +604,6 @@ async fn change_prefix(ctx: &Context, msg: &Message, mut args: Args) -> CommandR
{ {
let mut guild_data_opt = GuildData::get_from_id(*guild.id.as_u64(), pool.clone()).await; let mut guild_data_opt = GuildData::get_from_id(*guild.id.as_u64(), pool.clone()).await;
if guild_data_opt.is_none() {
guild_data_opt = Some(GuildData::create_from_guild(guild, pool.clone()).await.unwrap())
}
guild_data = guild_data_opt.unwrap(); guild_data = guild_data_opt.unwrap();
} }