report.py

from aiohttp import ClientSession as HTTP
from ext import Extension as Super
import load
import mwclient

FLAG_REPORT = 0
FLAG_UNREPORT = 1
FLAG_RESOLVE = 2

class Extension(Super):
    def __init__(self, bot, config):
        super(Extension, self).__init__(bot, config)
        self.temp = load.load_data('report')
        self.register_commands('report', 'unreport', 'resolve', 'kocka')
        self.mw = mwclient.Site('vstf.wikia.com', path='/')
        self.mw.login(config['username'], config['password'])
        for t in ['w', 's', 'p', 'b']:
            if not t in self.temp:
                if t == 'w' or t == 'b':
                    self.temp[t] = []
                else:
                    self.temp[t] = {}
        self.message = False

    def report_message(self, t, p=''):
        return {
            'w': '{{{{badwiki|{0}}}}}'.format('}}\n{{badwiki|'.join(self.temp['w'])),
            's': '== {0} ==\n{{{{Report spam|{0}|Spam|{1}|{{{{subst:REVISIONUSER}}}}|~~~~~}}}}'.format(p, '|'.join(self.temp['s'].get(p, []))),
            'p': '== {0} ==\n{{{{Report profile|{0}|Spam|{1}|{{{{subst:REVISIONUSER}}}}|~~~~~}}}}'.format(p, '|'.join(self.temp['p'].get(p, []))),
            'b': '\n'.join([ ('* [[w:c:{0}:Special:Contribs/{1}|{1}]]'.format(x.split(':')[0], ':'.join(x.split(':')[1:]))) for x in self.temp['b'] ])
        }[t]

    async def update_report(self, msg=None):
        load.write_data('report', self.temp)
        m = """__**Report message**__\nThis message is where new reports accumulate\n"""
        if len(self.temp['w']) > 0:
            m += '**Spam wikis**```%s```' % self.report_message('w')
        if len(self.temp['s']) > 0:
            for k, v in self.temp['s'].items():
                if len(v) > 0:
                    m += """**Spam users: %s**```%s```""" % (k, self.report_message('s', k))
        if len(self.temp['p']) > 0:
            for k, v in self.temp['p'].items():
                if len(v) > 0:
                    m += '**Spam profiles: %s**```%s```' % (k, self.report_message('p', k))
        if len(self.temp['b']) > 0:
            m += """**XRumer spam**```%s```""" % self.report_message('b')
        if not self.message:
            self.message = await self.bot.get_channel(self.config['bind_channel']).send(m)
        else:
            await self.message.edit(content=m)
        if(msg):
            await self.reply(msg, 'Reports updated!', True)

    def modify_array(self, array, el, flag, nosplit=False):
        # Don't ask. Please just don't ask.
        if nosplit:
            el = [el]
        else:
            el = el.split('|')
        # Python y u do dis
        def temp(array):
            del array[:]
        for e in el:
            [
                lambda: array.append(e.replace('_', ' ')) if not e in array else False,
                lambda: array.remove(e.replace('_', ' ')) if e in array else False,
                lambda: temp(array)
            ][flag]()

    async def webhook(self, user, flag, target):
        action = ''
        if flag == FLAG_REPORT:
            action = 'reported'
        elif flag == FLAG_UNREPORT:
            action = 'unreported'
        elif flag == FLAG_RESOLVE:
            action = 'resolved'
        await self.bot.get_channel(self.config['log_channel']).send('%s %s %s.' % (user.name, action, target))

    async def base_report(self, message, params, flag):
        params.append('')
        t = params[0].lower()
        if t == 'w':
            for e in params[1].split('|'):
                split = e.split(':')
                if len(split) > 1 and split[1] == 'f':
                    split[0] = split[0] + '||fandom'
                self.modify_array(self.temp[t], split[0], flag, nosplit=True)
            if flag == FLAG_RESOLVE:
                await self.webhook(message.author, flag, 'wiki reports')
            else:
                await self.webhook(message.author, flag, 'wiki(s): %s' % params[1])
            await self.update_report(message)
        elif t == 's' or t == 'p':
            wiki = params[1].lower()
            if not wiki in self.temp[t]:
                self.temp[t][wiki] = []
            name = 'spam'
            if t == 'p':
                name = 'spam profiles'
            users = ' '.join(params[2:]).strip()
            self.modify_array(self.temp[t][wiki], users, flag)
            if flag == FLAG_RESOLVE:
                await self.webhook(message.author, flag, '%s from %s' % (name, wiki))
            else:
                await self.webhook(message.author, flag, '%s from %s: %s' % (name, wiki, users))
            await self.update_report(message)
        elif t == 'b' or t == 'x':
            wiki = params[1]
            split = wiki.split(':')
            if len(split) >= 2:
                wiki = split[0].lower()
                params[2] = ':'.join(split[1:])
                users = wiki + ':' + ' '.join(params[2:]).strip()
                self.modify_array(self.temp['b'], users, flag)
                await self.webhook(message.author, flag, 'XRumer spam: %s' % users)
                await self.update_report(message)
            elif flag == FLAG_RESOLVE:
                self.modify_array(self.temp['b'], wiki + ':' + ' '.join(params[2:]).strip(), flag)
                await self.webhook(message.author, flag, 'XRumer spam')
                await self.update_report(message)
            else:
                await self.reply(message, 'Format for biglist spam is `!report b wiki:user`!')
        else:
            await self.reply(message, 'Invalid report type!', True)

    async def command_report(self, message, params):
        await self.base_report(message, params, FLAG_REPORT)

    async def command_unreport(self, message, params):
        await self.base_report(message, params, FLAG_UNREPORT)

    async def command_resolve(self, message, params):
        await self.base_report(message, params, FLAG_RESOLVE)

    async def command_kocka(self, message, params):
        if(message.author == self.bot.master):
            params.append('')
            page = self.mw.Pages['Report:' + {
                'w': 'Wiki',
                's': 'Spam',
                'p': 'User profile headers',
                'b': 'Spam/Biglist'
            }[params[0]]]
            text = page.text()
            text += '\n' + self.report_message(params[0], params[1])
            page.save(text, summary='Moving reports of Wikia Watchers')
            await self.base_report(message, params, FLAG_RESOLVE)