26.0-build-2

Structures/AIDatabase.js

@@ -0,0 +1,289 @@
+const Database = require("better-sqlite3");
+const path = require("path");
+
+const db = new Database(path.join(__dirname, "../Data/ai-memory.db"));
+
+db.pragma("journal_mode = WAL");
+
+db.prepare(
+  `
+  CREATE TABLE IF NOT EXISTS ai_conversations (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    guild_id TEXT,
+    channel_id TEXT,
+    user_id TEXT,
+    role TEXT NOT NULL,
+    content TEXT NOT NULL,
+    created_at INTEGER NOT NULL
+  )
+`,
+).run();
+
+db.prepare(
+  `
+  CREATE TABLE IF NOT EXISTS ai_summaries (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    guild_id TEXT,
+    channel_id TEXT,
+    summary TEXT NOT NULL,
+    created_at INTEGER NOT NULL,
+    updated_at INTEGER NOT NULL
+  )
+`,
+).run();
+
+db.prepare(
+  `
+  CREATE TABLE IF NOT EXISTS ai_facts (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    guild_id TEXT,
+    channel_id TEXT,
+    user_id TEXT,
+    fact_type TEXT NOT NULL,
+    fact_key TEXT NOT NULL,
+    fact_value TEXT NOT NULL,
+    created_at INTEGER NOT NULL,
+    updated_at INTEGER NOT NULL
+  )
+`,
+).run();
+
+db.prepare(
+  `
+  CREATE TABLE IF NOT EXISTS ai_knowledge (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    source_type TEXT NOT NULL,
+    source_name TEXT NOT NULL,
+    title TEXT NOT NULL,
+    content TEXT NOT NULL,
+    created_at INTEGER NOT NULL,
+    updated_at INTEGER NOT NULL
+  )
+`,
+).run();
+
+db.prepare(
+  `
+  CREATE TABLE IF NOT EXISTS ai_sessions (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    guild_id TEXT,
+    channel_id TEXT,
+    user_id TEXT,
+    model TEXT,
+    mode TEXT,
+    last_message_at INTEGER,
+    created_at INTEGER NOT NULL,
+    updated_at INTEGER NOT NULL
+  )
+`,
+).run();
+
+try {
+  db.prepare(`ALTER TABLE ai_facts ADD COLUMN user_id TEXT`).run();
+} catch (e) {
+  if (!String(e).includes("duplicate column name")) throw e;
+}
+
+module.exports = {
+
+  saveUserFact({ guildId, userId, factType, factKey, factValue }) {
+    const existing = db
+      .prepare(
+        `
+      SELECT id FROM ai_facts
+      WHERE guild_id IS ? AND user_id IS ? AND fact_type = ? AND fact_key = ?
+      LIMIT 1
+    `,
+      )
+      .get(guildId || null, userId || null, factType, factKey);
+
+    if (existing) {
+      db.prepare(
+        `
+        UPDATE ai_facts
+        SET fact_value = ?, updated_at = ?
+        WHERE id = ?
+      `,
+      ).run(factValue, Date.now(), existing.id);
+      return;
+    }
+
+    db.prepare(
+      `
+      INSERT INTO ai_facts (
+        guild_id, channel_id, user_id, fact_type, fact_key, fact_value, created_at, updated_at
+      ) VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+    `,
+    ).run(
+      guildId || null,
+      null,
+      userId || null,
+      factType,
+      factKey,
+      factValue,
+      Date.now(),
+      Date.now(),
+    );
+  },
+
+  getUserFacts(guildId, userId) {
+    return db
+      .prepare(
+        `
+      SELECT * FROM ai_facts
+      WHERE guild_id IS ? AND user_id IS ?
+      ORDER BY updated_at DESC
+    `,
+      )
+      .all(guildId || null, userId || null);
+  },
+
+  addMessage({ guildId, channelId, userId, role, content }) {
+    db.prepare(
+      `
+      INSERT INTO ai_conversations (
+        guild_id, channel_id, user_id, role, content, created_at
+      ) VALUES (?, ?, ?, ?, ?, ?)
+    `,
+    ).run(
+      guildId || null,
+      channelId || null,
+      userId || null,
+      role,
+      content,
+      Date.now(),
+    );
+  },
+
+  getRecentMessages(channelId, limit = 15) {
+    return db
+      .prepare(
+        `
+      SELECT * FROM ai_conversations
+      WHERE channel_id = ?
+      ORDER BY created_at DESC
+      LIMIT ?
+    `,
+      )
+      .all(channelId, limit)
+      .reverse();
+  },
+
+  saveSummary({ guildId, channelId, summary }) {
+    const existing = db
+      .prepare(
+        `
+      SELECT id FROM ai_summaries
+      WHERE guild_id IS ? AND channel_id IS ?
+      ORDER BY updated_at DESC
+      LIMIT 1
+    `,
+      )
+      .get(guildId || null, channelId || null);
+
+    if (existing) {
+      db.prepare(
+        `
+        UPDATE ai_summaries
+        SET summary = ?, updated_at = ?
+        WHERE id = ?
+      `,
+      ).run(summary, Date.now(), existing.id);
+      return;
+    }
+
+    db.prepare(
+      `
+      INSERT INTO ai_summaries (
+        guild_id, channel_id, summary, created_at, updated_at
+      ) VALUES (?, ?, ?, ?, ?)
+    `,
+    ).run(guildId || null, channelId || null, summary, Date.now(), Date.now());
+  },
+
+  getLatestSummary(channelId) {
+    return db
+      .prepare(
+        `
+      SELECT * FROM ai_summaries
+      WHERE channel_id = ?
+      ORDER BY updated_at DESC
+      LIMIT 1
+    `,
+      )
+      .get(channelId);
+  },
+
+  saveFact({ guildId, channelId, factType, factKey, factValue }) {
+    const existing = db
+      .prepare(
+        `
+      SELECT id FROM ai_facts
+      WHERE guild_id IS ? AND channel_id IS ? AND fact_type = ? AND fact_key = ?
+      LIMIT 1
+    `,
+      )
+      .get(guildId || null, channelId || null, factType, factKey);
+
+    if (existing) {
+      db.prepare(
+        `
+        UPDATE ai_facts
+        SET fact_value = ?, updated_at = ?
+        WHERE id = ?
+      `,
+      ).run(factValue, Date.now(), existing.id);
+      return;
+    }
+
+    db.prepare(
+      `
+      INSERT INTO ai_facts (
+        guild_id, channel_id, fact_type, fact_key, fact_value, created_at, updated_at
+      ) VALUES (?, ?, ?, ?, ?, ?, ?)
+    `,
+    ).run(
+      guildId || null,
+      channelId || null,
+      factType,
+      factKey,
+      factValue,
+      Date.now(),
+      Date.now(),
+    );
+  },
+
+  getFacts(channelId) {
+    return db
+      .prepare(
+        `
+      SELECT * FROM ai_facts
+      WHERE channel_id = ?
+      ORDER BY updated_at DESC
+    `,
+      )
+      .all(channelId);
+  },
+
+  addKnowledge({ sourceType, sourceName, title, content }) {
+    db.prepare(
+      `
+      INSERT INTO ai_knowledge (
+        source_type, source_name, title, content, created_at, updated_at
+      ) VALUES (?, ?, ?, ?, ?, ?)
+    `,
+    ).run(sourceType, sourceName, title, content, Date.now(), Date.now());
+  },
+
+  getKnowledge(limit = 25) {
+    return db
+      .prepare(
+        `
+      SELECT * FROM ai_knowledge
+      ORDER BY updated_at DESC
+      LIMIT ?
+    `,
+      )
+      .all(limit);
+  },
+};

