IntelliJ IDEA插件开发实战：如何高效接入Claude API实现智能编程助手

2次阅读

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

在开发 IDE 插件集成 AI 服务时，开发者常遇到几个核心挑战：

实时响应难题：传统 HTTP 请求的延迟难以满足代码补全的即时性要求，尤其当网络状况不稳定时体验更差
上下文保持成本高：每次 API 调用都需要重新发送完整上下文（如当前文件内容、光标位置等），增加了带宽消耗和响应时间
认证流程复杂：OAuth2 的安全要求与 IDE 插件的用户体验存在天然矛盾，需要精心设计授权流程

使用 IntelliJ 的 PasswordSafe 配合自定义 OAuth2 流程：

// 使用 CredentialStore 保存 refresh token
public class AuthManager {
    private static final String SERVICE_NAME = "ClaudePlugin";

    public void saveToken(String key, String token) {PasswordSafe.getInstance().setPassword(new PasswordSafe.ProtectionContext(),
            SERVICE_NAME,
            key,
            token
        );
    }

    // 获取 AccessToken 的完整流程
    public String getAccessToken() {
        // 检查本地缓存的 token 是否有效
        // 无效则使用 refresh token 获取新 access token
        // 全部失效时启动 OAuth2 授权流程
    }
}

建立带自动重连的 WebSocket 客户端：

class ClaudeWebSocketClient : WebSocketListener() {
    private var client: WebSocket? = null
    private val retryStrategy = ExponentialBackoffRetry()

    fun connect() {val request = Request.Builder()
            .url("wss://api.claude.ai/v1/stream")
            .addHeader("Authorization", "Bearer ${accessToken}")
            .build()

        OkHttpClient().newWebSocket(request, this)
    }

    override fun onFailure(webSocket: WebSocket, t: Throwable, response: Response?) {retryStrategy.retry(::connect)
    }

    // 实现心跳检测
    private fun startHeartbeat() {Executors.newSingleThreadScheduledExecutor().scheduleAtFixedRate({ client?.send("{\"type\":\"heartbeat\"}") },
            0, 30, TimeUnit.SECONDS
        )
    }
}

基于 PSI 树的上下文收集：

public class CodeContextCollector {public String collectContext(Editor editor, PsiFile file) {
        // 获取当前光标前后各 200 个字符
        int offset = editor.getCaretModel().getOffset();
        String text = file.getText();
        int start = Math.max(0, offset - 200);
        int end = Math.min(text.length(), offset + 200);

        // 提取当前方法信息
        PsiElement element = file.findElementAt(offset);
        PsiMethod containingMethod = PsiTreeUtil.getParentOfType(element, PsiMethod.class);

        // 构建包含类名、方法名、变量类型的上下文
        return String.format("Class: %s\nMethod: %s\nCode:\n%s", 
            containingMethod.getContainingClass().getName(),
            containingMethod.getName(),
            text.substring(start, end));
    }
}

本地缓存：对高频请求的代码建议建立 LRU 缓存
请求批处理：当用户快速输入时合并多个连续请求
上下文差异计算：只发送与前次请求的差异部分

使用 IntelliJ Platform 的 PersistentStateComponent 保存配置
敏感信息始终通过 PasswordSafe 存储
所有 API 调用启用 HTTPS 证书固定

UI 操作必须通过 ApplicationManager.getApplication().invokeLater() 执行
网络请求必须在后台线程进行
PSI 访问需要保证在 ReadAction 中

class RateLimiter {private val requestQueue = ArrayDeque<Long>()
    private val lock = ReentrantLock()

    fun acquire(): Boolean {
        lock.withLock {val now = System.currentTimeMillis()
            // 移除 1 分钟前的记录
            while (requestQueue.isNotEmpty() && 
                  now - requestQueue.first > 60_000) {requestQueue.removeFirst()
            }

            if (requestQueue.size < 30) { // Claude API 限制
                requestQueue.addLast(now)
                return true
            }
            return false
        }
    }
}

我们提供了一个可运行的 Demo 项目：Claude-IntelliJ-Plugin 包含：