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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of auto-coder might be problematic. Click here for more details.

Files changed (574) hide show
  1. auto_coder-2.0.0.dist-info/LICENSE +158 -0
  2. auto_coder-2.0.0.dist-info/METADATA +558 -0
  3. auto_coder-2.0.0.dist-info/RECORD +795 -0
  4. {auto_coder-1.0.0.dist-info → auto_coder-2.0.0.dist-info}/WHEEL +1 -1
  5. {auto_coder-1.0.0.dist-info → auto_coder-2.0.0.dist-info}/entry_points.txt +3 -3
  6. autocoder/__init__.py +31 -0
  7. autocoder/agent/auto_filegroup.py +32 -13
  8. autocoder/agent/auto_learn_from_commit.py +9 -1
  9. autocoder/agent/base_agentic/__init__.py +3 -0
  10. autocoder/agent/base_agentic/agent_hub.py +1 -1
  11. autocoder/agent/base_agentic/base_agent.py +235 -136
  12. autocoder/agent/base_agentic/default_tools.py +119 -118
  13. autocoder/agent/base_agentic/test_base_agent.py +1 -1
  14. autocoder/agent/base_agentic/tool_registry.py +32 -20
  15. autocoder/agent/base_agentic/tools/read_file_tool_resolver.py +24 -3
  16. autocoder/agent/base_agentic/tools/write_to_file_tool_resolver.py +24 -11
  17. autocoder/agent/base_agentic/types.py +42 -0
  18. autocoder/agent/entry_command_agent/chat.py +73 -59
  19. autocoder/auto_coder.py +31 -40
  20. autocoder/auto_coder_rag.py +11 -1084
  21. autocoder/auto_coder_runner.py +970 -2345
  22. autocoder/auto_coder_terminal.py +26 -0
  23. autocoder/auto_coder_terminal_v3.py +190 -0
  24. autocoder/chat/conf_command.py +224 -124
  25. autocoder/chat/models_command.py +361 -299
  26. autocoder/chat/rules_command.py +79 -31
  27. autocoder/chat_auto_coder.py +988 -398
  28. autocoder/chat_auto_coder_lang.py +23 -732
  29. autocoder/commands/auto_command.py +25 -8
  30. autocoder/commands/auto_web.py +1 -1
  31. autocoder/commands/tools.py +44 -44
  32. autocoder/common/__init__.py +150 -128
  33. autocoder/common/ac_style_command_parser/__init__.py +39 -2
  34. autocoder/common/ac_style_command_parser/config.py +422 -0
  35. autocoder/common/ac_style_command_parser/parser.py +292 -78
  36. autocoder/common/ac_style_command_parser/test_parser.py +241 -16
  37. autocoder/common/ac_style_command_parser/test_typed_parser.py +342 -0
  38. autocoder/common/ac_style_command_parser/typed_parser.py +653 -0
  39. autocoder/common/action_yml_file_manager.py +25 -13
  40. autocoder/common/agent_events/__init__.py +52 -0
  41. autocoder/common/agent_events/agent_event_emitter.py +193 -0
  42. autocoder/common/agent_events/event_factory.py +177 -0
  43. autocoder/common/agent_events/examples.py +307 -0
  44. autocoder/common/agent_events/types.py +113 -0
  45. autocoder/common/agent_events/utils.py +68 -0
  46. autocoder/common/agent_hooks/__init__.py +44 -0
  47. autocoder/common/agent_hooks/examples.py +582 -0
  48. autocoder/common/agent_hooks/hook_executor.py +217 -0
  49. autocoder/common/agent_hooks/hook_manager.py +288 -0
  50. autocoder/common/agent_hooks/types.py +133 -0
  51. autocoder/common/agent_hooks/utils.py +99 -0
  52. autocoder/common/agent_query_queue/queue_executor.py +324 -0
  53. autocoder/common/agent_query_queue/queue_manager.py +325 -0
  54. autocoder/common/agents/__init__.py +11 -0
  55. autocoder/common/agents/agent_manager.py +323 -0
  56. autocoder/common/agents/agent_parser.py +189 -0
  57. autocoder/common/agents/example_usage.py +344 -0
  58. autocoder/common/agents/integration_example.py +330 -0
  59. autocoder/common/agents/test_agent_parser.py +545 -0
  60. autocoder/common/async_utils.py +101 -0
  61. autocoder/common/auto_coder_lang.py +23 -972
  62. autocoder/common/autocoderargs_parser/__init__.py +14 -0
  63. autocoder/common/autocoderargs_parser/parser.py +184 -0
  64. autocoder/common/autocoderargs_parser/tests/__init__.py +1 -0
  65. autocoder/common/autocoderargs_parser/tests/test_args_parser.py +235 -0
  66. autocoder/common/autocoderargs_parser/tests/test_token_parser.py +195 -0
  67. autocoder/common/autocoderargs_parser/token_parser.py +290 -0
  68. autocoder/common/buildin_tokenizer.py +2 -4
  69. autocoder/common/code_auto_generate.py +149 -74
  70. autocoder/common/code_auto_generate_diff.py +163 -70
  71. autocoder/common/code_auto_generate_editblock.py +179 -89
  72. autocoder/common/code_auto_generate_strict_diff.py +167 -72
  73. autocoder/common/code_auto_merge_editblock.py +13 -6
  74. autocoder/common/code_modification_ranker.py +1 -1
  75. autocoder/common/command_completer.py +3 -3
  76. autocoder/common/command_file_manager/manager.py +183 -47
  77. autocoder/common/command_file_manager/test_command_file_manager.py +507 -0
  78. autocoder/common/command_templates.py +1 -1
  79. autocoder/common/conf_utils.py +2 -4
  80. autocoder/common/conversations/config.py +11 -3
  81. autocoder/common/conversations/get_conversation_manager.py +100 -2
  82. autocoder/common/conversations/llm_stats_models.py +264 -0
  83. autocoder/common/conversations/manager.py +112 -28
  84. autocoder/common/conversations/models.py +16 -2
  85. autocoder/common/conversations/storage/index_manager.py +134 -10
  86. autocoder/common/core_config/__init__.py +63 -0
  87. autocoder/common/core_config/agentic_mode_manager.py +109 -0
  88. autocoder/common/core_config/base_manager.py +123 -0
  89. autocoder/common/core_config/compatibility.py +151 -0
  90. autocoder/common/core_config/config_manager.py +156 -0
  91. autocoder/common/core_config/conversation_manager.py +31 -0
  92. autocoder/common/core_config/exclude_manager.py +72 -0
  93. autocoder/common/core_config/file_manager.py +177 -0
  94. autocoder/common/core_config/human_as_model_manager.py +129 -0
  95. autocoder/common/core_config/lib_manager.py +54 -0
  96. autocoder/common/core_config/main_manager.py +81 -0
  97. autocoder/common/core_config/mode_manager.py +126 -0
  98. autocoder/common/core_config/models.py +70 -0
  99. autocoder/common/core_config/test_memory_manager.py +1056 -0
  100. autocoder/common/env_manager.py +282 -0
  101. autocoder/common/env_manager_usage_example.py +211 -0
  102. autocoder/common/file_checkpoint/conversation_checkpoint.py +19 -19
  103. autocoder/common/file_checkpoint/manager.py +264 -48
  104. autocoder/common/file_checkpoint/test_backup.py +1 -18
  105. autocoder/common/file_checkpoint/test_manager.py +270 -1
  106. autocoder/common/file_checkpoint/test_store.py +1 -17
  107. autocoder/common/file_handler/__init__.py +23 -0
  108. autocoder/common/file_handler/active_context_handler.py +159 -0
  109. autocoder/common/file_handler/add_files_handler.py +409 -0
  110. autocoder/common/file_handler/chat_handler.py +180 -0
  111. autocoder/common/file_handler/coding_handler.py +401 -0
  112. autocoder/common/file_handler/commit_handler.py +200 -0
  113. autocoder/common/file_handler/lib_handler.py +156 -0
  114. autocoder/common/file_handler/list_files_handler.py +111 -0
  115. autocoder/common/file_handler/mcp_handler.py +268 -0
  116. autocoder/common/file_handler/models_handler.py +493 -0
  117. autocoder/common/file_handler/remove_files_handler.py +172 -0
  118. autocoder/common/git_utils.py +44 -8
  119. autocoder/common/global_cancel.py +15 -6
  120. autocoder/common/ignorefiles/test_ignore_file_utils.py +1 -1
  121. autocoder/common/international/__init__.py +31 -0
  122. autocoder/common/international/demo_international.py +92 -0
  123. autocoder/common/international/message_manager.py +157 -0
  124. autocoder/common/international/messages/__init__.py +56 -0
  125. autocoder/common/international/messages/async_command_messages.py +507 -0
  126. autocoder/common/international/messages/auto_coder_messages.py +2208 -0
  127. autocoder/common/international/messages/chat_auto_coder_messages.py +1547 -0
  128. autocoder/common/international/messages/command_help_messages.py +986 -0
  129. autocoder/common/international/messages/conversation_command_messages.py +191 -0
  130. autocoder/common/international/messages/git_helper_plugin_messages.py +159 -0
  131. autocoder/common/international/messages/queue_command_messages.py +751 -0
  132. autocoder/common/international/messages/rules_command_messages.py +77 -0
  133. autocoder/common/international/messages/sdk_messages.py +1707 -0
  134. autocoder/common/international/messages/token_helper_plugin_messages.py +361 -0
  135. autocoder/common/international/messages/tool_display_messages.py +1212 -0
  136. autocoder/common/international/messages/workflow_exception_messages.py +473 -0
  137. autocoder/common/international/test_international.py +612 -0
  138. autocoder/common/linter_core/__init__.py +28 -0
  139. autocoder/common/linter_core/base_linter.py +61 -0
  140. autocoder/common/linter_core/config_loader.py +271 -0
  141. autocoder/common/linter_core/formatters/__init__.py +0 -0
  142. autocoder/common/linter_core/formatters/base_formatter.py +38 -0
  143. autocoder/common/linter_core/formatters/raw_formatter.py +17 -0
  144. autocoder/common/linter_core/linter.py +166 -0
  145. autocoder/common/linter_core/linter_factory.py +216 -0
  146. autocoder/common/linter_core/linter_manager.py +333 -0
  147. autocoder/common/linter_core/linters/__init__.py +9 -0
  148. autocoder/common/linter_core/linters/java_linter.py +342 -0
  149. autocoder/common/linter_core/linters/python_linter.py +115 -0
  150. autocoder/common/linter_core/linters/typescript_linter.py +119 -0
  151. autocoder/common/linter_core/models/__init__.py +7 -0
  152. autocoder/common/linter_core/models/lint_result.py +91 -0
  153. autocoder/common/linter_core/models.py +33 -0
  154. autocoder/common/linter_core/tests/__init__.py +3 -0
  155. autocoder/common/linter_core/tests/test_config_loader.py +323 -0
  156. autocoder/common/linter_core/tests/test_config_loading.py +308 -0
  157. autocoder/common/linter_core/tests/test_factory_manager.py +234 -0
  158. autocoder/common/linter_core/tests/test_formatters.py +147 -0
  159. autocoder/common/linter_core/tests/test_integration.py +317 -0
  160. autocoder/common/linter_core/tests/test_java_linter.py +496 -0
  161. autocoder/common/linter_core/tests/test_linters.py +265 -0
  162. autocoder/common/linter_core/tests/test_models.py +81 -0
  163. autocoder/common/linter_core/tests/verify_config_loading.py +296 -0
  164. autocoder/common/linter_core/tests/verify_fixes.py +183 -0
  165. autocoder/common/llm_friendly_package/__init__.py +31 -0
  166. autocoder/common/llm_friendly_package/base_manager.py +102 -0
  167. autocoder/common/llm_friendly_package/docs_manager.py +121 -0
  168. autocoder/common/llm_friendly_package/library_manager.py +171 -0
  169. autocoder/common/{llm_friendly_package.py → llm_friendly_package/main_manager.py} +204 -231
  170. autocoder/common/llm_friendly_package/models.py +40 -0
  171. autocoder/common/llm_friendly_package/test_llm_friendly_package.py +536 -0
  172. autocoder/common/llms/__init__.py +15 -0
  173. autocoder/common/llms/demo_error_handling.py +85 -0
  174. autocoder/common/llms/factory.py +142 -0
  175. autocoder/common/llms/manager.py +264 -0
  176. autocoder/common/llms/pricing.py +121 -0
  177. autocoder/common/llms/registry.py +288 -0
  178. autocoder/common/llms/schema.py +77 -0
  179. autocoder/common/llms/simple_demo.py +45 -0
  180. autocoder/common/llms/test_quick_model.py +116 -0
  181. autocoder/common/llms/test_remove_functionality.py +182 -0
  182. autocoder/common/llms/tests/__init__.py +1 -0
  183. autocoder/common/llms/tests/test_manager.py +330 -0
  184. autocoder/common/llms/tests/test_registry.py +364 -0
  185. autocoder/common/mcp_tools/__init__.py +62 -0
  186. autocoder/common/{mcp_tools.py → mcp_tools/executor.py} +49 -40
  187. autocoder/common/{mcp_hub.py → mcp_tools/hub.py} +42 -68
  188. autocoder/common/{mcp_server_install.py → mcp_tools/installer.py} +16 -28
  189. autocoder/common/{mcp_server.py → mcp_tools/server.py} +176 -48
  190. autocoder/common/mcp_tools/test_keyboard_interrupt.py +93 -0
  191. autocoder/common/mcp_tools/test_mcp_tools.py +391 -0
  192. autocoder/common/{mcp_server_types.py → mcp_tools/types.py} +121 -48
  193. autocoder/common/mcp_tools/verify_functionality.py +202 -0
  194. autocoder/common/model_speed_tester.py +32 -26
  195. autocoder/common/priority_directory_finder/__init__.py +142 -0
  196. autocoder/common/priority_directory_finder/examples.py +230 -0
  197. autocoder/common/priority_directory_finder/finder.py +283 -0
  198. autocoder/common/priority_directory_finder/models.py +236 -0
  199. autocoder/common/priority_directory_finder/test_priority_directory_finder.py +431 -0
  200. autocoder/common/project_scanner/__init__.py +18 -0
  201. autocoder/common/project_scanner/compat.py +77 -0
  202. autocoder/common/project_scanner/scanner.py +436 -0
  203. autocoder/common/project_tracker/__init__.py +27 -0
  204. autocoder/common/project_tracker/api.py +228 -0
  205. autocoder/common/project_tracker/demo.py +272 -0
  206. autocoder/common/project_tracker/tracker.py +487 -0
  207. autocoder/common/project_tracker/types.py +53 -0
  208. autocoder/common/pruner/__init__.py +67 -0
  209. autocoder/common/pruner/agentic_conversation_pruner.py +651 -102
  210. autocoder/common/pruner/conversation_message_ids_api.py +386 -0
  211. autocoder/common/pruner/conversation_message_ids_manager.py +347 -0
  212. autocoder/common/pruner/conversation_message_ids_pruner.py +473 -0
  213. autocoder/common/pruner/conversation_normalizer.py +347 -0
  214. autocoder/common/pruner/conversation_pruner.py +26 -6
  215. autocoder/common/pruner/test_agentic_conversation_pruner.py +554 -112
  216. autocoder/common/pruner/test_conversation_normalizer.py +502 -0
  217. autocoder/common/pruner/test_tool_content_detector.py +324 -0
  218. autocoder/common/pruner/tool_content_detector.py +227 -0
  219. autocoder/common/pruner/tools/__init__.py +18 -0
  220. autocoder/common/pruner/tools/query_message_ids.py +264 -0
  221. autocoder/common/pruner/tools/test_agentic_pruning_logic.py +432 -0
  222. autocoder/common/pruner/tools/test_message_ids_pruning_only.py +192 -0
  223. autocoder/common/pull_requests/__init__.py +9 -1
  224. autocoder/common/pull_requests/utils.py +122 -1
  225. autocoder/common/rag_manager/rag_manager.py +36 -40
  226. autocoder/common/rulefiles/__init__.py +53 -1
  227. autocoder/common/rulefiles/api.py +250 -0
  228. autocoder/common/rulefiles/core/__init__.py +14 -0
  229. autocoder/common/rulefiles/core/manager.py +241 -0
  230. autocoder/common/rulefiles/core/selector.py +805 -0
  231. autocoder/common/rulefiles/models/__init__.py +20 -0
  232. autocoder/common/rulefiles/models/index.py +16 -0
  233. autocoder/common/rulefiles/models/init_rule.py +18 -0
  234. autocoder/common/rulefiles/models/rule_file.py +18 -0
  235. autocoder/common/rulefiles/models/rule_relevance.py +14 -0
  236. autocoder/common/rulefiles/models/summary.py +16 -0
  237. autocoder/common/rulefiles/test_rulefiles.py +776 -0
  238. autocoder/common/rulefiles/utils/__init__.py +34 -0
  239. autocoder/common/rulefiles/utils/monitor.py +86 -0
  240. autocoder/common/rulefiles/utils/parser.py +230 -0
  241. autocoder/common/save_formatted_log.py +67 -10
  242. autocoder/common/search_replace.py +8 -1
  243. autocoder/common/search_replace_patch/__init__.py +24 -0
  244. autocoder/common/search_replace_patch/base.py +115 -0
  245. autocoder/common/search_replace_patch/manager.py +248 -0
  246. autocoder/common/search_replace_patch/patch_replacer.py +304 -0
  247. autocoder/common/search_replace_patch/similarity_replacer.py +306 -0
  248. autocoder/common/search_replace_patch/string_replacer.py +181 -0
  249. autocoder/common/search_replace_patch/tests/__init__.py +3 -0
  250. autocoder/common/search_replace_patch/tests/run_tests.py +126 -0
  251. autocoder/common/search_replace_patch/tests/test_base.py +188 -0
  252. autocoder/common/search_replace_patch/tests/test_empty_line_insert.py +233 -0
  253. autocoder/common/search_replace_patch/tests/test_integration.py +389 -0
  254. autocoder/common/search_replace_patch/tests/test_manager.py +351 -0
  255. autocoder/common/search_replace_patch/tests/test_patch_replacer.py +316 -0
  256. autocoder/common/search_replace_patch/tests/test_regex_replacer.py +306 -0
  257. autocoder/common/search_replace_patch/tests/test_similarity_replacer.py +384 -0
  258. autocoder/common/shell_commands/__init__.py +197 -0
  259. autocoder/common/shell_commands/background_process_notifier.py +346 -0
  260. autocoder/common/shell_commands/command_executor.py +1127 -0
  261. autocoder/common/shell_commands/error_recovery.py +541 -0
  262. autocoder/common/shell_commands/exceptions.py +120 -0
  263. autocoder/common/shell_commands/interactive_executor.py +476 -0
  264. autocoder/common/shell_commands/interactive_pexpect_process.py +623 -0
  265. autocoder/common/shell_commands/interactive_process.py +744 -0
  266. autocoder/common/shell_commands/interactive_session_manager.py +1014 -0
  267. autocoder/common/shell_commands/monitoring.py +529 -0
  268. autocoder/common/shell_commands/process_cleanup.py +386 -0
  269. autocoder/common/shell_commands/process_manager.py +606 -0
  270. autocoder/common/shell_commands/test_interactive_pexpect_process.py +281 -0
  271. autocoder/common/shell_commands/tests/__init__.py +6 -0
  272. autocoder/common/shell_commands/tests/conftest.py +118 -0
  273. autocoder/common/shell_commands/tests/test_background_process_notifier.py +703 -0
  274. autocoder/common/shell_commands/tests/test_command_executor.py +448 -0
  275. autocoder/common/shell_commands/tests/test_error_recovery.py +305 -0
  276. autocoder/common/shell_commands/tests/test_exceptions.py +299 -0
  277. autocoder/common/shell_commands/tests/test_execute_batch.py +588 -0
  278. autocoder/common/shell_commands/tests/test_indented_batch_commands.py +244 -0
  279. autocoder/common/shell_commands/tests/test_integration.py +664 -0
  280. autocoder/common/shell_commands/tests/test_monitoring.py +546 -0
  281. autocoder/common/shell_commands/tests/test_performance.py +632 -0
  282. autocoder/common/shell_commands/tests/test_process_cleanup.py +397 -0
  283. autocoder/common/shell_commands/tests/test_process_manager.py +606 -0
  284. autocoder/common/shell_commands/tests/test_timeout_config.py +343 -0
  285. autocoder/common/shell_commands/tests/test_timeout_manager.py +520 -0
  286. autocoder/common/shell_commands/timeout_config.py +315 -0
  287. autocoder/common/shell_commands/timeout_manager.py +352 -0
  288. autocoder/common/terminal_paste/__init__.py +14 -0
  289. autocoder/common/terminal_paste/demo.py +145 -0
  290. autocoder/common/terminal_paste/demo_paste_functionality.py +95 -0
  291. autocoder/common/terminal_paste/paste_handler.py +200 -0
  292. autocoder/common/terminal_paste/paste_manager.py +118 -0
  293. autocoder/common/terminal_paste/tests/__init__.py +1 -0
  294. autocoder/common/terminal_paste/tests/test_paste_handler.py +182 -0
  295. autocoder/common/terminal_paste/tests/test_paste_manager.py +126 -0
  296. autocoder/common/terminal_paste/utils.py +163 -0
  297. autocoder/common/test_autocoder_args.py +232 -0
  298. autocoder/common/test_env_manager.py +173 -0
  299. autocoder/common/test_env_manager_integration.py +159 -0
  300. autocoder/common/text_similarity/__init__.py +9 -0
  301. autocoder/common/text_similarity/demo.py +216 -0
  302. autocoder/common/text_similarity/examples.py +266 -0
  303. autocoder/common/text_similarity/test_text_similarity.py +306 -0
  304. autocoder/common/text_similarity/text_similarity.py +194 -0
  305. autocoder/common/text_similarity/utils.py +125 -0
  306. autocoder/common/todos/__init__.py +61 -0
  307. autocoder/common/todos/cache/__init__.py +16 -0
  308. autocoder/common/todos/cache/base_cache.py +89 -0
  309. autocoder/common/todos/cache/cache_manager.py +228 -0
  310. autocoder/common/todos/cache/memory_cache.py +225 -0
  311. autocoder/common/todos/config.py +155 -0
  312. autocoder/common/todos/exceptions.py +35 -0
  313. autocoder/common/todos/get_todo_manager.py +161 -0
  314. autocoder/common/todos/manager.py +537 -0
  315. autocoder/common/todos/models.py +239 -0
  316. autocoder/common/todos/storage/__init__.py +14 -0
  317. autocoder/common/todos/storage/base_storage.py +76 -0
  318. autocoder/common/todos/storage/file_storage.py +278 -0
  319. autocoder/common/tokens/counter.py +24 -2
  320. autocoder/common/tools_manager/__init__.py +17 -0
  321. autocoder/common/tools_manager/examples.py +162 -0
  322. autocoder/common/tools_manager/manager.py +385 -0
  323. autocoder/common/tools_manager/models.py +39 -0
  324. autocoder/common/tools_manager/test_tools_manager.py +303 -0
  325. autocoder/common/tools_manager/utils.py +191 -0
  326. autocoder/common/v2/agent/agentic_callbacks.py +270 -0
  327. autocoder/common/v2/agent/agentic_edit.py +2699 -1856
  328. autocoder/common/v2/agent/agentic_edit_change_manager.py +474 -0
  329. autocoder/common/v2/agent/agentic_edit_tools/__init__.py +35 -1
  330. autocoder/common/v2/agent/agentic_edit_tools/ac_mod_list_tool_resolver.py +279 -0
  331. autocoder/common/v2/agent/agentic_edit_tools/ac_mod_write_tool_resolver.py +10 -1
  332. autocoder/common/v2/agent/agentic_edit_tools/background_task_tool_resolver.py +1167 -0
  333. autocoder/common/v2/agent/agentic_edit_tools/base_tool_resolver.py +2 -2
  334. autocoder/common/v2/agent/agentic_edit_tools/conversation_message_ids_read_tool_resolver.py +214 -0
  335. autocoder/common/v2/agent/agentic_edit_tools/conversation_message_ids_write_tool_resolver.py +299 -0
  336. autocoder/common/v2/agent/agentic_edit_tools/count_tokens_tool_resolver.py +290 -0
  337. autocoder/common/v2/agent/agentic_edit_tools/execute_command_tool_resolver.py +564 -29
  338. autocoder/common/v2/agent/agentic_edit_tools/execute_workflow_tool_resolver.py +485 -0
  339. autocoder/common/v2/agent/agentic_edit_tools/extract_to_text_tool_resolver.py +225 -0
  340. autocoder/common/v2/agent/agentic_edit_tools/lint_report.py +79 -0
  341. autocoder/common/v2/agent/agentic_edit_tools/linter_config_models.py +343 -0
  342. autocoder/common/v2/agent/agentic_edit_tools/linter_enabled_tool_resolver.py +189 -0
  343. autocoder/common/v2/agent/agentic_edit_tools/list_files_tool_resolver.py +169 -101
  344. autocoder/common/v2/agent/agentic_edit_tools/load_extra_document_tool_resolver.py +349 -0
  345. autocoder/common/v2/agent/agentic_edit_tools/read_file_tool_resolver.py +243 -50
  346. autocoder/common/v2/agent/agentic_edit_tools/replace_in_file_tool_resolver.py +667 -147
  347. autocoder/common/v2/agent/agentic_edit_tools/run_named_subagents_tool_resolver.py +691 -0
  348. autocoder/common/v2/agent/agentic_edit_tools/search_files_tool_resolver.py +410 -86
  349. autocoder/common/v2/agent/agentic_edit_tools/session_interactive_tool_resolver.py +115 -0
  350. autocoder/common/v2/agent/agentic_edit_tools/session_start_tool_resolver.py +190 -0
  351. autocoder/common/v2/agent/agentic_edit_tools/session_stop_tool_resolver.py +76 -0
  352. autocoder/common/v2/agent/agentic_edit_tools/test_write_to_file_tool_resolver.py +207 -192
  353. autocoder/common/v2/agent/agentic_edit_tools/todo_read_tool_resolver.py +80 -63
  354. autocoder/common/v2/agent/agentic_edit_tools/todo_write_tool_resolver.py +237 -233
  355. autocoder/common/v2/agent/agentic_edit_tools/use_mcp_tool_resolver.py +2 -2
  356. autocoder/common/v2/agent/agentic_edit_tools/web_crawl_tool_resolver.py +557 -0
  357. autocoder/common/v2/agent/agentic_edit_tools/web_search_tool_resolver.py +600 -0
  358. autocoder/common/v2/agent/agentic_edit_tools/write_to_file_tool_resolver.py +56 -121
  359. autocoder/common/v2/agent/agentic_edit_types.py +343 -9
  360. autocoder/common/v2/agent/runner/__init__.py +3 -3
  361. autocoder/common/v2/agent/runner/base_runner.py +12 -26
  362. autocoder/common/v2/agent/runner/{event_runner.py → file_based_event_runner.py} +3 -2
  363. autocoder/common/v2/agent/runner/sdk_runner.py +150 -8
  364. autocoder/common/v2/agent/runner/terminal_runner.py +170 -57
  365. autocoder/common/v2/agent/runner/tool_display.py +557 -159
  366. autocoder/common/v2/agent/test_agentic_callbacks.py +265 -0
  367. autocoder/common/v2/agent/test_agentic_edit.py +194 -0
  368. autocoder/common/v2/agent/tool_caller/__init__.py +24 -0
  369. autocoder/common/v2/agent/tool_caller/default_tool_resolver_map.py +135 -0
  370. autocoder/common/v2/agent/tool_caller/integration_test.py +172 -0
  371. autocoder/common/v2/agent/tool_caller/plugins/__init__.py +14 -0
  372. autocoder/common/v2/agent/tool_caller/plugins/base_plugin.py +126 -0
  373. autocoder/common/v2/agent/tool_caller/plugins/examples/__init__.py +13 -0
  374. autocoder/common/v2/agent/tool_caller/plugins/examples/logging_plugin.py +164 -0
  375. autocoder/common/v2/agent/tool_caller/plugins/examples/security_filter_plugin.py +198 -0
  376. autocoder/common/v2/agent/tool_caller/plugins/plugin_interface.py +141 -0
  377. autocoder/common/v2/agent/tool_caller/test_tool_caller.py +278 -0
  378. autocoder/common/v2/agent/tool_caller/tool_call_plugin_manager.py +331 -0
  379. autocoder/common/v2/agent/tool_caller/tool_caller.py +337 -0
  380. autocoder/common/v2/agent/tool_caller/usage_example.py +193 -0
  381. autocoder/common/v2/code_agentic_editblock_manager.py +4 -4
  382. autocoder/common/v2/code_auto_generate.py +136 -78
  383. autocoder/common/v2/code_auto_generate_diff.py +135 -79
  384. autocoder/common/v2/code_auto_generate_editblock.py +174 -99
  385. autocoder/common/v2/code_auto_generate_strict_diff.py +151 -71
  386. autocoder/common/v2/code_auto_merge.py +1 -1
  387. autocoder/common/v2/code_auto_merge_editblock.py +13 -1
  388. autocoder/common/v2/code_diff_manager.py +3 -3
  389. autocoder/common/v2/code_editblock_manager.py +4 -14
  390. autocoder/common/v2/code_manager.py +1 -1
  391. autocoder/common/v2/code_strict_diff_manager.py +2 -2
  392. autocoder/common/wrap_llm_hint/__init__.py +10 -0
  393. autocoder/common/wrap_llm_hint/test_wrap_llm_hint.py +1067 -0
  394. autocoder/common/wrap_llm_hint/utils.py +432 -0
  395. autocoder/common/wrap_llm_hint/wrap_llm_hint.py +323 -0
  396. autocoder/completer/__init__.py +8 -0
  397. autocoder/completer/command_completer_v2.py +1051 -0
  398. autocoder/default_project/__init__.py +501 -0
  399. autocoder/dispacher/__init__.py +4 -12
  400. autocoder/dispacher/actions/action.py +165 -7
  401. autocoder/dispacher/actions/plugins/action_regex_project.py +2 -2
  402. autocoder/index/entry.py +116 -124
  403. autocoder/{agent → index/filter}/agentic_filter.py +322 -333
  404. autocoder/index/filter/normal_filter.py +5 -11
  405. autocoder/index/filter/quick_filter.py +1 -1
  406. autocoder/index/index.py +36 -9
  407. autocoder/index/tests/__init__.py +1 -0
  408. autocoder/index/tests/run_tests.py +195 -0
  409. autocoder/index/tests/test_entry.py +303 -0
  410. autocoder/index/tests/test_index_manager.py +314 -0
  411. autocoder/index/tests/test_module_integration.py +300 -0
  412. autocoder/index/tests/test_symbols_utils.py +183 -0
  413. autocoder/inner/__init__.py +4 -0
  414. autocoder/inner/agentic.py +932 -0
  415. autocoder/inner/async_command_handler.py +992 -0
  416. autocoder/inner/conversation_command_handlers.py +623 -0
  417. autocoder/inner/merge_command_handler.py +213 -0
  418. autocoder/inner/queue_command_handler.py +684 -0
  419. autocoder/models.py +95 -266
  420. autocoder/plugins/git_helper_plugin.py +31 -29
  421. autocoder/plugins/token_helper_plugin.py +65 -46
  422. autocoder/pyproject/__init__.py +32 -29
  423. autocoder/rag/agentic_rag.py +215 -75
  424. autocoder/rag/cache/simple_cache.py +1 -2
  425. autocoder/rag/loaders/image_loader.py +1 -1
  426. autocoder/rag/long_context_rag.py +42 -26
  427. autocoder/rag/qa_conversation_strategy.py +1 -1
  428. autocoder/rag/terminal/__init__.py +17 -0
  429. autocoder/rag/terminal/args.py +581 -0
  430. autocoder/rag/terminal/bootstrap.py +61 -0
  431. autocoder/rag/terminal/command_handlers.py +653 -0
  432. autocoder/rag/terminal/formatters/__init__.py +20 -0
  433. autocoder/rag/terminal/formatters/base.py +70 -0
  434. autocoder/rag/terminal/formatters/json_format.py +66 -0
  435. autocoder/rag/terminal/formatters/stream_json.py +95 -0
  436. autocoder/rag/terminal/formatters/text.py +28 -0
  437. autocoder/rag/terminal/init.py +120 -0
  438. autocoder/rag/terminal/utils.py +106 -0
  439. autocoder/rag/test_agentic_rag.py +389 -0
  440. autocoder/rag/test_doc_filter.py +3 -3
  441. autocoder/rag/test_long_context_rag.py +1 -1
  442. autocoder/rag/test_token_limiter.py +517 -10
  443. autocoder/rag/token_counter.py +3 -0
  444. autocoder/rag/token_limiter.py +19 -15
  445. autocoder/rag/tools/__init__.py +26 -2
  446. autocoder/rag/tools/bochaai_example.py +343 -0
  447. autocoder/rag/tools/bochaai_sdk.py +541 -0
  448. autocoder/rag/tools/metaso_example.py +268 -0
  449. autocoder/rag/tools/metaso_sdk.py +417 -0
  450. autocoder/rag/tools/recall_tool.py +28 -7
  451. autocoder/rag/tools/run_integration_tests.py +204 -0
  452. autocoder/rag/tools/test_all_providers.py +318 -0
  453. autocoder/rag/tools/test_bochaai_integration.py +482 -0
  454. autocoder/rag/tools/test_final_integration.py +215 -0
  455. autocoder/rag/tools/test_metaso_integration.py +424 -0
  456. autocoder/rag/tools/test_metaso_real.py +171 -0
  457. autocoder/rag/tools/test_web_crawl_tool.py +639 -0
  458. autocoder/rag/tools/test_web_search_tool.py +509 -0
  459. autocoder/rag/tools/todo_read_tool.py +202 -0
  460. autocoder/rag/tools/todo_write_tool.py +412 -0
  461. autocoder/rag/tools/web_crawl_tool.py +634 -0
  462. autocoder/rag/tools/web_search_tool.py +558 -0
  463. autocoder/rag/tools/web_tools_example.py +119 -0
  464. autocoder/rag/types.py +16 -0
  465. autocoder/rag/variable_holder.py +4 -2
  466. autocoder/rags.py +86 -79
  467. autocoder/regexproject/__init__.py +23 -21
  468. autocoder/sdk/__init__.py +46 -190
  469. autocoder/sdk/api.py +370 -0
  470. autocoder/sdk/async_runner/__init__.py +26 -0
  471. autocoder/sdk/async_runner/async_executor.py +650 -0
  472. autocoder/sdk/async_runner/async_handler.py +356 -0
  473. autocoder/sdk/async_runner/markdown_processor.py +595 -0
  474. autocoder/sdk/async_runner/task_metadata.py +284 -0
  475. autocoder/sdk/async_runner/worktree_manager.py +438 -0
  476. autocoder/sdk/cli/__init__.py +2 -5
  477. autocoder/sdk/cli/formatters.py +28 -204
  478. autocoder/sdk/cli/handlers.py +77 -44
  479. autocoder/sdk/cli/main.py +154 -171
  480. autocoder/sdk/cli/options.py +95 -22
  481. autocoder/sdk/constants.py +139 -51
  482. autocoder/sdk/core/auto_coder_core.py +484 -109
  483. autocoder/sdk/core/bridge.py +297 -115
  484. autocoder/sdk/exceptions.py +18 -12
  485. autocoder/sdk/formatters/__init__.py +19 -0
  486. autocoder/sdk/formatters/input.py +64 -0
  487. autocoder/sdk/formatters/output.py +247 -0
  488. autocoder/sdk/formatters/stream.py +54 -0
  489. autocoder/sdk/models/__init__.py +6 -5
  490. autocoder/sdk/models/options.py +55 -18
  491. autocoder/sdk/utils/formatters.py +27 -195
  492. autocoder/suffixproject/__init__.py +28 -25
  493. autocoder/terminal/__init__.py +14 -0
  494. autocoder/terminal/app.py +454 -0
  495. autocoder/terminal/args.py +32 -0
  496. autocoder/terminal/bootstrap.py +178 -0
  497. autocoder/terminal/command_processor.py +521 -0
  498. autocoder/terminal/command_registry.py +57 -0
  499. autocoder/terminal/help.py +97 -0
  500. autocoder/terminal/tasks/__init__.py +5 -0
  501. autocoder/terminal/tasks/background.py +77 -0
  502. autocoder/terminal/tasks/task_event.py +70 -0
  503. autocoder/terminal/ui/__init__.py +13 -0
  504. autocoder/terminal/ui/completer.py +268 -0
  505. autocoder/terminal/ui/keybindings.py +75 -0
  506. autocoder/terminal/ui/session.py +41 -0
  507. autocoder/terminal/ui/toolbar.py +64 -0
  508. autocoder/terminal/utils/__init__.py +13 -0
  509. autocoder/terminal/utils/errors.py +18 -0
  510. autocoder/terminal/utils/paths.py +19 -0
  511. autocoder/terminal/utils/shell.py +43 -0
  512. autocoder/terminal_v3/__init__.py +10 -0
  513. autocoder/terminal_v3/app.py +201 -0
  514. autocoder/terminal_v3/handlers/__init__.py +5 -0
  515. autocoder/terminal_v3/handlers/command_handler.py +131 -0
  516. autocoder/terminal_v3/models/__init__.py +6 -0
  517. autocoder/terminal_v3/models/conversation_buffer.py +214 -0
  518. autocoder/terminal_v3/models/message.py +50 -0
  519. autocoder/terminal_v3/models/tool_display.py +247 -0
  520. autocoder/terminal_v3/ui/__init__.py +7 -0
  521. autocoder/terminal_v3/ui/keybindings.py +56 -0
  522. autocoder/terminal_v3/ui/layout.py +141 -0
  523. autocoder/terminal_v3/ui/styles.py +43 -0
  524. autocoder/tsproject/__init__.py +23 -23
  525. autocoder/utils/auto_coder_utils/chat_stream_out.py +1 -1
  526. autocoder/utils/llms.py +88 -80
  527. autocoder/utils/math_utils.py +101 -0
  528. autocoder/utils/model_provider_selector.py +16 -4
  529. autocoder/utils/operate_config_api.py +33 -5
  530. autocoder/utils/thread_utils.py +2 -2
  531. autocoder/version.py +4 -2
  532. autocoder/workflow_agents/__init__.py +84 -0
  533. autocoder/workflow_agents/agent.py +143 -0
  534. autocoder/workflow_agents/exceptions.py +573 -0
  535. autocoder/workflow_agents/executor.py +489 -0
  536. autocoder/workflow_agents/loader.py +737 -0
  537. autocoder/workflow_agents/runner.py +267 -0
  538. autocoder/workflow_agents/types.py +172 -0
  539. autocoder/workflow_agents/utils.py +434 -0
  540. autocoder/workflow_agents/workflow_manager.py +211 -0
  541. auto_coder-1.0.0.dist-info/METADATA +0 -396
  542. auto_coder-1.0.0.dist-info/RECORD +0 -442
  543. auto_coder-1.0.0.dist-info/licenses/LICENSE +0 -201
  544. autocoder/auto_coder_server.py +0 -672
  545. autocoder/benchmark.py +0 -138
  546. autocoder/common/ac_style_command_parser/example.py +0 -7
  547. autocoder/common/cleaner.py +0 -31
  548. autocoder/common/command_completer_v2.py +0 -615
  549. autocoder/common/context_pruner.py +0 -477
  550. autocoder/common/conversation_pruner.py +0 -132
  551. autocoder/common/directory_cache/__init__.py +0 -1
  552. autocoder/common/directory_cache/cache.py +0 -192
  553. autocoder/common/directory_cache/test_cache.py +0 -190
  554. autocoder/common/file_checkpoint/examples.py +0 -217
  555. autocoder/common/llm_friendly_package_example.py +0 -138
  556. autocoder/common/llm_friendly_package_test.py +0 -63
  557. autocoder/common/pull_requests/test_module.py +0 -1
  558. autocoder/common/rulefiles/autocoderrules_utils.py +0 -484
  559. autocoder/common/text.py +0 -30
  560. autocoder/common/v2/agent/agentic_edit_tools/list_package_info_tool_resolver.py +0 -42
  561. autocoder/common/v2/agent/agentic_edit_tools/test_execute_command_tool_resolver.py +0 -70
  562. autocoder/common/v2/agent/agentic_edit_tools/test_search_files_tool_resolver.py +0 -163
  563. autocoder/common/v2/agent/agentic_tool_display.py +0 -183
  564. autocoder/plugins/dynamic_completion_example.py +0 -148
  565. autocoder/plugins/sample_plugin.py +0 -160
  566. autocoder/sdk/cli/__main__.py +0 -26
  567. autocoder/sdk/cli/completion_wrapper.py +0 -38
  568. autocoder/sdk/cli/install_completion.py +0 -301
  569. autocoder/sdk/models/messages.py +0 -209
  570. autocoder/sdk/session/__init__.py +0 -32
  571. autocoder/sdk/session/session.py +0 -106
  572. autocoder/sdk/session/session_manager.py +0 -56
  573. {auto_coder-1.0.0.dist-info → auto_coder-2.0.0.dist-info}/top_level.txt +0 -0
  574. /autocoder/{sdk/example.py → common/agent_query_queue/__init__.py} +0 -0
