import asyncio
import code
import io
import pyrasite as pyr
import re
import sys
import traceback as tb

import discord as d
from discord.ext import commands

from misc import checks
from misc import exceptions as exc

nl = re.compile('\n')

class Tools:

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

    def format(self, i='', o=''):
        if len(o) > 1: return '>>> {}\n{}'.format(i, o)
        else: return '>>> {}'.format(i)
    async def generate(self, d, i='', o=''):
        return await d.send('```python\n{}```'.format(self.format(i, o)))
    async def refresh(self, m, i='', o=''):
        global nl
        output = m.content[10:-3]
        if len(nl.findall(output)) <= 20: await m.edit(content='```python\n{}\n{}\n>>>```'.format(output, self.format(i, o)))
        else: await m.edit(content='```python\n{}```'.format(self.format(i, o)))

    async def generate_err(self, d, o=''):
        return await d.send('```\n{}```'.format(o))
    async def refresh_err(self, m, o=''):
        await m.edit(content='```\n{}```'.format(o))

    @commands.command(name=',console', aliases=[',con', ',c'], hidden=True)
    @commands.is_owner()
    @checks.del_ctx()
    async def console(self, ctx):
        def execute(msg):
            if msg.content == ',exit' and msg.author is ctx.message.author:
                raise exc.CheckFail
            elif msg.author is ctx.message.author and msg.channel is ctx.message.channel: return True
            else: return False

        try:
            console = await self.generate(ctx)
            exception = await self.generate_err(ctx)
            while True:
                exe = await self.bot.wait_for('message', check=execute)
                await exe.delete()
                sys.stdout = io.StringIO()
                sys.stderr = io.StringIO()
                try: exec(exe.content)
                except Exception: tb.print_exc(limit=1)
                await self.refresh(console, exe.content, sys.stdout.getvalue())
                await self.refresh_err(exception, sys.stderr.getvalue())
                await ctx.send(console.content[10:-3])
                sys.stdout = sys.__stdout__
                sys.stderr = sys.__stderr__
        except exc.CheckFail:
            await ctx.send('↩️ **Exited console.**')
        except Exception:
            await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
            tb.print_exc(limit=1, file=sys.__stderr__)
        finally:
            sys.stdout = sys.__stdout__
            sys.stderr = sys.__stderr__
            print('Reset sys output.')

    @commands.command(name='arbitrary', aliases=[',arbit', ',ar'])
    @commands.is_owner()
    @checks.del_ctx()
    async def arbitrary(self, ctx, *, exe):
        try:
            sys.stdout = io.StringIO()
            exec(exe)
            await self.generate(ctx, exe, sys.stdout.getvalue())
        except Exception:
            await ctx.send('{}\n```{}```'.format(exc.base, tb.format_exc(limit=1)))
            tb.print_exc(limit=1)
        finally:
            sys.stdout = sys.__stdout__
            print('Reset stdout.')

    @commands.group(aliases=[',db'], hidden=True)
    @commands.is_owner()
    @checks.del_ctx()
    async def debug(self, ctx):
        console = await self.generate(ctx)
    @debug.command(name='inject', aliases=['inj'])
    async def _inject(self, ctx, *, input_):
        pass
    @debug.command(name='inspect', aliases=['ins'])
    async def _inspect(self, ctx, *, input_):
        pass