refactor: Improve user context retrieval and enhance channel information in AiChat

Author: 0-don

src/events/ai/ai.ts

@@ -1,7 +1,7 @@
 import { google } from "@ai-sdk/google";
 import { generateText, ModelMessage, StepResult } from "ai";
 import console, { error } from "console";
-import { Message, OmitPartialGroupDMChannel } from "discord.js";
+import { Message, OmitPartialGroupDMChannel, ThreadChannel } from "discord.js";
 import type { ArgsOf, Client } from "discordx";
 import { Discord, On } from "discordx";
 import { ConfigValidator } from "../../lib/config-validator";
@@ -60,10 +60,7 @@ export class AiChat {
     const messages = channelMessages.get(message.channel.id) || [];
 
     // Get user context
-    const userContext = await this.getUserContext(
-      message.author.id,
-      message.guildId!
-    );
+    const userContext = await this.getUserContext(message.author.id, message);
     const fullMessage = `${userMsg}${replyContext}${userContext}`;
 
     try {
@@ -124,21 +121,42 @@ export class AiChat {
 
   private async getUserContext(
     memberId: string,
-    guildId: string
+    message: Message
   ): Promise<string> {
     try {
-      const userStats = await StatsService.getUserStatsEmbed(memberId, guildId);
+      const userStats = await StatsService.getUserStatsEmbed(
+        memberId,
+        message.guildId!
+      );
 
       if (!userStats) {
         return "\n\n[User Context: New member, no stats available]";
       }
 
       const { embed, roles } = userStats;
 
-      // Simple JSON stringify approach
+      // Get channel information
+      let channelInfo = "";
+      const channel = message.channel;
+
+      if (channel.isThread()) {
+        const thread = channel as ThreadChannel;
+        try {
+          const starterMessage = await thread.fetchStarterMessage();
+          channelInfo = `Thread: "${thread.name}" in #${thread.parent?.name || "unknown"}\nOriginal post: "${starterMessage?.content?.substring(0, 200) || "No content"}${starterMessage?.content && starterMessage.content.length > 200 ? "..." : ""}"`;
+        } catch (error) {
+          channelInfo = `Thread: "${thread.name}" in #${thread.parent?.name || "unknown"}`;
+        }
+      } else if ("name" in channel) {
+        channelInfo = `Channel: #${channel.name}`;
+      } else {
+        channelInfo = "Channel: DM or unknown";
+      }
+
       const contextData = {
         embed,
         roles: roles?.filter(Boolean) || [],
+        location: channelInfo,
       };
 
       return `\n\n[User Context: ${JSON.stringify(contextData, null, 2)}]`;
@@ -186,7 +204,8 @@ export class AiChat {
         // Get context for the replied user too
         const repliedUserContext = await this.getUserContext(
           repliedUser.id,
-          message.guildId!
+
+          message
         );
 
         return {

src/events/ai/prompt.ts

@@ -26,8 +26,9 @@ SERVER FEATURES & COMMANDS:
 - **Leveling System**: Users earn XP from messages and get roles (Copy Paster → Script Kiddie → Vibe Coder → Intern → Junior Dev → Mid Dev → Senior Dev → Lead Dev → Tech Lead)
 - **Helper System**: React with ✅ in threads to give/receive helper points, earn helper roles
 - **Role Management**: Verified, VoiceOnly, Jail status roles with automatic assignment
+- **Jail System**: Spammers and rule violators get jailed - they can't see other channels or members, isolated until moderation review
 - **Voice Tracking**: Detailed voice channel time statistics and logging
-- **Spam Detection**: AI-powered spam detection for promotional content
+- **Spam Detection**: AI-powered spam detection for promotional content - first-message spammers get auto-jailed
 - **Translation**: DeepL integration for translating messages
 - **Member Analytics**: Growth tracking, join/leave events, activity charts
 
@@ -37,19 +38,21 @@ SERVER FEATURES & COMMANDS:
 - Comprehensive user statistics with charts
 - GIF reactions via Tenor API integration
 - Automatic role restoration for returning members
+- Smart spam detection that identifies promotional content and business solicitation
 
 CONTEXT AWARENESS:
 - This is a serious programming community, not a help desk
 - Members are encouraged to stay and contribute, not just ask questions and leave
 - React to technical discussions with appropriate programming knowledge
 - Reference server features when relevant (stats, leveling, helper system)
+- Spammers get isolated in jail where they can't bother other members
 
 RESPONSES:
 - Coding help: accurate + light sarcasm
 - Feature questions: explain server systems naturally
 - Stats requests: suggest using /me, /top, or /user commands
 - Thanks: "sure, whatever" 
-- Capabilities: "I help with programming, manage server features, and keep things lively"
+- Capabilities: "I help with programming, manage server features, keep spammers in jail, and keep things lively"
 - Avoid: politics, religion, adult content
 
 GIF USAGE: