1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

//! Shuttle service integration for the Poise discord bot framework.
//!
//! ## Example
//!
//! ```rust,no_run
//! use poise::serenity_prelude as serenity;
//! use shuttle_secrets::SecretStore;
//! use shuttle_poise::ShuttlePoise;
//!
//! struct Data {} // User data, which is stored and accessible in all command invocations
//! type Error = Box<dyn std::error::Error + Send + Sync>;
//! type Context<'a> = poise::Context<'a, Data, Error>;
//!
//! /// Responds with "world!"
//! #[poise::command(slash_command)]
//! async fn hello(ctx: Context<'_>) -> Result<(), Error> {
//!     ctx.say("world!").await?;
//!     Ok(())
//! }
//!
//! #[shuttle_runtime::main]
//! async fn poise(#[shuttle_secrets::Secrets] secret_store: SecretStore) -> ShuttlePoise<Data, Error> {
//!     // Get the discord token set in `Secrets.toml`
//!     let discord_token = secret_store
//!         .get("DISCORD_TOKEN")
//!         .expect("'DISCORD_TOKEN' was not found");
//!
//!     let framework = poise::Framework::builder()
//!         .options(poise::FrameworkOptions {
//!             commands: vec![hello()],
//!             ..Default::default()
//!         })
//!         .token(discord_token)
//!         .intents(serenity::GatewayIntents::non_privileged())
//!         .setup(|ctx, _ready, framework| {
//!             Box::pin(async move {
//!                 poise::builtins::register_globally(ctx, &framework.options().commands).await?;
//!                 Ok(Data {})
//!             })
//!         })
//!         .build()
//!         .await
//!         .map_err(shuttle_runtime::CustomError::new)?;
//!
//!     Ok(framework.into())
//! }
//! ```
use std::net::SocketAddr;
use std::sync::Arc;

/// A wrapper type for [poise::Framework] so we can implement [shuttle_runtime::Service] for it.
pub struct PoiseService<T, E>(pub Arc<poise::Framework<T, E>>);

#[shuttle_runtime::async_trait]
impl<T, E> shuttle_runtime::Service for PoiseService<T, E>
where
    T: Send + Sync + 'static,
    E: Send + Sync + 'static,
{
    async fn bind(mut self, _addr: SocketAddr) -> Result<(), shuttle_runtime::Error> {
        self.0
            .start_autosharded()
            .await
            .map_err(shuttle_runtime::CustomError::new)?;

        Ok(())
    }
}

impl<T, E> From<Arc<poise::Framework<T, E>>> for PoiseService<T, E> {
    fn from(framework: Arc<poise::Framework<T, E>>) -> Self {
        Self(framework)
    }
}

/// Return type from the `[shuttle_runtime::main]` macro for a Poise-based service.
///
/// ## Example
///
/// ```rust,no_run
/// use poise::serenity_prelude as serenity;
/// use shuttle_secrets::SecretStore;
/// use shuttle_poise::ShuttlePoise;
///
/// struct Data {} // User data, which is stored and accessible in all command invocations
/// type Error = Box<dyn std::error::Error + Send + Sync>;
/// type Context<'a> = poise::Context<'a, Data, Error>;
///
///
/// #[poise::command(slash_command)]
/// async fn hello(ctx: Context<'_>) -> Result<(), Error> {
///     ctx.say("world!").await?;
///     Ok(())
/// }
///
/// #[shuttle_runtime::main]
/// async fn poise(#[shuttle_secrets::Secrets] secret_store: SecretStore) -> ShuttlePoise<Data, Error> {
///
///     let discord_token = secret_store
///         .get("DISCORD_TOKEN")
///         .expect("'DISCORD_TOKEN' was not found");
///
///     let framework = poise::Framework::builder()
///         .options(poise::FrameworkOptions {
///             commands: vec![hello()],
///             ..Default::default()
///         })
///         .token(discord_token)
///         .intents(serenity::GatewayIntents::non_privileged())
///         .setup(|ctx, _ready, framework| {
///             Box::pin(async move {
///                 poise::builtins::register_globally(ctx, &framework.options().commands).await?;
///                 Ok(Data {})
///             })
///         })
///         .build()
///         .await
///         .map_err(shuttle_runtime::CustomError::new)?;
///
///     Ok(framework.into())
/// }
/// ```
pub type ShuttlePoise<T, E> = Result<PoiseService<T, E>, shuttle_runtime::Error>;