mirror of
https://github.com/JonnyBro/JaBa.git
synced 2024-12-27 22:03:03 +05:00
feat: rewritten two helpers, birthdays and checkreminds
This commit is contained in:
Slincnik
parent
823aea886f
commit
b45f3dcc35
2 changed files with 94 additions and 99 deletions
|
|
@ -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",
|
||||
{
|
||||
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,
|
||||
),
|
||||
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}** ${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("🎉"));
|
||||
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);
|
||||
}),
|
||||
);
|
||||
} 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");
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,66 +1,59 @@
|
|||
/**
|
||||
*
|
||||
* @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;
|
||||
|
||||
if (mustSent.length > 0) {
|
||||
mustSent.forEach(r => {
|
||||
const embed = client.embed({
|
||||
author: client.translate("general/remindme:EMBED_TITLE"),
|
||||
fields: [
|
||||
{
|
||||
name: client.translate("general/remindme:EMBED_CREATED"),
|
||||
value: `<t:${r.createdAt}:f>`,
|
||||
inline: true,
|
||||
},
|
||||
{
|
||||
name: client.translate("general/remindme:EMBED_TIME"),
|
||||
value: `<t:${r.sendAt}:f>`,
|
||||
inline: true,
|
||||
},
|
||||
{
|
||||
name: client.translate("common:MESSAGE"),
|
||||
value: r.message,
|
||||
},
|
||||
],
|
||||
});
|
||||
const reminds = user.reminds,
|
||||
mustSent = reminds.filter(r => r.sendAt < Math.floor(Date.now() / 1000));
|
||||
|
||||
cachedUser.send({
|
||||
embeds: [embed],
|
||||
});
|
||||
});
|
||||
if (!mustSent.length) return;
|
||||
|
||||
user.reminds = user.reminds.filter(r => r.sendAt >= dateNow);
|
||||
mustSent.forEach(r => {
|
||||
const embed = client.embed({
|
||||
author: client.translate("general/remindme:EMBED_TITLE"),
|
||||
fields: [
|
||||
{
|
||||
name: client.translate("general/remindme:EMBED_CREATED"),
|
||||
value: `<t:${r.createdAt}:f>`,
|
||||
inline: true,
|
||||
},
|
||||
{
|
||||
name: client.translate("general/remindme:EMBED_TIME"),
|
||||
value: `<t:${r.sendAt}:f>`,
|
||||
inline: true,
|
||||
},
|
||||
{
|
||||
name: client.translate("common:MESSAGE"),
|
||||
value: r.message,
|
||||
},
|
||||
],
|
||||
});
|
||||
|
||||
await user.save();
|
||||
cachedUser.send({ embeds: [embed] }).then(() => {
|
||||
client.adapter.updateOne(UserModel, { id: user.id }, { $pull: { reminds: { _id: r._id } } });
|
||||
});
|
||||
});
|
||||
|
||||
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);
|
||||
}
|
||||
};
|
||||
Loading…
Reference in a new issue