enwnbot2

index.js (5720B)

---

 const ES = require("eventsource");
 
 const irc = require("irc");
 
 const moment = require("moment-timezone");
 
 const {
   channels,
   ircBotName,
   ircServer,
   password,
   RCAPI,
   RQAPI,
   URAPI,
 } = require("./config.js");
 
 const {
   fetchData,
   fullUrl,
   getFullLink,
   getFullTemplate,
   linkRegex,
   streamError,
   streamMessage,
   templateRegex,
   thanksRegex,
 } = require("./utils.js");
 
 const seenLookup = {};
 
 const ircClient = new irc.Client(ircServer, ircBotName, {
   channels,
   userName: "enwnbot",
   realName: "enwnbot",
   sasl: true,
   password,
   floodProtection: true,
   floodProtectionDelay: 1000,
 });
 
 console.log("Connecting to the event stream...");
 
 const eventSource = new ES(RCAPI);
 
 eventSource.onopen = function (event) {
   console.log("--- Opened connection.");
 };
 
 eventSource.onerror = streamError;
 
 eventSource.onmessage = function (event) {
   let msg = streamMessage(event);
   if (msg) channels.forEach((channel) => ircClient.say(channel, msg));
 };
 
 ircClient.addListener("error", function (message) {
   console.log("error: ", message);
 });
 
 ircClient.addListener("pm", function (sender, msg) {
   if (
     msg == "KILL" &&
     ["pizero", "pizero|afk", "acagastya"].indexOf(sender) >= 0
   )
     process.abort();
   ircClient.say(sender, "I am a bot.");
 });
 
 ircClient.addListener("message", groupChat);
 
 ircClient.addListener("names", getNames);
 
 ircClient.addListener("nick", handleNickChange);
 
 ircClient.addListener("join", handleJoin);
 
 ircClient.addListener("part", handlePart);
 
 ircClient.addListener("quit", handleQuit);
 
 function handleJoin(channel, nick) {
   if (seenLookup[channel] === undefined) seenLookup[channel] = {};
   seenLookup[channel][nick] = "";
 }
 
 function getNames(channel, nicks) {
   seenLookup[channel] = { ...nicks };
   Object.keys(seenLookup[channel]).forEach(
     (nick) => (seenLookup[channel][nick] = "")
   );
 }
 
 function handlePart(channel, nick) {
   if (seenLookup[channel] === undefined) seenLookup[channel] = {};
   seenLookup[channel][nick] = +new Date();
 }
 
 function handleQuit(nick, reason, channels) {
   channels.forEach((channel) => {
     if (seenLookup[channel] !== undefined) {
       seenLookup[channel][nick] = +new Date();
     }
   });
 }
 
 function handleNickChange(oldNick, newNick, channels) {
   channels.forEach((channel) => {
     if (seenLookup[channel] !== undefined) {
       seenLookup[channel][oldNick] = +new Date();
       seenLookup[channel][newNick] = "";
     }
   });
 }
 
 function groupChat(sender, channel, msg) {
   if (thanksRegex.test(msg))
     ircClient.say(channel, `You are welcome, ${sender}.`);
   if (msg.includes(`${ircBotName} !RQ`)) announceRQ(sender, channel);
   if (msg.includes(`${ircBotName} !UR`)) announceUR(sender, channel);
   if (msg.startsWith("@seen ")) handleSeen(channel, msg);
 
   const links = msg.match(linkRegex);
   const templates = msg.match(templateRegex);
 
   if (!msg.endsWith("--ignore") && links) {
     const nonEmptyLinks = links.filter((el) => el.length > 4);
     const fullLinks = nonEmptyLinks.map(getFullLink);
     if (fullLinks.length) sayUrls(false, fullLinks, channel);
   }
 
   if (!msg.endsWith("--ignore") && templates) {
     const nonEmptyTemplates = templates.filter((el) => el.length > 4);
     const fullLinks = nonEmptyTemplates.map(getFullTemplate);
     if (fullLinks.length) sayUrls(false, fullLinks, channel);
   }
 }
 
 async function announceRQ(sender, channel) {
   const data = await fetchData(RQAPI);
 
   if (data.error)
     ircClient.say(
       channel,
       `Error occurred, ${sender}.  Try this instead: "[[CAT:REV]]"`
     );
   else {
     const { list } = data;
     if (!list.length)
       ircClient.say(channel, `Review queue is empty, ${sender}.`);
     else {
       ircClient.say(
         channel,
         `${list.length} articles to review, ${sender}.  They are:`
       );
       const titles = list.map(({ title }) => title);
       const times = list.map(({ timestamp }) => moment().to(moment(timestamp)));
       const urls = titles.map(fullUrl);
       sayUrls(true, urls, channel, titles, times);
     }
   }
 }
 
 async function announceUR(sender, channel) {
   const data = await fetchData(URAPI);
 
   if (data.error)
     ircClient.say(
       channel,
       `Error occurred, ${sender}.  Try this instead: "[[CAT:Under Review]]"`
     );
   else {
     const { list } = data;
     if (!list.length)
       ircClient.say(channel, `No articles are under review, ${sender}.`);
     else {
       ircClient.say(
         channel,
         `${list.length} articles are under review, ${sender}.  They are:`
       );
       const titles = list.map(({ title }) => title);
       const times = list.map(({ timestamp }) => moment().to(moment(timestamp)));
       const urls = titles.map(fullUrl);
       sayUrls(true, urls, channel, titles, times);
     }
   }
 }
 
 function handleSeen(channel, msg) {
   const username = msg.split("@seen ")[1];
   if (seenLookup[channel] === undefined) return;
   if (seenLookup[channel][username] === undefined) {
     ircClient.say(channel, `I have never seen ${username} on ${channel}.`);
     return;
   }
   if (seenLookup[channel][username] === "") {
     ircClient.say(channel, `${username} is online.`);
     return;
   }
   const time = new Date(seenLookup[channel][username]).toUTCString();
   ircClient.say(
     channel,
     `${username} was last onilne on ${channel} at ${time}.`
   );
 }
 
 function sayUrls(
   review = false,
   urlList,
   channel,
   titles = [],
   times = [],
   pending = []
 ) {
   urlList.forEach((url, idx) => {
     let msg = url;
     if (review) msg += " submitted for review";
     if (times.length) msg += ` *${times[idx]}*`;
     if (titles.length) msg += ` -- ${titles[idx]}`;
     if (pending[idx]) msg += " *under review*";
     ircClient.say(channel, msg);
   });
 }