120 lines
4.7 KiB
Python
120 lines
4.7 KiB
Python
"""HTML 渲染服务 - 通过 config.json 配置，统一文件入口"""
|
||
import json
|
||
import sys
|
||
from pathlib import Path
|
||
from oss.plugin.types import Plugin, register_plugin_type, Response
|
||
|
||
|
||
class _Log:
|
||
_TTY = sys.stdout.isatty()
|
||
_C = {"reset": "\033[0m", "white": "\033[0;37m", "yellow": "\033[1;33m", "blue": "\033[1;34m", "red": "\033[1;31m"}
|
||
@classmethod
|
||
def _c(cls, t, c):
|
||
return f"{cls._C.get(c,'')}{t}{cls._C['reset']}" if cls._TTY else t
|
||
@classmethod
|
||
def info(cls, m): print(f"{cls._c('[html-render]', 'white')} {cls._c(m, 'white')}")
|
||
@classmethod
|
||
def warn(cls, m): print(f"{cls._c('[html-render]', 'yellow')} {cls._c('⚠', 'yellow')} {cls._c(m, 'yellow')}")
|
||
@classmethod
|
||
def error(cls, m): print(f"{cls._c('[html-render]', 'red')} {cls._c('✗', 'red')} {cls._c(m, 'red')}")
|
||
|
||
|
||
class HtmlRenderPlugin(Plugin):
|
||
"""HTML 渲染插件 - 渲染服务由 html-render 提供"""
|
||
|
||
def __init__(self):
|
||
self.http_api = None
|
||
self.storage = None # plugin-storage 入口
|
||
self.config = {}
|
||
self.root_dir = None # 解析后的网站根目录
|
||
|
||
def init(self, deps: dict = None):
|
||
"""初始化 - 读取 config.json 并解析网站根目录"""
|
||
self._load_config()
|
||
_Log.info(f"配置加载完成: root_dir={self.root_dir}")
|
||
|
||
def start(self):
|
||
"""启动 - 注册路由到 http-api，共享配置给 web-toolkit"""
|
||
# 注册首页路由
|
||
if self.http_api and hasattr(self.http_api, 'router'):
|
||
self.http_api.router.get("/", self._serve_html)
|
||
_Log.info("已注册路由到 http-api")
|
||
else:
|
||
_Log.warn("http-api 未加载")
|
||
|
||
# 将配置共享给 web-toolkit（通过 plugin-storage 的 DCIM 共享存储）
|
||
if self.storage:
|
||
shared = self.storage.get_shared()
|
||
shared.set_shared("html-render-config", {
|
||
"root_dir": str(self.root_dir),
|
||
"index_file": self.config.get("index_file", "index.html"),
|
||
"static_prefix": self.config.get("static_prefix", "/static"),
|
||
})
|
||
_Log.info("配置已共享到 DCIM")
|
||
|
||
def stop(self):
|
||
"""停止"""
|
||
pass
|
||
|
||
def set_http_api(self, instance):
|
||
"""设置 http-api 实例"""
|
||
self.http_api = instance
|
||
|
||
def set_plugin_storage(self, instance):
|
||
"""设置 plugin-storage 实例（唯一文件读写入口）"""
|
||
self.storage = instance
|
||
|
||
def _load_config(self):
|
||
"""读取 config.json，解析根目录"""
|
||
config_path = Path("./data/html-render/config.json")
|
||
if not config_path.exists():
|
||
_Log.warn("config.json 不存在，使用默认配置")
|
||
self.config = {"root_dir": "../website", "index_file": "index.html"}
|
||
else:
|
||
with open(config_path, "r", encoding="utf-8") as f:
|
||
self.config = json.load(f)
|
||
|
||
# 解析根目录（相对于 config.json 的路径）
|
||
root_relative = self.config.get("root_dir", "../website")
|
||
self.root_dir = (config_path.parent / root_relative).resolve()
|
||
|
||
def _serve_html(self, request):
|
||
"""提供 HTML 页面 - 通过 plugin-storage 读取并注入静态资源路径"""
|
||
index_file = self.config.get("index_file", "index.html")
|
||
|
||
# 安全检查：防止路径穿越
|
||
if ".." in index_file or index_file.startswith("/"):
|
||
return Response(status=403, body="Forbidden")
|
||
|
||
if self.storage:
|
||
storage = self.storage.get_storage("html-render")
|
||
if storage.file_exists(index_file):
|
||
content = storage.read_file(index_file)
|
||
if content:
|
||
# 注入静态资源路径（相对路径 → /website/ 前缀）
|
||
content = self._inject_static_paths(content)
|
||
return Response(
|
||
status=200,
|
||
headers={"Content-Type": "text/html; charset=utf-8"},
|
||
body=content
|
||
)
|
||
return Response(status=404, body="Not Found")
|
||
|
||
def _inject_static_paths(self, html: str) -> str:
|
||
"""将相对静态资源路径替换为 /website/ 前缀"""
|
||
import re
|
||
# href="css/xxx" → href="/website/css/xxx"
|
||
html = re.sub(r'(href\s*=\s*["\'])css/', r'\1/website/css/', html)
|
||
# src="js/xxx" → src="/website/js/xxx"
|
||
html = re.sub(r'(src\s*=\s*["\'])js/', r'\1/website/js/', html)
|
||
# src="logo.svg" → src="/website/logo.svg"
|
||
html = re.sub(r'(src\s*=\s*["\'])(?!https?://|/)([\w.-]+\.(svg|png|jpg|gif|ico|webp))', r'\1/website/\2', html)
|
||
return html
|
||
|
||
|
||
register_plugin_type("HtmlRenderPlugin", HtmlRenderPlugin)
|
||
|
||
|
||
def New():
|
||
return HtmlRenderPlugin()
|