62 lines
1.8 KiB
JavaScript
62 lines
1.8 KiB
JavaScript
const Core = require("../discord-bot-core");
|
|
const GetUrls = require("get-urls");
|
|
const GuildData = require("./models/guild-data.js");
|
|
// @ts-ignore
|
|
const Config = require("./config.json");
|
|
|
|
const guildsIterator = (function* () {
|
|
while (true) {
|
|
if (client.guilds.size === 0)
|
|
yield null;
|
|
else
|
|
for (let i = 0; i < client.guilds.size; i++)
|
|
yield [...client.guilds.values()][i];
|
|
}
|
|
})();
|
|
|
|
const token = require("../" + process.argv[2]).token,
|
|
dataFile = process.argv[3];
|
|
|
|
const client = new Core.Client(token, dataFile, __dirname + "/commands", GuildData);
|
|
|
|
client.on("beforeLogin", () =>
|
|
setInterval(doGuildIteration, Config.feedCheckInterval));
|
|
|
|
client.on("ready", () => {
|
|
parseLinksInGuilds().then(doGuildIteration);
|
|
require("./legacy-upgrader.js")(); //upgrade legacy json into new database format
|
|
});
|
|
|
|
client.on("message", message => {
|
|
if (message.channel.type !== "text" || !message.member)
|
|
return;
|
|
|
|
client.guildDataModel.findOne({ guildID: message.guild.id })
|
|
.then(guildData => {
|
|
if (guildData) {
|
|
guildData.feeds.forEach(feedData =>
|
|
message.channel.id === feedData.channelID && feedData.cache(...GetUrls(message.content)));
|
|
guildData.save();
|
|
}
|
|
});
|
|
});
|
|
|
|
client.bootstrap();
|
|
|
|
//INTERNAL FUNCTIONS//
|
|
function doGuildIteration() {
|
|
const guild = guildsIterator.next().value;
|
|
guild && client.guildDataModel.findOne({ guildID: guild.id })
|
|
.then(guildData => guildData && guildData.checkFeeds(guild).then(() => guildData.save()));
|
|
}
|
|
|
|
function parseLinksInGuilds() {
|
|
const promises = [];
|
|
client.guildDataModel.find().then(guildDatas =>
|
|
guildDatas.forEach(guildData => {
|
|
if (client.guilds.get(guildData.guildID))
|
|
promises.push(guildData.cachePastPostedLinks(client.guilds.get(guildData.guildID)));
|
|
}));
|
|
|
|
return Promise.all(promises);
|
|
} |