@@ -0,0 +1,236 @@
1
+ """
2
+ 优先级目录查找器的数据模型
3
+
4
+ 定义了查找策略、验证规则、结果类型等核心数据结构。
5
+ """
6
+
7
+ import os
8
+ from dataclasses import dataclass, field
9
+ from enum import Enum
10
+ from typing import Dict, List, Optional, Set, Callable, Any, Union
11
+ from pathlib import Path
12
+
13
+
14
+ class SearchStrategy(Enum):
15
+ """查找策略枚举"""
16
+ FIRST_MATCH = "first_match" # 顺序查找,找到第一个有效目录就返回
17
+ MERGE_ALL = "merge_all" # 合并查找,合并所有有效目录的内容
18
+ LIST_ALL = "list_all" # 全部查找,返回所有有效目录列表
19
+
20
+
21
+ class ValidationMode(Enum):
22
+ """验证模式枚举"""
23
+ EXISTS = "exists" # 目录存在即可
24
+ HAS_FILES = "has_files" # 目录包含文件
25
+ HAS_SPECIFIC_FILES = "has_specific_files" # 目录包含特定类型文件
26
+ CUSTOM = "custom" # 自定义验证函数
27
+
28
+
29
+ @dataclass
30
+ class FileTypeFilter:
31
+ """文件类型过滤器"""
32
+ extensions: Set[str] = field(default_factory=set) # 文件扩展名,如 {'.md', '.txt'}
33
+ patterns: Set[str] = field(default_factory=set) # 文件名模式,如 {'*.yml', 'config.*'}
34
+ include_hidden: bool = False # 是否包含隐藏文件
35
+ recursive: bool = True # 是否递归搜索子目录
36
+ min_count: int = 1 # 最小文件数量
37
+ max_count: Optional[int] = None # 最大文件数量
38
+
39
+ def matches_file(self, file_path: str) -> bool:
40
+ """检查文件是否匹配过滤器"""
41
+ import fnmatch
42
+
43
+ file_name = os.path.basename(file_path)
44
+
45
+ # 检查隐藏文件
46
+ if not self.include_hidden and file_name.startswith('.'):
47
+ return False
48
+
49
+ # 检查扩展名和模式匹配(OR逻辑:满足任一条件即可)
50
+ extension_matches = False
51
+ pattern_matches = False
52
+
53
+ # 检查扩展名
54
+ if self.extensions:
55
+ _, ext = os.path.splitext(file_name)
56
+ # 将扩展名转换为小写并检查是否在集合中(扩展名集合应该都是小写的)
57
+ ext_lower = ext.lower()
58
+ # 同时检查原始扩展名和小写扩展名(为了兼容性)
59
+ extension_matches = ext_lower in self.extensions or ext in self.extensions
60
+
61
+ # 检查模式匹配
62
+ if self.patterns:
63
+ pattern_matches = any(fnmatch.fnmatch(file_name, pattern) for pattern in self.patterns)
64
+
65
+ # 如果设置了扩展名或模式,则至少要匹配其中一个
66
+ if self.extensions or self.patterns:
67
+ return extension_matches or pattern_matches
68
+
69
+ # 如果既没有设置扩展名也没有设置模式,则匹配所有文件
70
+ return True
71
+
72
+
73
+ @dataclass
74
+ class DirectoryConfig:
75
+ """目录配置"""
76
+ path: str # 目录路径(支持环境变量和~展开)
77
+ priority: int = 0 # 优先级(数字越小优先级越高)
78
+ validation_mode: ValidationMode = ValidationMode.HAS_FILES
79
+ file_filter: Optional[FileTypeFilter] = None
80
+ custom_validator: Optional[Callable[[str], bool]] = None
81
+ description: str = "" # 目录描述
82
+
83
+ def get_expanded_path(self) -> str:
84
+ """获取展开后的路径"""
85
+ expanded = os.path.expanduser(self.path)
86
+ expanded = os.path.expandvars(expanded)
87
+ return os.path.abspath(expanded)
88
+
89
+ def validate_directory(self) -> bool:
90
+ """验证目录是否有效"""
91
+ expanded_path = self.get_expanded_path()
92
+
93
+ if not os.path.exists(expanded_path) or not os.path.isdir(expanded_path):
94
+ return False
95
+
96
+ if self.validation_mode == ValidationMode.EXISTS:
97
+ return True
98
+
99
+ elif self.validation_mode == ValidationMode.HAS_FILES:
100
+ return self._has_any_files(expanded_path)
101
+
102
+ elif self.validation_mode == ValidationMode.HAS_SPECIFIC_FILES:
103
+ if self.file_filter is None:
104
+ return False
105
+ return self._has_matching_files(expanded_path)
106
+
107
+ elif self.validation_mode == ValidationMode.CUSTOM:
108
+ if self.custom_validator is None:
109
+ return False
110
+ return self.custom_validator(expanded_path)
111
+
112
+ return False
113
+
114
+ def _has_any_files(self, directory: str) -> bool:
115
+ """检查目录是否包含任何文件"""
116
+ try:
117
+ for item in os.listdir(directory):
118
+ item_path = os.path.join(directory, item)
119
+ if os.path.isfile(item_path):
120
+ return True
121
+ if os.path.isdir(item_path) and self._has_any_files(item_path):
122
+ return True
123
+ return False
124
+ except (OSError, PermissionError):
125
+ return False
126
+
127
+ def _has_matching_files(self, directory: str) -> bool:
128
+ """检查目录是否包含匹配的文件"""
129
+ if self.file_filter is None:
130
+ return False
131
+
132
+ matching_files = []
133
+
134
+ try:
135
+ self._collect_matching_files(directory, matching_files)
136
+
137
+ file_count = len(matching_files)
138
+ if file_count < self.file_filter.min_count:
139
+ return False
140
+
141
+ if self.file_filter.max_count is not None and file_count > self.file_filter.max_count:
142
+ return False
143
+
144
+ return True
145
+
146
+ except (OSError, PermissionError):
147
+ return False
148
+
149
+ def _collect_matching_files(self, directory: str, matching_files: List[str]) -> None:
150
+ """收集匹配的文件"""
151
+ for item in os.listdir(directory):
152
+ item_path = os.path.join(directory, item)
153
+
154
+ if os.path.isfile(item_path) and self.file_filter.matches_file(item_path):
155
+ matching_files.append(item_path)
156
+
157
+ elif (os.path.isdir(item_path) and
158
+ self.file_filter.recursive and
159
+ not os.path.basename(item).startswith('.')):
160
+ self._collect_matching_files(item_path, matching_files)
161
+
162
+
163
+ @dataclass
164
+ class SearchResult:
165
+ """查找结果"""
166
+ strategy: SearchStrategy
167
+ success: bool
168
+ selected_directories: List[str] = field(default_factory=list)
169
+ all_valid_directories: List[str] = field(default_factory=list)
170
+ errors: Dict[str, str] = field(default_factory=dict)
171
+ metadata: Dict[str, Any] = field(default_factory=dict)
172
+
173
+ @property
174
+ def primary_directory(self) -> Optional[str]:
175
+ """获取主要目录(第一个选中的目录)"""
176
+ return self.selected_directories[0] if self.selected_directories else None
177
+
178
+ @property
179
+ def has_errors(self) -> bool:
180
+ """是否有错误"""
181
+ return len(self.errors) > 0
182
+
183
+ def add_error(self, path: str, error_message: str) -> None:
184
+ """添加错误信息"""
185
+ self.errors[path] = error_message
186
+
187
+
188
+ @dataclass
189
+ class FinderConfig:
190
+ """查找器配置"""
191
+ directories: List[DirectoryConfig] = field(default_factory=list)
192
+ strategy: SearchStrategy = SearchStrategy.FIRST_MATCH
193
+ sort_by_priority: bool = True # 是否按优先级排序
194
+ include_metadata: bool = True # 是否包含元数据
195
+ fail_on_no_match: bool = False # 没有匹配时是否失败
196
+ default_fallback: Optional[str] = None # 默认回退目录
197
+
198
+ def add_directory(self, path: str, priority: int = 0, **kwargs) -> 'FinderConfig':
199
+ """添加目录配置"""
200
+ config = DirectoryConfig(path=path, priority=priority, **kwargs)
201
+ self.directories.append(config)
202
+ return self
203
+
204
+ def add_standard_paths(self, base_name: str, current_dir: Optional[str] = None) -> 'FinderConfig':
205
+ """添加标准路径配置"""
206
+ if current_dir is None:
207
+ current_dir = os.getcwd()
208
+
209
+ # 项目级目录
210
+ self.add_directory(
211
+ path=os.path.join(current_dir, base_name),
212
+ priority=1,
213
+ description=f"项目级{base_name}目录"
214
+ )
215
+
216
+ # 项目.auto-coder目录
217
+ self.add_directory(
218
+ path=os.path.join(current_dir, ".auto-coder", base_name),
219
+ priority=2,
220
+ description=f"项目.auto-coder/{base_name}目录"
221
+ )
222
+
223
+ # 全局目录
224
+ self.add_directory(
225
+ path=f"~/.auto-coder/{base_name}",
226
+ priority=3,
227
+ description=f"全局{base_name}目录"
228
+ )
229
+
230
+ return self
231
+
232
+ def get_sorted_directories(self) -> List[DirectoryConfig]:
233
+ """获取排序后的目录列表"""
234
+ if self.sort_by_priority:
235
+ return sorted(self.directories, key=lambda d: d.priority)
236
+ return self.directories.copy()
@@ -0,0 +1,431 @@
1
+ """
2
+ 优先级目录查找器测试套件
3
+
4
+ 测试所有主要功能,包括查找策略、验证规则、文件过滤等。
5
+ """
6
+
7
+ import os
8
+ import tempfile
9
+ import unittest
10
+ from unittest.mock import patch, MagicMock
11
+ from pathlib import Path
12
+
13
+ from .models import (
14
+ SearchStrategy, ValidationMode, FileTypeFilter, DirectoryConfig,
15
+ SearchResult, FinderConfig
16
+ )
17
+ from .finder import PriorityDirectoryFinder, QuickFinder
18
+ from . import find_command_directories, create_file_filter
19
+
20
+
21
+ class TestFileTypeFilter(unittest.TestCase):
22
+ """测试文件类型过滤器"""
23
+
24
+ def setUp(self):
25
+ self.filter = FileTypeFilter(
26
+ extensions={'.md', '.txt'},
27
+ patterns={'config.*', '*.yml'},
28
+ min_count=1,
29
+ recursive=True,
30
+ include_hidden=False
31
+ )
32
+
33
+ def test_matches_file_by_extension(self):
34
+ """测试按扩展名匹配文件"""
35
+ self.assertTrue(self.filter.matches_file("test.md"))
36
+ self.assertTrue(self.filter.matches_file("test.txt"))
37
+ self.assertFalse(self.filter.matches_file("test.py"))
38
+ self.assertFalse(self.filter.matches_file("test"))
39
+
40
+ def test_matches_file_by_pattern(self):
41
+ """测试按模式匹配文件"""
42
+ filter_patterns = FileTypeFilter(patterns={'*.yml', 'config.*'})
43
+ self.assertTrue(filter_patterns.matches_file("test.yml"))
44
+ self.assertTrue(filter_patterns.matches_file("config.json"))
45
+ self.assertTrue(filter_patterns.matches_file("config.py"))
46
+ self.assertFalse(filter_patterns.matches_file("test.py"))
47
+
48
+ def test_hidden_files(self):
49
+ """测试隐藏文件处理"""
50
+ filter_no_hidden = FileTypeFilter(include_hidden=False)
51
+ self.assertFalse(filter_no_hidden.matches_file(".hidden.txt"))
52
+
53
+ filter_with_hidden = FileTypeFilter(include_hidden=True)
54
+ self.assertTrue(filter_with_hidden.matches_file(".hidden.txt"))
55
+
56
+
57
+ class TestDirectoryConfig(unittest.TestCase):
58
+ """测试目录配置"""
59
+
60
+ def setUp(self):
61
+ self.temp_dir = tempfile.mkdtemp()
62
+ self.test_file = os.path.join(self.temp_dir, "test.md")
63
+ with open(self.test_file, 'w') as f:
64
+ f.write("# Test")
65
+
66
+ def tearDown(self):
67
+ import shutil
68
+ shutil.rmtree(self.temp_dir, ignore_errors=True)
69
+
70
+ def test_get_expanded_path(self):
71
+ """测试路径展开"""
72
+ config = DirectoryConfig("~/test")
73
+ expanded = config.get_expanded_path()
74
+ self.assertTrue(expanded.startswith(os.path.expanduser("~")))
75
+ self.assertTrue(os.path.isabs(expanded))
76
+
77
+ def test_validate_directory_exists(self):
78
+ """测试目录存在验证"""
79
+ config = DirectoryConfig(self.temp_dir, validation_mode=ValidationMode.EXISTS)
80
+ self.assertTrue(config.validate_directory())
81
+
82
+ config_nonexistent = DirectoryConfig("/nonexistent", validation_mode=ValidationMode.EXISTS)
83
+ self.assertFalse(config_nonexistent.validate_directory())
84
+
85
+ def test_validate_directory_has_files(self):
86
+ """测试目录包含文件验证"""
87
+ config = DirectoryConfig(self.temp_dir, validation_mode=ValidationMode.HAS_FILES)
88
+ self.assertTrue(config.validate_directory())
89
+
90
+ empty_dir = tempfile.mkdtemp()
91
+ try:
92
+ config_empty = DirectoryConfig(empty_dir, validation_mode=ValidationMode.HAS_FILES)
93
+ self.assertFalse(config_empty.validate_directory())
94
+ finally:
95
+ os.rmdir(empty_dir)
96
+
97
+ def test_validate_directory_specific_files(self):
98
+ """测试特定文件类型验证"""
99
+ file_filter = FileTypeFilter(extensions={'.md'})
100
+ config = DirectoryConfig(
101
+ self.temp_dir,
102
+ validation_mode=ValidationMode.HAS_SPECIFIC_FILES,
103
+ file_filter=file_filter
104
+ )
105
+ self.assertTrue(config.validate_directory())
106
+
107
+ filter_py = FileTypeFilter(extensions={'.py'})
108
+ config_py = DirectoryConfig(
109
+ self.temp_dir,
110
+ validation_mode=ValidationMode.HAS_SPECIFIC_FILES,
111
+ file_filter=filter_py
112
+ )
113
+ self.assertFalse(config_py.validate_directory())
114
+
115
+ def test_validate_directory_custom(self):
116
+ """测试自定义验证"""
117
+ def always_true(path):
118
+ return True
119
+
120
+ def always_false(path):
121
+ return False
122
+
123
+ config_true = DirectoryConfig(
124
+ self.temp_dir,
125
+ validation_mode=ValidationMode.CUSTOM,
126
+ custom_validator=always_true
127
+ )
128
+ self.assertTrue(config_true.validate_directory())
129
+
130
+ config_false = DirectoryConfig(
131
+ self.temp_dir,
132
+ validation_mode=ValidationMode.CUSTOM,
133
+ custom_validator=always_false
134
+ )
135
+ self.assertFalse(config_false.validate_directory())
136
+
137
+
138
+ class TestFinderConfig(unittest.TestCase):
139
+ """测试查找器配置"""
140
+
141
+ def test_add_directory(self):
142
+ """测试添加目录"""
143
+ config = FinderConfig()
144
+ config.add_directory("/test", priority=1, description="Test dir")
145
+
146
+ self.assertEqual(len(config.directories), 1)
147
+ self.assertEqual(config.directories[0].path, "/test")
148
+ self.assertEqual(config.directories[0].priority, 1)
149
+ self.assertEqual(config.directories[0].description, "Test dir")
150
+
151
+ def test_add_standard_paths(self):
152
+ """测试添加标准路径"""
153
+ config = FinderConfig()
154
+ config.add_standard_paths(".test", "/tmp")
155
+
156
+ self.assertEqual(len(config.directories), 3)
157
+
158
+ # 检查路径
159
+ paths = [d.path for d in config.directories]
160
+ self.assertIn("/tmp/.test", paths)
161
+ self.assertIn("/tmp/.auto-coder/.test", paths)
162
+ self.assertIn("~/.auto-coder/.test", paths)
163
+
164
+ def test_get_sorted_directories(self):
165
+ """测试目录排序"""
166
+ config = FinderConfig()
167
+ config.add_directory("/test3", priority=3)
168
+ config.add_directory("/test1", priority=1)
169
+ config.add_directory("/test2", priority=2)
170
+
171
+ sorted_dirs = config.get_sorted_directories()
172
+ priorities = [d.priority for d in sorted_dirs]
173
+ self.assertEqual(priorities, [1, 2, 3])
174
+
175
+
176
+ class TestPriorityDirectoryFinder(unittest.TestCase):
177
+ """测试优先级目录查找器"""
178
+
179
+ def setUp(self):
180
+ self.temp_dir = tempfile.mkdtemp()
181
+
182
+ # 创建测试目录结构
183
+ self.test_dirs = []
184
+ for i in range(3):
185
+ dir_path = os.path.join(self.temp_dir, f"test_{i}")
186
+ os.makedirs(dir_path)
187
+
188
+ # 在前两个目录创建文件
189
+ if i < 2:
190
+ with open(os.path.join(dir_path, f"file_{i}.md"), 'w') as f:
191
+ f.write(f"Test {i}")
192
+
193
+ self.test_dirs.append(dir_path)
194
+
195
+ def tearDown(self):
196
+ import shutil
197
+ shutil.rmtree(self.temp_dir, ignore_errors=True)
198
+
199
+ def test_find_directories_first_match(self):
200
+ """测试顺序查找策略"""
201
+ config = FinderConfig(strategy=SearchStrategy.FIRST_MATCH)
202
+ for i, dir_path in enumerate(self.test_dirs):
203
+ config.add_directory(dir_path, priority=i)
204
+
205
+ finder = PriorityDirectoryFinder(config)
206
+ result = finder.find_directories()
207
+
208
+ self.assertTrue(result.success)
209
+ self.assertEqual(len(result.selected_directories), 1)
210
+ self.assertEqual(result.selected_directories[0], self.test_dirs[0])
211
+ self.assertEqual(len(result.all_valid_directories), 2) # 只有前两个有文件
212
+
213
+ def test_find_directories_list_all(self):
214
+ """测试全部查找策略"""
215
+ config = FinderConfig(strategy=SearchStrategy.LIST_ALL)
216
+ for i, dir_path in enumerate(self.test_dirs):
217
+ config.add_directory(dir_path, priority=i)
218
+
219
+ finder = PriorityDirectoryFinder(config)
220
+ result = finder.find_directories()
221
+
222
+ self.assertTrue(result.success)
223
+ self.assertEqual(len(result.selected_directories), 2) # 只有前两个有文件
224
+ self.assertEqual(set(result.selected_directories), set(self.test_dirs[:2]))
225
+
226
+ def test_find_directories_merge_all(self):
227
+ """测试合并查找策略"""
228
+ config = FinderConfig(strategy=SearchStrategy.MERGE_ALL)
229
+ for i, dir_path in enumerate(self.test_dirs):
230
+ config.add_directory(dir_path, priority=i)
231
+
232
+ finder = PriorityDirectoryFinder(config)
233
+ result = finder.find_directories()
234
+
235
+ self.assertTrue(result.success)
236
+ self.assertEqual(len(result.selected_directories), 2) # 只有前两个有文件
237
+
238
+ def test_find_directories_with_fallback(self):
239
+ """测试回退目录"""
240
+ empty_dir = tempfile.mkdtemp()
241
+ try:
242
+ config = FinderConfig(default_fallback=empty_dir)
243
+ config.add_directory("/nonexistent", priority=1)
244
+
245
+ finder = PriorityDirectoryFinder(config)
246
+ result = finder.find_directories()
247
+
248
+ self.assertTrue(result.success)
249
+ self.assertEqual(result.primary_directory, empty_dir)
250
+ finally:
251
+ os.rmdir(empty_dir)
252
+
253
+ def test_find_first_directory(self):
254
+ """测试查找第一个目录的便捷方法"""
255
+ config = FinderConfig()
256
+ config.add_directory(self.test_dirs[0], priority=1)
257
+
258
+ finder = PriorityDirectoryFinder(config)
259
+ first_dir = finder.find_first_directory()
260
+
261
+ self.assertEqual(first_dir, self.test_dirs[0])
262
+
263
+ def test_find_all_directories(self):
264
+ """测试查找所有目录的便捷方法"""
265
+ config = FinderConfig()
266
+ for i, dir_path in enumerate(self.test_dirs[:2]): # 只添加有文件的目录
267
+ config.add_directory(dir_path, priority=i)
268
+
269
+ finder = PriorityDirectoryFinder(config)
270
+ all_dirs = finder.find_all_directories()
271
+
272
+ self.assertEqual(len(all_dirs), 2)
273
+ self.assertEqual(set(all_dirs), set(self.test_dirs[:2]))
274
+
275
+ def test_collect_files_from_directories(self):
276
+ """测试从目录收集文件"""
277
+ finder = PriorityDirectoryFinder()
278
+
279
+ md_filter = FileTypeFilter(extensions={'.md'})
280
+ files = finder.collect_files_from_directories(self.test_dirs[:2], md_filter)
281
+
282
+ self.assertEqual(len(files), 2)
283
+ for file_path in files:
284
+ self.assertTrue(file_path.endswith('.md'))
285
+
286
+ def test_metadata_inclusion(self):
287
+ """测试元数据包含"""
288
+ config = FinderConfig(include_metadata=True)
289
+ config.add_directory(self.test_dirs[0], priority=1, description="Test directory")
290
+
291
+ finder = PriorityDirectoryFinder(config)
292
+ result = finder.find_directories()
293
+
294
+ self.assertIn('total_configured', result.metadata)
295
+ self.assertIn('total_valid', result.metadata)
296
+ self.assertIn('strategy_used', result.metadata)
297
+ self.assertIn('directory_configs', result.metadata)
298
+
299
+ self.assertEqual(result.metadata['total_configured'], 1)
300
+ self.assertEqual(result.metadata['total_valid'], 1)
301
+
302
+
303
+ class TestQuickFinder(unittest.TestCase):
304
+ """测试快速查找器"""
305
+
306
+ def setUp(self):
307
+ self.temp_dir = tempfile.mkdtemp()
308
+
309
+ # 创建标准目录结构
310
+ self.cmd_dir = os.path.join(self.temp_dir, ".autocodercommands")
311
+ os.makedirs(self.cmd_dir)
312
+ with open(os.path.join(self.cmd_dir, "test.md"), 'w') as f:
313
+ f.write("# Test command")
314
+
315
+ def tearDown(self):
316
+ import shutil
317
+ shutil.rmtree(self.temp_dir, ignore_errors=True)
318
+
319
+ def test_find_standard_directories(self):
320
+ """测试查找标准目录"""
321
+ result = QuickFinder.find_standard_directories(
322
+ base_name=".autocodercommands",
323
+ current_dir=self.temp_dir,
324
+ file_extensions=["md"]
325
+ )
326
+
327
+ self.assertTrue(result.success)
328
+ self.assertEqual(result.primary_directory, self.cmd_dir)
329
+
330
+ def test_find_command_directories(self):
331
+ """测试查找命令目录"""
332
+ with patch('os.getcwd', return_value=self.temp_dir):
333
+ result = QuickFinder.find_command_directories()
334
+ self.assertTrue(result.success)
335
+ self.assertEqual(result.primary_directory, self.cmd_dir)
336
+
337
+
338
+ class TestConvenienceFunctions(unittest.TestCase):
339
+ """测试便捷函数"""
340
+
341
+ def test_create_file_filter(self):
342
+ """测试创建文件过滤器"""
343
+ filter_obj = create_file_filter(
344
+ extensions=[".md", ".txt"],
345
+ patterns=["*.yml"],
346
+ min_count=2
347
+ )
348
+
349
+ self.assertEqual(filter_obj.extensions, {".md", ".txt"})
350
+ self.assertEqual(filter_obj.patterns, {"*.yml"})
351
+ self.assertEqual(filter_obj.min_count, 2)
352
+
353
+ def test_find_command_directories_function(self):
354
+ """测试查找命令目录便捷函数"""
355
+ with patch('autocoder.common.priority_directory_finder.finder.QuickFinder.find_command_directories') as mock_find:
356
+ mock_result = MagicMock()
357
+ mock_find.return_value = mock_result
358
+
359
+ result = find_command_directories("/test")
360
+
361
+ mock_find.assert_called_once_with("/test")
362
+ self.assertEqual(result, mock_result)
363
+
364
+
365
+ class TestIntegration(unittest.TestCase):
366
+ """集成测试"""
367
+
368
+ def setUp(self):
369
+ self.temp_dir = tempfile.mkdtemp()
370
+
371
+ # 创建复杂的测试场景
372
+ self.project_cmd = os.path.join(self.temp_dir, ".autocodercommands")
373
+ self.auto_coder_cmd = os.path.join(self.temp_dir, ".auto-coder", ".autocodercommands")
374
+
375
+ os.makedirs(self.project_cmd)
376
+ os.makedirs(self.auto_coder_cmd)
377
+
378
+ # 只在.auto-coder目录创建文件
379
+ with open(os.path.join(self.auto_coder_cmd, "command.md"), 'w') as f:
380
+ f.write("# Auto-coder command")
381
+
382
+ def tearDown(self):
383
+ import shutil
384
+ shutil.rmtree(self.temp_dir, ignore_errors=True)
385
+
386
+ def test_priority_selection(self):
387
+ """测试优先级选择逻辑"""
388
+ result = QuickFinder.find_standard_directories(
389
+ base_name=".autocodercommands",
390
+ current_dir=self.temp_dir,
391
+ file_extensions=["md"],
392
+ strategy=SearchStrategy.FIRST_MATCH
393
+ )
394
+
395
+ # 应该选择.auto-coder目录(因为项目级目录为空)
396
+ self.assertTrue(result.success)
397
+ self.assertEqual(result.primary_directory, self.auto_coder_cmd)
398
+
399
+ def test_complete_workflow(self):
400
+ """测试完整工作流程"""
401
+ # 1. 配置查找器
402
+ config = FinderConfig(strategy=SearchStrategy.LIST_ALL, include_metadata=True)
403
+ config.add_standard_paths(".autocodercommands", self.temp_dir)
404
+
405
+ # 设置文件过滤器
406
+ md_filter = FileTypeFilter(extensions={'.md'})
407
+ for dir_config in config.directories:
408
+ dir_config.validation_mode = ValidationMode.HAS_SPECIFIC_FILES
409
+ dir_config.file_filter = md_filter
410
+
411
+ # 2. 执行查找
412
+ finder = PriorityDirectoryFinder(config)
413
+ result = finder.find_directories()
414
+
415
+ # 3. 验证结果
416
+ self.assertTrue(result.success)
417
+ self.assertEqual(len(result.selected_directories), 1)
418
+ self.assertEqual(result.selected_directories[0], self.auto_coder_cmd)
419
+
420
+ # 4. 收集文件
421
+ files = finder.collect_files_from_directories(result.selected_directories, md_filter)
422
+ self.assertEqual(len(files), 1)
423
+ self.assertTrue(files[0].endswith("command.md"))
424
+
425
+ # 5. 检查元数据
426
+ self.assertIn('total_configured', result.metadata)
427
+ self.assertIn('directory_configs', result.metadata)
428
+
429
+
430
+ if __name__ == '__main__':
431
+ unittest.main()
@@ -0,0 +1,18 @@
1
+ # -*- coding: utf-8 -*-
2
+ """
3
+ Project Scanner Module
4
+
5
+ 提供项目文件和目录扫描功能,支持文件监控和忽略规则过滤。
6
+ """
7
+
8
+ from .scanner import (
9
+ ProjectScanner,
10
+ SymbolItem,
11
+ get_project_scanner
12
+ )
13
+
14
+ __all__ = [
15
+ 'ProjectScanner',
16
+ 'SymbolItem',
17
+ 'get_project_scanner'
18
+ ]