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

jarvis/jarvis_code_agent/after_change.py

@@ -0,0 +1,233 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+"""
+文件变更后处理工具配置模块
+
+提供文件变更后处理工具配置和命令生成功能（如格式化、自动修复等）。
+"""
+
+import os
+from typing import Dict
+from typing import List
+from typing import Optional
+from typing import Tuple
+
+import yaml
+
+from jarvis.jarvis_utils.config import get_data_dir
+
+# 文件变更后处理工具命令模板映射（文件扩展名/文件名 -> 命令模板列表）
+# 占位符说明：
+# - {file_path}: 单个文件的完整路径
+# - {file_name}: 文件名（不含路径）
+AFTER_CHANGE_COMMAND_TEMPLATES_BY_FILE: Dict[str, List[str]] = {
+    # Python
+    ".py": ["ruff format {file_path}"],
+    ".pyw": ["ruff format {file_path}"],
+    ".pyi": ["ruff format {file_path}"],
+    ".pyx": ["ruff format {file_path}"],
+    ".pxd": ["ruff format {file_path}"],
+    # JavaScript/TypeScript
+    ".js": ["prettier --write {file_path}"],
+    ".mjs": ["prettier --write {file_path}"],
+    ".cjs": ["prettier --write {file_path}"],
+    ".jsx": ["prettier --write {file_path}"],
+    ".ts": ["prettier --write {file_path}"],
+    ".tsx": ["prettier --write {file_path}"],
+    ".cts": ["prettier --write {file_path}"],
+    ".mts": ["prettier --write {file_path}"],
+    # Rust
+    ".rs": ["rustfmt {file_path}"],
+    # Go
+    ".go": ["gofmt -w {file_path}"],
+    # Java
+    ".java": ["google-java-format -i {file_path}"],
+    # C/C++
+    ".c": ["clang-format -i {file_path}"],
+    ".cpp": ["clang-format -i {file_path}"],
+    ".cc": ["clang-format -i {file_path}"],
+    ".cxx": ["clang-format -i {file_path}"],
+    ".h": ["clang-format -i {file_path}"],
+    ".hpp": ["clang-format -i {file_path}"],
+    ".hxx": ["clang-format -i {file_path}"],
+    ".inl": ["clang-format -i {file_path}"],
+    ".ipp": ["clang-format -i {file_path}"],
+    # HTML/CSS
+    ".html": ["prettier --write {file_path}"],
+    ".htm": ["prettier --write {file_path}"],
+    ".xhtml": ["prettier --write {file_path}"],
+    ".css": ["prettier --write {file_path}"],
+    ".scss": ["prettier --write {file_path}"],
+    ".sass": ["prettier --write {file_path}"],
+    ".less": ["prettier --write {file_path}"],
+    # JSON/YAML
+    ".json": ["prettier --write {file_path}"],
+    ".jsonl": ["prettier --write {file_path}"],
+    ".json5": ["prettier --write {file_path}"],
+    ".yaml": ["prettier --write {file_path}"],
+    ".yml": ["prettier --write {file_path}"],
+    # Markdown
+    ".md": ["prettier --write {file_path}"],
+    ".markdown": ["prettier --write {file_path}"],
+    # SQL
+    ".sql": ["sqlfluff format {file_path}"],
+    # Shell/Bash
+    ".sh": ["shfmt -w {file_path}"],
+    ".bash": ["shfmt -w {file_path}"],
+    # XML
+    ".xml": [
+        "xmllint --format {file_path} > {file_path}.tmp && mv {file_path}.tmp {file_path}"
+    ],
+    ".xsd": [
+        "xmllint --format {file_path} > {file_path}.tmp && mv {file_path}.tmp {file_path}"
+    ],
+    ".dtd": [
+        "xmllint --format {file_path} > {file_path}.tmp && mv {file_path}.tmp {file_path}"
+    ],
+}
+
+
+def load_after_change_tools_config() -> Dict[str, List[str]]:
+    """从yaml文件加载全局文件变更后处理工具配置
+
+    Returns:
+        Dict[str, List[str]]: 文件扩展名/文件名 -> 命令模板列表
+    """
+    config_path = os.path.join(get_data_dir(), "after_change_tools.yaml")
+    if not os.path.exists(config_path):
+        return {}
+
+    with open(config_path, "r") as f:
+        config = yaml.safe_load(f) or {}
+        result = {}
+        for k, v in config.items():
+            k_lower = k.lower()
+            # 支持格式: ["template1", "template2"] 或 [("tool1", "template1"), ("tool2", "template2")]
+            if isinstance(v, list) and v:
+                if isinstance(v[0], str):
+                    # 新格式：直接是命令模板列表
+                    result[k_lower] = v
+                elif isinstance(v[0], (list, tuple)) and len(v[0]) == 2:
+                    # 旧格式：需要提取模板
+                    result[k_lower] = [template for _, template in v]
+        return result
+
+
+def load_project_after_change_tools_config(project_root: str) -> Dict[str, List[str]]:
+    """从项目根目录加载文件变更后处理工具配置
+
+    Args:
+        project_root: 项目根目录
+
+    Returns:
+        Dict[str, List[str]]: 文件扩展名/文件名 -> 命令模板列表
+    """
+    project_config_path = os.path.join(
+        project_root, ".jarvis", "after_change_tools.yaml"
+    )
+    if not os.path.exists(project_config_path):
+        return {}
+
+    with open(project_config_path, "r") as f:
+        config = yaml.safe_load(f) or {}
+        result = {}
+        for k, v in config.items():
+            k_lower = k.lower()
+            # 支持格式: ["template1", "template2"] 或 [("tool1", "template1"), ("tool2", "template2")]
+            if isinstance(v, list) and v:
+                if isinstance(v[0], str):
+                    # 新格式：直接是命令模板列表
+                    result[k_lower] = v
+                elif isinstance(v[0], (list, tuple)) and len(v[0]) == 2:
+                    # 旧格式：需要提取模板
+                    result[k_lower] = [template for _, template in v]
+        return result
+
+
+# 合并默认配置和全局yaml配置（项目级配置在运行时动态加载）
+AFTER_CHANGE_COMMAND_TEMPLATES_BY_FILE.update(load_after_change_tools_config())
+
+
+def _format_after_change_command(
+    template: str,
+    file_path: str,
+    project_root: Optional[str] = None,
+) -> Optional[str]:
+    """
+    格式化命令模板（内部函数）
+
+    Args:
+        template: 命令模板字符串
+        file_path: 文件路径（相对或绝对路径）
+        project_root: 项目根目录（可选，用于处理相对路径）
+
+    Returns:
+        命令字符串，如果无法生成则返回None
+    """
+    # 如果是绝对路径，直接使用；否则转换为绝对路径
+    if os.path.isabs(file_path):
+        abs_file_path = file_path
+    elif project_root:
+        abs_file_path = os.path.join(project_root, file_path)
+    else:
+        abs_file_path = os.path.abspath(file_path)
+
+    # 准备占位符替换字典
+    placeholders = {
+        "file_path": abs_file_path,
+        "file_name": os.path.basename(abs_file_path),
+    }
+
+    # 替换占位符
+    try:
+        command = template.format(**placeholders)
+    except KeyError:
+        # 如果模板中有未定义的占位符，返回None
+        return None
+
+    return command
+
+
+def get_after_change_commands_for_files(
+    files: List[str], project_root: Optional[str] = None
+) -> List[Tuple[str, str]]:
+    """
+    获取多个文件的变更后处理命令列表
+
+    Args:
+        files: 文件路径列表
+        project_root: 项目根目录（可选），如果提供则加载项目级配置
+
+    Returns:
+        [(file_path, command), ...] 格式的命令列表
+    """
+    # 加载项目级配置（如果提供项目根目录）
+    # 项目级配置会覆盖全局配置
+    config = AFTER_CHANGE_COMMAND_TEMPLATES_BY_FILE.copy()
+    if project_root:
+        project_config = load_project_after_change_tools_config(project_root)
+        config.update(project_config)  # 项目配置覆盖全局配置
+
+    commands = []
+
+    for file_path in files:
+        # 从文件扩展名/文件名直接获取命令模板列表
+        filename = os.path.basename(file_path)
+        filename_lower = filename.lower()
+
+        # 优先尝试完整文件名匹配
+        templates = config.get(filename_lower, [])
+
+        # 如果文件名匹配失败，再尝试扩展名匹配
+        if not templates:
+            ext = os.path.splitext(filename)[1]
+            if ext:
+                templates = config.get(ext.lower(), [])
+
+        for template in templates:
+            command = _format_after_change_command(template, file_path, project_root)
+            if command:
+                commands.append((file_path, command))
+
+    return commands

