import quart

from ... import group


@group.group_class('webchat', '/api/v1/pipelines/<pipeline_uuid>/chat')
class WebChatDebugRouterGroup(group.RouterGroup):
    async def initialize(self) -> None:
        @self.route('/send', methods=['POST'])
        async def send_message(pipeline_uuid: str) -> str:
            """Send a message to the pipeline for debugging"""
            try:
                data = await quart.request.get_json()
                session_type = data.get('session_type', 'person')
                message_chain_obj = data.get('message', [])

                if not message_chain_obj:
                    return self.http_status(400, -1, 'message is required')

                if session_type not in ['person', 'group']:
                    return self.http_status(400, -1, 'session_type must be person or group')

                webchat_adapter = self.ap.platform_mgr.webchat_proxy_bot.adapter

                if not webchat_adapter:
                    return self.http_status(404, -1, 'WebChat adapter not found')

                result = await webchat_adapter.send_webchat_message(pipeline_uuid, session_type, message_chain_obj)

                return self.success(
                    data={
                        'message': result,
                    }
                )

            except Exception as e:
                return self.http_status(500, -1, f'Internal server error: {str(e)}')

        @self.route('/messages/<session_type>', methods=['GET'])
        async def get_messages(pipeline_uuid: str, session_type: str) -> str:
            """Get the message history of the pipeline for debugging"""
            try:
                if session_type not in ['person', 'group']:
                    return self.http_status(400, -1, 'session_type must be person or group')

                webchat_adapter = self.ap.platform_mgr.webchat_proxy_bot.adapter

                if not webchat_adapter:
                    return self.http_status(404, -1, 'WebChat adapter not found')

                messages = webchat_adapter.get_webchat_messages(pipeline_uuid, session_type)

                return self.success(data={'messages': messages})

            except Exception as e:
                return self.http_status(500, -1, f'Internal server error: {str(e)}')

        @self.route('/reset/<session_type>', methods=['POST'])
        async def reset_session(session_type: str) -> str:
            """Reset the debug session"""
            try:
                if session_type not in ['person', 'group']:
                    return self.http_status(400, -1, 'session_type must be person or group')

                webchat_adapter = None
                for bot in self.ap.platform_mgr.bots:
                    if hasattr(bot.adapter, '__class__') and bot.adapter.__class__.__name__ == 'WebChatAdapter':
                        webchat_adapter = bot.adapter
                        break

                if not webchat_adapter:
                    return self.http_status(404, -1, 'WebChat adapter not found')

                webchat_adapter.reset_debug_session(session_type)

                return self.success(data={'message': 'Session reset successfully'})

            except Exception as e:
                return self.http_status(500, -1, f'Internal server error: {str(e)}')