const { Model } = require('flitter-orm') /* * Team Model * ------------------------------------------------------------- */ class Team extends Model { static get services() { return [...super.services, 'output', 'models'] } /* * Define the flitter-orm schema of the model. */ static get schema() { return { user_id: String, team_name: String, team_num: Number, player_ids: [String], } } /** * Look up or create a team for the given user. * @param {User} user * @return {Promise} */ static async getForUser(user) { // try to find an existing team first const existing_team = await this.findOne({ user_id: user.id }) if ( existing_team ) return existing_team // otherwise create a team for the user const new_team = new this({ user_id: user.id, team_name: `${user.uid}'s team`, player_ids: [], }) // Generate the next team number const highest_num_team = await this.sort('-team_num').findOne() if ( highest_num_team ) { new_team.team_num = highest_num_team.team_num + 1 } else { new_team.team_num = 1 } await new_team.save() return new_team } /** * returns the lineup */ async lineup() { const Lineup = this.models.get('Lineup') return Lineup.get_and_update_for_team(this) } /** * Returns the players associated with the team. */ async players() { const Player = this.models.get('Player') return Player.find({ _id: { $in: this.player_ids.map(x => Player.to_object_id(x)) } }) } /** * Get the cumulative data for the team (total wins, losses, points scored & allowed) * @return object */ async cumulative_data() { const Matchup = this.models.get('Matchup') const home_matchups = await Matchup.find({ home_team_id: this.id }) const visitor_matchups = await Matchup.find({ visitor_team_id: this.id }) const data = { wins: 0, losses: 0, points_scored: 0, points_allowed: 0, } for ( const matchup of home_matchups ) { if ( !matchup.complete ) continue data.points_scored += matchup.home_team_score data.points_allowed += matchup.visitor_team_score if ( matchup.home_team_score > matchup.visitor_team_score ) { data.wins += 1 } else { data.losses += 0 } } for ( const matchup of visitor_matchups ) { if ( !matchup.complete ) continue data.points_scored += matchup.visitor_team_score data.points_allowed += matchup.home_team_score if ( matchup.visitor_team_score > matchup.home_team_score ) { data.wins += 1 } else { data.losses += 0 } } return data } /** * updates the API's data */ async to_api() { const User = this.models.get('auth:User') return { user_id: this.user_id, user_display: (await User.findById(this.user_id))?.uid || 'Unknown User', team_name: this.team_name, team_num: this.team_num, } } } module.exports = exports = Team