feat: add vote api

Author: JayDwee

api/Cargo.lock

@@ -16,12 +16,14 @@ twilight = "0.16.0"
 twilight-http = "0.16.0"
 twilight-model = "0.16.0"
 aws-sdk-dynamodb = "1.89.0"
+aws-sdk-scheduler = "1.84.0"
 aws-config = "1.8.5"
 reqwest = { version = "0.12.23", features = ["json", "native-tls-vendored"] }
-chrono = "0.4.41"
+chrono = {  version = "0.4.41", features = ["serde"] }
 ed25519-dalek = "2.2.0"
 once_cell = "1.21.3"
 hex = "0.4.3"
 http-body-util = "0.1.3"
 regex = "1.11.1"
 cached = { version = "0.56.0", features = ["async"] }
+uuid = { version = "1.18.0", features = ["v4"] }

api/Cargo.toml

@@ -6,9 +6,11 @@ use tokio::time::sleep;
 use twilight_http::{Client, Error, Response};
 use twilight_http::api_error::ApiError;
 use twilight_http::error::ErrorType;
+use twilight_model::channel::Message;
+use twilight_model::channel::message::Component;
 use twilight_model::guild::{Guild, Member, Role};
 use twilight_model::id::Id;
-use twilight_model::id::marker::{GuildMarker, RoleMarker, UserMarker};
+use twilight_model::id::marker::{ChannelMarker, GuildMarker, MessageMarker, RoleMarker, UserMarker};
 use twilight_model::user::{CurrentUser, CurrentUserGuild};
 
 pub fn ise<T: std::fmt::Debug>(e: T) -> StatusCode {
@@ -46,8 +48,14 @@ where
 
 pub fn as_http_err(e: Error) -> StatusCode {
     match e.kind() {
-        ErrorType::Response{ status, .. } => StatusCode::from_u16(status.get()).map_err(ise).unwrap(),
-        _ => StatusCode::INTERNAL_SERVER_ERROR
+        ErrorType::Response{ status, error, .. } => {
+            error!("Discord Api Error: {} {:?}", status, error);
+            StatusCode::from_u16(status.get()).map_err(ise).unwrap()
+        },
+        _ => {
+            error!("Internal Server Error: {:?}", e);
+            StatusCode::INTERNAL_SERVER_ERROR
+        }
     }
 }
 
@@ -93,3 +101,36 @@ pub async fn get_guild_role(guild_id: Id<GuildMarker>, role_id: Id<RoleMarker>,
     retry_on_rl(|| async { client.role(guild_id, role_id).await}).await.map_err(as_http_err)?.model().await.map_err(ise)
 }
 
+pub async fn create_message(channel_id: Id<ChannelMarker>, 
+                            content: Option<&str>, 
+                            components: Option<&[Component]>,
+                            client: &Client) -> Result<Message,StatusCode> {
+    retry_on_rl(|| async {
+        let mut msg_builder = client.create_message(channel_id);
+        if let Some(content) = content {
+            msg_builder = msg_builder.content(content);
+        }
+        if let Some(components) = components {
+            msg_builder = msg_builder.components(components);
+        }
+        msg_builder.await
+    
+    }).await.map_err(as_http_err)?.model().await.map_err(ise)
+}
+
+pub async fn update_message(channel_id: Id<ChannelMarker>,
+                            message_id: Id<MessageMarker>,
+                          content: Option<Option<&str>>,
+                          components: Option<Option<&[Component]>>,
+                          client: &Client) -> Result<Message,StatusCode> {
+                          retry_on_rl(|| async {
+                              let mut msg_builder = client.update_message(channel_id, message_id);
+                              if let Some(content) = content {
+                                  msg_builder = msg_builder.content(content);
+                              }
+                              if let Some(components) = components {
+                                  msg_builder = msg_builder.components(components);
+                              }
+                              msg_builder.await
+                          }).await.map_err(as_http_err)?.model().await.map_err(ise)
+      }

api/src/discord.rs

@@ -1,5 +1,5 @@
 use aws_sdk_dynamodb::types::AttributeValue;
-use std::collections::HashMap;
+use std::collections::{HashMap, HashSet};
 
 pub fn as_string_opt(val: Option<&AttributeValue>) -> Option<String> {
     if let Some(v) = val {
@@ -95,3 +95,15 @@ pub fn as_map(val: Option<&AttributeValue>) -> Option<&HashMap<String, Attribute
     }
     None
 }
+
+pub fn as_u64_set(val: Option<&AttributeValue>) -> HashSet<u64> {
+    if let Some(val) = val {
+        if let Ok(val) = val.as_l() {
+            return val
+                .iter()
+                .map(|v| as_u64(Some(v), 0))
+                .collect();
+        }
+    }
+    HashSet::new()
+}

api/src/dynamo.rs

@@ -1,6 +1,6 @@
 use std::collections::{HashSet};
 use crate::AppState;
-use crate::guilds::models::{Guild, VerifyRole};
+use crate::guilds::models::Guild;
 use axum::extract::State;
 use axum::{Extension, Json, Router, extract::Path, routing::get};
 use http::StatusCode;
@@ -16,18 +16,21 @@ use twilight_model::id::marker::GuildMarker;
 use twilight_model::user::{CurrentUser, CurrentUserGuild};
 use crate::discord::{get_guild, get_guild_member};
 use crate::guilds::tasks::{add_role_to_guild, remove_role_from_guild};
+use crate::guilds::verify::models::VerifyRole;
 use crate::utils::{admin_guilds, is_client_admin_guild};
 
 pub mod models;
 pub mod verify;
 pub(crate) mod utils;
 pub mod tasks;
+pub(crate) mod votes;
 
 pub fn router() -> Router<AppState> {
     Router::new()
         .route("/", get(get_guilds).post(post_guilds))
         .route("/{guild_id}", get(get_guilds_id).put(put_guilds_id).post(post_guilds_id))
-        .nest("/{guild_id}/verify", verify::router())
+        .nest("/{guild_id}/verify", verify::controllers::router())
+        .nest("/{guild_id}/votes", votes::controllers::router())
         .layer(CorsLayer::permissive())
 }