From 4a042694d1f832b65ab1a2583ce393af916d9c63 Mon Sep 17 00:00:00 2001
From: Joseph Manley <j@cloudsumu.com>
Date: Sun, 13 Sep 2020 02:06:16 -0400
Subject: [PATCH] Enable and disable

---
 discordnotifier/__init__.py | 38 +++++++++++++++++++++++++++++--------
 1 file changed, 30 insertions(+), 8 deletions(-)

diff --git a/discordnotifier/__init__.py b/discordnotifier/__init__.py
index bf8eca5..fa2d1b8 100644
--- a/discordnotifier/__init__.py
+++ b/discordnotifier/__init__.py
@@ -6,6 +6,8 @@ version = "0.0.0"
 config = configparser.ConfigParser()
 config_path = "discord_config.ini"
 
+currently_enabled = True
+
 if not os.path.exists(config_path):
     with open(config_path, "w") as file_writer:
         file_writer.write("""[discord]
@@ -39,15 +41,35 @@ class Notifier(discord.Client):
             print(f"I see {message.author}")
             await message.channel.send(":eye: You have been seen! :eye:")
 
-        if str(message.author) in [str(self.user), "The Genuine Wonder#2859"]:
-            return
-        
-        # Format and print mesage
-        formatted_message = f"<{message.author}> \"{message.content}\" from #{message.channel} on {message.guild}"
-        print(formatted_message)
+        if str(message.author) == config["discord"]["ignore_user"]:
+            if str(message.channel) == "Direct Message":
+                if message.content.lower() == "start":
+                    currently_enabled = True
+                    message.channel.send("Notifications were enabled")
+                    print("Notifications were enabled")
+                elif message.content.lower() == "stop":
+                    currently_enabled = False
+                    print("Notifications were disabled")
+                    message.channel.send("Notifications were disabled")
+                elif message.content.lower() == "status":
+                    if currently_enabled:
+                        message.channel.send("Notifications are currently enabled!")
+                    else:
+                        message.channel.send("Notifications are currently disabled!")
+                else:
+                    print("Valid commands are 'START', 'STATUS', and 'STOP'")
 
-        # Send notification to SNS
-        sns.publish(Message=formatted_message)
+
+        if str(message.author) in [str(self.user), config["discord"]["ignore_user"]]:
+            return
+
+        if currently_enabled:
+            # Format and print mesage
+            formatted_message = f"<{message.author}> \"{message.content}\" from #{message.channel} on {message.guild}"
+            print(formatted_message)
+
+            # Send notification to SNS
+            sns.publish(Message=formatted_message)
 
 
 def main():