Add an action to query teams from a database (#19)

config/config-example.js

@@ -155,3 +155,8 @@ exports.standings = {
 	"30": "Permaban",
 	"100": "Disabled",
 };
+
+/**
+ * @type {import('pg').PoolConfig | null}
+ */
+exports.postgres = null;

package.json

@@ -15,6 +15,7 @@
   },
   "dependencies": {
     "@types/node": "^15.12.4",
+    "@types/pg": "^8.10.3",
     "bcrypt": "^5.0.1",
     "eslint-plugin-import": "^2.24.2",
     "google-auth-library": "^3.1.2",

src/actions.ts

@@ -654,6 +654,22 @@ export const actions: {[k: string]: QueryHandler} = {
 		await tables.oauthTokens.deleteAll()`WHERE client = ${client.id} and owner = ${this.user.id}`;
 		return {success: true};
 	},
+
+	async getteams(params) {
+		if (!this.user.loggedIn || this.user.id === 'guest') {
+			return {teams: []}; // don't wanna nag people with popups if they aren't logged in
+		}
+		let teams = [];
+		try {
+			teams = await tables.pgdb.query(
+				'SELECT teamid, team, format, title as name FROM teams WHERE ownerid = $1', [this.user.id]
+			) ?? [];
+		} catch (e) {
+			Server.crashlog(e, 'a teams database query', params);
+			throw new ActionError('The server could not load your teams. Please try again later.');
+		}
+		return {teams};
+	},
 };
 
 if (Config.actions) {

src/database.ts

@@ -7,6 +7,7 @@
  */
 
 import * as mysql from 'mysql2';
+import * as pg from 'pg';
 
 export type BasicSQLValue = string | number | null;
 export type SQLRow = {[k: string]: BasicSQLValue};
@@ -290,3 +291,16 @@ export class DatabaseTable<Row> {
 		return this.updateAll(data)`WHERE \`${this.primaryKeyName}\` = ${primaryKey} LIMIT 1`;
 	}
 }
+
+export class PGDatabase {
+	database: pg.Pool | null;
+	constructor(config: pg.PoolConfig | null) {
+		this.database = config ? new pg.Pool(config) : null;
+	}
+	async query<O = any>(query: string, values: BasicSQLValue[]) {
+		if (!this.database) return null;
+		const result = await this.database.query(query, values);
+		return result.rows as O[];
+	}
+}
+

src/tables.ts

@@ -1,14 +1,15 @@
 /**
  * Login server database tables
  */
-import {Database, DatabaseTable} from './database';
+import {Database, DatabaseTable, PGDatabase} from './database';
 import {Config} from './config-loader';
 
 import type {LadderEntry} from './ladder';
 import type {ReplayData} from './replays';
 
 // direct access
 export const psdb = new Database(Config.mysql);
+export const pgdb = new PGDatabase(Config.postgres);
 export const replaysDB = Config.replaysdb ? new Database(Config.replaysdb!) : psdb;
 export const ladderDB = Config.ladderdb ? new Database(Config.ladderdb!) : psdb;