from __future__ import annotations
from collections.abc import Sequence
from typing import Any
from litestar.connection import Request
from litestar.openapi.plugins import OpenAPIRenderPlugin
class ScalarRenderPlugin(OpenAPIRenderPlugin):
def __init__(
self,
*,
version: str = "1.19.5",
js_url: str | None = None,
css_url: str | None = None,
path: str | Sequence[str] = "/scalar",
**kwargs: Any,
) -> None:
self.js_url = js_url or f"https://cdn.jsdelivr.net/npm/@scalar/api-reference@{version}"
self.css_url = css_url
super().__init__(path=path, **kwargs)
def render(self, request: Request, openapi_schema: dict[str, Any]) -> bytes:
style_sheet_link = f'' if self.css_url else ""
head = f"""
{openapi_schema["info"]["title"]}
{self.style}
{self.favicon}
{style_sheet_link}
"""
body = f"""
"""
return f"""
{head}
{body}
""".encode()