From 61f449079ed1283cdbf901b6ca451485f0c38f4a Mon Sep 17 00:00:00 2001
From: luke <lukelrobles@gmail.com>
Date: Sun, 2 Jun 2019 16:42:35 -0700
Subject: [PATCH] Dice rolling method

---
 app/dice.py         | 12 ++++++++++++
 app/dragon-bot.py   |  7 +++++++
 app/help_methods.py |  6 +++++-
 3 files changed, 24 insertions(+), 1 deletion(-)
 create mode 100644 app/dice.py

diff --git a/app/dice.py b/app/dice.py
new file mode 100644
index 00000000..7508073a
--- /dev/null
+++ b/app/dice.py
@@ -0,0 +1,12 @@
+from random import randint
+
+def roll_logic(sides):
+  return randint(1, sides)
+
+def roll(n, sides):
+  results = tuple(roll_logic(int(sides)) for _ in range(int(n)))
+  message = "\n\n%s\nRolling %s %s sided die\n\n%s" % (':game_die:' * int(n), n, sides, results)
+  return message
+
+def parse_message(message):
+  return roll(message.split()[1], message.split()[2])
\ No newline at end of file
diff --git a/app/dragon-bot.py b/app/dragon-bot.py
index 0ca447f4..2c327576 100644
--- a/app/dragon-bot.py
+++ b/app/dragon-bot.py
@@ -11,6 +11,7 @@ import requests
 
 import animals
 import core_utils
+import dice
 import decide
 import define_word
 import discord
@@ -269,6 +270,12 @@ async def on_message(message):
             embed=quake.parse_message(message)
         )
 
+    if message.content.startswith('!roll'):
+        await client.send_message(
+            message.channel,
+            dice.parse_message(message.content)
+            )
+
     if message.content.startswith('!smug'):
         await client.send_message(
             message.channel,
diff --git a/app/help_methods.py b/app/help_methods.py
index 85b04754..5e7cc570 100644
--- a/app/help_methods.py
+++ b/app/help_methods.py
@@ -116,6 +116,10 @@ def get_help_message(method):
             'Returns player stats for a quake champions account\n'
             'Usage: !quake <player name>'
             ],
+        'roll': [
+            'Rolls N number of Y sided die'
+            'Usage: !roll 3 20'
+            ],
         'smug': [
             'Returns the URL for smug anime girl'
             ],
@@ -137,7 +141,7 @@ def get_help_message(method):
 
 def get_help_embed(client):
     categories = {
-        'fun': ['clap', 'birb', 'dog', 'excuse', 'greentext', 'lewd', 'message', 'meme', 'homepage', 'pout', 'smug', 'quake'],
+        'fun': ['clap', 'birb', 'dog', 'excuse', 'greentext', 'lewd', 'message', 'meme', 'homepage', 'pout', 'roll', 'smug', 'quake'],
         'util': ['8ball', 'decide', 'wallpaper', 'weather', 'stock', 'tts', 'issue'],
         'users': ['help', 'invite', 'purge', 'roles', 'source', 'minecraft'],
         'admin': ['emoji', 'cleanup']