jarvis/jarvis_code_agent/build_validation_config.py

@@ -1,6 +1,3 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-
 """
 构建验证配置管理模块
 
@@ -8,69 +5,80 @@
 """
 
 import os
-import yaml
-from typing import Optional, Dict, Any
 from pathlib import Path
+from typing import Any
+
+from jarvis.jarvis_utils.output import PrettyOutput
+
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
 
+from typing import Dict
+from typing import Optional
+from typing import cast
+
+import yaml
 
 CONFIG_FILE_NAME = "build_validation_config.yaml"
 
 
 class BuildValidationConfig:
     """构建验证配置管理器"""
-    
+
     def __init__(self, project_root: str):
         self.project_root = project_root
         self.config_dir = os.path.join(project_root, ".jarvis")
         self.config_path = os.path.join(self.config_dir, CONFIG_FILE_NAME)
         self._config: Optional[Dict[str, Any]] = None
-    
+
     def _ensure_config_dir(self) -> None:
         """确保配置目录存在"""
         if not os.path.exists(self.config_dir):
             os.makedirs(self.config_dir, exist_ok=True)
-    
+
     def _load_config(self) -> Dict[str, Any]:
         """加载配置文件"""
         if self._config is not None:
             return self._config
-        
+
         if not os.path.exists(self.config_path):
             self._config = {}
             return self._config
