import shelve
import os
from .config import DATA, ROLLBACK


def create_report_text(chat_id):
    report = ''
    top_users = ''
    flooders = []
    if os.path.exists(f'{DATA}{chat_id}.db'):
        with shelve.open(f'{DATA}{chat_id}') as s:
            for user_id in ROLLBACK:
                if user_id in s['Messages'] and s['Messages'][user_id]['Count'] > 10:
                    s['Messages'][user_id]['Count'] //= 3

            for n in range(min(3, len(s['Messages']))):
                top_user = s['Messages'][
                    sorted(s['Messages'], key=lambda a: s['Messages'][a]['Count'], reverse=True)[n]]
                if top_user['Count'] >= 5:
                    flooders.append(top_user)

            for i, flooder in enumerate(flooders):
                user = flooder["User"]
                name = f"{user.first_name} {user.last_name}" if user.last_name else user.first_name
                top_users += f'\n  {i + 1}. {name} [<code>{flooder["Count"]}</code>]'

            if len(flooders) >= 3 or s['Banned'] or s['Deleted']:
                report = f"<code>Hello, World!</code>\n"
            if len(flooders) >= 3:
                report += f"\n<b>Топ</b> 🐍{top_users}\n"
            if s['Banned'] or s['Deleted']:
                report += f"""
<b>Заблокировано</b>
├ Пользователей: <code>{s['Banned']}</code>
└ Сообщений: <code>{s['Deleted']}</code>
"""
        return report
    else:
        return f'Невозможно получить статистику.\n{DATA}{chat_id}.db не существует, возможно запрос в приватном чате. '


def reset_report_stats(chat_id):
    with shelve.open(f'{DATA}{chat_id}') as s:
        s['Messages'] = {}
        s['Banned'] = 0
        s['Deleted'] = 0
        return f"""Chat ID: {chat_id}

{s['Messages']=}
{s['Banned']=}
{s['Deleted']=}"""