feat(paginate): copy paginate logic over and restrict to the author id

Author: DevYukine

src/abstraction/command.rs

@@ -1,9 +1,15 @@
 use crate::built_info;
 use lazy_static::lazy_static;
+use poise::CreateReply;
+use poise::serenity_prelude::CreateActionRow;
 use reqwest::header::HeaderMap;
 use serenity::all::RoleId;
+use serenity::builder::{
+	CreateButton, CreateEmbed, CreateInteractionResponse, CreateInteractionResponseMessage,
+};
 use std::env;
 use std::sync::Arc;
+use std::time::Duration;
 
 pub type CommandError = anyhow::Error;
 pub type CommandContext<'a> = poise::Context<'a, CommandData, CommandError>;
@@ -89,3 +95,67 @@ pub async fn is_user_trusted_or_above(ctx: CommandContext<'_>) -> CheckResult {
 
 	Ok(false)
 }
+
+pub async fn paginate<U, E>(
+	ctx: poise::Context<'_, U, E>,
+	pages: &[&str],
+) -> Result<(), serenity::Error> {
+	// Define some unique identifiers for the navigation buttons
+	let ctx_id = ctx.id();
+	let author_id = ctx.author().id;
+	let prev_button_id = format!("{}prev", ctx_id);
+	let next_button_id = format!("{}next", ctx_id);
+
+	// Send the embed with the first page as content
+	let reply = {
+		let components = CreateActionRow::Buttons(vec![
+			CreateButton::new(&prev_button_id).emoji('◀'),
+			CreateButton::new(&next_button_id).emoji('▶'),
+		]);
+
+		CreateReply::default()
+			.embed(CreateEmbed::default().description(pages[0]))
+			.components(vec![components])
+	};
+
+	ctx.send(reply).await?;
+
+	// Loop through incoming interactions with the navigation buttons
+	let mut current_page = 0;
+	while let Some(press) = serenity::collector::ComponentInteractionCollector::new(ctx)
+		// We defined our button IDs to start with `ctx_id`. If they don't, some other command's
+		// button was pressed
+		.filter(move |press| {
+			press.data.custom_id.starts_with(&ctx_id.to_string()) && press.user.id == author_id
+		})
+		// Timeout when no navigation button has been pressed for 24 hours
+		.timeout(Duration::from_secs(3600 * 24))
+		.await
+	{
+		// Depending on which button was pressed, go to next or previous page
+		if press.data.custom_id == next_button_id {
+			current_page += 1;
+			if current_page >= pages.len() {
+				current_page = 0;
+			}
+		} else if press.data.custom_id == prev_button_id {
+			current_page = current_page.checked_sub(1).unwrap_or(pages.len() - 1);
+		} else {
+			// This is an unrelated button interaction
+			continue;
+		}
+
+		// Update the message with the new page contents
+		press
+			.create_response(
+				ctx.serenity_context(),
+				CreateInteractionResponse::UpdateMessage(
+					CreateInteractionResponseMessage::new()
+						.embed(CreateEmbed::new().description(pages[current_page])),
+				),
+			)
+			.await?;
+	}
+
+	Ok(())
+}

src/abstraction/playmatch.rs

@@ -1,4 +1,4 @@
-use crate::abstraction::command::CommandContext;
+use crate::abstraction::command::{CommandContext, paginate};
 use crate::util::create_discord_markdown_table;
 use playmatch_client::types::{
 	CompanyMetadataResponse, ExternalMetadata, MatchType, MetadataProvider,
@@ -109,7 +109,7 @@ where
 		.map(|s| s.as_str())
 		.collect::<Vec<&str>>();
 
-	poise::builtins::paginate(ctx, markdown_pages_ref.as_slice()).await?;
+	paginate(ctx, markdown_pages_ref.as_slice()).await?;
 
 	Ok(())
 }

src/command/playmatch.rs

@@ -838,11 +838,7 @@ async fn handle_suggestion_message(data: SuggestionMessageHandleData) -> Command
 				.await?;
 
 			let message = match data.r#type {
-				SuggestionType::Platform => format!(
-					"Your Playmatch Metadata Suggestion for {display_type} **{}** was approved! Thank you very much for your contribution 🎉!",
-					data.name
-				),
-				SuggestionType::Company => format!(
+				SuggestionType::Platform | SuggestionType::Company => format!(
 					"Your Playmatch Metadata Suggestion for {display_type} **{}** was approved! Thank you very much for your contribution 🎉!",
 					data.name
 				),