Add database handlers (#12)

* Create database service file

* Separate populating guilds from creating table handler

* Add config for using database functions

* Cleanup

Author: vexuas

src/events/ready/index.ts

@@ -1,6 +1,7 @@
 import { REST, Routes } from 'discord.js';
 import { AppCommands } from '../../commands/commands';
-import { BOT_TOKEN, ENV, GUILD_IDS } from '../../config/environment';
+import { BOT_TOKEN, ENV, GUILD_IDS, USE_DATABASE } from '../../config/environment';
+import { createGuildTable, populateGuilds } from '../../services/database';
 import { EventModule } from '../events';
 
 const rest = new REST({ version: '9' }).setToken(BOT_TOKEN);
@@ -40,6 +41,10 @@ export default function ({ app, appCommands }: EventModule) {
   app.once('ready', async () => {
     try {
       await registerApplicationCommands(appCommands);
+      if (USE_DATABASE) {
+        await createGuildTable();
+        await populateGuilds(app.guilds.cache);
+      }
       console.log("I'm booting up! (◕ᴗ◕✿)");
     } catch (error) {
       console.log(error);

src/services/database.ts

@@ -0,0 +1,102 @@
+import { Collection, Guild } from 'discord.js';
+import { Pool, PoolClient } from 'pg';
+import { databaseConfig } from '../config/database';
+const pool: Pool = new Pool(databaseConfig);
+
+type GuildRecord = {
+  uuid: string;
+  name: string;
+  member_count: number;
+  owner_id: string;
+};
+
+export async function createGuildTable() {
+  const client: PoolClient = await pool.connect();
+  if (client) {
+    try {
+      await client.query('BEGIN');
+      const createGuildTableQuery =
+        'CREATE TABLE IF NOT EXISTS Guild(uuid TEXT NOT NULL PRIMARY KEY, name TEXT NOT NULL, member_count INTEGER NOT NULL, owner_id TEXT NOT NULL)';
+      await client.query(createGuildTableQuery);
+      await client.query('COMMIT');
+    } catch (error) {
+      await client.query('ROLLBACK');
+      console.log(error);
+      //TODO: Add error handling
+    } finally {
+      client.release();
+    }
+  }
+}
+export async function getGuilds() {
+  const client: PoolClient = await pool.connect();
+  if (client) {
+    try {
+      await client.query('BEGIN');
+      const getAllGuildsQuery = 'SELECT * FROM Guild';
+      const allGuilds = await client.query(getAllGuildsQuery);
+      return allGuilds;
+    } catch (error) {
+      await client.query('ROLLBACK');
+      console.log(error);
+      //TODO: Add error handling
+    } finally {
+      client.release();
+    }
+  }
+}
+export async function populateGuilds(existingGuilds: Collection<string, Guild>) {
+  try {
+    const guildsInDatabase = await getGuilds();
+    existingGuilds.forEach(async (guild: Guild) => {
+      const isInDatabase =
+        guildsInDatabase &&
+        guildsInDatabase.rows.find((guildDb: GuildRecord) => guildDb.uuid === guild.id);
+      if (!isInDatabase) {
+        await insertNewGuild(guild);
+      }
+    });
+  } catch (error) {
+    console.log(error);
+    //TODO: Add error handling
+  }
+}
+export async function insertNewGuild(newGuild: Guild) {
+  const client: PoolClient = await pool.connect();
+  if (client) {
+    try {
+      await client.query('BEGIN');
+      const insertNewGuildQuery =
+        'INSERT INTO Guild (uuid, name, member_count, owner_id) VALUES ($1, $2, $3, $4)';
+      await client.query(insertNewGuildQuery, [
+        newGuild.id,
+        newGuild.name,
+        newGuild.memberCount,
+        newGuild.ownerId,
+      ]);
+      await client.query('COMMIT');
+    } catch (error) {
+      await client.query('ROLLBACK');
+      console.log(error);
+      //TODO: Add error handling
+    } finally {
+      client.release();
+    }
+  }
+}
+export async function deleteGuild(existingGuild: Guild) {
+  const client: PoolClient = await pool.connect();
+  if (client) {
+    try {
+      await client.query('BEGIN');
+      const deleteGuildQuery = 'DELETE from Guild WHERE uuid = ($1)';
+      await client.query(deleteGuildQuery, [existingGuild.id]);
+      await client.query('COMMIT');
+    } catch (error) {
+      await client.query('ROLLBACK');
+      console.log(error);
+    } finally {
+      client.release();
+    }
+  }
+}