profiles.py

# -*- coding: utf-8 -*-
from datetime import datetime as d
from typing import Union

from discord import User
from discord.ext.commands import Cog, command, Context

from enigma.emebds.core import ErrorEmbed, SuccessEmbed, InfoEmbed
from enigma.emebds.misc import PleaseWaitEmbed, DevelopmentEmbed
from enigma.settings import in_production
from enigma.utils.database import get_single_user, update_profile, user_get_cash
from enigma.utils.strings import f_btc


class Profiles(Cog):
    def __init__(self, bot):
        self.bot = bot

    @command(
        name='profile',
        brief='Shows user\'s profile',
        description='Show user\'s profile from database (XP, cash, etc.)',
        usage='[user]',
        aliases=['prof'],
        enabled=in_production()
    )
    async def profile(self, ctx: Context, user: User = None):
        if not user:
            user = ctx.author
        if user.bot is True:
            await ctx.send(embed=ErrorEmbed(
                author=ctx.author,
                title=':x: Bots do not have profiles'
            ))
        else:
            msg = await ctx.send(embed=PleaseWaitEmbed(author=ctx.author))
            result_user = get_single_user(user.id if user else ctx.author.id)
            await msg.edit(embed=SuccessEmbed(
                author=ctx.author,
                title=':card_box: User\'s data',
                description='```py\n'
                            '{0.display_name}#{0.discriminator}\n'
                            '```'.format(ctx.guild.get_member(result_user.user_id))
            ).add_field(
                name='XP',
                value=str(result_user.user_xp)
            ).add_field(
                name='Cash',
                value=f_btc(result_user.user_cash)
            ))

    @command(
        name='daily',
        brief='Collect daily cash bonus',
        help='Cooldown resets on 00:00.',
        enabled=in_production()
    )
    async def daily(self, ctx: Context):
        base_cash = 200
        user = get_single_user(ctx.author.id)
        now = d.now()
        able = False
        time_travel = False
        if user.last_daily.year == now.year:
            if user.last_daily.month == now.month:
                if user.last_daily.day == now.day:
                    pass
                elif user.last_daily.day < now.day:
                    able = True
                else:
                    time_travel = True
            elif user.last_daily.month < now.month:
                able = True
            else:
                time_travel = True
        elif user.last_daily.year < now.year:
            able = True
        else:
            time_travel = True
        if time_travel:
            await ctx.send(embed=ErrorEmbed(
                author=ctx.author,
                title=':x: Hello time traveler',
                description='Anyway, there\'s no cash for you.'
            ))
        elif able:
            new_user = user_get_cash(ctx.author.id, base_cash)
            await ctx.send(embed=SuccessEmbed(
                author=ctx.author,
                title=':moneybag: Daily bonus gained!',
                description=f'You\'ve earned **{f_btc(base_cash)}**,'
                            f' so now you have **{f_btc(new_user.user_cash)}**.'
            ))
        else:
            await ctx.send(embed=ErrorEmbed(
                author=ctx.author,
                title=':x: You\'ve already collected your daily cash',
                description='Please come back tomorrow.'
            ))

    @command(
        name='reputation',
        brief='[ WIP ] Gives someone reputation',
        help='Available once a day, resets on 00:00.',
        usage='<user>',
        aliases=['rep'],
        enabled=not in_production()
    )
    async def reputation(self, ctx: Context, user: User):
        # TODO - Reputation
        await ctx.send(embed=DevelopmentEmbed(
            author=ctx.author
        ).add_field(
            name='USER',
            value=str(user)
        ))

    @command(
        name='manage',
        brief='Manages users\' data',
        description='Only true Anonymous can use this command',
        help='Available options:\n'
             '- xp\n'
             '- cash',
        usage='<user> <option> <value>',
        enabled=in_production(),
        hidden=True
    )
    async def manage(self, ctx: Context, user: User = None, option: str = None, value: int = None):
        if ctx.author.id != self.bot.owner_id:
            st = ':x: You\'re not authorized'
        elif not user:
            st = ':x: No user provided'
        elif user.bot is True:
            st = ':x: Bots do not have profiles'
        elif not option:
            st = ':x: No option specified'
        elif option not in ['xp', 'cash']:
            st = ':x: Invalid option'
        elif value is None:
            st = ':x: No amount provided'
        else:
            st = None
            msg = await ctx.send(embed=PleaseWaitEmbed(author=ctx.author))
            managed_user = get_single_user(user.id)
            if not managed_user:
                await msg.edit(embed=ErrorEmbed(
                    author=ctx.author,
                    title=':x: User do not exists in database'
                ))
            else:
                if option == 'xp':
                    updated_user = update_profile(user.id, xp=value)
                elif option == 'cash':
                    updated_user = update_profile(user.id, cash=value)
                else:
                    raise
                text = '```py\n{0.display_name}#{0.discriminator}\n'.format(ctx.guild.get_member(updated_user.user_id))
                if option == 'xp':
                    text += f'XP:         {value}\n'
                elif option == 'cash':
                    text += f'Cash:       {f_btc(value)}\n'
                text += '```'
                await msg.edit(embed=SuccessEmbed(
                    author=ctx.author,
                    title=':incoming_envelope: Profile has been updated',
                    description=text
                ))
        if st:
            await ctx.send(embed=ErrorEmbed(
                author=ctx.author,
                title=st
            ))

    @command(
        name='avatar',
        brief='Shows user\'s avatar',
        description='No arg will return your\'s avatar and adding user ID or mention will show other user\'s avatar',
        usage='[user]',
        aliases=['avk'],
        enabled=in_production()
    )
    async def avatar(self, ctx: Context, user: User = None):
        if not user:
            user = ctx.author
        await ctx.send(embed=SuccessEmbed(
            author=ctx.author,
            title=f':bust_in_silhouette: {user} avatar',
            description=f'As you wish, {ctx.author.mention}.'
        ).set_image(
            url=user.avatar_url
        ))

    @command(
        name='whois',
        brief='Short info about user',
        usage='<user>',
        enabled=in_production()
    )
    async def whois(self, ctx: Context, user: User = None):
        if not user:
            await ctx.send(embed=ErrorEmbed(
                author=ctx.author,
                title=':x: No user specified'
            ))
        else:
            await ctx.send(embed=InfoEmbed(
                author=ctx.author,
                title=f':face_with_monocle: Who is {user.display_name}'
            ).add_field(
                name='Original name',
                value='{0.name}#{0.discriminator}'.format(user),
                inline=False
            ).add_field(
                name='User ID',
                value=str(user.id),
                inline=False
            ).add_field(
                name='Joined Discord at',
                value=str(user.created_at)[:19],
                inline=False
            ).add_field(
                name='Bot account',
                value=str(user.bot),
                inline=False
            ).add_field(
                name='Public flags',
                value=', '.join(map(lambda z: f'`{z.upper()}`' if z else None, list(filter(
                    lambda y: y is not None, map(lambda x: x[0] if x[1] else None, user.public_flags)
                )))) or 'None'
            ).set_thumbnail(
                url=user.avatar_url
            ))

    @command(
        name='notes',
        brief='Manages your notes',
        description='Every user can save upto 3 notes.',
        help='Available options:'
             '- create <note content>'
             '- delete <note UID>'
             '- edit <note UID>',
        usage='<option [...]>',
        enabled=not in_production()
    )
    async def notes(self, ctx: Context, option: str = None, arg: Union[int, str] = None):
        # TODO - Notes
        await ctx.send(embed=DevelopmentEmbed(
            author=ctx.author
        ).add_field(
            name='OPTION',
            value=option
        ).add_field(
            name='ARG',
            value=f'{str(arg)} ({type(arg).__class__.__name__})'
        ))

    # TODO - Quests

    # TODO - Top reputations


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