main.js 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326
  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 fs = require("fs");
  8. const Jimp = require("jimp");
  9. const Clarifai = require("clarifai");
  10. const ClarifaiTOKEN = require("./clarifai_keys.js");
  11. const EMOTE_GUILD = "505333548694241281";
  12. const VALID_EXTENSIONS = new Set([
  13. "png",
  14. "jpg",
  15. "jpeg",
  16. "tiff",
  17. "tif",
  18. "bmp",
  19. "webp"
  20. ]);
  21. const clarifaiApp = new Clarifai.App({
  22. apiKey: ClarifaiTOKEN
  23. });
  24. const turndown = new TurndownService();
  25. turndown.addRule("image", {
  26. filter: "img",
  27. replacement: () => ""
  28. });
  29. turndown.addRule("link", {
  30. filter: node => node.nodeName === "A" &&node.getAttribute("href"),
  31. replacement: (content, node) => node.getAttribute("href")
  32. });
  33. const statsFile = fs.openSync(__dirname + "/imagestats.csv", "a");
  34. const client = new Discord.Client();
  35. const parser = new RSSParser();
  36. const RSS_UPDATE_INTERVAL_MIN = 5;
  37. const REACT_PROBABILITY = 0.6;
  38. function isValidImage(fileName) {
  39. let extPosition = fileName.lastIndexOf(".");
  40. if(extPosition < 0)
  41. return false;
  42. let ext = fileName.substring(extPosition + 1).toLowerCase();
  43. return VALID_EXTENSIONS.has(ext);
  44. }
  45. function getThreadId(url) {
  46. let result = url.substring(url.lastIndexOf(".") + 1);
  47. if(result.endsWith("/"))
  48. result = result.substring(0, result.length - 1);
  49. return result;
  50. }
  51. async function checkFeeds() {
  52. console.log(`Checking feeds on ${new Date().toISOString()}`);
  53. let feeds = db.get("rssFeeds").value();
  54. let outlets = db.get("feedOutputs").value();
  55. let oldNews = db.get("postedNewsGuids");
  56. for(let feedEntry of feeds) {
  57. let feed = await parser.parseURL(feedEntry.url);
  58. if(feed.items.length == 0)
  59. continue;
  60. let printableItems = feed.items.filter(i => i.isoDate > feedEntry.lastUpdate).sort((a, b) => a.isoDate.localeCompare(b.isoDate));
  61. if(printableItems.length > 0) {
  62. printableItems.forEach(item => {
  63. let itemID = getThreadId(item.guid);
  64. if(oldNews.has(itemID).value())
  65. return;
  66. outlets.forEach(ch => {
  67. 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, "")}`);
  68. });
  69. oldNews.set(itemID, true).write();
  70. });
  71. let lastUpdateDate = printableItems[printableItems.length - 1].isoDate;
  72. console.log(`Setting last update marker on ${feedEntry.url} to ${lastUpdateDate}`);
  73. db.get("rssFeeds").find({ url: feedEntry.url}).assign({lastUpdate: lastUpdateDate}).write();
  74. }
  75. }
  76. }
  77. function isAuthorised(member) {
  78. if (db.get("editors.users").includes(member.id).value())
  79. return true;
  80. if (db.get("editors.roles").intersectionWith(member.roles.keyArray()).isEmpty().value())
  81. return false;
  82. return true;
  83. }
  84. async function processFaceSwap(message, attachmentUrl) {
  85. let result = await clarifaiApp.models.predict(Clarifai.FACE_DETECT_MODEL, attachmentUrl);
  86. if(result.outputs[0].data.regions === undefined || result.outputs[0].data.regions.length == 0)
  87. return;
  88. let image = await Jimp.read(attachmentUrl);
  89. let w = image.getWidth();
  90. let h = image.getHeight();
  91. let emojiKeys = [...client.guilds.get(EMOTE_GUILD).emojis.filter(e => !e.animated).keys()];
  92. for(let region of result.outputs[0].data.regions) {
  93. let bb = region.region_info.bounding_box;
  94. let bw = (bb.right_col - bb.left_col) * w;
  95. let bh = (bb.bottom_row - bb.top_row) * h;
  96. let dx = (bb.right_col + bb.left_col) * w / 2;
  97. let dy = (bb.bottom_row + bb.top_row) * h / 2;
  98. let emojiKey = emojiKeys[Math.floor(Math.random() * emojiKeys.length)];
  99. let emoji = client.emojis.get(emojiKey);
  100. let emojiImage = await Jimp.read(emoji.url);
  101. let ew = emojiImage.getWidth();
  102. let eh = emojiImage.getHeight();
  103. let scaleFactor = Math.max(bw, bh) / Math.max(ew, eh);
  104. ew *= scaleFactor;
  105. eh *= scaleFactor;
  106. emojiImage = emojiImage.scale(scaleFactor);
  107. image = image.composite(emojiImage, dx - ew / 2, dy - eh / 2);
  108. }
  109. image.quality(90);
  110. let buffer = await image.getBufferAsync(Jimp.MIME_JPEG);
  111. message.channel.send(`I noticed a face in the image. I think this looks better ${client.emojis.get("505076258753740810").toString()}`, {
  112. files: [ buffer ]
  113. });
  114. }
  115. function faceMorph(message) {
  116. let probValue = db.get("faceEditChannels").get(message.channel.id);
  117. if(probValue.isUndefined().value() || probValue.isNull().value())
  118. return;
  119. if(Math.random() > probValue.value())
  120. return;
  121. let imageAttachment = message.attachments.find(v => isValidImage(v.filename));
  122. processFaceSwap(message, imageAttachment.url).catch(err => console.log(`Failed to run faceapp because ${err}`));
  123. }
  124. const commands = {
  125. "make guide": msg => {
  126. if (!isAuthorised(msg.member)) return;
  127. let content = msg.content.substring(msg.content.indexOf("make guide") + "make guide ".length);
  128. let guideName = content.substring(0, content.indexOf("\n")).trim();
  129. let guideContent = content.substring(content.indexOf("\n")).trim();
  130. let guide = db.get("guides").find({ name: guideName });
  131. if (!guide.isUndefined().value()) {
  132. guide.assign({ content: guideContent }).write();
  133. } else {
  134. db.get("guides")
  135. .push({
  136. name: guideName,
  137. content: guideContent
  138. })
  139. .write();
  140. }
  141. msg.channel.send(
  142. `${msg.author.toString()} Added/updated "${guideName}"!`
  143. );
  144. },
  145. "delete guide": (msg, s) => {
  146. if (!isAuthorised(msg.member)) return;
  147. let guideName = s.substring("delete guide ".length).trim();
  148. let val = db.get("guides").find({ name: guideName });
  149. if (val.isUndefined().value()) {
  150. msg.channel.send(`${msg.author.toString()} No guide "${guideName}"!`);
  151. return;
  152. }
  153. db.get("guides")
  154. .remove({ name: guideName })
  155. .write();
  156. msg.channel.send(
  157. `${msg.author.toString()} Removed guide "${guideName}"!`
  158. );
  159. },
  160. "react to": (msg, s) => {
  161. if (!isAuthorised(msg.member)) return;
  162. const pattern = /^react to\s+"([^"]+)"\s+with\s+\<:[^:]+:([^\>]+)\>$/i;
  163. let contents = pattern.exec(s);
  164. if(contents != null) {
  165. let reactable = contents[1].trim().toLowerCase();
  166. let reactionEmoji = contents[2];
  167. if(!client.emojis.has(reactionEmoji)){
  168. msg.channel.send(`${msg.author.toString()} I cannot react with this emoji :(`);
  169. return;
  170. }
  171. db.get("messageReactions").set(reactable, reactionEmoji).write();
  172. msg.channel.send(`${msg.author.toString()} Added reaction!`);
  173. }
  174. },
  175. "remove reaction to": (msg, s) => {
  176. if (!isAuthorised(msg.member)) return;
  177. let content = s.substring("remove reaction to ".length).trim().toLowerCase();
  178. if(!db.get("messageReactions").has(content).value()) {
  179. msg.channel.send(`${msg.author.toString()} No such reaction available!`);
  180. return;
  181. }
  182. db.get("messageReactions").unset(content).write();
  183. msg.channel.send(`${msg.author.toString()} Removed reaction!`);
  184. },
  185. "reactions": msg => {
  186. let reactions = db.get("messageReactions").keys().value().reduce((p, c) => `${p}\n${c}`, "");
  187. msg.channel.send(`I'll react to the following messages:\n\`\`\`${reactions}\`\`\``);
  188. },
  189. "help": msg => {
  190. let guides = db
  191. .get("guides")
  192. .map(g => g.name)
  193. .reduce((p, c) => `${p}\n${c}`, "")
  194. .value();
  195. msg.channel.send(
  196. `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}\`\`\``
  197. );
  198. }
  199. };
  200. client.on("ready", () => {
  201. console.log("Ready!");
  202. client.user.setActivity("@NoctBot help", { type: "PLAYING" });
  203. interval(checkFeeds, RSS_UPDATE_INTERVAL_MIN * 60 * 1000);
  204. });
  205. client.on("message", m => {
  206. if (m.author.id == client.user.id) return;
  207. let imagesCount = m.attachments.filter(v => isValidImage(v.filename)).size;
  208. if(imagesCount > 0) {
  209. let now = new Date();
  210. fs.writeSync(statsFile, `${now.getUTCFullYear()}-${now.getUTCMonth()+1}-${now.getUTCDate()} ${now.getUTCHours()}:${now.getUTCMinutes()};${imagesCount};${m.channel.name}\n`);
  211. faceMorph(m);
  212. }
  213. let content = m.cleanContent.trim();
  214. let lowerContent = content.toLowerCase();
  215. if(db.get("messageReactions").has(lowerContent).value()) {
  216. m.react(client.emojis.get(db.get("messageReactions").get(lowerContent).value()));
  217. return;
  218. }
  219. if(m.mentions.users.size == 0)
  220. return;
  221. if (!db.get("reactableMentionedUsers").intersectionWith(m.mentions.users.map(u => u.id)).isEmpty().value()) {
  222. const emoteId = db
  223. .get("emotes")
  224. .get("angery")
  225. .randomElement()
  226. .value();
  227. m.react(client.emojis.find(e => e.id == emoteId));
  228. return;
  229. }
  230. if (m.mentions.users.first().id == client.user.id) {
  231. if(content.startsWith(`@${client.user.username}`)) {
  232. content = content
  233. .substring(`@${client.user.username} `.length);
  234. let lowerCaseContent = content.toLowerCase().trim();
  235. for (let c in commands) {
  236. if (lowerCaseContent.startsWith(c)) {
  237. commands[c](m, content);
  238. return;
  239. }
  240. }
  241. if (lowerCaseContent.length > 0) {
  242. let parts = lowerCaseContent.trim().split(" ");
  243. let guide = db
  244. .get("guides")
  245. .map(g => Object.assign({parts: g.name.toLowerCase().split(" ")}, g))
  246. .sortBy(g => g.parts.length)
  247. .maxBy(k => db._.intersection(parts, k.parts).length)
  248. .value();
  249. let hits =
  250. guide !== undefined &&
  251. db._.intersection(guide.name.toLowerCase().split(" "), parts).length > 0;
  252. if (hits) {
  253. m.channel.send(guide.content);
  254. return;
  255. }
  256. }
  257. }
  258. let emoteType = "angery";
  259. if(db.get("specialUsers").includes(m.author.id).value())
  260. emoteType = "hug";
  261. else if(db.get("bigUsers").includes(m.author.id).value())
  262. emoteType = "big";
  263. else if(db.get("dedUsers").includes(m.author.id).value())
  264. emoteType = "ded";
  265. const id = db
  266. .get("emotes")
  267. .get(emoteType)
  268. .randomElement()
  269. .value();
  270. m.channel.send(client.emojis.find(e => e.id == id).toString());
  271. } else if (m.content.includes("Noct")) {
  272. m.channel.send(
  273. client.emojis.find(e => e.name == "mukuNeighWaaaaaa").toString()
  274. );
  275. }
  276. });
  277. client.on("messageReactionAdd", (r, u) => {
  278. if(Math.random() <= REACT_PROBABILITY && !r.me)
  279. r.message.react(r.emoji);
  280. });
  281. client.login(TOKEN);