You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.2 KiB
Rust
41 lines
1.2 KiB
Rust
use crate::error::SerenityUtilsResult;
|
|
use serenity::builder::CreateMessage;
|
|
use std::future::Future;
|
|
use std::pin::Pin;
|
|
use std::sync::Arc;
|
|
|
|
pub type MessageBuildOutput<'b> =
|
|
Pin<Box<dyn Future<Output = SerenityUtilsResult<CreateMessage<'b>>> + Send + 'b>>;
|
|
pub type MessageBuilderFn<'b> = Arc<dyn Fn() -> MessageBuildOutput<'b> + Send + Sync>;
|
|
|
|
#[derive(Clone)]
|
|
/// A page that stores a builder function for message pages
|
|
/// or static pages
|
|
pub enum Page<'b> {
|
|
Builder(MessageBuilderFn<'b>),
|
|
Static(CreateMessage<'b>),
|
|
}
|
|
|
|
impl<'b> Page<'b> {
|
|
/// Creates a new page with the given builder function
|
|
pub fn new_builder<F: 'static>(builder_fn: F) -> Self
|
|
where
|
|
F: Fn() -> MessageBuildOutput<'b> + Send + Sync,
|
|
{
|
|
Self::Builder(Arc::new(builder_fn))
|
|
}
|
|
|
|
/// Creates a new page with a static message
|
|
pub fn new_static(page: CreateMessage<'b>) -> Self {
|
|
Self::Static(page)
|
|
}
|
|
|
|
/// Returns the CreateMessage of the page
|
|
pub async fn get(&self) -> SerenityUtilsResult<CreateMessage<'b>> {
|
|
match self {
|
|
Page::Builder(b) => b().await,
|
|
Page::Static(inner) => Ok(inner.clone()),
|
|
}
|
|
}
|
|
}
|