rewrite in rust :)

1 files changed, 158 insertions, 0 deletions

diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..4565bd4
--- /dev/null
+++ b/src/main.rs
@@ -0,0 +1,158 @@
+use serenity::async_trait;
+use serenity::framework::standard::macros::{command, group};
+use serenity::framework::standard::{CommandResult, StandardFramework};
+use serenity::model::application::interaction::{Interaction, InteractionResponseType};
+use serenity::model::channel::Message;
+use serenity::model::id::GuildId;
+use serenity::model::prelude::Ready;
+use serenity::prelude::*;
+use std::{env, vec};
+
+mod api;
+mod commands;
+mod utils;
+
+const GUILD: u64 = 1055663552679137310;
+const BOT: u64 = 1056467120986271764;
+
+#[group]
+#[commands(bing, ask, random_teawie, teawiespam)]
+struct General;
+
+struct Handler;
+
+#[async_trait]
+impl EventHandler for Handler {
+	/*
+	 * echo some messages when they're sent
+	 */
+	async fn message(&self, ctx: Context, msg: Message) {
+		let author = msg.author.id.as_u64();
+
+		if author == &BOT {
+			return;
+		}
+
+		let mut echo_msgs = vec!["🗿", "Twitter's Recommendation Algorithm"];
+
+		for emoji in utils::TEAMOJIS {
+			// i was also lazy here
+			echo_msgs.push(emoji);
+		}
+
+		for echo in echo_msgs {
+			if msg.content.as_str() == echo {
+				let send = msg.reply(&ctx, echo);
+				if let Err(why) = send.await {
+					println!("error when replying to {:?}: {:?}", msg.content, why);
+				}
+			}
+		}
+	}
+
+	async fn interaction_create(&self, ctx: Context, interaction: Interaction) {
+		if let Interaction::ApplicationCommand(command) = interaction {
+			println!("Received command interaction: {:#?}", command);
+			let content = match command.data.name.as_str() {
+				"ask" => commands::ask::run(&command.data.options).await,
+				"copypasta" => {
+					commands::copypasta::run(&command.data.options, command.channel_id, &ctx.http)
+						.await
+				}
+				"random_teawie" => commands::random_teawie::run(&command.data.options).await,
+				_ => "not implemented :(".to_string(),
+			};
+
+			if let Err(why) = command
+				.create_interaction_response(&ctx.http, |response| {
+					response
+						.kind(InteractionResponseType::ChannelMessageWithSource)
+						.interaction_response_data(|message| message.content(content))
+				})
+				.await
+			{
+				println!("cannot respond to slash command: {}", why);
+			}
+		}
+	}
+
+	async fn ready(&self, ctx: Context, ready: Ready) {
+		println!("connected as {:?}", ready.user.name);
+
+		let guild_id = GuildId(GUILD);
+
+		let commands = GuildId::set_application_commands(&guild_id, &ctx.http, |commands| {
+			commands
+				.create_application_command(|command| commands::ask::register(command))
+				.create_application_command(|command| commands::random_teawie::register(command))
+				.create_application_command(|command| commands::copypasta::register(command))
+		})
+		.await;
+
+		println!("registered commands: {:#?}", commands);
+	}
+}
+
+#[tokio::main]
+async fn main() {
+	let framework = StandardFramework::new()
+		.configure(|c| c.prefix("!"))
+		.group(&GENERAL_GROUP);
+
+	let token = env::var("TOKEN").expect("couldn't find token in environment.");
+
+	let intents = GatewayIntents::all();
+	let mut client = Client::builder(token, intents)
+		.event_handler(Handler)
+		.framework(framework)
+		.await
+		.expect("error creating client");
+
+	if let Err(why) = client.start().await {
+		println!("an error occurred: {:?}", why);
+	}
+}
+
+#[command]
+async fn bing(ctx: &Context, msg: &Message) -> CommandResult {
+	msg.channel_id
+		.send_message(&ctx.http, |m| m.content("bong"))
+		.await?;
+
+	Ok(())
+}
+
+#[command]
+async fn ask(ctx: &Context, msg: &Message) -> CommandResult {
+	let resp = utils::get_random_response().await;
+	msg.channel_id
+		.send_message(&ctx.http, |m| m.content(resp))
+		.await?;
+
+	Ok(())
+}
+
+#[command]
+async fn random_teawie(ctx: &Context, msg: &Message) -> CommandResult {
+	let resp = api::guzzle::get_random_teawie().await;
+	msg.channel_id
+		.send_message(&ctx.http, |m| m.content(resp))
+		.await?;
+
+	Ok(())
+}
+
+#[command]
+async fn teawiespam(ctx: &Context, msg: &Message) -> CommandResult {
+	let mut resp = String::new();
+
+	for _ in 0..50 {
+		resp += "<:teawiesmile:1056438046440042546>";
+	}
+
+	msg.channel_id
+		.send_message(&ctx.http, |m| m.content(resp))
+		.await?;
+
+	Ok(())
+}