Structures/Client.js

@@ -49,9 +49,10 @@ class Client extends Discord.Client {
         GatewayIntentBits.Guilds,
         GatewayIntentBits.GuildMembers,
         GatewayIntentBits.GuildMessages,
+        GatewayIntentBits.GuildMessageReactions,
         GatewayIntentBits.MessageContent,
       ],
-      partials: [Partials.Message, Partials.Channel],
+      partials: [Partials.Message, Partials.Channel, Partials.Reaction],
     });
 
     this.commands = new Discord.Collection();

Structures/Database.js

@@ -52,6 +52,14 @@ const safeAlter = (sql) => {
 safeAlter(`ALTER TABLE tickets ADD COLUMN voiceChannelId TEXT`);
 safeAlter(`ALTER TABLE tickets ADD COLUMN department TEXT DEFAULT 'general'`);
 safeAlter(`ALTER TABLE tickets ADD COLUMN priority TEXT DEFAULT 'medium'`);
+safeAlter(`ALTER TABLE tickets ADD COLUMN aiModeEnabled INTEGER DEFAULT 0`);
+safeAlter(`ALTER TABLE tickets ADD COLUMN aiChannelId TEXT`);
+safeAlter(`ALTER TABLE tickets ADD COLUMN aiType TEXT`);
+safeAlter(`ALTER TABLE tickets ADD COLUMN aiEnabledBy TEXT`);
+safeAlter(`ALTER TABLE tickets ADD COLUMN aiEnabledAt INTEGER`);
+safeAlter(`ALTER TABLE tickets ADD COLUMN questionAnswers TEXT`);
+safeAlter(`ALTER TABLE tickets ADD COLUMN ticketName TEXT`);
+safeAlter(`ALTER TABLE tickets ADD COLUMN ticketNumber TEXT`);
 
 db.prepare(
   `
@@ -84,18 +92,46 @@ db.prepare(
 ).run();
 
 function ensureProfile(userId) {
-  const existing = db.prepare(`
+  const existing = db
+    .prepare(
+      `
     SELECT user_id FROM profiles WHERE user_id = ?
-  `).get(userId);
+  `,
+    )
+    .get(userId);
 
   if (!existing) {
-    db.prepare(`
+    db.prepare(
+      `
       INSERT INTO profiles (user_id, bio, timezone, clockedIn)
       VALUES (?, '', '', 0)
-    `).run(userId);
+    `,
+    ).run(userId);
   }
 }
 
+function normaliseTicketRow(row) {
+  if (!row) return null;
+
+  let questionAnswers = [];
+
+  if (row.questionAnswers) {
+    try {
+      questionAnswers = JSON.parse(row.questionAnswers);
+    } catch {
+      questionAnswers = [];
+    }
+  }
+
+  return {
+    ...row,
+    aiModeEnabled: Boolean(row.aiModeEnabled),
+    clockedIn:
+      row.clockedIn !== undefined ? Boolean(row.clockedIn) : row.clockedIn,
+    questionAnswers,
+  };
+}
+
 module.exports = {
   addTicket(ticket) {
     const now = Date.now();
@@ -110,7 +146,15 @@ module.exports = {
         created_at,
         updated_at,
         department,
-        priority
+        priority,
+        aiModeEnabled,
+        aiChannelId,
+        aiType,
+        aiEnabledBy,
+        aiEnabledAt,
+        questionAnswers,
+        ticketName,
+        ticketNumber
       )
       VALUES (
         @ticket_id,
@@ -120,7 +164,15 @@ module.exports = {
         @created_at,
         @updated_at,
         @department,
-        @priority
+        @priority,
+        @aiModeEnabled,
+        @aiChannelId,
+        @aiType,
+        @aiEnabledBy,
+        @aiEnabledAt,
+        @questionAnswers,
+        @ticketName,
+        @ticketNumber
       )
       ON CONFLICT(ticket_id) DO NOTHING
     `,
@@ -133,6 +185,28 @@ module.exports = {
       updated_at: now,
       department: ticket.department || "general",
       priority: ticket.priority || "medium",
+      aiModeEnabled: ticket.aiModeEnabled ? 1 : 0,
+      aiChannelId: ticket.aiChannelId || null,
+      aiType: ticket.aiType || null,
+      aiEnabledBy: ticket.aiEnabledBy || null,
+      aiEnabledAt: ticket.aiEnabledAt || null,
+      questionAnswers: JSON.stringify(ticket.questionAnswers || []),
+      ticketName: ticket.name || null,
+      ticketNumber: ticket.number || null,
+    });
+  },
+
+  updateTicketQuestionAnswers(ticketId, questionAnswers = []) {
+    db.prepare(
+      `
+      UPDATE tickets
+      SET questionAnswers=@questionAnswers, updated_at=@updatedAt
+      WHERE ticket_id=@ticketId
+    `,
+    ).run({
+      ticketId,
+      questionAnswers: JSON.stringify(questionAnswers),
+      updatedAt: Date.now(),
     });
   },
 
