From 64d25950e3957849278927d0fa99f0efa39e6204 Mon Sep 17 00:00:00 2001
From: flifloo <flifloo@gmail.com>
Date: Wed, 8 Apr 2020 22:34:03 +0200
Subject: [PATCH] Setup basic xp and show stats

---
 extensions/__init__.py |   1 +
 extensions/xp.py       | 104 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 105 insertions(+)
 create mode 100644 extensions/xp.py

diff --git a/extensions/__init__.py b/extensions/__init__.py
index 9953568..bad2fb2 100644
--- a/extensions/__init__.py
+++ b/extensions/__init__.py
@@ -4,3 +4,4 @@ bot.load_extension("extensions.help")
 bot.load_extension("extensions.speak")
 bot.load_extension("extensions.extension")
 bot.load_extension("extensions.purge")
+bot.load_extension("extensions.xp")
diff --git a/extensions/xp.py b/extensions/xp.py
new file mode 100644
index 0000000..c6fc717
--- /dev/null
+++ b/extensions/xp.py
@@ -0,0 +1,104 @@
+from discord.ext import commands
+from discord import Embed, Message
+from threading import RLock
+from shelve import open
+from datetime import datetime, timedelta
+from re import compile
+
+from bot_bde.logger import logger
+
+
+extension_name = "xp"
+logger = logger.getChild(extension_name)
+
+url_re = compile(r"http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\), ]|(?:%[0-9a-fA-F][0-9a-fA-F]))+")
+XP = {"message": 2,
+      "image": 15,
+      "file": 20,
+      "link": 10}
+
+class Xp(commands.Cog):
+    def __init__(self, bot: commands.Bot):
+        self.bot = bot
+        self.lock = RLock()
+
+    @commands.group("xp", pass_context=True)
+    @commands.guild_only()
+    async def xp(self, ctx: commands.Context):
+        if ctx.invoked_subcommand is None:
+            player = None
+            with self.lock:
+                with open("xp", writeback=True) as data:
+                    if str(ctx.author.id) in data:
+                        player = data[str(ctx.author.id)]
+            embed = Embed(title="Current stats")
+            embed.add_field(name="level", value=player["level"])
+            embed.add_field(name="xp", value=player["xp"])
+            embed.add_field(name="message", value=player["message"])
+            embed.add_field(name="image", value=player["image"])
+            embed.add_field(name="file", value=player["file"])
+            embed.add_field(name="link", value=player["link"])
+            await ctx.send(embed=embed)
+
+    @xp.group("help", pass_context=True)
+    @commands.guild_only()
+    async def xp_help(self, ctx: commands.Context):
+        embed = Embed(title="Xp help")
+        embed.add_field(name="xp", value="Show your current cp", inline=False)
+        await ctx.send(embed=embed)
+
+    @commands.Cog.listener()
+    async def on_message(self, message: Message):
+        if not message.author.bot:
+            with self.lock:
+                with open("xp", writeback=True) as data:
+                    can_xp = False
+                    if not str(message.author.id) in data:
+                        can_xp = True
+                        data[str(message.author.id)] = {"xp": 0,
+                                                        "level": 0,
+                                                        "message": 0,
+                                                        "image": 0,
+                                                        "file": 0,
+                                                        "link": 0,
+                                                        "last_message": message.created_at}
+                    if message.created_at - data[str(message.author.id)]["last_message"] >= timedelta(minutes=5):
+                        can_xp = True
+                    data[str(message.author.id)]["message"] += 1
+                    if can_xp:
+                        data[str(message.author.id)]["xp"] += XP["message"]*len(message.content)
+                    if url_re.match(message.content):
+                        data[str(message.author.id)]["link"] += 1
+                        if can_xp:
+                            data[str(message.author.id)]["xp"] += XP["link"]
+                    if message.attachments:
+                        for a in message.attachments:
+                            if a.width is not None:
+                                data[str(message.author.id)]["image"] += 1
+                                if can_xp:
+                                    data[str(message.author.id)]["xp"] += XP["image"]
+                            else:
+                                data[str(message.author.id)]["file"] += 1
+                                if can_xp:
+                                    data[str(message.author.id)]["xp"] += XP["file"]
+                    data[str(message.author.id)]["last_message"] = message.created_at
+
+
+def setup(bot):
+    logger.info(f"Loading...")
+    try:
+        bot.add_cog(Xp(bot))
+    except Exception as e:
+        logger.error(f"Error loading: {e}")
+    else:
+        logger.info(f"Load successful")
+
+
+def teardown(bot):
+    logger.info(f"Unloading...")
+    try:
+        bot.remove_cog("Speak")
+    except Exception as e:
+        logger.error(f"Error unloading: {e}")
+    else:
+        logger.info(f"Unload successful")