From f88487b27bd624a535fbd793e275aa495031ce97 Mon Sep 17 00:00:00 2001
From: Skillet <kilerskyman@gmail.com>
Date: Mon, 13 Feb 2023 06:33:08 -0500
Subject: [PATCH] Create db-log-addOn.js

---
 squadjsPlugins/db-log-addOn.js | 134 +++++++++++++++++++++++++++++++++
 1 file changed, 134 insertions(+)
 create mode 100644 squadjsPlugins/db-log-addOn.js

diff --git a/squadjsPlugins/db-log-addOn.js b/squadjsPlugins/db-log-addOn.js
new file mode 100644
index 0000000..b49607e
--- /dev/null
+++ b/squadjsPlugins/db-log-addOn.js
@@ -0,0 +1,134 @@
+import Sequelize from 'sequelize';
+
+import DBLog from './db-log.js';
+
+const { DataTypes } = Sequelize;
+
+export default class DBLogPlayerTime extends DBLog {
+  static get description() {
+    return (
+                'replacement add-on to dblog for player join/seeding times'
+    );
+  }
+
+  static get defaultEnabled() {
+    return false;
+  }
+
+  static get optionsSpecification() {
+    return {
+      ...DBLog.optionsSpecification,
+          seedingThreshold: {
+                required: false,
+                description: 'seeding Threshold.',
+                default: 50
+          }
+    };
+  }
+
+  constructor(server, options, connectors) {
+    super(server, options, connectors);
+
+        this.seeding = false;
+
+        this.createModel(
+                'PlayerTime',
+                {
+                        id: {
+                                type: DataTypes.INTEGER,
+                                primaryKey: true,
+                                autoIncrement: true
+                        },
+                        playerName: {
+                                type: DataTypes.STRING
+                        },
+                        joinTime: {
+                                type: DataTypes.DATE
+                        },
+                        leaveTime: {
+                                type: DataTypes.DATE
+                        },
+                        seedTime: {
+                                type: DataTypes.DATE
+                        },
+                        joinedSeeding: {
+                                type: DataTypes.BOOLEAN
+                        }
+                },
+                {
+                        charset: 'utf8mb4',
+                        collate: 'utf8mb4_unicode_ci'
+                }
+        );
+
+        this.models.Server.hasMany(this.models.PlayerTime, {
+                foreignKey: { name: 'server', allowNull: false },
+                onDelete: 'CASCADE'
+        });
+
+        this.models.SteamUser.hasMany(this.models.PlayerTime, {
+                foreignKey: {name: 'player' },
+                onDelete: 'CASCADE'
+        });
+
+        this.onPlayerConnected = this.onPlayerConnected.bind(this);
+        this.onPlayerDisconnected = this.onPlayerDisconnected.bind(this);
+  }
+
+  async prepareToMount() {
+    await super.prepareToMount();
+        await this.models.PlayerTime.sync();
+  }
+
+  async mount() {
+    await super.mount();
+        this.server.on('PLAYER_CONNECTED', this.onPlayerConnected);
+        this.server.on('PLAYER_DISCONNECTED', this.onPlayerDisconnected);
+  }
+
+  async unmount() {
+        this.models.PlayerTime.update(
+                { leaveTime: 0 },
+                { where: { leaveTime: null , server: this.options.overrideServerID || this.server.id } }
+        );
+    await super.unmount();
+        this.server.removeEventListener('PLAYER_CONNECTED', this.onPlayerConnected);
+        this.server.removeEventListener('PLAYER_DISCONNECTED', this.onPlayerDisconnected);
+  }
+
+  async onUpdatedA2SInformation(info) {
+    await super.onUpdatedA2SInformation(info);
+
+        if(info.a2sPlayerCount >= this.options.seedingThreshold && seeding === true) {
+                await this.models.PlayerTime.update(
+                        { seedTime: info.time },
+                        { where: { seedTime: null, joinedSeeding: 1, leaveTime: null, server: this.options.overrideServerID || this.server.id } }
+                );
+        }
+
+        this.seeding = info.a2sPlayerCount >= this.options.seedingThreshold ? false : true;
+  }
+
+  async onPlayerConnected(info) {
+        if(info.player)
+                await this.models.SteamUser.upsert({
+                        steamID: info.player.steamID,
+                        lastName: info.player.name
+                });
+
+        await this.models.PlayerTime.create({
+                server: this.options.overrideServerID || this.server.id,
+                player: info.player ? info.player.steamID : null,
+                playerName: info.player ? info.player.name : null,
+                joinTime: info.time,
+                joinedSeeding: this.seeding
+        });
+  }
+
+  async onPlayerDisconnected(info) {
+        await this.models.PlayerTime.update(
+                { leaveTime: info.time },
+                { where: { player: info.player.steamID, leaveTime: null, server: this.options.overrideServerID || this.server.id } }
+        );
+  }
+}