-        
+
         try:
             with open(self.config_path, "r", encoding="utf-8") as f:
                 self._config = yaml.safe_load(f) or {}
             return self._config
         except Exception as e:
             # 配置文件损坏时，返回空配置
-            print(f"⚠️ 加载构建验证配置失败: {e}，使用默认配置")
+            PrettyOutput.auto_print(f"⚠️ 加载构建验证配置失败: {e}，使用默认配置")
             self._config = {}
             return self._config
-    
+
     def _save_config(self) -> bool:
         """保存配置文件"""
         try:
             self._ensure_config_dir()
             with open(self.config_path, "w", encoding="utf-8") as f:
-                yaml.safe_dump(self._config, f, allow_unicode=True, default_flow_style=False)
+                yaml.safe_dump(
+                    self._config, f, allow_unicode=True, default_flow_style=False
+                )
             return True
         except Exception as e:
-            print(f"❌ 保存构建验证配置失败: {e}")
+            PrettyOutput.auto_print(f"❌ 保存构建验证配置失败: {e}")
             return False
-    
+
     def is_build_validation_disabled(self) -> bool:
         """检查是否已禁用构建验证"""
         config = self._load_config()
-        return config.get("disable_build_validation", False)
-    
+        return cast(bool, config.get("disable_build_validation", False))
+
     def disable_build_validation(self, reason: Optional[str] = None) -> bool:
         """禁用构建验证
-        
+
         Args:
             reason: 禁用原因（可选）
-        
+
         Returns:
             bool: 是否成功保存配置
         """
