from aiogram.utils.keyboard import ReplyKeyboardBuilder
from aiogram.types import KeyboardButton
from locks_api import fetch_locks


async def get_locks_keyboard(allowed_locks: list):
    kb = ReplyKeyboardBuilder()
    locks = await fetch_locks()
    for zone in allowed_locks:
        lock_info = locks.get(zone)
        btn_text = lock_info["name"] if lock_info and "name" in lock_info else zone
        kb.button(text=btn_text)
    return kb.as_markup(resize_keyboard=True)


def get_contact_keyboard():
    kb = ReplyKeyboardBuilder()
    kb.add(KeyboardButton(text="Поделиться контактом", request_contact=True))
    return kb.as_markup(resize_keyboard=True)