jarvis-ai-assistant 0.7.0__py3-none-any.whl → 0.7.8__py3-none-any.whl

jarvis/jarvis_c2rust/collector.py

@@ -1,27 +1,25 @@
 # -*- coding: utf-8 -*-
 """
-Lightweight function-name collector for header files using libclang.
+使用libclang收集头文件中函数名的轻量级工具。
 
-Purpose:
-- Given one or more C/C++ header files (.h/.hh/.hpp/.hxx), parse each file with libclang
-  and collect function names.
-- Prefer qualified names when available; fall back to unqualified names.
-- Write unique names (de-duplicated, preserving first-seen order) to the specified output file (one per line).
+用途:
+- 给定一个或多个C/C++头文件(.h/.hh/.hpp/.hxx)，使用libclang解析每个文件并收集函数名
+- 优先使用限定名称(qualified names)，否则回退到非限定名称
+- 将唯一名称(去重并保留首次出现顺序)写入指定输出文件(每行一个)
 
-Design:
-- Reuse scanner utilities:
+设计:
+- 复用扫描工具:
   - _try_import_libclang()
   - find_compile_commands()
   - load_compile_commands()
-  - scan_file()  (note: scan_file collects only definitions; inline headers are often definitions)
-- If compile_commands.json exists, use its args; otherwise, fall back to minimal include of file.parent.
-- For header parsing, ensure a language flag (-x c-header / -x c++-header) is present if args do not specify one.
+  - scan_file() (注意: scan_file只收集定义；内联头文件通常是定义)
+- 如果存在compile_commands.json，使用其参数；否则回退到包含文件父目录的最小参数集
+- 对于头文件解析，确保语言标志(-x c-header / -x c++-header)存在，如果参数中未指定
 
-Notes:
-- This module focuses on correctness/robustness over performance.
-- It does not attempt to discover transitive includes; it only assists the parser by adding -I <file.parent>.
+注意事项:
+- 本模块注重正确性/鲁棒性而非性能
+- 不尝试发现传递包含；仅通过添加-I <file.parent>来辅助解析器
 """
-
 from __future__ import annotations
 
 from pathlib import Path
@@ -41,27 +39,26 @@ HEADER_EXTS = {".h", ".hh", ".hpp", ".hxx"}
 
 def _guess_lang_header_flag(file: Path) -> List[str]:
     """
-    Guess appropriate -x language flag for header if not specified in compile args.
+    如果编译参数中未指定，则猜测头文件合适的-x语言标志
     """
     ext = file.suffix.lower()
     if ext in {".hh", ".hpp", ".hxx"}:
         return ["-x", "c++-header"]
-    # Default to C header for .h (conservative)
+    # 对于.h文件默认使用C头文件(保守选择)
     return ["-x", "c-header"]
 
-
 def _ensure_parse_args_for_header(file: Path, base_args: Optional[List[str]]) -> List[str]:
     """
-    Ensure minimal args for header parsing:
-    - include file.parent via -I
-    - add a language flag -x c-header/c++-header if none exists
+    确保头文件解析所需的最小参数集:
+    - 通过-I包含file.parent目录
+    - 如果没有语言标志则添加-x c-header/c++-header
     """
     args = list(base_args or [])
-    # Detect if a language flag already exists (-x <lang>)
+    # 检测是否已存在语言标志(-x <lang>)
     has_lang = False
     for i, a in enumerate(args):
         if a == "-x":
-            # If '-x' present and followed by a value, treat as language specified
+            # 如果存在'-x'且后面有值，则认为已指定语言
             if i + 1 < len(args):
                 has_lang = True
                 break
@@ -72,7 +69,7 @@ def _ensure_parse_args_for_header(file: Path, base_args: Optional[List[str]]) ->
     if not has_lang:
         args.extend(_guess_lang_header_flag(file))
 
-    # Ensure -I <file.parent> is present
+    # 确保存在-I <file.parent>
     inc_dir = str(file.parent)
     has_inc = False
     i = 0
@@ -85,7 +82,7 @@ def _ensure_parse_args_for_header(file: Path, base_args: Optional[List[str]]) ->
             i += 2
             continue
         elif a.startswith("-I"):
-            # Could be like -I/path
+            # 可能是-I/path格式
             if a[2:] == inc_dir:
                 has_inc = True
                 break
@@ -95,11 +92,10 @@ def _ensure_parse_args_for_header(file: Path, base_args: Optional[List[str]]) ->
 
     return args
 
-
 def _collect_decl_function_names(cindex, file: Path, args: List[str]) -> List[str]:
     """
-    Fallback for headers without inline definitions:
-    collect function declarations (prototypes/methods) defined in this header file.
+    对于没有内联定义的头文件的回退方案:
+    收集此头文件中定义的函数声明(原型/方法)
     """
     try:
         index = cindex.Index.create()
