Commit

old/bot/combat_handlers.py

@@ -0,0 +1,165 @@
+"""
+Combat-related action handlers.
+"""
+import logging
+from . import keyboards
+from .api_client import api_client
+from .utils import format_stat_bar
+from data.world_loader import game_world
+
+logger = logging.getLogger(__name__)
+
+
+async def handle_combat_attack(query, user_id: int, player: dict, data: list = None):
+    """Handle player attack action in combat."""
+    from bot import combat
+    await query.answer()
+    
+    message, npc_died, turn_ended = await combat.player_attack(user_id)
+    
+    if npc_died:
+        # Combat ended - return to main menu
+        location = game_world.get_location(player['location_id'])
+        location_image = location.image_path if location else None
+        
+        from .handlers import send_or_edit_with_image
+        await send_or_edit_with_image(
+            query,
+            text=message,
+            reply_markup=keyboards.main_menu_keyboard(),
+            image_path=location_image
+        )
+    elif turn_ended:
+        # NPC's turn - auto-attack
+        npc_message, player_died = await combat.npc_attack(user_id)
+        message += "\n\n" + npc_message
+        
+        if player_died:
+            from .handlers import send_or_edit_with_image
+            await send_or_edit_with_image(query, text=message, reply_markup=None)
+        else:
+            combat_data = await api_client.get_combat(user_id)
+            if combat_data:
+                from data.npcs import NPCS
+                npc_def = NPCS.get(combat_data['npc_id'])
+                keyboard = await keyboards.combat_keyboard(user_id)
+                
+                from .handlers import send_or_edit_with_image
+                await send_or_edit_with_image(
+                    query,
+                    text=message,
+                    reply_markup=keyboard,
+                    image_path=npc_def.image_url if npc_def else None
+                )
+    else:
+        await query.answer(message, show_alert=False)
+
+
+async def handle_combat_flee(query, user_id: int, player: dict, data: list = None):
+    """Handle flee attempt from combat."""
+    from bot import combat
+    await query.answer()
+    
+    message, fled, turn_ended = await combat.flee_attempt(user_id)
+    
+    if fled:
+        # Successfully fled - return to main menu
+        location = game_world.get_location(player['location_id'])
+        location_image = location.image_path if location else None
+        
+        from .handlers import send_or_edit_with_image
+        await send_or_edit_with_image(
+            query,
+            text=message,
+            reply_markup=keyboards.main_menu_keyboard(),
+            image_path=location_image
+        )
+    elif turn_ended:
+        # Failed to flee - NPC attacks
+        npc_message, player_died = await combat.npc_attack(user_id)
+        message += "\n\n" + npc_message
+        
+        if player_died:
+            from .handlers import send_or_edit_with_image
+            await send_or_edit_with_image(query, text=message, reply_markup=None)
+        else:
+            combat_data = await api_client.get_combat(user_id)
+            if combat_data:
+                from data.npcs import NPCS
+                npc_def = NPCS.get(combat_data['npc_id'])
+                keyboard = await keyboards.combat_keyboard(user_id)
+                
+                from .handlers import send_or_edit_with_image
+                await send_or_edit_with_image(
+                    query,
+                    text=message,
+                    reply_markup=keyboard,
+                    image_path=npc_def.image_url if npc_def else None
+                )
+    else:
+        await query.answer(message, show_alert=False)
+
+
+async def handle_combat_use_item_menu(query, user_id: int, player: dict, data: list = None):
+    """Show menu of usable items during combat."""
+    await query.answer()
+
+
+async def handle_combat_use_item(query, user_id: int, player: dict, data: list):
+    """Use an item during combat."""
+    from bot import combat
+    item_db_id = int(data[1])
+    
+    message, turn_ended = await combat.use_item_in_combat(user_id, item_db_id)
+    await query.answer(message, show_alert=False)
+    
+    if turn_ended:
+        # NPC's turn
+        npc_message, player_died = await combat.npc_attack(user_id)
+        
+        if player_died:
+            from .handlers import send_or_edit_with_image
+            await send_or_edit_with_image(
+                query,
+                text=message + "\n\n" + npc_message,
+                reply_markup=None
+            )
+        else:
+            combat_data = await api_client.get_combat(user_id)
+            if combat_data:
+                from data.npcs import NPCS
+                npc_def = NPCS.get(combat_data['npc_id'])
+                keyboard = await keyboards.combat_keyboard(user_id)
+                full_message = message + "\n\n" + npc_message + "\n\n🎯 Your turn!"
+                
+                from .handlers import send_or_edit_with_image
+                await send_or_edit_with_image(
+                    query,
+                    text=full_message,
+                    reply_markup=keyboard,
+                    image_path=npc_def.image_url if npc_def else None
+                )
+
+
+async def handle_combat_back(query, user_id: int, player: dict, data: list = None):
+    """Return to combat menu from item selection."""
+    await query.answer()
+    combat_data = await api_client.get_combat(user_id)
+    
+    if combat_data:
+        from data.npcs import NPCS
+        npc_def = NPCS.get(combat_data['npc_id'])
+        keyboard = await keyboards.combat_keyboard(user_id)
+        
+        message = f"⚔️ Combat with {npc_def.emoji} {npc_def.name}!\n"
+        message += format_stat_bar("Your HP", "❤️", player['hp'], player['max_hp']) + "\n"
+        message += format_stat_bar("Enemy HP", npc_def.emoji, combat_data['npc_hp'], combat_data['npc_max_hp']) + "\n\n"
+        message += "🎯 Your turn!" if combat_data['turn'] == 'player' else "⏳ Enemy's turn..."
+        
+        from .handlers import send_or_edit_with_image
+        await send_or_edit_with_image(
+            query,
+            text=message,
+            reply_markup=keyboard,
+            image_path=npc_def.image_url if npc_def else None
+        )