Firebase changes

src/__mocks__/discord.js.ts

@@ -1,8 +1,9 @@
-import { Client, ClientOptions, RESTOptions, UserFlagsBitField } from "discord.js";
+import { Client, ClientOptions, GuildManager, RESTOptions, UserFlagsBitField } from "discord.js";
 
 const discordJS = jest.requireActual<typeof import("discord.js")>("discord.js");
 const mockDiscordJS = jest.createMockFromModule<typeof import("discord.js")>("discord.js");
 
+mockDiscordJS.MessageFlags = discordJS.MessageFlags;
 mockDiscordJS.EmbedBuilder = discordJS.EmbedBuilder;
 mockDiscordJS.ApplicationCommandOptionType = discordJS.ApplicationCommandOptionType;
 mockDiscordJS.GatewayIntentBits = discordJS.GatewayIntentBits;
@@ -32,12 +33,29 @@ class MockREST extends discordJS.REST {
     }
 }
 
+
+class MockGuildManager {
+    create: jest.Mock;
+    fetch: jest.Mock;
+    setIncidentActions: jest.Mock;
+    widgetImageURL: jest.Mock;
+
+    constructor() {
+        this.create = jest.fn().mockResolvedValue(null);
+        this.fetch = jest.fn().mockResolvedValue(new mockDiscordJS.Collection());
+        this.setIncidentActions = jest.fn().mockResolvedValue(null);
+        this.widgetImageURL = jest.fn().mockResolvedValue(null);
+    }
+}
+
 // @ts-ignore
 class MockClient {
     _ready: true = true;
     on: jest.Mock;
     options: ClientOptions;
     token: string;
+    guilds: GuildManager;
+
     constructor(options?: ClientOptions) {
         this.on = MockClient.prototype.on;
         this.login = MockClient.prototype.login;
@@ -47,6 +65,7 @@ class MockClient {
             } as any
         }
         this.token = "test-token";
+        this.guilds = new MockGuildManager() as unknown as GuildManager;
     }
 
     public login(token: string): Promise<void> {
@@ -98,5 +117,6 @@ class MockClientUser {
 mockDiscordJS.REST = MockREST;
 mockDiscordJS.Client = MockClient as unknown as typeof discordJS.Client;
 mockDiscordJS.ClientUser = MockClientUser as unknown as typeof discordJS.ClientUser;
+mockDiscordJS.GuildManager = MockGuildManager as unknown as typeof discordJS.GuildManager;
 
 module.exports = mockDiscordJS;

src/__mocks__/firebase/database.ts

@@ -0,0 +1,30 @@
+import { DatabaseReference, DataSnapshot } from "firebase/database";
+
+const firebaseDatabase = jest.requireActual<typeof import("firebase/database")>("firebase/database");
+const mockFirebaseDatabase = jest.createMockFromModule<typeof import("firebase/database")>("firebase/database");
+
+const mockSnapshot = {
+    exists: () => true,
+    val: () => ({ count: 2 }),
+    forEach: () => true
+};
+
+mockFirebaseDatabase.get = jest.fn().mockResolvedValue(mockSnapshot);
+
+mockFirebaseDatabase.child = jest.fn((_, path): DatabaseReference => {
+    return `${path}` as unknown as DatabaseReference;
+});
+mockFirebaseDatabase.ref = jest.fn().mockReturnValue('mock-ref');
+mockFirebaseDatabase.set = jest.fn().mockResolvedValue(undefined);
+mockFirebaseDatabase.remove = jest.fn().mockResolvedValue(undefined);
+
+class mockDataSnapshot {
+    forEach: jest.Mock;
+    constructor() {
+        this.forEach = jest.fn().mockReturnValue(true);
+    }
+}
+
+mockFirebaseDatabase.DataSnapshot = mockDataSnapshot as unknown as typeof DataSnapshot;
+
+module.exports = mockFirebaseDatabase;

src/__mocks__/helpers.ts

@@ -0,0 +1,7 @@
+const helpers = jest.requireActual<typeof import("../helpers")>("../helpers");
+const mockHelpers = jest.createMockFromModule<typeof import("../helpers")>("../helpers");
+
+module.exports = {
+	...helpers,
+	MINIMUM_MOOD_LIFESPAN: 0
+};

src/discordApp.ts

@@ -1,6 +1,7 @@
 import "dotenv/config";
 import { Client, GatewayIntentBits, Events } from "discord.js";
 import { clarify, embed, ping, tone, requestAnonymousClarification, mood } from "./interactions"
+import { cleanupMoods } from "./helpers";
 
 export async function launchBot(): Promise<Client> {
     // the client has to declare the features it uses up front so discord.js kno9ws if it can
@@ -9,7 +10,8 @@ export async function launchBot(): Promise<Client> {
     // discord API: https://discord.com/developers/docs/topics/gateway#list-of-intents
     const client = new Client({
         intents: [
-            GatewayIntentBits.GuildEmojisAndStickers,
+            GatewayIntentBits.GuildExpressions,
+            GatewayIntentBits.GuildMembers,
             GatewayIntentBits.GuildVoiceStates,
             GatewayIntentBits.Guilds,
             GatewayIntentBits.GuildMessages,
@@ -25,6 +27,13 @@ export async function launchBot(): Promise<Client> {
     client.on(Events.ClientReady, () => {
         if (client.user) {
             console.log(`client "ready": Logged in as ${client.user.tag}!`);
+
+            client.guilds.fetch().then(guilds => {
+                guilds.map((_, id) => {
+                    console.log(`cleaning up roles in guild with id ${id}`);
+                    cleanupMoods(client, id);
+                })
+            });
         } else {
             console.error(`client "ready": client.user is null!`);
         }

src/gptRequests.ts

@@ -72,7 +72,7 @@ export async function analyzeMoodColor(mood: string): Promise<string> {
                 "text": `
                     This text is supposed to show a text messaging app user's mood. The text might be an emotion, or a symbol for an emotion (such as a smiley)
                     or it could be something arbitrary. Try to come up with a color hexcode that depicts the mood, and return that as hexcode without the pound (#) symbol 
-                    (example, 000000) Only return the 6 character hexcode that is appropriate for the mood, and nothing else.
+                    (example, 000000) Only return the 6 character hexcode that is appropriate for the mood, and nothing else. Use saturated colors.
                 `
                 }
             ]