SquadJSDocker/squadjsPlugins/db-log-addOn.js
2023-02-15 02:28:21 -05:00

139 lines
4.5 KiB
JavaScript

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 && this.seeding === true) {
await this.models.PlayerTime.update(
{ seedTime: info.time },
{ where: { seedTime: null, joinedSeeding: 1, leaveTime: null, server: this.options.overrideServerID || this.server.id } }
);
}
if(this.seeding === true && info.a2sPlayerCount >= this.options.seedingThreshold) this.seeding = false;
else if(this.seeding === false && (info.a2sPlayerCount-20) < this.options.seedingThreshold) this.seeding = 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) {
if(info.player){
await this.models.PlayerTime.update(
{ leaveTime: info.time },
{ where: { player: info.player.steamID, leaveTime: null, server: this.options.overrideServerID || this.server.id } }
);
}
}
}