From c3cc3820d41474aed6d028bf28266dd568c27a8c Mon Sep 17 00:00:00 2001
From: Tobias <thm.frey@gmail.com>
Date: Mon, 22 Jul 2019 02:55:55 +0200
Subject: [PATCH] server.js: Add the functionality to auto-moderate media in a
 channel (#59)

This makes it possible to auto-delete text-messages, sent in a channel meant for images and other media.
---
 env.json      |  3 +++
 src/server.js | 17 +++++++++++++++++
 2 files changed, 20 insertions(+)

diff --git a/env.json b/env.json
index b7afb11..bd4763a 100644
--- a/env.json
+++ b/env.json
@@ -17,6 +17,9 @@
     "DISCORD_LOG_CHANNEL": {
       "description": "The unique ID for the channel the bot will output logs to."
     },
+    "DISCORD_MEDIA_CHANNEL": {
+      "description": "The unique ID for the channel the bot will moderate images in."
+    },
     "DISCORD_RULES_CHANNEL": {
       "description": "The unique ID for the channel that the bot will manage rules from."
     },
diff --git a/src/server.js b/src/server.js
index 49ae834..6939228 100644
--- a/src/server.js
+++ b/src/server.js
@@ -16,6 +16,8 @@ var cachedModules = [];
 var cachedTriggers = [];
 var client = new discord.Client();
 
+let mediaUsers = new Map();
+
 logger.info('Application startup. Configuring environment.');
 
 process.on('unhandledRejection', (error, promise) => {
@@ -94,6 +96,21 @@ client.on('message', message => {
 
   logger.verbose(`${message.author.username} ${message.author} [Channel: ${message.channel.name} ${message.channel}]: ${message.content}`);
 
+  if (message.channel.id === process.env.DISCORD_MEDIA_CHANNEL && !message.author.bot) {
+    const AllowedMediaRoles = ['Administrators', 'Moderators', 'Team', 'VIP'];
+    if (!findArray(message.member.roles.map(function (x) { return x.name; }), AllowedMediaRoles)) {
+      const urlRegex = new RegExp(/https?:\/\/(www\.)?[-a-zA-Z0-9@:%._+~#=]{2,256}\.[a-z]{2,4}\b([-a-zA-Z0-9@:%_+.~#?&//=]*)/gi);
+      if (message.attachments.size > 0 || message.content.match(urlRegex)) {
+        mediaUsers.set(message.author.id, true);
+      } else if (mediaUsers.get(message.author.id)) {
+        mediaUsers.set(message.author.id, false);
+      } else {
+        message.delete();
+        mediaUsers.set(message.author.id, false);
+      }
+    }
+  }
+
   // Check if the channel is #rules, if so we want to follow a different logic flow.
   if (message.channel.id === process.env.DISCORD_RULES_CHANNEL) {
     if (message.content.toLowerCase().includes(process.env.DISCORD_RULES_TRIGGER)) {