eecs448-project-4/app/models/Team.model.js

80 lines
2.0 KiB
JavaScript
Raw Normal View History

const { Model } = require('flitter-orm')
/*
* Team Model
* -------------------------------------------------------------
*/
class Team extends Model {
static get services() {
2020-11-05 03:18:47 +00:00
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],
}
}
2020-11-05 03:18:47 +00:00
/**
* Look up or create a team for the given user.
* @param {User} user
* @return {Promise<Team>}
*/
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
}
async lineup() {
const Lineup = this.models.get('Lineup')
return Lineup.get_and_update_for_team(this)
}
async players() {
const Player = this.models.get('Player')
return Player.find({
_id: {
$in: this.player_ids.map(x => Player.to_object_id(x))
}
})
}
2020-11-05 03:18:47 +00:00
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