from discord.ext import commands
from .utils import config
from .utils import checks
import re


class Tags:
    """This class contains all the commands for custom tags"""

    def __init__(self, bot):
        self.bot = bot

    @commands.command(pass_context=True, no_pm=True)
    @checks.customPermsOrRole(send_messages=True)
    async def tags(self, ctx):
        """Prints all the custom tags that this server currently has"""
        tags = config.getContent('tags') or {}
        fmt = "\n".join("{}".format(tag['tag']) for tag in tags if tag['server_id'] == ctx.message.server.id)
        await self.bot.say('```\n{}```'.format(fmt))

    @commands.group(pass_context=True, invoke_without_command=True, no_pm=True)
    @checks.customPermsOrRole(send_messages=True)
    async def tag(self, ctx, *, tag: str):
        """This can be used to call custom tags
         The format to call a custom tag is !tag <tag>"""
        tags = config.getContent('tags') or {}
        result = [t for t in tags if t['tag'] == tag and t['server_id'] == ctx.message.server.id]
        if len(result) == 0:
            await self.bot.say('That tag does not exist!')
            return
        await self.bot.say("{}".format(result[0]['result']))

    @tag.command(name='add', aliases=['create', 'start'], pass_context=True, no_pm=True)
    @checks.customPermsOrRole(kick_members=True)
    async def add_tag(self, ctx, *, result: str):
        """Use this to add a new tag that can be used in this server
        Format to add a tag is !tag add <tag> - <result>"""
        try:
            tag = re.search("(.*) - (.*)", result).group(1).strip()
            tag_result = re.search("(.*) - (.*)", result).group(2).strip()
        except AttributeError:
            await self.bot.say("Please provide the format for the tag in: !tag add <tag> - <result>")
            return
        if len(tag) == 0 or len(tag_result) == 0:
            await self.bot.say("Please provide the format for the tag in: !tag add <tag> - <result>")
            return
            
        tags = config.getContent('tags') or {}
        for t in tags:
            if t['tag'] == tag and t['server_id'] == ctx.message.server.id:
                t['result'] = tag_result
                await self.bot.say(
                    "I have just updated the tag `{0}`! You can call this tag by entering !tag {0}".format(tag))
                return
        tags.append({'server_id': ctx.message.server.id, 'tag': tag, 'result': tag_result})
        await self.bot.say(
            "I have just added the tag `{0}`! You can call this tag by entering !tag {0}".format(tag))

    @tag.command(name='delete', aliases=['remove', 'stop'], pass_context=True, no_pm=True)
    @checks.customPermsOrRole(kick_members=True)
    async def del_tag(self, ctx, *, tag: str):
        """Use this to remove a tag that from use for this server
        Format to delete a tag is !tag delete <tag>"""
        tags = config.getContent('tags') or {}
        result = [t for t in tags if t['tag'] == tag and t['server_id'] == ctx.message.server.id]
        if len(result) == 0:
            await self.bot.say(
                "The tag {} does not exist! You can't remove something if it doesn't exist...".format(tag))
            return
        for t in tags:
            if t['tag'] == tag and t['server_id'] == ctx.message.server.id:
                tags.remove(t)
                await self.bot.say('I have just removed the tag `{}`'.format(tag))


def setup(bot):
    bot.add_cog(Tags(bot))