discord-leveling-super
Version:
Easy and customizable leveling framework for your Discord bot.
144 lines (143 loc) • 6.75 kB
JavaScript
"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
const discord_js_1 = require("discord.js");
const Emitter_1 = __importDefault(require("../classes/Emitter"));
const LevelingError_1 = __importDefault(require("../classes/LevelingError"));
const DatabaseManager_1 = __importDefault(require("./DatabaseManager"));
const Errors_1 = __importDefault(require("../structures/Errors"));
/**
* Total XP manager methods class.
* @extends {Emitter}
*/
class TotalXPManager extends Emitter_1.default {
/**
* Database Manager.
* @type {DatabaseManager}
* @private
*/
database;
/**
* Total XP manager methods class.
* @param {LevelingOptions} options Leveling options object.
*/
constructor(options) {
super();
this.database = new DatabaseManager_1.default(options);
}
/**
* Gets the XP for specified user.
* @param {String | GuildMember | User} member Member or it's ID.
* @param {String | Guild} guild Guild or it's ID.
* @returns {Number} Amount of total XP.
*/
get(member, guild) {
const isUser = member instanceof discord_js_1.GuildMember || member instanceof discord_js_1.User;
const isGuild = guild instanceof discord_js_1.Guild;
if (typeof member !== 'string' && !isUser)
throw new LevelingError_1.default(Errors_1.default.invalidTypes.member + typeof member);
if (typeof guild !== 'string' && !isGuild)
throw new LevelingError_1.default(Errors_1.default.invalidTypes.guild + typeof guild);
const user = isUser ? member.id : member.toString();
const botGuild = isGuild ? guild.id : guild.toString();
const totalXP = this.database.fetch(`${botGuild}.${user}.totalXP`);
return totalXP;
}
/**
* Sets the XP for specified user.
* @fires Leveling#setTotalXP
* @param {Number} totalXP Amount of total XP.
* @param {String | GuildMember | User} member Member or it's ID.
* @param {String | Guild} guild Guild or it's ID.
* @returns {Boolean} If set successfully: true, else: false.
*/
set(totalXP, member, guild) {
const isUser = member instanceof discord_js_1.GuildMember || member instanceof discord_js_1.User;
const isGuild = guild instanceof discord_js_1.Guild;
if (typeof member !== 'string' && !isUser)
throw new LevelingError_1.default(Errors_1.default.invalidTypes.member + typeof member);
if (typeof guild !== 'string' && !isGuild)
throw new LevelingError_1.default(Errors_1.default.invalidTypes.guild + typeof guild);
const user = isUser ? member.id : member.toString();
const botGuild = isGuild ? guild.id : guild.toString();
const userLevelData = this.database.fetch(`${botGuild}.${user}`);
this.emit('setTotalXP', {
userData: userLevelData.userData,
guildID: botGuild,
userID: user,
xp: userLevelData.xp,
level: userLevelData.level,
totalXP,
maxXP: userLevelData.maxXP,
difference: userLevelData.difference,
multiplier: userLevelData.multiplier
});
return this.database.set(`${botGuild}.${user}.totalXP`, totalXP);
}
/**
* Adds the XP for specified user.
* @fires Leveling#addTotalXP
* @param {Number} totalXP Amount of total XP.
* @param {String | GuildMember | User} member Member or it's ID.
* @param {String | Guild} guild Guild or it's ID.
* @param {Boolean} onMessage The value will be true if the method was called on 'messageCreate' bot event.
* @returns {Boolean} If added successfully: true, else: false.
*/
add(totalXP, member, guild, onMessage = false) {
const isUser = member instanceof discord_js_1.GuildMember || member instanceof discord_js_1.User;
const isGuild = guild instanceof discord_js_1.Guild;
if (typeof member !== 'string' && !isUser)
throw new LevelingError_1.default(Errors_1.default.invalidTypes.member + typeof member);
if (typeof guild !== 'string' && !isGuild)
throw new LevelingError_1.default(Errors_1.default.invalidTypes.guild + typeof guild);
const user = isUser ? member.id : member.toString();
const botGuild = isGuild ? guild.id : guild.toString();
const userData = this.database.fetch(`${botGuild}.${user}`);
this.emit('addTotalXP', {
guildID: botGuild,
userID: user,
xp: userData.xp,
level: userData.level,
gainedXP: totalXP,
totalXP: userData.totalXP + totalXP,
maxXP: userData.maxXP,
difference: userData.difference,
multiplier: userData.multiplier,
onMessage
});
return this.database.add(`${botGuild}.${user}.totalXP`, totalXP);
}
/**
* Subtracts the XP for specified user.
* @fires Leveling#subtractTotalXP
* @param {Number} totalXP Amount of total XP.
* @param {String | GuildMember | User} member Member or it's ID.
* @param {String | Guild} guild Guild or it's ID.
* @returns {Boolean} If subtracted successfully: true, else: false.
*/
subtract(totalXP, member, guild) {
const isUser = member instanceof discord_js_1.GuildMember || member instanceof discord_js_1.User;
const isGuild = guild instanceof discord_js_1.Guild;
if (typeof member !== 'string' && !isUser)
throw new LevelingError_1.default(Errors_1.default.invalidTypes.member + typeof member);
if (typeof guild !== 'string' && !isGuild)
throw new LevelingError_1.default(Errors_1.default.invalidTypes.guild + typeof guild);
const user = isUser ? member.id : member.toString();
const botGuild = isGuild ? guild.id : guild.toString();
const userLevelData = this.database.fetch(`${botGuild}.${user}`);
this.emit('subtractTotalXP', {
userData: userLevelData.userData,
guildID: botGuild,
userID: user,
xp: userLevelData.xp,
level: userLevelData.level,
totalXP: userLevelData.totalXP - totalXP,
maxXP: userLevelData.maxXP,
difference: userLevelData.difference,
multiplier: userLevelData.multiplier
});
return this.database.subtract(`${botGuild}.${user}.totalXP`, totalXP);
}
}
module.exports = TotalXPManager;