修复了一些错误 更新了AI.md(给ai看的)
This commit is contained in:
@@ -1,48 +1,15 @@
|
||||
"""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 # 解析后的网站根目录
|
||||
|
||||
self.storage = None 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", {
|
||||
@@ -53,19 +20,9 @@ class HtmlRenderPlugin(Plugin):
|
||||
_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 不存在,使用默认配置")
|
||||
@@ -74,40 +31,13 @@ class HtmlRenderPlugin(Plugin):
|
||||
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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user