44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
"""
|
||
Controller 端点解析模块（纯 Python，无需 Java）。
|
||
"""
|
||
|
||
from pathlib import Path
|
||
from typing import Dict, List
|
||
|
||
from models import ApiEndpoint, ApiParameter
|
||
|
||
|
||
def endpoints_to_map(endpoints: List[ApiEndpoint]) -> Dict[str, ApiEndpoint]:
|
||
"""端点列表转字典，key 为 endpoint_key。"""
|
||
return {ep.endpoint_key: ep for ep in endpoints}
|
||
|
||
|
||
def filter_endpoints_by_files(
|
||
endpoints: List[ApiEndpoint], changed_files: List[str]
|
||
) -> List[ApiEndpoint]:
|
||
"""仅保留变更文件中的端点。"""
|
||
if not changed_files:
|
||
return endpoints
|
||
changed_set = {f.replace("\\", "/") for f in changed_files}
|
||
return [ep for ep in endpoints if ep.source_file in changed_set]
|
||
|
||
|
||
def parse_endpoints_from_files(
|
||
repo_root: Path,
|
||
source_subdir: str,
|
||
file_paths: List[str],
|
||
file_contents: Dict[str, str],
|
||
) -> List[ApiEndpoint]:
|
||
"""
|
||
解析指定 Controller 文件，提取接口参数（仅解析传入文件，不全量扫描）。
|
||
|
||
:param repo_root: 仓库根
|
||
:param source_subdir: 源码目录（相对仓库根）
|
||
:param file_paths: 文件路径列表
|
||
:param file_contents: 路径 -> 源码内容
|
||
:return: ApiEndpoint 列表
|
||
"""
|
||
from controller_ast_parser import parse_controller_files
|
||
|
||
return parse_controller_files(repo_root, source_subdir, file_paths, file_contents)
|