from fastapi import FastAPI
from lightrag.api.routers.document_routes import create_document_routes
from lightrag.lightrag import LightRAG
from unittest.mock import AsyncMock

# Create a simple app with document routes
app = FastAPI()


class DummyRagManager:
    def __init__(self, mapping):
        self.mapping = mapping

    async def get_rag_instance(self, tenant_id, kb_id, user_id=None):
        return self.mapping.get(tenant_id)


mock_rag_instances = {
    "tenant-a": AsyncMock(spec=LightRAG),
    "tenant-b": AsyncMock(spec=LightRAG),
}

dummy_rag = AsyncMock(spec=LightRAG)
dummy_doc_manager = AsyncMock()
rag_manager = DummyRagManager(mock_rag_instances)

doc_router = create_document_routes(
    dummy_rag, dummy_doc_manager, rag_manager=rag_manager
)
app.include_router(doc_router, prefix="/api")

# Print route info
for route in app.routes:
    if (
        hasattr(route, "methods")
        and "POST" in route.methods
        and "/api/documents/text" in route.path
    ):
        print("Path:", route.path)
        print("Name:", route.name)
        print("Endpoint signature:", route.dependant)
        print(
            "Endpoint params:",
            [
                p.name
                for p in route.dependant.path_params
                + route.dependant.query_params
                + route.dependant.header_params
                + route.dependant.cookie_params
                + route.dependant.body_params
                + route.dependant.dependency_params
            ],
        )

print("\nAll routes:")
for r in app.routes:
    print(r.path)