keyboards.py

from aiogram.types import InlineKeyboardMarkup
from aiogram.utils.keyboard import InlineKeyboardBuilder

import texts.buttons
from callbacks import ActionButton, RemindButton, RemindButtonAction, NavigateButton, NavigateButtonLocation, \
    ActionButtonAction
from entities.remind import Remind
from entities.remind_group import RemindGroup
from texts.buttons import back_to_menu, remind_creation_confirm, remind_creation_change_text, \
    remind_creation_change_time
from texts.messages import TIME_FORMAT


def get_menu_keyboard() -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()

    builder.button(
        text=texts.buttons.new_remind,
        callback_data=ActionButton(action=ActionButtonAction.new_remind)
    )
    builder.button(
        text=texts.buttons.new_group_remind,
        callback_data=ActionButton(action=ActionButtonAction.new_group_remind)
    )
    builder.button(
        text=texts.buttons.remind_from_voice,
        callback_data=ActionButton(action=ActionButtonAction.create_remind_from_voice)
    )
    builder.button(
        text=texts.buttons.my_remind_list,
        callback_data=ActionButton(action=ActionButtonAction.remind_list)
    )
    builder.button(
        text=texts.buttons.my_remind_groups_list,
        callback_data=ActionButton(action=ActionButtonAction.show_remind_groups_list)
    )
    builder.button(
        text=texts.buttons.help_section,
        callback_data=ActionButton(action=ActionButtonAction.show_help_section)
    )

    builder.adjust(1)
    return builder.as_markup()


def get_confirm_remind_creation_keyboard(raw: bool = False) -> InlineKeyboardBuilder:
    builder = InlineKeyboardBuilder()

    builder.button(text=remind_creation_change_text,
                   callback_data=ActionButton(action=ActionButtonAction.edit_remind_text))
    builder.button(text=remind_creation_change_time,
                   callback_data=ActionButton(action=ActionButtonAction.edit_remind_time))
    builder.button(text=texts.buttons.remind_creation_change_remind_group,
                   callback_data=ActionButton(action=ActionButtonAction.edit_remind_group))
    if raw:
        builder.button(
            text=texts.buttons.back_to_raw_remind_list,
            callback_data=ActionButton(action=ActionButtonAction.back_to_raw_remind_list))
    else:
        builder.button(
            text=remind_creation_confirm,
            callback_data=ActionButton(action=ActionButtonAction.confirm_remind_creation))

    builder.adjust(2, 1)

    return builder


def get_remind_list_keyboard(reminds: list[Remind]) -> InlineKeyboardBuilder:
    builder = InlineKeyboardBuilder()
    for i, remind in enumerate(reminds, 1):
        r = RemindButton(remind_id=remind.id, action=RemindButtonAction.show)
        builder.button(text=f"{i}. {remind.text}",
                       callback_data=r)

    builder.button(text=back_to_menu, callback_data=ActionButton(action=ActionButtonAction.show_menu))
    builder.adjust(1)

    return builder


def get_remind_menu_markup(remind: Remind) -> InlineKeyboardBuilder:
    builder = InlineKeyboardBuilder()
    builder.button(text="Удалить",
                   callback_data=RemindButton(remind_id=remind.id, action=RemindButtonAction.delete))
    builder.button(text="Редактировать текст",
                   callback_data=RemindButton(remind_id=remind.id, action=RemindButtonAction.edit_text))
    builder.button(text="Редактировать время",
                   callback_data=RemindButton(remind_id=remind.id, action=RemindButtonAction.edit_time))
    builder.button(text="🔙 Назад к списку",
                   callback_data=NavigateButton(location=NavigateButtonLocation.remind_list))
    builder.adjust(1)

    return builder


def get_groups_list_keyboard(groups: list[RemindGroup], action: ActionButtonAction) -> InlineKeyboardBuilder:
    kb = InlineKeyboardBuilder()
    kb.button(text=texts.buttons.create_remind_group, callback_data=ActionButton(action=ActionButtonAction.create_remind_group))
    kb.attach(get_groups_list_raw_keyboard(action, groups))
    kb.button(text=back_to_menu, callback_data=ActionButton(action=ActionButtonAction.show_menu))
    kb.adjust(1)
    return kb


def get_groups_list_raw_keyboard(action: ActionButtonAction, groups: list[RemindGroup]):
    kb = InlineKeyboardBuilder()
    for group in groups:
        text = f"{group.name} (#{group.id})"
        kb.button(text=text,
                  callback_data=ActionButton(action=action, data=f"{group.id}"))
    kb.adjust(1)
    return kb


