mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
268 lines
9.7 KiB
Python
268 lines
9.7 KiB
Python
|
from collections.abc import Callable
|
|||
|
|
import inspect
|
|||
|
|
from pathlib import Path
|
|||
|
|
from typing import Any
|
|||
|
|
|
|||
|
|
import aiofiles
|
|||
|
|
from jinja2 import (
|
|||
|
|
ChoiceLoader,
|
|||
|
|
Environment,
|
|||
|
|
FileSystemLoader,
|
|||
|
|
PrefixLoader,
|
|||
|
|
TemplateNotFound,
|
|||
|
|
select_autoescape,
|
|||
|
|
)
|
|||
|
|
import markdown
|
|||
|
|
from pydantic import BaseModel
|
|||
|
|
import ujson as json
|
|||
|
|
|
|||
|
|
from zhenxun.configs.path_config import THEMES_PATH
|
|||
|
|
from zhenxun.services.log import logger
|
|||
|
|
from zhenxun.services.renderer.models import TemplateManifest
|
|||
|
|
from zhenxun.services.renderer.protocols import Renderable
|
|||
|
|
from zhenxun.utils.exception import RenderingError
|
|||
|
|
from zhenxun.utils.pydantic_compat import model_dump
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Theme(BaseModel):
|
|||
|
|
name: str
|
|||
|
|
palette: dict[str, Any]
|
|||
|
|
style_css: str = ""
|
|||
|
|
assets_dir: Path
|
|||
|
|
default_assets_dir: Path
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ThemeManager:
|
|||
|
|
def __init__(
|
|||
|
|
self,
|
|||
|
|
plugin_template_paths: dict[str, Path],
|
|||
|
|
custom_filters: dict[str, Callable],
|
|||
|
|
custom_globals: dict[str, Callable],
|
|||
|
|
markdown_styles: dict[str, Path],
|
|||
|
|
):
|
|||
|
|
prefix_loader = PrefixLoader(
|
|||
|
|
{
|
|||
|
|
namespace: FileSystemLoader(str(path.absolute()))
|
|||
|
|
for namespace, path in plugin_template_paths.items()
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
theme_loader = FileSystemLoader(
|
|||
|
|
[
|
|||
|
|
str(THEMES_PATH / "current_theme_placeholder" / "templates"),
|
|||
|
|
str(THEMES_PATH / "default" / "templates"),
|
|||
|
|
]
|
|||
|
|
)
|
|||
|
|
final_loader = ChoiceLoader([prefix_loader, theme_loader])
|
|||
|
|
|
|||
|
|
self.jinja_env = Environment(
|
|||
|
|
loader=final_loader,
|
|||
|
|
enable_async=True,
|
|||
|
|
autoescape=select_autoescape(["html", "xml"]),
|
|||
|
|
)
|
|||
|
|
self.current_theme: Theme | None = None
|
|||
|
|
self._custom_filters = custom_filters
|
|||
|
|
self._custom_globals = custom_globals
|
|||
|
|
self._markdown_styles = markdown_styles
|
|||
|
|
|
|||
|
|
self.jinja_env.globals["resolve_template"] = self._resolve_component_template
|
|||
|
|
|
|||
|
|
self.jinja_env.filters["md"] = self._markdown_filter
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def _markdown_filter(text: str) -> str:
|
|||
|
|
"""一个将 Markdown 文本转换为 HTML 的 Jinja2 过滤器。"""
|
|||
|
|
if not isinstance(text, str):
|
|||
|
|
return ""
|
|||
|
|
return markdown.markdown(
|
|||
|
|
text,
|
|||
|
|
extensions=[
|
|||
|
|
"pymdownx.tasklist",
|
|||
|
|
"tables",
|
|||
|
|
"fenced_code",
|
|||
|
|
"codehilite",
|
|||
|
|
"mdx_math",
|
|||
|
|
"pymdownx.tilde",
|
|||
|
|
],
|
|||
|
|
extension_configs={"mdx_math": {"enable_dollar_delimiter": True}},
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
async def load_theme(self, theme_name: str = "default"):
|
|||
|
|
theme_dir = THEMES_PATH / theme_name
|
|||
|
|
if not theme_dir.is_dir():
|
|||
|
|
logger.error(f"主题 '{theme_name}' 不存在，将回退到默认主题。")
|
|||
|
|
if theme_name == "default":
|
|||
|
|
raise FileNotFoundError("默认主题 'default' 未找到！")
|
|||
|
|
theme_name = "default"
|
|||
|
|
theme_dir = THEMES_PATH / "default"
|
|||
|
|
|
|||
|
|
if self.jinja_env.loader and isinstance(self.jinja_env.loader, ChoiceLoader):
|
|||
|
|
current_loaders = list(self.jinja_env.loader.loaders)
|
|||
|
|
if len(current_loaders) > 1:
|
|||
|
|
current_loaders[1] = FileSystemLoader(
|
|||
|
|
[
|
|||
|
|
str(theme_dir / "templates"),
|
|||
|
|
str(THEMES_PATH / "default" / "templates"),
|
|||
|
|
]
|
|||
|
|
)
|
|||
|
|
self.jinja_env.loader = ChoiceLoader(current_loaders)
|
|||
|
|
else:
|
|||
|
|
logger.error("Jinja2 loader 不是 ChoiceLoader 或未设置，无法更新主题路径。")
|
|||
|
|
|
|||
|
|
palette_path = theme_dir / "palette.json"
|
|||
|
|
palette = (
|
|||
|
|
json.loads(palette_path.read_text("utf-8")) if palette_path.exists() else {}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
self.current_theme = Theme(
|
|||
|
|
name=theme_name,
|
|||
|
|
palette=palette,
|
|||
|
|
assets_dir=theme_dir / "assets",
|
|||
|
|
default_assets_dir=THEMES_PATH / "default" / "assets",
|
|||
|
|
)
|
|||
|
|
theme_context_dict = {
|
|||
|
|
"name": theme_name,
|
|||
|
|
"palette": palette,
|
|||
|
|
"assets_dir": theme_dir / "assets",
|
|||
|
|
"default_assets_dir": THEMES_PATH / "default" / "assets",
|
|||
|
|
}
|
|||
|
|
self.jinja_env.globals["theme"] = theme_context_dict
|
|||
|
|
logger.info(f"主题管理器已加载主题: {theme_name}")
|
|||
|
|
|
|||
|
|
async def _resolve_component_template(self, component_path: str) -> str:
|
|||
|
|
"""
|
|||
|
|
智能解析组件路径。
|
|||
|
|
如果路径是目录，则查找 manifest.json 以获取入口点。
|
|||
|
|
"""
|
|||
|
|
if Path(component_path).suffix:
|
|||
|
|
return component_path
|
|||
|
|
|
|||
|
|
manifest_path_str = f"{component_path}/manifest.json"
|
|||
|
|
|
|||
|
|
if not self.jinja_env.loader:
|
|||
|
|
raise TemplateNotFound(
|
|||
|
|
f"Jinja2 loader 未配置。无法查找 '{manifest_path_str}'"
|
|||
|
|
)
|
|||
|
|
try:
|
|||
|
|
_, full_path, _ = self.jinja_env.loader.get_source(
|
|||
|
|
self.jinja_env, manifest_path_str
|
|||
|
|
)
|
|||
|
|
if full_path and Path(full_path).exists():
|
|||
|
|
async with aiofiles.open(full_path, encoding="utf-8") as f:
|
|||
|
|
manifest_data = json.loads(await f.read())
|
|||
|
|
entrypoint = manifest_data.get("entrypoint")
|
|||
|
|
if not entrypoint:
|
|||
|
|
raise RenderingError(
|
|||
|
|
f"组件 '{component_path}' 的 manifest.json 中缺少 "
|
|||
|
|
f"'entrypoint' 键。"
|
|||
|
|
)
|
|||
|
|
return f"{component_path}/{entrypoint}"
|
|||
|
|
except TemplateNotFound:
|
|||
|
|
logger.debug(
|
|||
|
|
f"未找到 '{manifest_path_str}'，将回退到默认的 'main.html' 入口点。"
|
|||
|
|
)
|
|||
|
|
return f"{component_path}/main.html"
|
|||
|
|
raise TemplateNotFound(f"无法为组件 '{component_path}' 找到模板入口点。")
|
|||
|
|
|
|||
|
|
async def get_template_manifest(
|
|||
|
|
self, component_path: str
|
|||
|
|
) -> TemplateManifest | None:
|
|||
|
|
"""
|
|||
|
|
查找并解析组件的 manifest.json 文件。
|
|||
|
|
"""
|
|||
|
|
manifest_path_str = f"{component_path}/manifest.json"
|
|||
|
|
|
|||
|
|
if not self.jinja_env.loader:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
_, full_path, _ = self.jinja_env.loader.get_source(
|
|||
|
|
self.jinja_env, manifest_path_str
|
|||
|
|
)
|
|||
|
|
if full_path and Path(full_path).exists():
|
|||
|
|
async with aiofiles.open(full_path, encoding="utf-8") as f:
|
|||
|
|
manifest_data = json.loads(await f.read())
|
|||
|
|
return TemplateManifest(**manifest_data)
|
|||
|
|
except TemplateNotFound:
|
|||
|
|
return None
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def _resolve_markdown_style_path(self, style_name: str) -> Path | None:
|
|||
|
|
"""
|
|||
|
|
按照 注册 -> 主题约定 -> 默认约定 的顺序解析 Markdown 样式路径。
|
|||
|
|
"""
|
|||
|
|
if style_name in self._markdown_styles:
|
|||
|
|
logger.debug(f"找到已注册的 Markdown 样式: '{style_name}'")
|
|||
|
|
return self._markdown_styles[style_name]
|
|||
|
|
|
|||
|
|
logger.warning(f"样式 '{style_name}' 在注册表中未找到。")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
async def _render_component_to_html(
|
|||
|
|
self,
|
|||
|
|
component: Renderable,
|
|||
|
|
required_scripts: list[str] | None = None,
|
|||
|
|
required_styles: list[str] | None = None,
|
|||
|
|
**kwargs,
|
|||
|
|
) -> str:
|
|||
|
|
"""将 Renderable 组件渲染成 HTML 字符串，并处理异步数据。"""
|
|||
|
|
if not self.current_theme:
|
|||
|
|
await self.load_theme()
|
|||
|
|
|
|||
|
|
assert self.current_theme is not None, "主题加载失败"
|
|||
|
|
|
|||
|
|
data_dict = component.get_render_data()
|
|||
|
|
|
|||
|
|
custom_style_css = ""
|
|||
|
|
if hasattr(component, "get_extra_css"):
|
|||
|
|
css_result = component.get_extra_css(self)
|
|||
|
|
if inspect.isawaitable(css_result):
|
|||
|
|
custom_style_css = await css_result
|
|||
|
|
else:
|
|||
|
|
custom_style_css = css_result
|
|||
|
|
|
|||
|
|
def asset_loader(asset_path: str) -> str:
|
|||
|
|
"""[新增] 用于在Jinja2模板中解析静态资源的辅助函数。"""
|
|||
|
|
assert self.current_theme is not None
|
|||
|
|
current_theme_asset = self.current_theme.assets_dir / asset_path
|
|||
|
|
if current_theme_asset.exists():
|
|||
|
|
return current_theme_asset.relative_to(THEMES_PATH.parent).as_posix()
|
|||
|
|
|
|||
|
|
default_theme_asset = self.current_theme.default_assets_dir / asset_path
|
|||
|
|
if default_theme_asset.exists():
|
|||
|
|
return default_theme_asset.relative_to(THEMES_PATH.parent).as_posix()
|
|||
|
|
|
|||
|
|
logger.warning(
|
|||
|
|
f"资源文件在主题 '{self.current_theme.name}' 和 'default' 中均未找到: "
|
|||
|
|
f"{asset_path}"
|
|||
|
|
)
|
|||
|
|
return ""
|
|||
|
|
|
|||
|
|
theme_context_dict = model_dump(self.current_theme)
|
|||
|
|
theme_context_dict["asset"] = asset_loader
|
|||
|
|
|
|||
|
|
resolved_template_name = await self._resolve_component_template(
|
|||
|
|
str(component.template_name)
|
|||
|
|
)
|
|||
|
|
logger.debug(
|
|||
|
|
f"正在渲染组件 '{component.template_name}' "
|
|||
|
|
f"(主题: {self.current_theme.name})，解析模板: '{resolved_template_name}'",
|
|||
|
|
"RendererService",
|
|||
|
|
)
|
|||
|
|
if self._custom_filters:
|
|||
|
|
self.jinja_env.filters.update(self._custom_filters)
|
|||
|
|
if self._custom_globals:
|
|||
|
|
self.jinja_env.globals.update(self._custom_globals)
|
|||
|
|
template = self.jinja_env.get_template(resolved_template_name)
|
|||
|
|
|
|||
|
|
template_context = {
|
|||
|
|
"data": data_dict,
|
|||
|
|
"theme": theme_context_dict,
|
|||
|
|
"theme_css": "",
|
|||
|
|
"custom_style_css": custom_style_css,
|
|||
|
|
"required_scripts": required_scripts or [],
|
|||
|
|
"required_styles": required_styles or [],
|
|||
|
|
}
|
|||
|
|
template_context.update(kwargs)
|
|||
|
|
|
|||
|
|
return await template.render_async(**template_context)
|