AsgardEternal/DBLogExtension
1
0
Fork 0
mirror of https://github.com/AsgardEternal/DBLogExtension.git synced 2024-09-28 11:54:24 -05:00
Code Issues Packages Projects Releases Wiki Activity

Update db-log-addOn.js

Browse Source
This commit is contained in:
Skillet 2023-05-08 13:58:18 -04:00 committed by GitHub
parent d06e9d353d
commit d2a93a4436
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 329 additions and 255 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

172
db-log-addOn.js
View File

@ -1,8 +1,12 @@
import Sequelize from 'sequelize';
// eslint-disable-file all
import Sequelize, { QueryTypes } from 'sequelize';
import DBLog from './db-log.js';
import path from "path";
import fs from "fs";
import { open } from "node:fs/promises"
const {DataTypes} = Sequelize;
const { DataTypes } = Sequelize;
const ServerState = {
init: 0,
seeding: 1,
@ -11,9 +15,7 @@ const ServerState = {
export default class DBLogPlayerTime extends DBLog {
static get description() {
return (
'replacement add-on to dblog for player join/seeding times'
);
return 'replacement add-on to dblog for player join/seeding times';
}
static get defaultEnabled() {
@ -27,6 +29,21 @@ export default class DBLogPlayerTime extends DBLog {
required: false,
description: 'seeding Threshold.',
default: 50
},
whitelistfilepath: {
required: false,
description: 'path to a file to write out auto-wl',
default: null
},
incseed: {
required: false,
description: 'rate of increase as a percentage to whitelist',
default: 0
},
decseed: {
required: false,
description: 'rate of decrease as a percentage to whitelist',
default: 0
}
};
}
@ -64,12 +81,12 @@ export default class DBLogPlayerTime extends DBLog {
);
this.models.Server.hasMany(this.models.PlayerTime, {
foreignKey: {name: 'server', allowNull: false},
foreignKey: { name: 'server', allowNull: false },
onDelete: 'CASCADE'
});
this.models.SteamUser.hasMany(this.models.PlayerTime, {
foreignKey: {name: 'player'},
foreignKey: { name: 'player' },
onDelete: 'CASCADE'
});
@ -80,13 +97,12 @@ export default class DBLogPlayerTime extends DBLog {
async prepareToMount() {
await super.prepareToMount();
await this.models.PlayerTime.sync();
}
async mount() {
console.log('Mounting db-log');
if(this.server.currentLayer){
if(this.server.currentLayer.gamemode === "Seed"){
if (this.server.currentLayer) {
if (this.server.currentLayer.gamemode === 'Seed') {
console.log('starting to seeding');
this.seeding = ServerState.seeding;
} else {
@ -94,7 +110,7 @@ export default class DBLogPlayerTime extends DBLog {
this.seeding = ServerState.live;
}
} else {
if(this.server.currentLayerRcon.layer.includes("Seed")){
if (this.server.currentLayerRcon.layer.includes('Seed')) {
console.log('starting to seeding');
this.seeding = ServerState.seeding;
} else {
@ -116,16 +132,26 @@ export default class DBLogPlayerTime extends DBLog {
console.log('starting DB repair for addOn');
const lastTickTime = await this.models.TickRate.findOne({
where: {server: this.options.overrideServerID || this.server.id},
where: { server: this.options.overrideServerID || this.server.id },
order: [['id', 'DESC']],
logging: console.log
}
);
});
console.log('last tick found:', lastTickTime);
if(!lastTickTime) return;
const lastServerDate = lastTickTime.time;
const lastServerTime = lastServerDate.getFullYear() + '-' + (lastServerDate.getMonth() + 1) + '-' + lastServerDate.getDate() + ' ' + lastServerDate.getHours() + ':' + lastServerDate.getMinutes() + ':' + lastServerDate.getSeconds();
const lastServerTime =
lastServerDate.getFullYear() +
'-' +
(lastServerDate.getMonth() + 1) +
'-' +
lastServerDate.getDate() +
' ' +
lastServerDate.getHours() +
':' +
lastServerDate.getMinutes() +
':' +
lastServerDate.getSeconds();
console.log('last time found:', lastServerTime);
const playerOnlineID = [];
@ -135,22 +161,20 @@ export default class DBLogPlayerTime extends DBLog {
}
console.log('players online:', playerOnlineID);
const {notIn, is} = Sequelize.Op;
const updateVals = {endTime: lastServerTime};
const { notIn, is } = Sequelize.Op;
const updateVals = { endTime: lastServerTime };
const whereStuff = {
endTime: {[is]: null},
endTime: { [is]: null },
server: this.options.overrideServerID || this.server.id,
player: {[notIn]: playerOnlineID}
player: { [notIn]: playerOnlineID }
};
console.log(updateVals);
console.log(whereStuff);
const rowUpdate = await this.models.PlayerTime.update(
updateVals, {
const rowUpdate = await this.models.PlayerTime.update(updateVals, {
where: whereStuff,
logging: console.log
}
);
});
console.log('updated playerTimes row count: %i', rowUpdate[0]);
console.log('finish DB repair');
@ -158,17 +182,29 @@ export default class DBLogPlayerTime extends DBLog {
async unmount() {
this.models.PlayerTime.update(
{leaveTime: 0},
{where: {leaveTime: null, server: this.options.overrideServerID || this.server.id}}
{ 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 updateCurrentTimeState(date, oldState, newState){
if(oldState === newState) return;
const timeNow = date.getFullYear() + '-' + (date.getMonth() + 1) + '-' + date.getDate() + ' ' + date.getHours() + ':' + date.getMinutes() + ':' + date.getSeconds();
async updateCurrentTimeState(date, oldState, newState) {
if (oldState === newState) return;
this.seeding = newState;
const timeNow =
date.getFullYear() +
'-' +
(date.getMonth() + 1) +
'-' +
date.getDate() +
' ' +
date.getHours() +
':' +
date.getMinutes() +
':' +
date.getSeconds();
console.log(timeNow);
const curPlayer = await this.models.PlayerTime.findAll({
where: {
@ -178,8 +214,8 @@ export default class DBLogPlayerTime extends DBLog {
}
});
console.log(curPlayer);
let curplayerarr = [];
for (const oneplayer of curPlayer){
const curplayerarr = [];
for (const oneplayer of curPlayer) {
console.log(oneplayer);
curplayerarr.push({
startTime: timeNow,
@ -201,32 +237,64 @@ export default class DBLogPlayerTime extends DBLog {
}
}
);
await this.models.PlayerTime.bulkCreate(curplayerarr,{
fields: ['startTime', 'endTime','serverState','session','server','player']
await this.models.PlayerTime.bulkCreate(curplayerarr, {
fields: ['startTime', 'endTime', 'serverState', 'session', 'server', 'player']
});
this.seeding = newState;
await this.updateAutoWL();
}
async onUpdatedA2SInformation(info) {
await super.onUpdatedA2SInformation(info);
// const curDateTime = new Date();
// if ((this.seeding !== ServerState.live) && (info.a2sPlayerCount >= this.options.seedingThreshold)) {
// console.log('switching to Live');
// await this.updateCurrentTimeState(curDateTime, this.seeding, ServerState.live);
// } else if (this.seeding === false && (info.a2sPlayerCount - 20) < this.options.seedingThreshold) {
// console.log('switching to seeding');
// await this.updateCurrentTimeState(curDateTime, this.seeding, ServerState.seeding);
// }
// const curDateTime = new Date();
// if ((this.seeding !== ServerState.live) && (info.a2sPlayerCount >= this.options.seedingThreshold)) {
// console.log('switching to Live');
// await this.updateCurrentTimeState(curDateTime, this.seeding, ServerState.live);
// } else if (this.seeding === false && (info.a2sPlayerCount - 20) < this.options.seedingThreshold) {
// console.log('switching to seeding');
// await this.updateCurrentTimeState(curDateTime, this.seeding, ServerState.seeding);
// }
}
async onNewGame(info){
async updateAutoWL() {
if(!this.options.whitelistfilepath) return;
// eslint-disable-next-line no-unused-vars
const seedTimes = await this.query(
'select lastName as playername, discordID, steamID, ' +
'sum(time_to_sec(timediff(ifnull(endTime,now()), startTime))/3600) as seedTime ' +
'from DBLog_PlayerTimes join DBLog_SteamUsers DLSU on DBLog_PlayerTimes.player = DLSU.steamID ' +
'where startTime between (now() - INTERVAL 1 MONTH) and now() and server != 3 and serverState=1 ' +
'group by player ' +
'order by seedTime desc',
{ type: QueryTypes.SELECT }
);
const topTime = seedTimes[0].seedTime;
const seedid = [];
for(const seeder of seedTimes){
if(((this.options.incseed*seeder.seedTime) - (this.options.decseed*(topTime-seeder.seedTime))) > 100) seedid.push(seeder);
}
const lcladminpath = path.resolve(__dirname, "../../../", this.options.whitelistfilepath);
if(!fs.existsSync(lcladminpath)) {
this.verbose(1, "WARNING: auto whitelist admins file not found");
return;
}
const adminfile = await open(lcladminpath, 'rw');
await adminfile.write(`Group=server-${this.server.options.id}-autowl:reserve\n`);
for(const seeding of seedid){
await adminfile.write(`Admin=${seeding.steamID}:server-${this.server.options.id}-autowl //name:${seeding.playername}, discord ID: ${seeding.discordID}\n`);
}
}
async onNewGame(info) {
await super.onNewGame(info);
console.log(info);
const curDateTime = info.time;
if(info.layer){
if(info.layer.gamemode === 'Seed'){
if (info.layer) {
if (info.layer.gamemode === 'Seed') {
console.log('switching to seeding');
await this.updateCurrentTimeState(curDateTime, this.seeding, ServerState.seeding);
} else {
@ -234,7 +302,7 @@ export default class DBLogPlayerTime extends DBLog {
await this.updateCurrentTimeState(curDateTime, this.seeding, ServerState.live);
}
} else {
if(info.layerClassname.includes("Seed")){
if (info.layerClassname.includes('Seed')) {
console.log('switching to seeding');
await this.updateCurrentTimeState(curDateTime, this.seeding, ServerState.seeding);
} else {
@ -242,6 +310,10 @@ export default class DBLogPlayerTime extends DBLog {
await this.updateCurrentTimeState(curDateTime, this.seeding, ServerState.live);
}
}
// eslint-disable-next-line no-empty
if (this.seeding !== ServerState.seeding) {
}
}
async onPlayerConnected(info) {
@ -262,7 +334,8 @@ export default class DBLogPlayerTime extends DBLog {
}
async onPlayerDisconnected(info) {
await new Promise(r => setTimeout(r, 500));
// eslint-disable-next-line promise/param-names
await new Promise((r) => setTimeout(r, 500));
console.log(info);
if (info.player) {
await this.models.SteamUser.upsert({
@ -271,13 +344,14 @@ export default class DBLogPlayerTime extends DBLog {
});
}
const rowAffect = await this.models.PlayerTime.update(
{endTime: info.time},
{where:
{ endTime: info.time },
{
where: {
player: info.steamID,
endTime: null,
server: this.options.overrideServerID || this.server.id
}}
}
}
);
console.log('player disconnect rows update: %i', rowAffect[0]);
}