# myapp/sockets/namespaces.py
import json
from flask import current_app, request, session
from flask_socketio import Namespace, join_room, leave_room, emit
from ..services.memory_service import save_chapter_memory_by_uuid
from ..services.backboard_service import realtime_response
class VSCodeNamespace(Namespace):
    def on_login(self,data):        
        ex = current_app.extensions
        username2uuid    = ex["username2uuid"]
        backboard_manager = ex["backboard_manager"]
        print("VSCode client connected")
        print(data)
        dataconfig = data['config']
        user_id = dataconfig.get('user_id')
        path = dataconfig.get('path')
        course_id = dataconfig.get('course_id')
        lesson_id = dataconfig.get('chapter_id')
        print(f"User {user_id} connected with path: {path}")
        useruuid = username2uuid[user_id]
        join_room(useruuid, namespace='/vscode')
        if backboard_manager.get_backboard(useruuid) == None:
            backboard_manager.add_backboard(useruuid,user_id, course_id, lesson_id, path)


    def on_message(self, data):
        print(f"Received from VSCode client: {data}")
        dataconfig = data['config']
        realtime_response(dataconfig, data)
        # emit('response', {'message': 'Config data received and connection established'})
    def on_disconnect(self, data):
        print("VSCode client disconnected")
        print("Disconnect reason:"+str(data))


class AgentNamespace(Namespace):
    def on_login(self, data):
        ex = current_app.extensions
        username2uuid    = ex["username2uuid"]
        backboard_manager = ex["backboard_manager"]
        agent_manager = ex["agent_manager"]
        data = json.loads(data)
        user = data['username']
        course_id = data['course_id']
        chapter_id = data['chapter_id']
        user_uuid = username2uuid[user]
        session['user_id'] = user_uuid
        print(f'User connected with session user_id: {user_uuid}')
        # 从markdown、markdown_prompts、score_prompts中各读取数据folder.md并进行new_agent
        with open(f'books/markdown/{course_id}/{chapter_id}.md','r',encoding='UTF-8') as fmd,\
            open(f'books/markdown_prompts/{course_id}/{chapter_id}.md','r',encoding='UTF-8')as fmdp,\
            open(f'books/score_prompts/{course_id}/{chapter_id}.md','r',encoding='UTF-8') as fsp:
            markdown = fmd.read()
            markdown_prompts = fmdp.read()
            score_prompts = fsp.read()
            user_uuid, agent = agent_manager.new_agent(course_id, chapter_id, markdown, markdown_prompts, score_prompts, id=user_uuid, 
                                                       root_path=f'../../study/{user}/{course_id}/{chapter_id}')
            print(user_uuid)
            join_room(user_uuid, namespace='/agent')  # 将该用户加入以user_id为名的room
        backboard_manager.add_backboard(user_uuid, user, course_id, chapter_id, root_path=f'../../study/{user}/{course_id}/{chapter_id}')

    def on_language(self, language):
        ex = current_app.extensions
        agent_manager = ex["agent_manager"]
        id = session.get('user_id')
        agent_manager.change_language(id, language)

    def on_message(self, data):
        print(f"Message from client: {data}")
        ex = current_app.extensions
        agent_manager = ex["agent_manager"]
        backboard_manager = ex["backboard_manager"]
        id = session.get('user_id')
        if (type(data)==str):
            data = json.loads(data)
        print(id)
        if data['type'] == 'text':
            res = agent_manager.invoke(id, data['data'], backboard_manager.get_backboard(id).get_info_prompt())
            print('=*='*20)
            print(res.content)
            reply = f"{res.content['speak']}"
            with current_app.app_context(): 
                emit('message',reply, room=id, namespace='/agent')
                emit('request_function',res.content['function'], room=id, namespace='/agent')

        if data['type'] == 'function':
            agent_manager.function_call(id, data['data'])

    def on_initiative(self,data):
        print("User active function call")
        ex = current_app.extensions
        agent_manager = ex["agent_manager"]
        backboard_manager = ex["backboard_manager"]
        user_id = session['user_id']
        if data['name'] == 'sample_judge':
            agent_manager.sample_judge(user_id, backboard_manager.get_backboard(user_id))
        if data['name'] == 'judge':
            agent_manager.judge(user_id, backboard_manager.get_backboard(user_id))



    def on_disconnect(self,data):
        print("VSCode client disconnected")
        print("Disconnect reason:"+str(data))