import time
from fastapi import FastAPI
from pydantic import BaseModel
import uvicorn
from multiprocessing import Process, Event, freeze_support, Manager
from loguru import logger

from liveMan import DouyinLiveWebFetcher, DouyinLiveWebReply
from helper import PromptQueue, LiveChatConfig

logger.add('log.log', encoding='utf-8', rotation="500MB")

# FastAPI应用实例
app = FastAPI(title="直播状态控制接口")

# 请求体模型
class SpeakingStatus(BaseModel):
    is_speaking: bool

# 全局变量，用于在FastAPI进程中存储共享字典引用
app.state.shared_config = None

# 接口：更新is_speaking状态
@app.post("/speaking-status", summary="更新说话状态")
def update_speaking_status(status: SpeakingStatus):
    if app.state.shared_config is None:
        return {"error": "服务未初始化"}
    app.state.shared_config['is_speaking'] = status.is_speaking
    # logger.info(f"通过API更新is_speaking状态为: {status.is_speaking}")
    return {
        "message": "状态更新成功",
        "current_status": {
            "is_speaking": app.state.shared_config['is_speaking'],
        }
    }

# 启动FastAPI服务的函数，接收共享字典作为参数
def start_fastapi_server(shared_config):
    # 将共享字典存储到FastAPI应用的状态中
    app.state.shared_config = shared_config
    # 启动UVicorn服务器
    uvicorn.run(app, host="0.0.0.0", port=8000)

def fetch_user_chat_content(ws_open_event, queue):
    fetcher = DouyinLiveWebFetcher(ws_open_event, queue)
    fetcher.start()

def reply_user_chat_content(queue, config):
    reply = DouyinLiveWebReply(queue, config)
    reply()

if __name__ == '__main__':
    freeze_support()

    LiveChatConfig().update_chat_enable_status('启动中')
    with Manager() as manager:
        # 创建共享字典，包含所有需要的键
        shared_config = manager.dict({
            'is_speaking': False
        })
        queue = PromptQueue(10)
        ws_open_event = Event()

        # 创建进程时将共享字典传递给FastAPI服务
        api_process = Process(
            target=start_fastapi_server,
            args=(shared_config,),
            name="FastAPI Server"
        )
        fetch_process = Process(
            target=fetch_user_chat_content,
            args=(ws_open_event, queue)
        )
        reply_process = Process(
            target=reply_user_chat_content,
            args=(queue, shared_config)
        )

        # 启动所有进程
        api_process.start()
        fetch_process.start()
        ws_open_event.wait()
        reply_process.start()

        # 等待所有进程完成
        fetch_process.join()
        reply_process.join()
        api_process.join()