from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup


def ikb(rows=None, back=False, todo="start_back"):
    if rows is None:
        rows = []
    lines = []
    for row in rows:
        line = []
        try:
            for button in row:
                btn_text = button.split(".")[1].upper()
                button = btn(btn_text, button)  # InlineKeyboardButton
                line.append(button)
            lines.append(line)
        except AttributeError:
            for button in row:
                button = btn(*button)  # InlineKeyboardButton
                line.append(button)
            lines.append(line)
    if back:
        back_btn = [(btn("« Back", todo))]
        lines.append(back_btn)
    return InlineKeyboardMarkup(inline_keyboard=lines)


def btn(text, value, type="callback_data"):
    return InlineKeyboardButton(text, **{type: value})