87 lines
3.7 KiB
Python
87 lines
3.7 KiB
Python
"""读取前端组件目录"""
|
|
|
|
from pathlib import Path
|
|
|
|
from app.schemas.component_resource import (
|
|
ComponentCatalog,
|
|
ComponentQuery,
|
|
ComponentResult,
|
|
)
|
|
|
|
|
|
class ComponentProvider:
|
|
def __init__(self, catalog_path: str | Path | None = None):
|
|
# 保存可选的目录文件路径，未指定时由查询方法使用项目内的默认目录
|
|
self.catalog_path = catalog_path
|
|
|
|
def query(self, query: ComponentQuery) -> ComponentResult:
|
|
# TODO 以后迁移数据库，保持组件查询接口不变
|
|
# 优先使用调用方指定的路径，否则读取当前模块旁的 catalog/components.json
|
|
path = (
|
|
Path(self.catalog_path)
|
|
if self.catalog_path is not None
|
|
else Path(__file__).resolve().parent / "catalog" / "components.json"
|
|
)
|
|
|
|
# 如果传入的是相对路径，以项目根目录为基准，避免受启动时工作目录影响
|
|
if not path.is_absolute():
|
|
path = Path(__file__).resolve().parents[2] / path
|
|
|
|
try:
|
|
# 每次读取当前文件，避免更新目录后仍使用旧缓存
|
|
# utf-8-sig 同时兼容带 BOM 和不带 BOM 的 UTF-8 文件
|
|
content = path.read_text(encoding="utf-8-sig")
|
|
except FileNotFoundError:
|
|
# 如果目录文件不存在，返回未支持状态和原因，由调用方决定失败处理方式
|
|
return ComponentResult(
|
|
status="unsupported",
|
|
message="尚未配置前端组件目录，请先导出真实组件定义",
|
|
)
|
|
|
|
# 文件损坏或定义不合法直接报错，不能伪装成查无结果
|
|
catalog = ComponentCatalog.model_validate_json(content)
|
|
# 如果提供关键词，去除首尾空白并统一大小写，未提供时不启用关键词筛选
|
|
keyword = (
|
|
query.keyword.strip().casefold() if query.keyword is not None else None
|
|
)
|
|
|
|
# 收集同时满足所有已提供条件的组件，未设置任何条件时返回全量目录
|
|
matches = []
|
|
for component in catalog.components:
|
|
# 如果指定目录编号，则跳过编号不完全匹配的组件
|
|
if query.key is not None and component.key != query.key:
|
|
continue
|
|
# 如果指定前端组件类型，则跳过类型不完全匹配的组件
|
|
if query.type is not None and component.type != query.type:
|
|
continue
|
|
# 将编号、类型、名称、说明和标签拼成检索文本，并与关键词统一大小写
|
|
searchable = " ".join(
|
|
[
|
|
component.key,
|
|
component.type,
|
|
component.name,
|
|
component.description,
|
|
*component.tags,
|
|
]
|
|
).casefold()
|
|
# 如果设置了关键词，但检索文本中不包含该关键词，则跳过当前组件
|
|
if keyword is not None and keyword not in searchable:
|
|
continue
|
|
# 所有筛选条件通过，保留完整组件定义供后续规划和动作生成使用
|
|
matches.append(component)
|
|
|
|
# 根据是否有匹配项区分查询成功和未找到，同时保留目录版本、来源及结果说明
|
|
return ComponentResult(
|
|
status="resolved" if matches else "not_found",
|
|
components=matches,
|
|
catalog_version=catalog.version,
|
|
source=catalog.source,
|
|
message="已找到组件定义" if matches else "已查询组件目录，未找到匹配的组件",
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
provider = ComponentProvider()
|
|
result = provider.query(ComponentQuery())
|
|
print(result.model_dump_json(indent=2))
|