diff options
| -rw-r--r-- | .drone.yml | 8 | ||||
| -rw-r--r-- | Cargo.lock | 7 | ||||
| -rw-r--r-- | Cargo.toml | 1 | ||||
| -rw-r--r-- | build.rs | 5 | ||||
| -rw-r--r-- | sqlx-data.json | 132 | ||||
| -rw-r--r-- | src/handler.rs | 73 | ||||
| -rw-r--r-- | src/main.rs | 3 |
7 files changed, 221 insertions, 8 deletions
@@ -6,12 +6,4 @@ steps: - name: build image: rust:latest commands: - - apt update - - apt install postgresql -y - - pg_ctlcluster 13 main start - - cargo install sqlx-cli - - cargo sqlx migrate run - cargo check - - cargo fmt -- --check - environment: - DATABASE_URL: postgres://postgres@localhost/campmaster @@ -310,6 +310,9 @@ name = "either" version = "1.6.1" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "e78d4f1cc4ae33bbfc157ed5d5a5ef3bc29227303d595861deb238fcec4e9457" +dependencies = [ + "serde", +] [[package]] name = "encoding_rs" @@ -1301,6 +1304,7 @@ version = "1.0.79" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "8e8d9fa5c3b304765ce1fd9c4c8a3de2c8db365a5b91be52f186efc675681d95" dependencies = [ + "indexmap", "itoa", "ryu", "serde", @@ -1501,9 +1505,12 @@ dependencies = [ "dotenv", "either", "heck", + "hex", "once_cell", "proc-macro2", "quote", + "serde", + "serde_json", "sha2", "sqlx-core", "sqlx-rt", @@ -65,4 +65,5 @@ features = [ "chrono", "migrate", "macros", + "offline", ] diff --git a/build.rs b/build.rs new file mode 100644 index 0000000..7609593 --- /dev/null +++ b/build.rs @@ -0,0 +1,5 @@ +// generated by `sqlx migrate build-script` +fn main() { + // trigger recompilation when a new migration is added + println!("cargo:rerun-if-changed=migrations"); +}
\ No newline at end of file diff --git a/sqlx-data.json b/sqlx-data.json new file mode 100644 index 0000000..b6bb605 --- /dev/null +++ b/sqlx-data.json @@ -0,0 +1,132 @@ +{ + "db": "PostgreSQL", + "04383f28e708546beb642eb13c81048dc166b3a7994c110922abbced8470a93c": { + "query": "DELETE FROM reaction_roles WHERE id=$1", + "describe": { + "columns": [], + "parameters": { + "Left": [ + "Int4" + ] + }, + "nullable": [] + } + }, + "277984fe7b7a2aa1c3b02913db57caae607875e2389c9191d22a6efc46aa7e4a": { + "query": "SELECT * FROM reaction_roles WHERE message_id=$1", + "describe": { + "columns": [ + { + "ordinal": 0, + "name": "id", + "type_info": "Int4" + }, + { + "ordinal": 1, + "name": "channel_id", + "type_info": "Text" + }, + { + "ordinal": 2, + "name": "message_id", + "type_info": "Text" + }, + { + "ordinal": 3, + "name": "guild_id", + "type_info": "Text" + }, + { + "ordinal": 4, + "name": "reaction", + "type_info": "Text" + }, + { + "ordinal": 5, + "name": "role_id", + "type_info": "Text" + } + ], + "parameters": { + "Left": [ + "Text" + ] + }, + "nullable": [ + false, + false, + false, + false, + false, + false + ] + } + }, + "2900553c18f0b74b218a9ecf289cc4d74ceda4ebe1e5494f0b28a8160f5c651a": { + "query": "SELECT * FROM reaction_roles WHERE message_id=$1 AND reaction=$2", + "describe": { + "columns": [ + { + "ordinal": 0, + "name": "id", + "type_info": "Int4" + }, + { + "ordinal": 1, + "name": "channel_id", + "type_info": "Text" + }, + { + "ordinal": 2, + "name": "message_id", + "type_info": "Text" + }, + { + "ordinal": 3, + "name": "guild_id", + "type_info": "Text" + }, + { + "ordinal": 4, + "name": "reaction", + "type_info": "Text" + }, + { + "ordinal": 5, + "name": "role_id", + "type_info": "Text" + } + ], + "parameters": { + "Left": [ + "Text", + "Text" + ] + }, + "nullable": [ + false, + false, + false, + false, + false, + false + ] + } + }, + "f8d670f7e142e9658916ed389e2def233430420c4121aa473c0277c61c5cfcee": { + "query": "INSERT INTO reaction_roles (channel_id, message_id, guild_id, reaction, role_id) VALUES ($1, $2, $3, $4, $5)", + "describe": { + "columns": [], + "parameters": { + "Left": [ + "Text", + "Text", + "Text", + "Text", + "Text" + ] + }, + "nullable": [] + } + } +}
\ No newline at end of file diff --git a/src/handler.rs b/src/handler.rs new file mode 100644 index 0000000..ef8fc00 --- /dev/null +++ b/src/handler.rs @@ -0,0 +1,73 @@ +use poise::serenity_prelude as serenity; + +use crate::models::ReactionRole; +use crate::{Data, Error}; + +pub async fn event_handler( + ctx: &serenity::Context, + event: &poise::Event<'_>, + data: &Data, +) -> Result<(), Error> { + { + let pool = data.pg.lock().unwrap().clone(); + match event { + poise::Event::ReactionAdd { add_reaction } => { + let current_user = ctx.http.get_current_user().await?; + if add_reaction.user_id.unwrap() == current_user.id { + return Ok(()); + } + let rrole = sqlx::query_as!( + ReactionRole, + "SELECT * FROM reaction_roles WHERE message_id=$1 AND reaction=$2", + add_reaction.message_id.0.to_string(), + add_reaction.emoji.to_string() + ) + .fetch_one(&pool) + .await?; + let member = ctx + .http + .get_member( + rrole.guild_id.parse::<u64>()?, + add_reaction.user_id.unwrap().0, + ) + .await?; + let member_roles = member.roles; + let role_id = serenity::RoleId(rrole.role_id.parse::<u64>()?); + if member_roles.contains(&role_id) { + ctx.http + .remove_member_role( + member.guild_id.0, + member.user.id.0, + role_id.0, + Some("Reaction Role Menu"), + ) + .await?; + } else { + ctx.http + .add_member_role( + rrole.guild_id.parse::<u64>()?, + add_reaction.user_id.unwrap().0, + rrole.role_id.parse::<u64>()?, + Some("Reaction Role"), + ) + .await?; + } + + if let Ok(dm_chan) = add_reaction + .user_id + .unwrap() + .create_dm_channel(&ctx.http) + .await { + dm_chan.say(ctx, format!("Toggled the role!")).await?; + } else { + println!("Could not DM user, but we did the role anyways"); + } + + add_reaction.delete(&ctx.http).await?; + } + _ => (), + } + } + + Ok(()) +} diff --git a/src/main.rs b/src/main.rs index 2968f20..8693381 100644 --- a/src/main.rs +++ b/src/main.rs @@ -123,6 +123,9 @@ async fn main() { ) .await .expect("Couldn't connect to postgresql"); + sqlx::migrate!("./migrations") + .run(&pool) + .await.unwrap(); Ok(Data { pg: Mutex::new(pool), }) |