Merge b45f3dcc35 into 12054c35c5

src/adapters/cache/MapCache.js

@@ -1,4 +1,4 @@
-import ICacheAdapter from "./ICacheAdapter";
+import ICacheAdapter from "./ICacheAdapter.js";
 
 export default class MapCache extends ICacheAdapter {
 	constructor() {

src/helpers/birthdays.js

@@ -1,69 +1,71 @@
 import { CronJob } from "cron";
+import useClient from "../utils/use-client.js";
+import UserModel from "../models/UserModel.js";
+import { createEmbed } from "../utils/create-embed.js";
+import logger from "./logger.js";
+import { getNoun } from "./functions.js";
 
-/**
- *
- * @param {import("../base/Client")} client
- */
-async function checkBirthdays(client) {
-	for (const guild of client.guilds.cache.values()) {
+const checkBirthdays = async () => {
+	const client = useClient();
+
+	const guilds = client.guilds.cache.values();
+	const users = await client.adapter.find(UserModel, { birthdate: { $gt: 1 } });
+
+	const currentData = new Date();
+	const currentYear = currentData.getFullYear();
+	const currentMonth = currentData.getMonth();
+	const currentDate = currentData.getDate();
+
+	for (const guild of guilds) {
 		try {
-			const guildData = await client.getGuildData(guild.id);
-			const channel = guildData.plugins.birthdays ? await client.channels.fetch(guildData.plugins.birthdays) : null;
+			const data = await client.getGuildData(guild.id);
+			const channel = data.plugins.birthdays ? await client.channels.fetch(data.plugins.birthdays) : null;
 
-			if (channel) {
-				const date = new Date();
-				const currentDay = date.getDate();
-				const currentMonth = date.getMonth() + 1;
-				const currentYear = date.getFullYear();
+			if (!channel) return;
 
-				const users = await client.usersData.find({ birthdate: { $gt: 1 } });
+			const userIDs = users.filter(u => guild.members.cache.has(u.id)).map(u => u.id);
 
-				for (const user of users) {
-					if (!guild.members.cache.has(user.id)) continue;
+			await Promise.all(
+				userIDs.map(async userID => {
+					const user = users.find(u => u.id === userID);
+					const userData = new Date(user.birthdate).getFullYear() <= 1970 ? new Date(user.birthdate * 1000) : new Date(user.birthdate);
+					const userYear = userData.getFullYear();
+					const userMonth = userData.getMonth();
+					const userDate = userData.getDate();
 
-					const userDate = new Date(user.birthdate).getFullYear() <= 1970 ? new Date(user.birthdate * 1000) : new Date(user.birthdate);
-					const day = userDate.getDate();
-					const month = userDate.getMonth() + 1;
-					const year = userDate.getFullYear();
-					const age = currentYear - year;
+					const age = currentYear - userYear;
 
-					if (currentMonth === month && currentDay === day) {
-						const embed = client.embed({
-							author: client.user.getUsername(),
+					if (userDate === currentDate && userMonth === currentMonth) {
+						const embed = createEmbed({
+							author: client.user.username,
 							fields: [
 								{
-									name: client.translate("economy/birthdate:HAPPY_BIRTHDAY", null, guildData.language),
-									value: client.translate(
-										"economy/birthdate:HAPPY_BIRTHDAY_MESSAGE",
-										{
+									name: client.translate("economy/birthdate:HAPPY_BIRTHDAY", {
+										lng: data.language,
+									}),
+									value: client.translate("economy/birthdate:HAPPY_BIRTHDAY_MESSAGE", {
+										lng: data.language,
 										user: user.id,
-											age: `**${age}** ${client.functions.getNoun(
-												age,
-												client.translate("misc:NOUNS:AGE:1", null, guildData.language),
-												client.translate("misc:NOUNS:AGE:2", null, guildData.language),
-												client.translate("misc:NOUNS:AGE:5", null, guildData.language),
-											)}`,
-										},
-										guildData.language,
-									),
+										age: `**${age}** ${getNoun(age, [
+											client.translate("misc:NOUNS:AGE:1", null, data.language),
+											client.translate("misc:NOUNS:AGE:2", null, data.language),
+											client.translate("misc:NOUNS:AGE:5", null, data.language),
+										])}`,
+									}),
 								},
 							],
 						});
 
-						await channel.send({ embeds: [embed] }).then(m => m.react("🎉"));
-					}
-				}
-			}
-		} catch (e) {
-			if (e.code === 10003) console.log(`No channel found for ${guild.name}`);
-			else console.error(`Error processing birthdays for guild "${guild.name}":`, e);
-		}
+						await channel.send({ embeds: [embed] }).then(m => m.react(" "));
+					}
+				}),
+			);
+		} catch (error) {
+			logger.error(error);
 		}
 	}
+};
 
-export async function init(client) {
-	new CronJob("0 5 * * *", checkBirthdays(client), null, true, "Europe/Moscow");
-}
-export async function run(client) {
-	await checkBirthdays(client);
+export async function init() {
+	new CronJob("0 5 * * *", checkBirthdays(), null, true, "Europe/Moscow");
 }

src/helpers/checkReminds.js

@@ -1,25 +1,27 @@
-/**
- *
- * @param {import("../base/Client")} client
- */
-async function checkReminds(client) {
-	client.usersData.find({ reminds: { $gt: [] } }).then(users => {
+import UserModel from "../models/UserModel";
+import useClient from "../utils/use-client";
+
+const checkReminds = async () => {
+	const client = useClient();
+
+	client.adapter.find(UserModel, { reminds: { $gt: [] } }).then(users => {
 		for (const user of users) {
 			if (!client.users.cache.has(user.id)) client.users.fetch(user.id);
 
-			client.databaseCache.usersReminds.set(user.id, user);
+			client.cacheReminds.set(user.id, user);
 		}
 	});
 
-	client.databaseCache.usersReminds.forEach(async user => {
+	client.cacheReminds.forEach(async user => {
 		const cachedUser = client.users.cache.get(user.id);
 
-		if (cachedUser) {
-			const dateNow = Math.floor(Date.now() / 1000),
-				reminds = user.reminds,
-				mustSent = reminds.filter(r => r.sendAt < dateNow);
+		if (!cachedUser) return;
+
+		const reminds = user.reminds,
+			mustSent = reminds.filter(r => r.sendAt < Math.floor(Date.now() / 1000));
+
+		if (!mustSent.length) return;
 
-			if (mustSent.length > 0) {
 		mustSent.forEach(r => {
 			const embed = client.embed({
 				author: client.translate("general/remindme:EMBED_TITLE"),
@@ -41,26 +43,17 @@ async function checkReminds(client) {
 				],
 			});
 
-					cachedUser.send({
-						embeds: [embed],
+			cachedUser.send({ embeds: [embed] }).then(() => {
+				client.adapter.updateOne(UserModel, { id: user.id }, { $pull: { reminds: { _id: r._id } } });
 			});
 		});
 
-				user.reminds = user.reminds.filter(r => r.sendAt >= dateNow);
-
-				await user.save();
-
-				if (user.reminds.length === 0) client.databaseCache.usersReminds.delete(user.id);
-			}
-		}
+		if (!user.reminds.length) client.cacheReminds.delete(user.id);
 	});
-}
+};
 
-export async function init(client) {
+export const init = async () => {
 	setInterval(async () => {
-		await checkReminds(client);
+		await checkReminds();
 	}, 1000);
-}
-export async function run(client) {
-	await checkReminds(client);
-}
+};

src/structures/client.js

@@ -7,6 +7,9 @@ import logger from "../helpers/logger.js";
 import ConfigService from "../services/config/index.js";
 import InternationalizationService from "../services/languages/index.js";
 import { SUPER_CONTEXT } from "../constants/index.js";
+import GuildModel from "../models/GuildModel.js";
+import UserModel from "../models/UserModel.js";
+import MemberModel from "../models/MemberModel.js";
 
 export class ExtendedClient extends Client {
 	/**
@@ -21,6 +24,7 @@ export class ExtendedClient extends Client {
 		this.configService = new ConfigService();
 		this.adapter = new MongooseAdapter(this.configService.get("mongoDB"));
 		this.i18n = new InternationalizationService(this);
+		this.cacheReminds = new Map();
 		new Player(this);
 
 		SUPER_CONTEXT.enterWith(this);
@@ -37,4 +41,60 @@ export class ExtendedClient extends Client {
 			logger.error(error);
 		}
 	}
+
+	/**
+	 * Retrieves a guild data object from the database.
+	 * @param {string} guildId - The ID of the guild to find or create.
+	 * @returns {Promise<GuildModel>} The guild data object, either retrieved from the database or newly created.
+	 */
+	async getGuildData(guildId) {
+		let guildData = await this.adapter.findOne(GuildModel, { id: guildId });
+
+		if (!guildData) {
+			guildData = new GuildModel({ id: guildId });
+			await guildData.save();
+		}
+
+		return guildData;
+	}
+
+	/**
+	 * Returns a User data from the database.
+	 * @param {string} userID - The ID of the user to find or create.
+	 * @returns {Promise<UserModel>} The user data object, either retrieved from the database or newly created.
+	 */
+	async getUserData(userID) {
+		let userData = await this.adapter.findOne(UserModel, { id: userID });
+
+		if (!userData) {
+			userData = new UserModel({ id: userID });
+			await userData.save();
+		}
+
+		return userData;
+	}
+
+	/**
+	 * Returns a Member data from the database.
+	 * @param {string} memberId - The ID of the member to find or create.
+	 * @param {string} guildId - The ID of the guild the member belongs to.
+	 * @returns {Promise<MemberModel>} The member data object, either retrieved from the database or newly created.
+	 */
+	async getMemberData(memberId, guildId) {
+		let memberData = await this.adapter.findOne(MemberModel, { guildID: guildId, id: memberId });
+
+		if (!memberData) {
+			memberData = new MemberModel({ id: memberId, guildID: guildId });
+			await memberData.save();
+
+			const guildData = await this.getGuildData(guildId);
+
+			if (guildData) {
+				guildData.members.push(memberData._id);
+				await guildData.save();
+			}
+		}
+
+		return memberData;
+	}
 }