@@ -192,18 +266,103 @@ module.exports = {
     });
   },
 
+  setTicketAIState(ticketId, aiState = {}) {
+    db.prepare(
+      `
+      UPDATE tickets
+      SET
+        aiModeEnabled=@aiModeEnabled,
+        aiChannelId=@aiChannelId,
+        aiType=@aiType,
+        aiEnabledBy=@aiEnabledBy,
+        aiEnabledAt=@aiEnabledAt,
+        updated_at=@updatedAt
+      WHERE ticket_id=@ticketId
+    `,
+    ).run({
+      ticketId,
+      aiModeEnabled: aiState.enabled ? 1 : 0,
+      aiChannelId: aiState.aiChannelId || null,
+      aiType: aiState.aiType || null,
+      aiEnabledBy: aiState.enabledBy || null,
+      aiEnabledAt: aiState.enabledAt
+        ? new Date(aiState.enabledAt).getTime()
+        : aiState.enabled
+          ? Date.now()
+          : null,
+      updatedAt: Date.now(),
+    });
+  },
+
+  enableTicketAI(
+    ticketId,
+    aiChannelId = null,
+    aiType = null,
+    enabledBy = null,
+  ) {
+    db.prepare(
+      `
+      UPDATE tickets
+      SET
+        aiModeEnabled=1,
+        aiChannelId=@aiChannelId,
+        aiType=@aiType,
+        aiEnabledBy=@aiEnabledBy,
+        aiEnabledAt=@aiEnabledAt,
+        updated_at=@updatedAt
+      WHERE ticket_id=@ticketId
+    `,
+    ).run({
+      ticketId,
+      aiChannelId,
+      aiType,
+      aiEnabledBy,
+      aiEnabledAt: Date.now(),
+      updatedAt: Date.now(),
+    });
+  },
+
+  disableTicketAI(ticketId) {
+    db.prepare(
+      `
+      UPDATE tickets
+      SET
+        aiModeEnabled=0,
+        updated_at=@updatedAt
+      WHERE ticket_id=@ticketId
+    `,
+    ).run({
+      ticketId,
+      updatedAt: Date.now(),
+    });
+  },
+
   getTicket(ticketId) {
-    return db
+    const row = db
       .prepare(
         `
       SELECT * FROM tickets WHERE ticket_id=@ticketId
     `,
       )
       .get({ ticketId });
+
+    return normaliseTicketRow(row);
+  },
+
+  getTicketByAIChannel(aiChannelId) {
+    const row = db
+      .prepare(
+        `
+      SELECT * FROM tickets WHERE aiChannelId=@aiChannelId LIMIT 1
+    `,
+      )
+      .get({ aiChannelId });
+
+    return normaliseTicketRow(row);
   },
 
   getAllTickets() {
-    return db.prepare(`SELECT * FROM tickets`).all();
+    return db.prepare(`SELECT * FROM tickets`).all().map(normaliseTicketRow);
   },
 
   saveTicketPanel(messageId, channelId) {
@@ -267,9 +426,13 @@ module.exports = {
   getProfile(userId) {
     ensureProfile(userId);
 
-    const row = db.prepare(`
+    const row = db
+      .prepare(
+        `
       SELECT * FROM profiles WHERE user_id = ?
-    `).get(userId);
+    `,
+      )
+      .get(userId);
 
     if (!row) {
       return {
@@ -289,11 +452,13 @@ module.exports = {
   setClockedIn(userId, clockedIn) {
     ensureProfile(userId);
 
-    db.prepare(`
+    db.prepare(
+      `
       UPDATE profiles
       SET clockedIn = ?
       WHERE user_id = ?
-    `).run(clockedIn ? 1 : 0, userId);
+    `,
+    ).run(clockedIn ? 1 : 0, userId);
   },
 
   saveSettings(obj) {