auto-coder 1.0.0__py3-none-any.whl → 2.0.1__py3-none-any.whl

autocoder/chat/rules_command.py

@@ -9,15 +9,15 @@ from rich.panel import Panel
 from rich.table import Table
 from rich.text import Text
 from rich.markdown import Markdown
-from autocoder.auto_coder_runner import get_memory, save_memory
-from autocoder.common.rulefiles.autocoderrules_utils import AutocoderRulesManager
+from autocoder.common.rulefiles import AutocoderRulesManager, init_rule
 from autocoder.agent.auto_learn import AutoLearn
 from autocoder.common import SourceCode, SourceCodeList
 from autocoder.auto_coder_runner import get_final_config, get_single_llm
 from autocoder.chat_auto_coder_lang import get_message, get_message_with_format
-from autocoder.rag.token_counter import count_tokens
+from autocoder.common.tokens import count_string_tokens as count_tokens
 from autocoder.common.printer import Printer
 from autocoder.common.ac_style_command_parser import CommandParser
+from autocoder.common.core_config import get_memory_manager
 from loguru import logger
 
 printer = Printer()
@@ -69,9 +69,9 @@ def _print_rules_table(rules: Dict[str, str], pattern: str = "*"):
 
 # --- Command Handlers ---
 
-def _handle_list_rules(memory: Dict[str, Any], args: List[str]) -> str:
+def _handle_list_rules(args: List[str]) -> str:
     """Handles listing rules files, supports wildcard filtering."""
-    from autocoder.common.rulefiles.autocoderrules_utils import get_rules
+    from autocoder.common.rulefiles import get_rules
     
     # 获取项目根目录（当前工作目录）
     project_root = os.getcwd()
@@ -124,7 +124,7 @@ def _handle_list_rules(memory: Dict[str, Any], args: List[str]) -> str:
     else:
         return _print_rules_table(rules)
 
-def _handle_remove_rules(memory: Dict[str, Any], args: List[str]) -> str:
+def _handle_remove_rules(args: List[str]) -> str:
     """Handles removing rules files based on glob pattern."""
     if not args:
         message = get_message("rules_remove_param_required")
@@ -191,13 +191,13 @@ def _handle_remove_rules(memory: Dict[str, Any], args: List[str]) -> str:
     printer.print_str_in_terminal(message)
     return message
 
-def _handle_analyze_commit_rules(memory: Dict[str, Any], args: List[str],commit_id: str, coding_func=None) -> str:
+def _handle_analyze_commit_rules(args: List[str], commit_id: str, coding_func=None) -> str:
     """Handles analyzing current files with rules."""
     query = " ".join(args) if args else ""
     
-    args = get_final_config()    
-    llm = get_single_llm(args.model, product_mode=args.product_mode)    
-    auto_learn = AutoLearn(llm=llm, args=args)
+    config = get_final_config()    
+    llm = get_single_llm(config.model, product_mode=config.product_mode)    
+    auto_learn = AutoLearn(llm=llm, args=config)
     changes, _ = auto_learn.get_commit_changes(commit_id)        
     
     try:        