@@ -137,25 +133,24 @@ def _collect_decl_function_names(cindex, file: Path, args: List[str]) -> List[st
         pass
     return names
 
-
 def collect_function_names(
     files: List[Path],
     out_path: Path,
     compile_commands_root: Optional[Path] = None,
 ) -> Path:
     """
-    Collect function names from given header files and write unique names to out_path.
+    从给定的头文件中收集函数名并将唯一名称写入out_path
 
-    Parameters:
-    - files: list of header file paths (.h/.hh/.hpp/.hxx). Non-header files will be skipped.
-    - out_path: output file path. Will be created (parents too) and overwritten.
-    - compile_commands_root: optional root directory to search for compile_commands.json.
-      If not provided, we search upward from each file's directory.
+    参数:
+    - files: 头文件路径列表(.h/.hh/.hpp/.hxx)。非头文件将被跳过
+    - out_path: 输出文件路径。将被创建(包括父目录)并覆盖
+    - compile_commands_root: 可选，搜索compile_commands.json的根目录
+      如果未提供，则从每个文件的目录向上搜索
 
-    Returns:
-    - Path to the written out_path.
+    返回值:
+    - 写入的out_path路径
     """
-    # Normalize and filter header files
+    # 标准化和过滤头文件
     hdrs: List[Path] = []
     for p in files or []:
         try:
@@ -166,30 +161,30 @@ def collect_function_names(
             hdrs.append(fp)
 
     if not hdrs:
-        raise ValueError("No valid header files (.h/.hh/.hpp/.hxx) were provided.")
+        raise ValueError("未提供有效的头文件(.h/.hh/.hpp/.hxx)")
 
-    # Prepare libclang
+    # 准备libclang
     cindex = _try_import_libclang()
     if cindex is None:
         from clang import cindex as _ci  # type: ignore
         cindex = _ci
 
-    # Prepare compile_commands args map (either once for provided root, or per-file if None)
+    # 准备compile_commands参数映射(如果提供了根目录则全局一次，否则每个文件单独处理)
     cc_args_map_global: Optional[Dict[str, List[str]]] = None
     if compile_commands_root is not None:
         cc_file = find_compile_commands(Path(compile_commands_root))
         if cc_file:
             cc_args_map_global = load_compile_commands(cc_file)
 
-    # Collect names (preserving order)
+    # 收集名称(保持顺序)
     seen = set()
     ordered_names: List[str] = []
 
     for hf in hdrs:
-        # Determine args for this file
+        # 确定此文件的参数
         cc_args_map = cc_args_map_global
         if cc_args_map is None:
-            # Try to find compile_commands.json near this file
+            # 尝试在此文件附近查找compile_commands.json
             cc_file_local = find_compile_commands(hf.parent)
             if cc_file_local:
                 try:
@@ -205,7 +200,7 @@ def collect_function_names(
 
         args = _ensure_parse_args_for_header(hf, base_args)
 
-        # Attempt to scan. If error, try a fallback with minimal args.
+        # 尝试扫描。如果出错，尝试使用最小参数集的回退方案
         try:
             funcs = scan_file(cindex, hf, args)
         except Exception:
@@ -214,7 +209,7 @@ def collect_function_names(
             except Exception:
                 funcs = []
 
-        # Extract preferred name (qualified_name or name) from definitions
+        # 从定义中提取首选名称(qualified_name或name)
         added_count = 0
         for fn in funcs:
             name = ""
@@ -231,7 +226,7 @@ def collect_function_names(
             ordered_names.append(name)
             added_count += 1
 
-        # Fallback: if no definitions found in this header, collect declarations
+        # 回退: 如果在此头文件中未找到定义，则收集声明
         if not funcs or added_count == 0:
             try:
                 decl_names = _collect_decl_function_names(cindex, hf, args)
@@ -244,15 +239,12 @@ def collect_function_names(
                 seen.add(name)
                 ordered_names.append(name)
 
-    # Write out file (one per line)
+    # 写出文件(每行一个)
     out_path = Path(out_path)
     out_path.parent.mkdir(parents=True, exist_ok=True)
     try:
         out_path.write_text("\n".join(ordered_names) + ("\n" if ordered_names else ""), encoding="utf-8")
     except Exception as e:
-        raise RuntimeError(f"Failed to write output file: {out_path}: {e}")
-
-    return out_path
-
+        raise RuntimeError(f"写入输出文件失败: {out_path}: {e}")
 
-__all__ = ["collect_function_names"]
+    return out_path

jarvis/jarvis_c2rust/constants.py

@@ -0,0 +1,26 @@
+# -*- coding: utf-8 -*-
+"""
+C2Rust 转译器常量定义
+"""
+
+# 数据文件常量
+C2RUST_DIRNAME = ".jarvis/c2rust"
+
+SYMBOLS_JSONL = "symbols.jsonl"
+ORDER_JSONL = "translation_order.jsonl"
+PROGRESS_JSON = "progress.json"
+CONFIG_JSON = "config.json"
+SYMBOL_MAP_JSONL = "symbol_map.jsonl"
+
+# 配置常量
+ERROR_SUMMARY_MAX_LENGTH = 2000  # 错误信息摘要最大长度
+DEFAULT_PLAN_MAX_RETRIES = 0  # 规划阶段默认最大重试次数（0表示无限重试）
+DEFAULT_REVIEW_MAX_ITERATIONS = 0  # 审查阶段最大迭代次数（0表示无限重试）
+DEFAULT_CHECK_MAX_RETRIES = 0  # cargo check 阶段默认最大重试次数（0表示无限重试）
+DEFAULT_TEST_MAX_RETRIES = 0  # cargo test 阶段默认最大重试次数（0表示无限重试）
+
+# 回退与重试常量
+CONSECUTIVE_FIX_FAILURE_THRESHOLD = 10  # 连续修复失败次数阈值，达到此值将触发回退
+MAX_FUNCTION_RETRIES = 10  # 函数重新开始处理的最大次数
+DEFAULT_PLAN_MAX_RETRIES_ENTRY = 5  # run_transpile 入口函数的 plan_max_retries 默认值
+