IntelliJ IDEA插件开发实战：集成ChatGPT实现智能代码补全

2次阅读

没有评论

共计 2350 个字符，预计需要花费 6 分钟才能阅读完成。

在日常开发中，我们经常遇到传统代码补全的局限性：

上下文感知弱：常规补全只能基于当前文件词汇或简单语法分析
创造性建议缺失：无法自动生成符合业务逻辑的完整代码块
学习成本高：团队新成员需要长时间熟悉项目才能获得有效补全

AI 辅助开发的价值在于：

通过理解自然语言注释生成符合上下文的代码
自动补全复杂算法实现
根据错误堆栈建议修复方案

开箱即用的强大模型（GPT-3.5/4）
免维护基础设施
按需付费的成本可控

数据不出内网
无 API 调用限制
可定制微调

实战建议：个人开发者优先选用 OpenAI API，企业级开发建议部署 Llama2 等开源模型。

注册 Action：通过 <actions> 声明入口点

class ChatGPTAction : AnAction() {override fun actionPerformed(e: AnActionEvent) {// 打开交互面板}
}

创建 ToolWindow：作为主要交互界面

val toolWindow = ToolWindowManager.getInstance(project)
    .registerToolWindow(ChatGPTToolWindow.ID, true, ToolWindowAnchor.RIGHT)

带指数退避的重试机制实现：

suspend fun callWithRetry(
    prompt: String,
    maxRetries: Int = 3
): String {
    var delay = 1000L
    repeat(maxRetries) { attempt ->
        try {return openAIClient.completions(prompt)
        } catch (e: Exception) {if (attempt == maxRetries - 1) throw e
            delay(delay)
            delay *= 2 // 指数退避
        }
    }
    throw IllegalStateException()}

关键操作示例：

// 获取当前编辑器
val editor = e.getData(CommonDataKeys.EDITOR) ?: return

// 通过 PSI 获取当前类信息
val psiFile = PsiDocumentManager.getInstance(project).getPsiFile(editor.document)
val psiClass = (psiFile as? PsiJavaFile)?.classes?.firstOrNull()

// 在光标位置插入代码
WriteCommandAction.runWriteCommandAction(project) {val factory = PsiElementFactory.getInstance(project)
    val newMethod = factory.createMethodFromText(
        generatedCode, 
        psiClass
    )
    psiClass.add(newMethod)
}

建议项目目录：

/src/main/kotlin
  ├── actions
  │   └── ChatGPTAction.kt
  ├── toolwindow
  │   └── ChatGPTToolWindow.kt
  ├── clients
  │   └── OpenAIClient.kt
  └── handlers
      └── CodeInsertionHandler.kt

将多个相关请求合并：

fun batchComplete(prompts: List<String>): List<String> {
    val batchRequest = prompts.map { 
        CompletionRequest(
            model = "gpt-3.5-turbo",
            messages = listOf(Message(role = "user", content = it))
        )
    }
    return openAIClient.batchCompletions(batchRequest)
}

提取类关键信息而非完整代码
使用 AST 摘要代替原始文本
限制历史对话轮次

实现请求队列
缓存高频请求结果
客户端限流（如 Guava RateLimiter）

过滤敏感信息（API 密钥、IP 等）
企业版建议添加审计日志
可选本地预处理脚本

通过 PSI 获取更多语义信息：

fun getMethodContext(psiMethod: PsiMethod): String {
    return buildString {append("Return type: ${psiMethod.returnType?.presentableText}\n")
        append("Parameters: ${psiMethod.parameterList.parameters.joinToString { it.type.presentableText}}\n")
        psiMethod.parent?.let {append("In class: ${(it as? PsiClass)?.qualifiedName}\n")
        }
    }
}

完整可运行示例已上传 Github：
github.com/example/idea-chatgpt-plugin

包含：