Refactor Handler implementations and improve API key usage

Implemented `ChatHandler` trait for chat message handling in the discord bot. The `process_message` method now takes in a reference instead of a cloned `Message` object, improving performance. The `new` method in `Handler` implementation also takes a reference to the `String` API key now, enhancing memory safety and efficiency.

src/bin/bootstrap.rs

@@ -1,3 +1,4 @@
+use gpt_discord_bot::ChatHandler;
 use serenity::prelude::*;
 use std::env;
 use std::path::Path;
@@ -23,9 +24,9 @@ async fn main() {
 
     // Create a new client with the discord token
     let mut client = Client::builder(&discord_token, intents)
-        .event_handler(Handler::new(&gpt_api_key).await.unwrap())
+        .event_handler(Handler::new(&gpt_api_key))
         .await
-        .expect("Error creating client");
+        .expect("Err creating client");
 
     // Start listening for events
     println!("Bot is now running. Press Ctrl+C to stop.");

src/discord/bot.rs

@@ -7,23 +7,25 @@ use tokio::select;
 use tokio::time::interval;
 
 // Define a trait representing handler behavior
+#[async_trait]
 pub trait ChatHandler {
     // Create a new instance of the handler
-    fn new(api_key: String) -> Result<Self>
+    fn new(api_key: &String) -> Self
     where
         Self: Sized;
 
     // Process an incoming message and return a response
-    fn process_message(&self, msg: Message) -> Option<String>;
+    async fn process_message(&self, msg: &Message) -> Option<String>;
 }
 
 // Handler struct implementing the ChatHandler trait
 pub struct Handler {
     pub gpt_client: ChatGPT,
 }
 
-impl Handler {
-    pub async fn new(api_key: &String) -> Result<Self> {
+#[async_trait]
+impl ChatHandler for Handler {
+    fn new(api_key: &String) -> Self {
         let config: ModelConfiguration = ModelConfigurationBuilder::default()
             .engine(ChatGPTEngine::Gpt4)
             .timeout(Duration::from_secs(50))
@@ -32,11 +34,11 @@ impl Handler {
                 log::error!("Failed to build ModelConfiguration: {}", e);
                 ModelConfiguration::default()
             });
-        let gpt_client = ChatGPT::new_with_config(api_key, config)?;
-        Ok(Self { gpt_client })
+        let gpt_client = ChatGPT::new_with_config(api_key, config).unwrap();
+        Self { gpt_client }
     }
 
-    async fn process_message(&self, msg: Message) -> Option<String> {
+    async fn process_message(&self, msg: &Message) -> Option<String> {
         if msg.author.bot || !msg.content.starts_with(".") {
             return None;
         }
@@ -69,7 +71,7 @@ impl Handler {
 #[async_trait]
 impl SerenityEventHandler for Handler {
     async fn message(&self, ctx: Context, msg: Message) {
-        if let Some(result) = self.process_message(msg.clone()).await {
+        if let Some(result) = self.process_message(&msg).await {
             let processing_future = msg.channel_id.say(&ctx.http, "Processing...");
 
             match processing_future.await {

src/discord/mod.rs

@@ -1,2 +1,3 @@
 mod bot;
+pub use bot::ChatHandler;
 pub use bot::Handler;

src/lib.rs

@@ -1,2 +1,3 @@
 mod discord;
+pub use crate::discord::ChatHandler;
 pub use crate::discord::Handler;