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 } } ); } 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) { await this.models.PlayerTime.update( { leaveTime: info.time }, { where: { player: info.player.steamID, leaveTime: null, server: this.options.overrideServerID || this.server.id } } ); } }