109 lines
3.2 KiB
TypeScript
109 lines
3.2 KiB
TypeScript
import { getModelForClass, prop, ReturnModelType } from "@typegoose/typegoose";
|
|
import { Document } from "mongoose";
|
|
import { PlayerHistory } from "@ssr/common/types/player/player-history";
|
|
import { formatDateMinimal, getDaysAgoDate, getMidnightAlignedDate } from "@ssr/common/utils/time-utils";
|
|
|
|
/**
|
|
* The model for a player.
|
|
*/
|
|
export class Player {
|
|
/**
|
|
* The id of the player.
|
|
*/
|
|
@prop()
|
|
public _id!: string;
|
|
|
|
/**
|
|
* The player's statistic history.
|
|
*/
|
|
@prop()
|
|
private statisticHistory?: Record<string, PlayerHistory>;
|
|
|
|
@prop()
|
|
public lastTracked?: Date;
|
|
|
|
/**
|
|
* Gets the player's statistic history.
|
|
*/
|
|
public getStatisticHistory(): Record<string, PlayerHistory> {
|
|
if (this.statisticHistory === undefined) {
|
|
this.statisticHistory = {};
|
|
}
|
|
return this.statisticHistory;
|
|
}
|
|
|
|
/**
|
|
* Gets the player's history for a specific date.
|
|
*
|
|
* @param date the date to get the history for.
|
|
*/
|
|
public getHistoryByDate(date: Date): PlayerHistory {
|
|
if (this.statisticHistory === undefined) {
|
|
this.statisticHistory = {};
|
|
}
|
|
return this.getStatisticHistory()[formatDateMinimal(getMidnightAlignedDate(date))] || {};
|
|
}
|
|
|
|
/**
|
|
* Gets the player's history for the previous X days.
|
|
*
|
|
* @param days the number of days to get the history for.
|
|
*/
|
|
public getHistoryPreviousDays(days: number): Record<string, PlayerHistory> {
|
|
if (this.statisticHistory === undefined) {
|
|
this.statisticHistory = {};
|
|
}
|
|
const history: Record<string, PlayerHistory> = {};
|
|
for (let i = 0; i < days; i++) {
|
|
const date = formatDateMinimal(getMidnightAlignedDate(getDaysAgoDate(i)));
|
|
const playerHistory = this.getStatisticHistory()[date];
|
|
if (playerHistory === undefined || Object.keys(playerHistory).length === 0) {
|
|
continue;
|
|
}
|
|
history[date] = playerHistory;
|
|
}
|
|
return history;
|
|
}
|
|
|
|
/**
|
|
* Sets the player's statistic history.
|
|
*
|
|
* @param date the date to set it for.
|
|
* @param history the history to set.
|
|
*/
|
|
public setStatisticHistory(date: Date, history: PlayerHistory) {
|
|
if (this.statisticHistory === undefined) {
|
|
this.statisticHistory = {};
|
|
}
|
|
this.getStatisticHistory()[formatDateMinimal(getMidnightAlignedDate(date))] = history;
|
|
}
|
|
|
|
/**
|
|
* Sorts the player's statistic history by
|
|
* date in descending order. (oldest to newest)
|
|
*/
|
|
public sortStatisticHistory() {
|
|
if (this.statisticHistory === undefined) {
|
|
this.statisticHistory = {};
|
|
}
|
|
return Object.entries(this.getStatisticHistory())
|
|
.sort((a, b) => Date.parse(b[0]) - Date.parse(a[0]))
|
|
.reduce((acc, [key, value]) => ({ ...acc, [key]: value }), {});
|
|
}
|
|
|
|
/**
|
|
* Gets the number of days tracked.
|
|
*
|
|
* @returns the number of days tracked.
|
|
*/
|
|
public getDaysTracked(): number {
|
|
return Object.keys(this.getStatisticHistory()).length;
|
|
}
|
|
}
|
|
|
|
// This type defines a Mongoose document based on Player.
|
|
export type PlayerDocument = Player & Document;
|
|
|
|
// This type ensures that PlayerModel returns Mongoose documents (PlayerDocument) that have Mongoose methods (save, remove, etc.)
|
|
export const PlayerModel: ReturnModelType<typeof Player> = getModelForClass(Player);
|