main.js 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175
  1. const Discord = require("discord.js");
  2. const TOKEN = require("./token.js");
  3. const db = require("./db.js");
  4. const RSSParser = require("rss-parser");
  5. const interval = require("interval-promise");
  6. const TurndownService = require("turndown");
  7. const turndown = new TurndownService();
  8. turndown.addRule("image", {
  9. filter: "img",
  10. replacement: () => ""
  11. });
  12. turndown.addRule("link", {
  13. filter: node => node.nodeName === "A" &&node.getAttribute("href"),
  14. replacement: (content, node) => node.getAttribute("href")
  15. });
  16. const client = new Discord.Client();
  17. const parser = new RSSParser();
  18. const RSS_UPDATE_INTERVAL_MIN = 5;
  19. async function checkFeeds() {
  20. let feeds = db.get("rssFeeds").value();
  21. let outlets = db.get("feedOutputs").value();
  22. for(let feedEntry of feeds) {
  23. let feed = await parser.parseURL(feedEntry.url);
  24. if(feed.items.length == 0)
  25. continue;
  26. let printableItems = feed.items.filter(i => i.isoDate > feedEntry.lastUpdate).sort((a, b) => a.isoDate.localeCompare(b.isoDate));
  27. if(printableItems.length > 0) {
  28. printableItems.forEach(item => {
  29. outlets.forEach(ch => {
  30. client.channels.get(ch).send(`**${item.title}**\nPosted by ${item.creator}\n${item.link}\n\n${turndown.turndown(item[feedEntry.contentElement]).replace(/( {2}\n|\n\n){2,}/gm, "\n").replace(item.link, "")}`);
  31. });
  32. });
  33. let lastUpdateDate = printableItems[printableItems.length - 1].isoDate;
  34. console.log(`Setting last update marker on ${feedEntry.url} to ${lastUpdateDate}`);
  35. db.get("rssFeeds").find({ url: feedEntry.url}).assign({lastUpdate: lastUpdateDate}).write();
  36. }
  37. }
  38. }
  39. function isAuthorised(member) {
  40. if (db.get("editors.users").includes(member.id).value())
  41. return true;
  42. if (db.get("editors.roles").intersectionWith(member.roles.keyArray()).isEmpty().value())
  43. return false;
  44. return true;
  45. }
  46. const commands = {
  47. "make guide": msg => {
  48. if (!isAuthorised(msg.member)) return;
  49. let content = msg.content.substring(msg.content.indexOf("make guide") + "make guide ".length);
  50. let guideName = content.substring(0, content.indexOf("\n")).trim();
  51. let guideContent = content.substring(content.indexOf("\n")).trim();
  52. let guide = db.get("guides").find({ name: guideName });
  53. if (!guide.isUndefined().value()) {
  54. guide.assign({ content: guideContent }).write();
  55. } else {
  56. db.get("guides")
  57. .push({
  58. name: guideName,
  59. content: guideContent
  60. })
  61. .write();
  62. }
  63. msg.channel.send(
  64. `${msg.author.toString()} Added/updated "${guideName}"!`
  65. );
  66. },
  67. "delete guide": (msg, s) => {
  68. if (!isAuthorised(msg.member)) return;
  69. let guideName = s.substring("delete guide ".length).trim();
  70. let val = db.get("guides").find({ name: guideName });
  71. if (val.isUndefined().value()) {
  72. msg.channel.send(`${msg.author.toString()} No guide "${guideName}"!`);
  73. return;
  74. }
  75. db.get("guides")
  76. .remove({ name: guideName })
  77. .write();
  78. msg.channel.send(
  79. `${msg.author.toString()} Removed guide "${guideName}!"`
  80. );
  81. },
  82. help: msg => {
  83. let guides = db
  84. .get("guides")
  85. .map(g => g.name)
  86. .reduce((p, c) => `${p}\n${c}`, "")
  87. .value();
  88. msg.channel.send(
  89. `Hello! I am NoctBot! I have answers for the most common C(O)M-related questions.\nJust ping me with one of the following keywords:\n\`\`\`${guides}\`\`\``
  90. );
  91. }
  92. };
  93. client.on("ready", () => {
  94. console.log("Ready!");
  95. client.user.setActivity("@NoctBot help", { type: "PLAYING" });
  96. interval(checkFeeds, RSS_UPDATE_INTERVAL_MIN * 60 * 1000);
  97. });
  98. client.on("message", m => {
  99. if (m.author.id == client.user.id || m.mentions.users.size == 0) return;
  100. if (!db.get("reactableMentionedUsers").intersectionWith(m.mentions.users.map(u => u.id)).isEmpty().value()) {
  101. const emoteId = db
  102. .get("emotes")
  103. .get("angery")
  104. .randomElement()
  105. .value();
  106. m.react(client.emojis.find(e => e.id == emoteId));
  107. return;
  108. }
  109. if (m.mentions.users.first().id == client.user.id) {
  110. let content = m.cleanContent.trim();
  111. if(content.startsWith(`@${client.user.username}`)) {
  112. content = content
  113. .substring(`@${client.user.username} `.length);
  114. let lowerCaseContent = content.toLowerCase().trim();
  115. for (let c in commands) {
  116. if (lowerCaseContent.startsWith(c)) {
  117. commands[c](m, content);
  118. return;
  119. }
  120. }
  121. if (lowerCaseContent.length > 0) {
  122. let parts = lowerCaseContent.trim().split(" ");
  123. let guide = db
  124. .get("guides")
  125. .map(g => Object.assign({parts: g.name.toLowerCase().split(" ")}, g))
  126. .sortBy(g => g.parts.length)
  127. .maxBy(k => db._.intersection(parts, k.parts).length)
  128. .value();
  129. let hits =
  130. guide !== undefined &&
  131. db._.intersection(guide.name.toLowerCase().split(" "), parts).length > 0;
  132. if (hits) {
  133. m.channel.send(guide.content);
  134. return;
  135. }
  136. }
  137. }
  138. let emoteType = "angery";
  139. if(db.get("specialUsers").includes(m.author.id).value())
  140. emoteType = "hug";
  141. else if(db.get("bigUsers").includes(m.author.id).value())
  142. emoteType = "big";
  143. else if(db.get("dedUsers").includes(m.author.id).value())
  144. emoteType = "ded";
  145. const id = db
  146. .get("emotes")
  147. .get(emoteType)
  148. .randomElement()
  149. .value();
  150. m.channel.send(client.emojis.find(e => e.id == id).toString());
  151. } else if (m.content.includes("Noct")) {
  152. m.channel.send(
  153. client.emojis.find(e => e.name == "mukuNeighWaaaaaa").toString()
  154. );
  155. }
  156. });
  157. client.login(TOKEN);