import { getModelForClass, modelOptions, prop, ReturnModelType, Severity } 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. */ @modelOptions({ options: { allowMixed: Severity.ALLOW } }) export class Player { /** * The id of the player. */ @prop() public _id!: string; /** * The player's statistic history. */ @prop() private statisticHistory?: Record; /** * The date the player was last tracked. */ @prop() public lastTracked?: Date; /** * The date the player was first tracked. */ @prop() public trackedSince?: Date; /** * Gets the player's statistic history. */ public getStatisticHistory(): Record { 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 { const statisticHistory = this.getStatisticHistory(); const history: Record = {}; for (let i = 0; i < days; i++) { const date = formatDateMinimal(getMidnightAlignedDate(getDaysAgoDate(i))); const playerHistory = statisticHistory[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.statisticHistory[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 = {}; } this.statisticHistory = Object.fromEntries( Object.entries(this.statisticHistory).sort((a, b) => new Date(b[0]).getTime() - new Date(a[0]).getTime()) ); } /** * 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 = getModelForClass(Player);