from datetime import datetime as dt
from itertools import islice
from typing import Literal, Dict, List

from telebot import types  # noqa

from app.utils.system import readable_size


def chunk_dict(data: dict, size: int = 2):
    it = iter(data)
    for i in range(0, len(data), size):
        yield {k: data[k] for k in islice(it, size)}


class BotKeyboard:

    @staticmethod
    def main_menu():
        keyboard = types.InlineKeyboardMarkup()
        keyboard.add(
            types.InlineKeyboardButton(text="🔁 System Info", callback_data="system"),
            types.InlineKeyboardButton(text="♻️ Restart Xray", callback_data="restart"),
        )
        keyboard.add(
            types.InlineKeyboardButton(text="👥 Users", callback_data="users:1"),
            types.InlineKeyboardButton(
                text="✏️ Edit All Users", callback_data="edit_all"
            ),
        )
        # keyboard.add(
        #    types.InlineKeyboardButton(text='➕ Create User From Template', callback_data='template_add_user'))
        keyboard.add(
            types.InlineKeyboardButton(text="➕ Create User", callback_data="add_user")
        )
        return keyboard

    @staticmethod
    def edit_all_menu():
        keyboard = types.InlineKeyboardMarkup()
        keyboard.add(
            types.InlineKeyboardButton(
                text="🗑 Delete Expired", callback_data="delete_expired"
            ),
            types.InlineKeyboardButton(
                text="🗑 Delete Limited", callback_data="delete_limited"
            ),
        )
        keyboard.add(
            types.InlineKeyboardButton(
                text="🔋 Data (➕|➖)", callback_data="add_data"
            ),
            types.InlineKeyboardButton(
                text="📅 Time (➕|➖)", callback_data="add_time"
            ),
        )
        # keyboard.add(
        #    types.InlineKeyboardButton(text='➕ Add Inbound', callback_data='inbound_add'),
        #    types.InlineKeyboardButton(text='➖ Remove Inbound', callback_data='inbound_remove'))
        keyboard.add(types.InlineKeyboardButton(text="🔙 Back", callback_data="cancel"))
        return keyboard

    # @staticmethod
    # def inbounds_menu(action, inbounds):
    #    keyboard = types.InlineKeyboardMarkup()
    #    for inbound in inbounds:
    #        keyboard.add(types.InlineKeyboardButton(text=inbound, callback_data=f'confirm_{action}:{inbound}'))
    #    keyboard.add(types.InlineKeyboardButton(text='🔙 Back', callback_data='cancel'))
    #    return keyboard

    @staticmethod
    def random_username(template_id: str = ""):
        keyboard = types.InlineKeyboardMarkup()

        keyboard.add(
            types.InlineKeyboardButton(
                text="🔡 Random Username", callback_data=f"random:{template_id}"
            )
        )
        keyboard.add(
            types.InlineKeyboardButton(text="🔙 Cancel", callback_data="cancel")
        )
        return keyboard

    @staticmethod
    def user_menu(user_info, with_back: bool = True, page: int = 1, note: bool = False):
        keyboard = types.InlineKeyboardMarkup()
        keyboard.add(
            types.InlineKeyboardButton(
                text="❌ Disable" if user_info["status"] == "active" else "✅ Activate",
                callback_data=f"{'suspend' if user_info['status'] == 'active' else 'activate'}:{user_info['username']}",
            ),
            types.InlineKeyboardButton(
                text="🗑 Delete", callback_data=f"delete:{user_info['username']}"
            ),
        )
        if note:
            keyboard.add(
                types.InlineKeyboardButton(
                    text="🚫 Revoke Sub",
                    callback_data=f"revoke_sub:{user_info['username']}",
                ),
                types.InlineKeyboardButton(
                    text="✏️ Edit", callback_data=f"edit:{user_info['username']}"
                ),
            )
            keyboard.add(
                types.InlineKeyboardButton(
                    text="📝 Edit Note",
                    callback_data=f"edit_note:{user_info['username']}",
                ),
                types.InlineKeyboardButton(
                    text="📡 Links", callback_data=f"links:{user_info['username']}"
                ),
            )
        else:
            keyboard.add(
                types.InlineKeyboardButton(
                    text="📡 Links", callback_data=f"links:{user_info['username']}"
                ),
                types.InlineKeyboardButton(
                    text="✏️ Edit", callback_data=f"edit:{user_info['username']}"
                ),
            )
        keyboard.add(
            types.InlineKeyboardButton(
                text="🔁 Reset usage",
                callback_data=f"reset_usage:{user_info['username']}",
            ),
            types.InlineKeyboardButton(
                text="🔋 Charge", callback_data=f"charge:{user_info['username']}"
            ),
        )
        if with_back:
            keyboard.add(
                types.InlineKeyboardButton(
                    text="🔙 Back", callback_data=f"users:{page}"
                )
            )
        return keyboard

    @staticmethod
    def show_links(username: str):
        keyboard = types.InlineKeyboardMarkup()

        keyboard.add(
            types.InlineKeyboardButton(
                text="🖼 QR code", callback_data=f"genqr:{username}"
            )
        )
        keyboard.add(
            types.InlineKeyboardButton(text="🔙 Back", callback_data=f"user:{username}")
        )
        return keyboard

    @staticmethod
    def subscription_page(sub_url: str):
        keyboard = types.InlineKeyboardMarkup()
        if sub_url[:4] == "http":
            keyboard.add(
                types.InlineKeyboardButton(text="🚀 Subscription Page", url=sub_url)
            )
        return keyboard

    @staticmethod
    def confirm_action(action: str, username: str = None):
        keyboard = types.InlineKeyboardMarkup()
        keyboard.add(
            types.InlineKeyboardButton(
                text="Yes", callback_data=f"{action}:{username}"
            ),
            types.InlineKeyboardButton(text="No", callback_data=f"cancel"),
        )
        return keyboard

    @staticmethod
    def inline_cancel_action(callback_data: str = "cancel"):
        keyboard = types.InlineKeyboardMarkup()
        keyboard.add(
            types.InlineKeyboardButton(text="🔙 Cancel", callback_data=callback_data)
        )
        return keyboard

    @staticmethod
    def user_list(users: list, page: int, total_pages: int):
        keyboard = types.InlineKeyboardMarkup()
        if len(users) >= 2:
            users = [p for p in users]
            users = [users[i : i + 2] for i in range(0, len(users), 2)]
        else:
            users = [users]
        for user in users:
            row = []
            for p in user:
                status = {
                    "active": "✅",
                    "expired": "🕰",
                    "limited": "📵",
                    "disabled": "❌",
                }
                row.append(
                    types.InlineKeyboardButton(
                        text=f"{p.username} ({status[p.status]})",
                        callback_data=f"user:{p.username}:{page}",
                    )
                )
            keyboard.row(*row)
        # if there is more than one page
        if total_pages > 1:
            if page > 1:
                keyboard.add(
                    types.InlineKeyboardButton(
                        text="⬅️ Previous", callback_data=f"users:{page - 1}"
                    )
                )
            if page < total_pages:
                keyboard.add(
                    types.InlineKeyboardButton(
                        text="➡️ Next", callback_data=f"users:{page + 1}"
                    )
                )
        keyboard.add(types.InlineKeyboardButton(text="🔙 Back", callback_data="cancel"))
        return keyboard

    @staticmethod
    def select_services(
        services,
        selected_services: List[int],
        action: Literal["edit", "create"],
        username: str = None,
        data_limit: float = None,
        expire_date: dt = None,
    ):

        keyboard = types.InlineKeyboardMarkup()

        if action == "edit":
            keyboard.add(
                types.InlineKeyboardButton(
                    text="⚠️ Data Limit:", callback_data=f"help_edit"
                )
            )
            keyboard.add(
                types.InlineKeyboardButton(
                    text=f"{readable_size(data_limit) if data_limit else 'Unlimited'}",
                    callback_data=f"help_edit",
                ),
                types.InlineKeyboardButton(
                    text="✏️ Edit", callback_data=f"edit_user:{username}:data"
                ),
            )
            keyboard.add(
                types.InlineKeyboardButton(
                    text="📅 Expire Date:", callback_data=f"help_edit"
                )
            )
            keyboard.add(
                types.InlineKeyboardButton(
                    text=f"{expire_date.strftime('%Y-%m-%d') if expire_date else 'Never'}",
                    callback_data=f"help_edit",
                ),
                types.InlineKeyboardButton(
                    text="✏️ Edit", callback_data=f"edit_user:{username}:expire"
                ),
            )
        for s in services:
            keyboard.add(
                types.InlineKeyboardButton(
                    text=f"🌐 {s.name} {'✅' if s.id in selected_services else '❌'}",
                    callback_data=f"switch_service:{s.id}:{action}",
                )
            )
        keyboard.add(
            types.InlineKeyboardButton(
                text="Done",
                callback_data="edit_user_done" if action == "edit" else "add_user_done",
            )
        )
        keyboard.add(
            types.InlineKeyboardButton(
                text="Cancel",
                callback_data=f"user:{username}" if action == "edit" else "cancel",
            )
        )

        return keyboard