from typing import Dict, List, Tuple, Union

from aiogram.types import CallbackGame, InlineKeyboardButton, InlineKeyboardMarkup, LoginUrl
from aiogram.utils.callback_data import CallbackData

from ..keyboard_utils import schema_generator


class InlineConstructor:
    aliases = {
        'cb': 'callback_data'
    }
    available_properities = [
        'text', 'callback_data', 'login_url', 'switch_inline_query', 'switch_inline_query_current_chat',
        'callback_game', 'pay'
    ]
    properties_amount = 2

    @staticmethod
    def _create_kb(
            actions: List[Dict[
                str,
                Union[
                    str,
                    bool,
                    Tuple[Dict[str, str], CallbackData],
                    LoginUrl,
                    CallbackGame,
                ]
            ]],
            schema: List[int]
    ) -> InlineKeyboardMarkup:
        kb = InlineKeyboardMarkup()
        kb.row_width = max(schema)
        btns = []
        # noinspection DuplicatedCode
        for a in actions:
            data: Dict[str, Union[str, bool, Tuple[Dict[str, str], CallbackData], LoginUrl, CallbackGame]] = {}
            for k, v in InlineConstructor.aliases.items():
                if k in a:
                    a[v] = a[k]
                    del a[k]
            for k in a:
                if k in InlineConstructor.available_properities:
                    if len(data) < InlineConstructor.properties_amount:
                        data[k] = a[k]
                    else:
                        break
            if 'callback_data' in data:
                data['callback_data'] = data['callback_data'][1].new(**data['callback_data'][0])
            if 'pay' in data:
                if len(btns) != 0 and data['pay']:
                    raise ValueError('Платежная кнопка должна идти первой в клавиатуре')
                data['pay'] = a['pay']
            if len(data) != InlineConstructor.properties_amount:
                raise ValueError('Недостаточно данных для создания кнопки')
            btns.append(InlineKeyboardButton(**data))
        kb.inline_keyboard = schema_generator.create_keyboard_layout(btns, schema)
        return kb