from fastapi import APIRouter
from .routes import presetsRoute, chatsRoute, worldbooksRoute, apiConfigRoute, charactersRoute, chatWsRoute, tokenUsageRoute, imageGalleryRoute, regexRoute, chatSummaryRoute, studioRoute, fictionRoute
from utils.file_utils import get_all_roles_and_chats
from core.config import settings
from pathlib import Path

router = APIRouter()

# 注册子路由（HTTP路由）
router.include_router(presetsRoute.router)
router.include_router(chatsRoute.router)
router.include_router(worldbooksRoute.router)
router.include_router(apiConfigRoute.router)
router.include_router(charactersRoute.router)

# ✅ 注册新增路由
router.include_router(tokenUsageRoute.router)
router.include_router(imageGalleryRoute.router)
router.include_router(regexRoute.router)
router.include_router(chatSummaryRoute.router)
router.include_router(studioRoute.router)
router.include_router(fictionRoute.router)

# ✅ 注册 WebSocket 路由（必须在 HTTP 路由之后，避免路径冲突）
router.include_router(chatWsRoute.router)


# 保留原有的其他路由
@router.get("/tool_bar/get_all_role_and_chat")
def get_all_role_and_chat_endpoint():
	return get_all_roles_and_chats(Path(settings.DATA_PATH))