mirror of
https://github.com/ENSL/ensl_gathers.git
synced 2024-11-26 06:10:58 +00:00
132 lines
3.5 KiB
JavaScript
132 lines
3.5 KiB
JavaScript
"use strict";
|
|
|
|
/*
|
|
* Implements User Model
|
|
*
|
|
*/
|
|
|
|
var _ = require("lodash");
|
|
var async = require("async");
|
|
var mongoose = require("mongoose");
|
|
var Profile = mongoose.model("Profile");
|
|
var steam = require('steamidconvert')();
|
|
var enslClient = require("../ensl/client")();
|
|
var hiveClient = require("../hive/client")();
|
|
|
|
function User (user) {
|
|
this.id = user['id'];
|
|
this.online = true;
|
|
this.username = user['username'];
|
|
this.country = user['country'];
|
|
this.time_zone = user['time_zone'];
|
|
this.avatar = enslClient.getFullAvatarUri(user['avatar']);
|
|
this.admin = user['admin'];
|
|
this.moderator = user['moderator'];
|
|
this.team = user['team'];
|
|
this.bans = user['bans'];
|
|
if (user['steam']) {
|
|
this.steam = {
|
|
id: user['steam']['id'] || this.getSteamId(),
|
|
url: user['steam']['url'] || null,
|
|
nickname: user['steam']['nickname'] || null
|
|
};
|
|
} else {
|
|
this.steam = {
|
|
id: null,
|
|
url: null,
|
|
nickname: null
|
|
};
|
|
}
|
|
this.profile = null;
|
|
this.hive = {
|
|
id: null
|
|
};
|
|
|
|
if (this.steam.id) {
|
|
this.hive.id = this.getHiveId();
|
|
}
|
|
}
|
|
|
|
User.prototype.isChatAdmin = function () {
|
|
return this.admin || this.moderator;
|
|
};
|
|
|
|
User.prototype.isGatherAdmin = function () {
|
|
return this.admin || this.moderator;
|
|
};
|
|
|
|
User.prototype.isUserAdmin = function () {
|
|
return this.admin;
|
|
};
|
|
|
|
User.prototype.getSteamId = function () {
|
|
if (this.steam.url === null) return null;
|
|
var urlId = this.steam.url.match(/\d*$/);
|
|
if (!urlId || urlId[0].length === 0) return null;
|
|
return steam.convertToText(urlId[0]);
|
|
};
|
|
|
|
User.prototype.getHiveId = function () {
|
|
var steamId = this.steam.id;
|
|
if (!steamId) return null;
|
|
var index = steamId.match(/:0:/) ? 0 : 1;
|
|
var tailId = parseInt(steamId.match(/\d*$/), 10);
|
|
return index === 1 ? (tailId * 2) + 1 : tailId * 2;
|
|
};
|
|
|
|
var allowedAttributes = ["enslo", "division", "skill", "gatherMusic"];
|
|
var allowedAbilities = ["skulk", "lerk", "fade", "gorge", "onos", "commander"];
|
|
User.prototype.updateProfile = function (data, callback) {
|
|
let self = this;
|
|
Profile.findOne({userId: self.id}, (error, profile) => {
|
|
if (error) return callback(error);
|
|
allowedAttributes.forEach(function (attr) {
|
|
if (data[attr] !== undefined) profile[attr] = data[attr];
|
|
});
|
|
if (data.abilities) {
|
|
allowedAbilities.forEach(function (attr) {
|
|
let newAbility = data.abilities[attr];
|
|
let abilities = profile.abilities;
|
|
if (newAbility !== undefined) abilities[attr] = newAbility;
|
|
});
|
|
}
|
|
profile.save(function (error, profile) {
|
|
if (error) return callback(error);
|
|
self.profile = profile.toJson();
|
|
return callback(error, profile);
|
|
});
|
|
});
|
|
};
|
|
|
|
User.find = function (id, callback) {
|
|
enslClient.getUserById({
|
|
id: id
|
|
}, (error, response, body) => {
|
|
if (error) return callback(error);
|
|
if (response.statusCode !== 200) return callback(new Error("Unable to auth user against API"));
|
|
let user = new User(body);
|
|
async.parallel([
|
|
// Retrieve or create user profile from local store
|
|
callback => {
|
|
Profile.findOrCreate(user, (error, profile) => {
|
|
if (error) return callback(error);
|
|
user.profile = profile.toJson();
|
|
return callback(null, profile);
|
|
});
|
|
},
|
|
// Retrive Hive Stats
|
|
callback => {
|
|
hiveClient.getUserStatsV2(user, (error, response, userStats) => {
|
|
if (error || response.statusCode !== 200 || userStats.id === null) return callback();
|
|
_.assign(user.hive, userStats);
|
|
return callback(null, userStats);
|
|
});
|
|
}
|
|
], function (error, result) {
|
|
if (error) return callback(error);
|
|
return callback(null, user);
|
|
})
|
|
});
|
|
};
|
|
|
|
module.exports = User;
|