121 lines
3.4 KiB
JavaScript
121 lines
3.4 KiB
JavaScript
/*Canopy - The next generation of stoner streaming software
|
|
Copyright (C) 2024 Rainbownapkin and the TTN Community
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as
|
|
published by the Free Software Foundation, either version 3 of the
|
|
License, or (at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>.*/
|
|
|
|
//NPM Imports
|
|
const {mongoose} = require('mongoose');
|
|
|
|
//Local Imports
|
|
const {userSchema} = require('./userSchema');
|
|
|
|
const userBanSchema = new mongoose.Schema({
|
|
user: {
|
|
type: mongoose.SchemaTypes.ObjectID,
|
|
required: true,
|
|
ref: "user"
|
|
},
|
|
//To be used in future when ip-hashing/better session tracking is implemented
|
|
ips: {
|
|
type: [mongoose.SchemaTypes.String],
|
|
required: false
|
|
},
|
|
//To be used in future when alt-detection has been implemented
|
|
alts: {
|
|
type: [userSchema],
|
|
required: false
|
|
},
|
|
banDate: {
|
|
|
|
type: mongoose.SchemaTypes.Date,
|
|
required: true,
|
|
default: new Date()
|
|
},
|
|
expirationDays: {
|
|
type: mongoose.SchemaTypes.Number,
|
|
required: true,
|
|
default: 30
|
|
},
|
|
//If true, then expiration date deletes associated accounts instead of deleting the ban record
|
|
deleteAccountOnExpire: {
|
|
type: mongoose.SchemaTypes.Boolean,
|
|
required: true,
|
|
default: false
|
|
}
|
|
});
|
|
|
|
userBanSchema.statics.checkBanByUserDoc = async function(userDB){
|
|
const banDB = await this.find({});
|
|
var foundBan = null;
|
|
|
|
banDB.forEach((ban) => {
|
|
if(ban.user.toString() == userDB._id.toString()){
|
|
foundBan = ban;
|
|
}
|
|
});
|
|
|
|
return foundBan;
|
|
}
|
|
|
|
userBanSchema.statics.checkBan = async function(user){
|
|
const userDB = await userModel.findOne({user: user.user});
|
|
return this.checkBanByUserDoc(userDB);
|
|
}
|
|
|
|
userBanSchema.statics.banByUserDoc = async function(userDB){
|
|
if(await this.checkBanByUserDoc(userDB) != null){
|
|
throw new Error("User already banned");
|
|
}
|
|
|
|
return await this.create({user: userDB._id});
|
|
}
|
|
|
|
userBanSchema.statics.ban = async function(user){
|
|
const userDB = await userModel.findOne({user: user.user});
|
|
return this.banByUserDoc(userDB);
|
|
}
|
|
|
|
userBanSchema.statics.getBans = async function(){
|
|
const banDB = await this.find({}).populate('user');
|
|
var bans = [];
|
|
|
|
banDB.forEach((ban) => {
|
|
//Calcualte expiration date
|
|
var expirationDate = new Date(ban.banDate);
|
|
expirationDate.setDate(expirationDate.getDate() + ban.expirationDays);
|
|
|
|
const userObj = {
|
|
id: ban.user.id,
|
|
user: ban.user.user,
|
|
img: ban.user.img,
|
|
date: ban.user.date
|
|
}
|
|
|
|
const banObj = {
|
|
banDate: ban.banDate,
|
|
expirationDays: ban.expirationDays,
|
|
expirationDate: expirationDate,
|
|
user: userObj,
|
|
ips: ban.ips,
|
|
alts: ban.alts,
|
|
deleteAccountOnExpire: ban.deleteAccountOnExpire
|
|
}
|
|
|
|
bans.push(banObj);
|
|
});
|
|
|
|
return bans;
|
|
}
|
|
|
|
module.exports = mongoose.model("userBan", userBanSchema); |