init

bot/handlers/navigation.py

@@ -0,0 +1,93 @@
+from aiogram.fsm.context import FSMContext
+from aiogram.types import Message, CallbackQuery
+
+from .context import ContextData, CallbackCommand
+
+
+async def save_navigation_context(callback_data: ContextData, state: FSMContext) -> list[ContextData]:
+    data = await state.get_data()
+    stack = [ContextData.unpack(item) for item in data.get("navigation_stack", [])]
+    data_nc = data.get("navigation_context")
+    navigation_context = ContextData.unpack(data_nc) if data_nc else None
+    if callback_data.back:
+        callback_data.back = False 
+        if stack:
+            stack.pop()
+    else:
+        if stack and navigation_context and navigation_context.command == callback_data.command:
+            navigation_context = callback_data
+        elif navigation_context:
+            stack.append(navigation_context)
+
+    await state.update_data({"navigation_stack": [item.pack() for item in stack],
+                             "navigation_context": callback_data.pack()})
+    return stack
+
+
+def pop_navigation_context(stack: list[ContextData]) -> ContextData | None:
+    if stack:
+        data = stack[-1]
+        data.back = True
+        return data
+    
+
+async def get_navigation_context(state: FSMContext) -> tuple[list[ContextData], ContextData | None]:
+    data = await state.get_data()
+    data_nc = data.get("navigation_context")
+    context = ContextData.unpack(data_nc) if data_nc else None
+    return ([ContextData.unpack(item) for item in data.get("navigation_stack", [])],
+            context)
+
+
+async def clear_state(state: FSMContext, clear_nav: bool = False):
+    if clear_nav:
+        await state.clear()
+    else:
+        state_data = await state.get_data()
+        stack = state_data.get("navigation_stack")
+        context = state_data.get("navigation_context")
+        update_data = {}
+        if stack:
+            update_data["navigation_stack"] = stack
+        if context:
+            update_data["navigation_context"] = context
+        await state.clear()
+        await state.update_data(update_data)
+
+
+async def route_callback(message: Message | CallbackQuery, back: bool = True, **kwargs):
+
+    stack, context = await get_navigation_context(kwargs["state"])
+    if back:
+        context = pop_navigation_context(stack)
+        stack = await save_navigation_context(callback_data = context, state = kwargs["state"])
+    kwargs.update({"callback_data": context, "navigation_stack": stack})
+    if context:
+        if context.command == CallbackCommand.MENU_ENTRY_MAIN:
+            await main_menu(message, **kwargs)
+        elif context.command == CallbackCommand.MENU_ENTRY_SETTINGS:
+            await settings_menu(message, **kwargs)
+        elif context.command == CallbackCommand.MENU_ENTRY_PARAMETERS:
+            await parameters_menu(message, **kwargs)
+        elif context.command == CallbackCommand.MENU_ENTRY_LANGUAGE:
+            await language_menu(message, **kwargs)
+        elif context.command == CallbackCommand.MENU_ENTRY_ENTITIES:
+            await entities_menu(message, **kwargs)
+        elif context.command == CallbackCommand.ENTITY_LIST:
+            await entity_list(message, **kwargs)
+        elif context.command == CallbackCommand.ENTITY_ITEM:
+            await entity_item(message, **kwargs)
+
+        else:
+            raise ValueError(f"Unknown command {context.command}")
+    else:
+        raise ValueError("No navigation context")
+
+
+from .menu.main import main_menu
+from .menu.settings import settings_menu
+from .menu.parameters import parameters_menu
+from .menu.language import language_menu
+from .menu.entities import entities_menu
+from .forms.entity_list import entity_list
+from .forms.entity_form import entity_item