website/gazebot/src/main.rs

73 lines
2.0 KiB
Rust
Raw Normal View History

use reqwest::header::AUTHORIZATION;
2024-10-30 16:33:08 +03:00
use serenity::all::ActivityData;
2024-10-30 14:00:36 +03:00
use serenity::async_trait;
use serenity::json::json;
2024-10-30 14:00:36 +03:00
use serenity::model::channel::Message;
use serenity::model::prelude::*;
use serenity::prelude::*;
2024-10-30 16:33:08 +03:00
use shuttle_runtime::SecretStore;
2024-10-30 14:00:36 +03:00
struct Handler {
http: reqwest::Client,
secrets: SecretStore,
}
2024-10-30 14:00:36 +03:00
#[async_trait]
impl EventHandler for Handler {
async fn message(&self, _: Context, msg: Message) {
if msg.content.starts_with("do ") {
return;
}
let mut note_content = msg.content.clone();
const BSKY_TAG: &str = ".nobsky";
let no_bsky_posse = note_content.contains(BSKY_TAG);
if no_bsky_posse {
note_content = note_content.replace(BSKY_TAG, "");
}
let note_data = json!({"content": note_content.trim(), "bskyPosse": !no_bsky_posse});
let resp = self
.http
.post("https://gaze.systems/log/create")
.header(AUTHORIZATION, self.secrets.get("DISCORD_TOKEN").unwrap())
.json(&note_data)
.send()
.await
.and_then(|resp| resp.error_for_status());
if let Err(why) = resp {
tracing::error!("could not create note: {why}");
2024-10-30 14:00:36 +03:00
}
}
2024-10-30 16:33:08 +03:00
async fn ready(&self, ctx: Context, ready: Ready) {
tracing::info!("{} is connected!", ready.user.name);
2024-10-30 16:33:08 +03:00
ctx.set_presence(
Some(ActivityData::listening("messages")),
OnlineStatus::Online,
);
2024-10-30 14:00:36 +03:00
}
}
2024-10-30 16:33:08 +03:00
#[shuttle_runtime::main]
async fn serenity(
#[shuttle_runtime::Secrets] secrets: SecretStore,
) -> shuttle_serenity::ShuttleSerenity {
let token = secrets.get("DISCORD_TOKEN").unwrap();
let intents = GatewayIntents::DIRECT_MESSAGES | GatewayIntents::MESSAGE_CONTENT;
let client = Client::builder(&token, intents)
.event_handler(Handler {
http: reqwest::Client::new(),
secrets,
})
2024-10-30 16:33:08 +03:00
.await
.expect("Err creating client");
Ok(client.into())
}