cooking

Author: Tricked-dev

src/event_handler.rs

@@ -31,6 +31,10 @@ pub async fn handle_event(
                 return Ok(());
             }
 
+            // Increment message count for this channel
+            let count = locked_state.channel_message_counts.entry(msg.channel_id).or_insert(0);
+            *count += 1;
+
             if let Some(today_i) = locked_state.config.today_i_channel {
                 if msg.channel_id == Id::new(today_i) && !msg.content.clone().to_lowercase().starts_with("today i") {
                     http.delete_message(msg.channel_id, msg.id).exec().await?;

src/message_handler.rs

@@ -222,6 +222,11 @@ pub async fn handle_message(
                 || msg.referenced_message.clone().map(|msg| msg.author.id) == Some(Id::<UserMarker>::new(locked_state.config.id))
             ) =>
         {
+            // Check if we should create memories based on message count
+            let should_create_memory = locked_state.channel_message_counts
+                .get(&msg.channel_id)
+                .map(|count| *count >= 30)
+                .unwrap_or(false);
             let name = msg.author.name.clone();
 
             let mut context = String::new();
@@ -303,13 +308,18 @@ pub async fn handle_message(
                         Arc::clone(http),
                     ));
 
-                    // Spawn background task to create memories
-                    tokio::spawn(memory_creator::create_memories_background(
-                        locked_state.db.clone(),
-                        context.clone(),
-                        user_mentions_clone,
-                        locked_state.config.clone(),
-                    ));
+                    // Spawn background task to create memories only if we've reached the threshold
+                    if should_create_memory {
+                        tokio::spawn(memory_creator::create_memories_background(
+                            locked_state.db.clone(),
+                            context.clone(),
+                            user_mentions_clone,
+                            locked_state.config.clone(),
+                        ));
+
+                        // Reset the message counter for this channel
+                        locked_state.channel_message_counts.insert(msg.channel_id, 0);
+                    }
 
                     Ok(Command::nothing())
                 }

src/structs.rs

@@ -126,6 +126,8 @@ pub struct State {
     pub pending_math_tests: HashMap<u64, PendingMathTest>,
     /// Pending color tests
     pub pending_color_tests: HashMap<u64, PendingColorTest>,
+    /// Message count per channel since last memory creation
+    pub channel_message_counts: HashMap<Id<ChannelMarker>, i32>,
 }
 // i hate fixing error
 unsafe impl Send for State {}
@@ -151,6 +153,7 @@ impl State {
             config,
             pending_math_tests: HashMap::new(),
             pending_color_tests: HashMap::new(),
+            channel_message_counts: HashMap::new(),
         }
     }
 }