def get_grop_management_keyboard(group: RemindGroup, is_owner: bool = False) -> InlineKeyboardBuilder:
    kb = InlineKeyboardBuilder()

    kb.button(text=texts.buttons.leave_from_remind_group,
              callback_data=ActionButton(action=ActionButtonAction.leave_from_remind_group, data=f"{group.id}"))
    kb.button(text=texts.buttons.change_remind_group_name,
              callback_data=ActionButton(action=ActionButtonAction.change_remind_group_name, data=f"{group.id}"))
    if is_owner:
        kb.button(text=texts.buttons.remind_group_member_management,
                  callback_data=ActionButton(action=ActionButtonAction.remind_group_member_management, data=f"{group.id}"))
        kb.button(text=texts.buttons.delete_remind_group,
                  callback_data=ActionButton(action=ActionButtonAction.delete_remind_group, data=f"{group.id}"))

    kb.button(text=texts.buttons.back_to_group_list,
              callback_data=ActionButton(action=ActionButtonAction.show_group_list))

    kb.adjust(1)

    return kb


def get_new_group_remind_keyboard(groups: list[RemindGroup]) -> InlineKeyboardBuilder:
    kb = InlineKeyboardBuilder()
    kb.attach(get_groups_list_raw_keyboard(ActionButtonAction.select_group_for_new_remind, groups))
    kb.button(text=texts.buttons.back, callback_data=ActionButton(action=ActionButtonAction.show_menu))
    kb.adjust(1)
    return kb


def get_edit_remind_group_keyboard(groups: list[RemindGroup]) -> InlineKeyboardBuilder:
    kb = InlineKeyboardBuilder()
    kb.attach(get_groups_list_raw_keyboard(ActionButtonAction.edit_group_for_new_remind, groups))
    kb.button(text=texts.buttons.back,
              callback_data=ActionButton(action=ActionButtonAction.show_confirm_remind_creation_menu))
    kb.adjust(1)
    return kb


def get_edit_remind_group_for_all_raw_keyboard(groups: list[RemindGroup]) -> InlineKeyboardBuilder:
    kb = InlineKeyboardBuilder()
    kb.attach(get_groups_list_raw_keyboard(ActionButtonAction.edit_group_for_all_raw_reminds, groups))
    kb.button(text=texts.buttons.back,
              callback_data=ActionButton(action=ActionButtonAction.back_to_raw_remind_list))
    kb.adjust(1)
    return kb

def get_remind_creation_successful_keyboard() -> InlineKeyboardBuilder:
    kb = InlineKeyboardBuilder()
    kb.button(
        text=texts.buttons.create_another_remind,
        callback_data=ActionButton(action=ActionButtonAction.new_remind)
    )
    kb.button(text=back_to_menu, callback_data=ActionButton(action=ActionButtonAction.show_menu))
    kb.adjust(1)
    return kb


def get_entering_remind_creation_keyboard() -> InlineKeyboardBuilder:
    kb = InlineKeyboardBuilder()
    kb.button(text=back_to_menu, callback_data=ActionButton(action=ActionButtonAction.show_menu))
    return kb


def get_create_remind_group_markup() -> InlineKeyboardBuilder:
    kb = InlineKeyboardBuilder()
    kb.button(text=texts.buttons.back_to_group_list, callback_data=ActionButton(action=ActionButtonAction.show_remind_groups_list))
    return kb


def get_help_section_keyboard() -> InlineKeyboardBuilder:
    kb = InlineKeyboardBuilder()
    kb.button(text=back_to_menu, callback_data=ActionButton(action=ActionButtonAction.show_menu))
    return kb


def get_success_group_creating_keyboard() -> InlineKeyboardBuilder:
    kb = InlineKeyboardBuilder()
    kb.button(text=texts.buttons.back_to_group_list, callback_data=ActionButton(action=ActionButtonAction.show_remind_groups_list))
    return kb


def get_parsed_reminds_keyboard(raw_reminds: list[list[str]]) -> InlineKeyboardBuilder:
    kb = InlineKeyboardBuilder()

    for i, raw_remind in enumerate(raw_reminds):
        time = raw_remind[0].strftime(TIME_FORMAT)
        text = raw_remind[1]
        button_text = texts.buttons.raw_remind_button.format(time=time, text=text)
        kb.button(text=button_text,
                  callback_data=ActionButton(action=ActionButtonAction.edit_raw_remind, data=f"{i}"))

    kb.button(text=texts.buttons.remind_creation_change_remind_group,
                   callback_data=ActionButton(action=ActionButtonAction.edit_remind_group_for_all_raw))
    kb.button(text=texts.buttons.approve_all_reminds,
              callback_data=ActionButton(action=ActionButtonAction.approve_all_reminds))
    kb.button(text=back_to_menu, callback_data=ActionButton(action=ActionButtonAction.show_menu))
    kb.adjust(1)
    return kb


def get_navigation_keyboard(text: str, action: ActionButtonAction) -> InlineKeyboardBuilder:
    kb = InlineKeyboardBuilder()
    kb.button(text=text, callback_data=ActionButton(action=action))
    return kb


def get_create_remind_from_voice_keyboard() -> InlineKeyboardBuilder:
    return get_navigation_keyboard(texts.buttons.back_to_menu, ActionButtonAction.show_menu)