from vkbottle import Keyboard, Text
from vkbottle.tools.keyboard import Callback


def back_to_main():
    kb = Keyboard(one_time=False)
    kb.add(Text("Главное меню"))
    return kb.get_json()


def build_keyboard(buttons, back: bool = False):
    kb = Keyboard(one_time=False)

    for i, btn in enumerate(buttons):
        kb.add(Text(btn["title"]))
        if (i + 1) % 2 == 0:
            kb.row()

    if back:
        kb.row()
        kb.add(Text("Назад"))

    return kb.get_json()


def pause_keyboard():
    kb = Keyboard(one_time=False, inline=True)
    kb.add(Callback("Да, помогло", payload={"pause": "yes"}))
    kb.add(Callback("Нет, дальше", payload={"pause": "no"}))
    return kb.get_json()


def tech_feedback_keyboard():
    kb = Keyboard(one_time=False, inline=True)
    kb.add(Callback("Да", payload={"tech_feedback": "yes"}))
    kb.add(Callback("Нет", payload={"tech_feedback": "no"}))
    return kb.get_json()