summaryrefslogtreecommitdiff
path: root/crates/bot-client/src
diff options
context:
space:
mode:
authorseth <[email protected]>2024-06-16 07:15:13 -0400
committerGitHub <[email protected]>2024-06-16 07:15:13 -0400
commitd25129d829e0ebd70b4e60e399fe91c0d80aa1ad (patch)
tree2a62992f2980f9fed2204ef5ef708a0228998cf1 /crates/bot-client/src
parenta0bfcc1587e3cef1b8f6fa0508a280fc48c82231 (diff)
use libgit2 to track PRs (#10)v0.2.0
* nix: don't depend on registry for nixpkgs input * use libgit2 to track PRs * nix: don't use ci devShell as defaul * crates: bump serenity from `9ad74d4` to `0.12.2 * nix: fix cross compiled builds * crates: split more from client * bot-jobs: update remote refs more efficiently * git-tracker: account for HEAD commits * bot-config: use nixpkgs branches from environment * bot-commands: don't display branches prs haven't landed in * git-tracker: return false when commits aren't found this is annoying as a hard error since it turns out github will report garbage merge commit SHAs for PRs that *haven't* been merged yet. yay * bot: improve docs in some places * bot-client: display invite link on start * bot-http: add TeawieClientExt * bot-commands: add /about * docs: update readme todos * nix: enable StateDirectory in module * crates: bump to 0.2.0
Diffstat (limited to 'crates/bot-client/src')
-rw-r--r--crates/bot-client/src/handler.rs112
-rw-r--r--crates/bot-client/src/lib.rs79
2 files changed, 191 insertions, 0 deletions
diff --git a/crates/bot-client/src/handler.rs b/crates/bot-client/src/handler.rs
new file mode 100644
index 0000000..2cd0082
--- /dev/null
+++ b/crates/bot-client/src/handler.rs
@@ -0,0 +1,112 @@
+use crate::{SharedConfig, SharedHttp};
+use bot_error::Error;
+
+use log::{debug, error, info, trace, warn};
+use serenity::all::CreateBotAuthParameters;
+use serenity::async_trait;
+use serenity::builder::{
+ CreateEmbed, CreateInteractionResponse, CreateInteractionResponseFollowup,
+ CreateInteractionResponseMessage,
+};
+use serenity::model::{
+ application::{Command, CommandInteraction, Interaction},
+ colour::Colour,
+ gateway::Ready,
+};
+use serenity::prelude::{Context, EventHandler};
+
+#[derive(Clone, Copy, Debug)]
+pub struct Handler;
+
+impl Handler {
+ async fn register_commands(&self, ctx: &Context) -> Result<(), Error> {
+ let commands = bot_commands::to_vec();
+ let commands_len = commands.len();
+ for command in commands {
+ Command::create_global_command(&ctx.http, command).await?;
+ }
+
+ debug!("Registered {} commands", commands_len);
+ Ok(())
+ }
+
+ /// Dispatch our commands from a [`CommandInteraction`]
+ async fn dispatch_command(ctx: &Context, command: &CommandInteraction) -> Result<(), Error> {
+ let command_name = command.data.name.as_str();
+
+ // grab our configuration & http client from the aether
+ let (http, config) = {
+ let read = ctx.data.read().await;
+ let http = read
+ .get::<SharedHttp>()
+ .ok_or("Couldn't get shared HTTP client! WHY??????")?
+ .clone();
+ let config = read
+ .get::<SharedConfig>()
+ .ok_or("Couldn't get shared bot configuration!")?
+ .clone();
+ (http, config)
+ };
+
+ match command_name {
+ "about" => bot_commands::about::respond(ctx, &http, command).await?,
+ "ping" => bot_commands::ping::respond(ctx, command).await?,
+ "track" => bot_commands::track::respond(ctx, &http, &config, command).await?,
+ _ => {
+ let message = CreateInteractionResponseMessage::new().content(format!(
+ "It doesn't look like you can use `{command_name}`. Sorry :("
+ ));
+ let response = CreateInteractionResponse::Message(message);
+ command.create_response(&ctx, response).await?;
+ }
+ };
+
+ Ok(())
+ }
+
+ async fn invite_link(ctx: &Context) {
+ if let Ok(invite_link) = CreateBotAuthParameters::new().auto_client_id(ctx).await {
+ let link = invite_link.build();
+ info!("You can install me as an app at {link}");
+ } else {
+ warn!("Couldn't figure out our own client ID! Something might be wrong");
+ }
+ }
+}
+
+#[async_trait]
+impl EventHandler for Handler {
+ /// Dispatch our commands and try to handle errors from them
+ async fn interaction_create(&self, ctx: Context, interaction: Interaction) {
+ if let Interaction::Command(command) = interaction {
+ let command_name = &command.data.name;
+ trace!("Received command: {}", command_name);
+
+ if let Err(why) = Handler::dispatch_command(&ctx, &command).await {
+ error!(
+ "Ran into an error while dispatching command {}:\n{why:?}",
+ command_name
+ );
+
+ let embed = CreateEmbed::new()
+ .title("An error occurred")
+ .description("Sorry about that!")
+ .color(Colour::RED);
+ let response = CreateInteractionResponseFollowup::new().embed(embed);
+
+ if let Err(why) = command.create_followup(&ctx.http, response).await {
+ error!("Ran into an error while trying to recover from an error!\n{why:?}");
+ }
+ }
+ }
+ }
+
+ async fn ready(&self, ctx: Context, ready: Ready) {
+ info!("Connected as {}!", ready.user.name);
+ Handler::invite_link(&ctx).await;
+
+ if let Err(why) = self.register_commands(&ctx).await {
+ error!("Couldn't register commands!\n{why:?}");
+ };
+ }
+}
diff --git a/crates/bot-client/src/lib.rs b/crates/bot-client/src/lib.rs
new file mode 100644
index 0000000..851b853
--- /dev/null
+++ b/crates/bot-client/src/lib.rs
@@ -0,0 +1,79 @@
+use bot_config::Config;
+use bot_error::Error;
+use bot_http as http;
+
+use std::sync::Arc;
+
+use log::trace;
+use serenity::prelude::{Client, GatewayIntents, TypeMapKey};
+
+mod handler;
+
+use handler::Handler;
+
+/// Container for [`http::Client`]
+struct SharedHttp;
+
+impl TypeMapKey for SharedHttp {
+ type Value = Arc<http::Client>;
+}
+
+/// Container for [`Config`]
+struct SharedConfig;
+
+impl TypeMapKey for SharedConfig {
+ type Value = Arc<Config>;
+}
+
+/// Fetch our bot token
+fn token() -> Result<String, Error> {
+ let token = std::env::var("DISCORD_BOT_TOKEN")?;
+ Ok(token)
+}
+
+/// Create our client
+///
+/// # Errors
+///
+/// Will return [`Err`] if a [`Client`] cannot be created or configuration
+/// cannot be created from the environment.
+///
+/// # Panics
+///
+/// Will [`panic!`] if the bot token isn't found or the ctrl+c handler can't be made
+pub async fn get() -> Result<Client, Error> {
+ let token = token().expect("Couldn't find token in environment! Is DISCORD_BOT_TOKEN set?");
+
+ let intents = GatewayIntents::default();
+ trace!("Creating client");
+ let client = Client::builder(token, intents)
+ .event_handler(Handler)
+ .await?;
+
+ // add state stuff
+ let http_client = <http::Client as http::ClientExt>::default();
+ let config = Config::from_env()?;
+
+ {
+ let mut data = client.data.write().await;
+
+ data.insert::<SharedHttp>(Arc::new(http_client));
+ data.insert::<SharedConfig>(Arc::new(config.clone()));
+ }
+
+ let shard_manager = client.shard_manager.clone();
+
+ // gracefully shutdown on ctrl+c
+ tokio::spawn(async move {
+ #[cfg(target_family = "unix")]
+ tokio::signal::ctrl_c()
+ .await
+ .expect("Couldn't register ctrl+c handler!");
+ shard_manager.shutdown_all().await;
+ });
+
+ // run our jobs
+ bot_jobs::dispatch(config)?;
+
+ Ok(client)
+}