add /refactor.api

This commit is contained in:
bobo.yang 2025-03-11 13:32:05 +08:00
parent 99b7aca0fc
commit 7e15ab24cd
3 changed files with 135 additions and 0 deletions

View File

@ -0,0 +1,3 @@
# /refactor.api
对选中代码进行API重构

127
community/refactor/api/command.py Executable file
View File

@ -0,0 +1,127 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import sys
import json
from typing import Dict, Any, Tuple
from lib.ide_service import IDEService
from lib.chatmark import Button, Form, TextEditor
from lib.workflow import workflow_call
from devchat.llm import chat_json
# 步骤3: 使用AI识别API路径和METHOD的提示词
API_ANALYSIS_PROMPT = """
分析以下代码识别其中的API路径和HTTP方法
代码:
```
{code}
```
请提取代码中定义或使用的API路径和HTTP方法(GET, POST, PUT, DELETE等)
如果代码中有多个API请识别最主要的一个
如果无法确定HTTP方法请使用"GET"作为默认值
返回JSON格式如下:
{{
"api_path": "识别到的API路径例如/api/users",
"method": "识别到的HTTP方法例如GET、POST、PUT、DELETE等"
}}
"""
@chat_json(prompt=API_ANALYSIS_PROMPT)
def analyze_api(code: str) -> Dict[str, str]:
"""使用AI分析代码中的API路径和HTTP方法"""
pass
def main() -> None:
"""API重构工作流主函数"""
try:
# 步骤1: 获取用户输入的重构目标
if len(sys.argv) < 2:
print("错误: 请提供重构目标")
sys.exit(1)
refactor_target = sys.argv[1]
# 步骤2: 获取用户选中的代码
selected_code = IDEService().get_selected_range()
if not selected_code or not selected_code.text.strip():
print("错误: 请先选择需要重构的代码")
sys.exit(1)
# 步骤3: 使用AI识别API路径和METHOD
print("正在分析选中代码中的API信息...")
api_info = analyze_api(code=selected_code.text)
if not api_info or "api_path" not in api_info or "method" not in api_info:
print("错误: 无法识别API信息")
sys.exit(1)
api_path = api_info["api_path"]
method = api_info["method"]
# 步骤4: 显示识别结果并让用户确认
print(f"识别到的API信息:")
print(f"API路径: {api_path}")
print(f"HTTP方法: {method}")
api_path_editor = TextEditor(api_path)
form = Form([
"### 请确认API信息",
"API路径:",
api_path_editor,
f"HTTP方法: {method}",
"请确认或修改API路径然后点击下方按钮继续"
])
form.render()
# 获取用户确认后的API路径
confirmed_api_path = api_path_editor.new_text
# 步骤5: 调用重构工作流进行代码重构
print(f"正在重构API: {confirmed_api_path}...")
refactor_result = workflow_call(f"/refactor {refactor_target}")
if refactor_result != 0:
print("错误: API重构失败")
sys.exit(1)
print("API重构成功!")
# 步骤6: 显示按钮让用户确认是否继续
continue_button = Button(["提交修改并测试API", "结束重构"])
continue_button.render()
if continue_button.clicked == 1: # 用户选择结束
print("API重构已完成未提交修改")
return
# 步骤7: 调用GitHub提交工作流提交修改
print("正在提交修改...")
commit_result = workflow_call("/github.commit")
if commit_result != 0:
print("警告: 代码提交失败但将继续进行API测试")
else:
print("代码提交成功!")
# 步骤8: 调用API测试工作流对重构API进行测试
print("正在准备API测试...")
test_command = f"/test.api.upload {confirmed_api_path} {method} {refactor_target}"
test_result = workflow_call(test_command)
if test_result != 0:
print("警告: API测试可能未成功完成")
print("API重构工作流执行完毕!")
except Exception as e:
print(f"错误: 执行过程中发生异常: {str(e)}")
sys.exit(1)
if __name__ == "__main__":
main()

View File

@ -0,0 +1,5 @@
description: 对选中代码进行API重构
input: required
help: README.md
steps:
- run: $devchat_python $command_path/command.py "$input"