@@ -81,7 +89,7 @@ class BuildValidationConfig:
         config["disabled_at"] = str(Path(self.project_root).resolve())
         self._config = config
         return self._save_config()
-    
+
     def enable_build_validation(self) -> bool:
         """重新启用构建验证"""
         config = self._load_config()
@@ -89,39 +97,39 @@ class BuildValidationConfig:
         # 保留历史信息，但清除禁用标志
         self._config = config
         return self._save_config()
-    
+
     def get_disable_reason(self) -> Optional[str]:
         """获取禁用原因"""
         config = self._load_config()
         return config.get("disable_reason")
-    
+
     def has_been_asked(self) -> bool:
         """检查是否已经询问过用户"""
         config = self._load_config()
-        return config.get("has_been_asked", False)
-    
+        return cast(bool, config.get("has_been_asked", False))
+
     def mark_as_asked(self) -> bool:
         """标记为已询问"""
         config = self._load_config()
         config["has_been_asked"] = True
         self._config = config
         return self._save_config()
-    
+
     def get_selected_build_system(self) -> Optional[str]:
         """获取用户选择的构建系统
-        
+
         Returns:
             构建系统名称（如 "rust", "python"），如果未选择则返回None
         """
         config = self._load_config()
         return config.get("selected_build_system")
-    
+
     def set_selected_build_system(self, build_system: str) -> bool:
         """保存用户选择的构建系统
-        
+
         Args:
             build_system: 构建系统名称（如 "rust", "python"）
-        
+
         Returns:
             bool: 是否成功保存配置
         """
@@ -129,4 +137,3 @@ class BuildValidationConfig:
         config["selected_build_system"] = build_system
         self._config = config
         return self._save_config()
-

jarvis/jarvis_code_agent/builtin_rules.py

@@ -0,0 +1,68 @@
+# -*- coding: utf-8 -*-
+"""CodeAgent 内置规则模块
+
+提供一些优秀的开发实践规则，如 TDD、代码审查、重构等。
+"""
+
+from pathlib import Path
+
+# 内置规则字典：规则名称 -> 规则内容
+# 此字典将在模块加载时通过 _load_builtin_rules() 函数动态填充
+BUILTIN_RULES: dict[str, str] = {}
+
+
+def _load_rules_from_directory(directory: Path) -> None:
+    """从指定目录加载规则文件
+
+    参数:
+        directory: 规则文件所在目录
+    """
+    if not directory.exists():
+        return
+
+    for rule_file in directory.glob("*.md"):
+        rule_name = rule_file.stem  # 去掉 .md 后缀
+        try:
+            with open(rule_file, "r", encoding="utf-8") as f:
+                rule_content = f.read().strip()
+                if rule_content:
+                    BUILTIN_RULES[rule_name] = rule_content
+        except Exception:
+            # 忽略加载失败的文件，不影响其他规则
+            continue
+
+
+def _load_builtin_rules() -> None:
+    """加载所有内置规则"""
+    rules_dir = Path(__file__).parent
+
+    # 加载通用规则（rules 目录）
+    _load_rules_from_directory(rules_dir / "rules")
+
+    # 加载测试规则（test_rules 目录）
+    _load_rules_from_directory(rules_dir / "test_rules")
+
+
+# 在模块加载时自动加载所有规则
+_load_builtin_rules()
+
+
+def get_builtin_rule(rule_name: str) -> str | None:
+    """获取内置规则
+
+    参数:
+        rule_name: 规则名称（不区分大小写）
+
+    返回:
+        str: 规则内容，如果未找到则返回 None
+    """
+    return BUILTIN_RULES.get(rule_name.lower())
+
+
+def list_builtin_rules() -> list[str]:
+    """列出所有可用的内置规则名称
+
+    返回:
+        list[str]: 规则名称列表
+    """
+    return list(BUILTIN_RULES.keys())