@@ -217,15 +217,17 @@ def _handle_analyze_commit_rules(memory: Dict[str, Any], args: List[str],commit_
         printer.print_str_in_terminal(message)        
         return message
 
-def _handle_analyze_rules(memory: Dict[str, Any], args: List[str], coding_func=None) -> str:
+def _handle_analyze_rules(args: List[str], coding_func=None) -> str:
     """Handles analyzing current files with rules."""
     query = " ".join(args) if args else ""
     
-    args = get_final_config()    
-    llm = get_single_llm(args.model, product_mode=args.product_mode)    
-    auto_learn = AutoLearn(llm=llm, args=args)
+    config = get_final_config()    
+    llm = get_single_llm(config.model, product_mode=config.product_mode)    
+    auto_learn = AutoLearn(llm=llm, args=config)
     
-    files = memory.get("current_files", {}).get("files", [])
+    # Use memory manager to get current files
+    memory_manager = get_memory_manager()
+    files = memory_manager.get_current_files()
     if not files:
         message = get_message("rules_no_active_files")
         printer.print_str_in_terminal(message)
@@ -253,7 +255,7 @@ def _handle_analyze_rules(memory: Dict[str, Any], args: List[str], coding_func=N
         printer.print_str_in_terminal(message)
         return message
 
-def _handle_get_rules(memory: Dict[str, Any], args: List[str]) -> str:
+def _handle_get_rules(args: List[str]) -> str:
     """Handles displaying the content of rules files based on glob pattern."""
     if not args:
         message = get_message("rules_get_param_required")
@@ -263,7 +265,7 @@ def _handle_get_rules(memory: Dict[str, Any], args: List[str]) -> str:
     pattern = args[0]
     
     # 获取规则管理器
-    from autocoder.common.rulefiles.autocoderrules_utils import get_rules
+    from autocoder.common.rulefiles import get_rules
     project_root = os.getcwd()
     rules = get_rules(project_root)
     
@@ -327,7 +329,7 @@ def _handle_get_rules(memory: Dict[str, Any], args: List[str]) -> str:
     # 由于控制台直接打印，返回空字符串
     return ""
 
-def _handle_help(memory: Dict[str, Any], args: List[str]) -> str:
+def _handle_help(args: List[str]) -> str:
     """Provides help text for the /rules command."""
     help_text = get_message("rules_help_text")
     if not help_text:
@@ -339,6 +341,7 @@ def _handle_help(memory: Dict[str, Any], args: List[str]) -> str:
   /rules /remove [通配符]          - 删除规则文件，使用通配符匹配
   /rules /analyze [查询]           - 分析当前文件，可选提供查询内容
   /rules /commit <提交ID> /query <查询> - 分析特定提交，必须提供提交ID和查询内容
+  /rules /init [项目根目录]         - 使用 subagent 机制生成项目初始化规则
   /rules /help                    - 显示此帮助信息
   
 默认情况下，直接使用 /rules [查询] 相当于 /rules /analyze [查询]
@@ -347,7 +350,7 @@ def _handle_help(memory: Dict[str, Any], args: List[str]) -> str:
     printer.print_str_in_terminal(help_text)
     return help_text
 
-def _handle_commit_rules(memory: Dict[str, Any], args: List[str], coding_func=None) -> str:
+def _handle_commit_rules(args: List[str], coding_func=None) -> str:
     """处理 commit 命令，要求格式为 /commit <commit_id> /query <查询内容>"""
     if not args:
         message = get_message("rules_commit_param_required")
@@ -389,7 +392,7 @@ def _handle_commit_rules(memory: Dict[str, Any], args: List[str], coding_func=No
     
     # 实现实际的 commit 分析逻辑...
     try:        
-        _handle_analyze_commit_rules(memory, [query], commit_id=commit_id, coding_func=coding_func)
+        _handle_analyze_commit_rules([query], commit_id=commit_id, coding_func=coding_func)
         message = get_message_with_format("rules_commit_success", commit_id=commit_id, query=query)
         printer.print_str_in_terminal(message)
         return message
@@ -398,24 +401,68 @@ def _handle_commit_rules(memory: Dict[str, Any], args: List[str], coding_func=No
         printer.print_str_in_terminal(message)
         return message
 
-# Command dispatch table
-COMMAND_HANDLERS: Dict[str, Callable[[Dict[str, Any], List[str]], str]] = {
+def _handle_init_rules(args: List[str]) -> str:
+    """处理初始化规则生成命令"""
+    # 获取必要的配置
+    print("not support init rules now")
+    return 
+    config = get_final_config()    
+    llm = get_single_llm(config.model, product_mode=config.product_mode)
+    
+    # 可选的项目根目录参数
+    project_root = args[0] if args else None
+    
+    try:
+        printer.print_str_in_terminal(get_message("rule_cmd_analyzing_project"))
+        
+        # 调用 init_rule 函数
+        result = init_rule(llm=llm, args=config, project_root=project_root)
+        
+        if result is None:
+            message = get_message("rule_cmd_init_failed")
+            printer.print_str_in_terminal(message)
+            return message
+        
+        # 成功生成，输出结果信息
+        success_message = f"""{get_message("rule_cmd_init_success")}
+
+{get_message("rule_cmd_project_type")}: {result.project_type}
+{get_message("rule_cmd_rules_file")}: {result.file_path}
+{get_message("rule_cmd_detected_technologies")}: {', '.join(result.technologies)}
+{get_message("rule_cmd_detected_commands")}: {', '.join(result.commands)}
+
+{get_message("rule_cmd_rules_content_preview")}:
+{result.content[:300]}{'...' if len(result.content) > 300 else ''}
+
+{get_message("rule_cmd_edit_rules_note")}
+"""
+        printer.print_str_in_terminal(success_message)
+        return success_message
+        
+    except Exception as e:
+        error_message = get_message_with_format("rule_cmd_init_error", error=str(e))
+        printer.print_str_in_terminal(error_message)
+        logger.exception(e)
+        return error_message
+
+# Command dispatch table - note: analyze and commit handlers have extra coding_func parameter
+COMMAND_HANDLERS = {
     "list": _handle_list_rules,
     "remove": _handle_remove_rules,
     "get": _handle_get_rules,
     "analyze": _handle_analyze_rules,  # 默认行为
     "help": _handle_help,
     "commit": _handle_commit_rules,  # 添加新的命令处理函数
+    "init": _handle_init_rules,  # 添加初始化规则生成命令
 }
 
-def handle_rules_command(command_args: str, memory: Dict[str, Any], coding_func=None) -> str:
+def handle_rules_command(command_args: str, coding_func=None) -> str:
     """
     Handles the /rules command and its subcommands.
 
     Args:
         command_args: The arguments string following the /rules command.
                       Example: "analyze code quality", "/list", "/remove *.md"
-        memory: The current session memory dictionary.
 
     Returns:
         A string response to be displayed to the user.
@@ -423,7 +470,7 @@ def handle_rules_command(command_args: str, memory: Dict[str, Any], coding_func=
     rules_str = command_args.strip()
     # 处理空命令
     if not rules_str:
-        return _handle_help(memory, [])        
+        return _handle_help([])        
     
     # 处理子命令
     if rules_str.startswith("/"):
@@ -435,16 +482,17 @@ def handle_rules_command(command_args: str, memory: Dict[str, Any], coding_func=
         handler = COMMAND_HANDLERS.get(subcommand)
         if handler:
             try:
-                # 仅 analyze 需要 coding_func
+                # 仅 analyze 和 commit 需要 coding_func
                 if subcommand == "analyze":
-                    return handler(memory, args, coding_func=coding_func)
+                    return _handle_analyze_rules(args, coding_func=coding_func)
                 elif subcommand == "commit":
-                    return handler(memory, args, coding_func=coding_func)
+                    return _handle_commit_rules(args, coding_func=coding_func)
                 else:
-                    return handler(memory, args)
+                    return handler(args)
             except Exception as e:
                 message = get_message_with_format("rules_command_error", subcommand=subcommand, error=str(e))
                 printer.print_str_in_terminal(message)
+                logger.exception(f"Error in {subcommand} command: {e}")
                 return message
         else:
             message = get_message_with_format("rules_unknown_command", subcommand=subcommand)
@@ -452,7 +500,7 @@ def handle_rules_command(command_args: str, memory: Dict[str, Any], coding_func=
             return message
     elif rules_str.lower() == "help":
         # 处理无斜杠的 help 命令
-        return _handle_help(memory, [])
+        return _handle_help([])
     else:        
         # 将整个字符串作为查询参数传递给 analyze
-        return _handle_analyze_rules(memory, [rules_str], coding_func=coding_func) 
+        return _handle_analyze_rules([rules_str], coding_func=coding_func)