revanced-discord-bot/src/utils/poll.rs

use poise::serenity_prelude::{ButtonStyle, ReactionType, Timestamp};

use base64::Engine;
use reqwest::StatusCode;
use tracing::log::{error, trace};

use super::bot::get_data_lock;
use super::*;

pub async fn handle_poll(
    ctx: &serenity::Context,
    interaction: &serenity::Interaction,
    poll_id: u64,
    min_join_date: Timestamp,
) -> Result<(), crate::serenity::SerenityError> {
    trace!("Handling poll: {}.", poll_id);

    let data = get_data_lock(ctx).await;
    let data = data.read().await;

    let component = &interaction.clone().message_component().unwrap();

    let member = component.member.as_ref().unwrap();

    let eligible = member.joined_at.unwrap() <= min_join_date;
    let result = if eligible {
        match data
            .api
            .authenticate(&base64::engine::general_purpose::STANDARD_NO_PAD.encode(
                hmac_sha256::HMAC::mac(member.user.id.to_string(), &data.poll_secret),
            ))
            .await
        {
            Ok(auth) => Ok(auth.access_token),
            Err(err) => match err.status() {
                Some(StatusCode::PRECONDITION_FAILED) => Err("You can only vote once."),
                _ => {
                    error!("API Request error: {:?}", err);
                    Err("API Request failed. Please try again later.")
                },
            },
        }
    } else {
        Err("You are not eligible to vote on this poll.")
    };

    component
        .create_interaction_response(&ctx.http, |r| {
            r.interaction_response_data(|m| {
                if let Ok(token) = result.as_deref() {
                    let url = format!("https://revanced.app/poll#{token}");
                    m.components(|c| {
                        c.create_action_row(|r| {
                            r.create_button(|b| {
                                b.label("Vote")
                                    .emoji(ReactionType::Unicode("🗳️".to_string()))
                                    .style(ButtonStyle::Link)
                                    .url(&url)
                            })
                        })
                    })
                } else {
                    m
                }
                .ephemeral(true)
                .embed(|e| {
                    match result {
                        Ok(_) => e
                            .title("Cast your vote")
                            .description("You can now vote on the poll."),
                        Err(msg) => e.title("Error").description(msg),
                    }
                    .color(data.configuration.general.embed_color)
                    .thumbnail(member.user.face())
                })
            })
        })
        .await?;

    Ok(())
}