IntelliJ IDEA 集成 UNIT-00Berserk Interface 插件开发指南你是不是也想过要是写代码的时候IDE能像有个经验丰富的搭档在旁边随时帮你补全代码、解释逻辑、甚至重构优化那该多省心今天我们就来动手实现这个想法。我会带你一步步在 IntelliJ IDEA 里开发一个能集成 UNIT-00Berserk Interface 的智能插件让它成为你编码时的得力助手。这个插件能做什么呢简单来说就是让 UNIT-00 的能力直接嵌入到你的开发环境中。比如你写了一半的方法它能帮你智能补全你看不懂的复杂代码块它能生成清晰的注释甚至它还能分析你的代码结构给出重构建议。整个过程我们都会从零开始用最直白的方式讲清楚即使你之前没怎么接触过 IDEA 插件开发也能跟着做出来。1. 环境准备与项目创建工欲善其事必先利其器。在开始敲代码之前我们得先把“厨房”收拾好。1.1 安装 IntelliJ IDEA 与插件 SDK首先你需要一个 IntelliJ IDEA。如果你还没有可以去官网下载社区版Community Edition它对于插件开发来说完全够用而且是免费的。下载后按照提示安装即可这个过程和安装其他软件没什么两样。安装好 IDEA 后打开它我们还需要配置插件开发环境。IDEA 本身已经集成了插件开发所需的 SDK软件开发工具包我们只需要确认一下。打开 IDEA点击File文件菜单选择Project Structure...项目结构。在左侧找到Platform Settings平台设置下的SDKs。检查列表里是否有一个名为IntelliJ IDEA Plugin SDK的条目。通常IDEA 会自动检测并添加它。如果没有点击左上角的号选择Add IntelliJ Platform Plugin SDK...然后选择你 IDEA 的安装目录它会自动配置好。这就好比你要做木工IDEA 是你的工作台而这个 Plugin SDK 就是你工具箱里最核心的那套凿子和刨子。1.2 创建你的第一个插件项目环境好了我们来创建项目。这个过程 IDEA 提供了模板非常方便。在 IDEA 启动界面或通过File - New - Project...新建项目。在左侧选择IntelliJ Platform Plugin。右边会显示项目信息。给项目起个名字比如BerserkCoderAssistant。选择好项目存放的位置。最关键的一步是选择SDK。点击下拉框选择我们刚才确认过的IntelliJ IDEA Plugin SDK。点击Create创建。稍等片刻一个最基础的插件项目骨架就生成好了。你会看到项目里自动包含了一些文件其中最重要的是src/main/resources/META-INF/plugin.xml这是插件的“身份证”和“说明书”我们后面会详细配置它。2. 理解插件基础与连接 UNIT-00项目建好了但在动手连接 UNIT-00 之前我们得先花几分钟了解一下 IDEA 插件是怎么工作的以及我们打算让 UNIT-00 帮什么忙。2.1 IDEA 插件能干什么你可以把 IDEA 插件想象成给 IDEA 这个超级跑车加装的“定制配件”。通过插件我们可以监听事件比如用户敲击了键盘、点击了菜单、或者编辑器里的内容发生了变化。访问和操作代码读取当前文件的代码、分析语法结构、甚至直接插入或修改代码文本。扩展界面添加新的工具栏按钮、菜单项、侧边栏工具窗口等等。我们的智能插件核心就是监听用户在编辑器中的操作比如输入停顿、选中代码然后将相关的代码上下文发送给 UNIT-00 接口再把返回的智能建议补全的代码、生成的注释展示给用户或直接应用到代码中。2.2 规划插件功能与 API 调用为了让教程更聚焦我们主要实现三个核心功能你可以把它们看作插件的三个“技能”智能代码补全当用户输入时插件能调用 UNIT-00根据上下文预测并建议接下来的代码行。注释生成用户选中一段代码右键菜单可以选择“生成注释”插件会请求 UNIT-00 为这段代码生成解释性注释。重构建议在项目视图中右键点击文件或目录插件可以分析代码结构并给出 UNIT-00 提供的重构优化建议。要实现这些我们需要和 UNIT-00Berserk Interface 对话。这通常意味着通过 HTTP 请求调用其提供的 API。你需要准备 UNIT-00 的 API 访问地址Endpoint和认证密钥API Key。这部分信息你需要从 UNIT-00 的服务提供商那里获取。在代码中我们会把这些配置放在安全的地方。3. 搭建插件骨架与配置现在我们来给插件“装上轮子”和“方向盘”也就是配置核心文件并建立基础的代码结构。3.1 配置 plugin.xml打开plugin.xml文件它现在可能只有一些基础信息。我们需要对它进行增补告诉 IDEA 我们的插件是谁有什么能力。idea-plugin !-- 插件唯一ID通常用倒置域名格式 -- idcom.yourname.berserk.coder.assistant/id !-- 插件在插件市场显示的名字 -- nameBerserk Coder Assistant/name !-- 版本号 -- version1.0/version !-- 供应商信息 -- vendor emailyour.emailexample.com urlhttps://yourwebsite.comYour Name/vendor !-- 插件描述会显示在插件详情页 -- description![CDATA[ An intelligent coding assistant plugin that integrates UNIT-00: Berserk Interface into IntelliJ IDEA, providing AI-powered code completion, comment generation, and refactoring suggestions. ]]/description !-- 依赖声明我们的插件需要依赖哪些其他插件或平台功能 -- dependscom.intellij.modules.platform/depends !-- 如果你用到其他语言支持比如Python可能需要添加 dependscom.intellij.modules.python/depends -- !-- 扩展点声明这是我们插件扩展IDEA功能的地方 -- extensions defaultExtensionNscom.intellij !-- 例如我们可以在这里注册代码补全贡献器 -- !-- completion.contributor ... / 这个我们稍后在Java代码里用另一种方式实现 -- /extensions actions !-- 这里我们将定义具体的菜单动作比如“生成注释”的右键菜单项 -- !-- 下一节我们会具体添加 -- /actions /idea-plugin这个文件是插件的元数据核心先有个大概了解后面我们会一步步填充actions部分。3.2 创建 API 服务类我们需要一个专门负责和 UNIT-00 通信的“信使”。在src/main/java下你可能需要先创建这些目录新建一个包比如com.berserk.assistant.service然后在这个包里创建一个 Java 类BerserkApiService。这个类主要做两件事构建发送给 UNIT-00 的请求数据以及解析 UNIT-00 返回的结果。这里我们先写一个骨架并模拟一个简单的注释生成请求。package com.berserk.assistant.service; import com.google.gson.Gson; import com.google.gson.JsonObject; import okhttp3.*; import java.io.IOException; public class BerserkApiService { private static final String API_BASE_URL https://api.your-unit00-service.com/v1; // 替换为实际API地址 private static final String API_KEY your-api-key-here; // 替换为你的API密钥 private static final OkHttpClient client new OkHttpClient(); private static final Gson gson new Gson(); // 一个示例方法为选中的代码生成注释 public String generateCommentForCode(String selectedCode, String language) throws IOException { // 1. 构建请求体JSON格式根据UNIT-00的API文档来构造 JsonObject requestBody new JsonObject(); requestBody.addProperty(model, unit-00-codex); // 假设的模型名 requestBody.addProperty(prompt, Generate a concise inline comment for the following language code:\n selectedCode); requestBody.addProperty(max_tokens, 100); RequestBody body RequestBody.create( gson.toJson(requestBody), MediaType.parse(application/json; charsetutf-8) ); // 2. 构建HTTP请求 Request request new Request.Builder() .url(API_BASE_URL /completions) // 假设的补全端点 .post(body) .addHeader(Authorization, Bearer API_KEY) .addHeader(Content-Type, application/json) .build(); // 3. 发送请求并获取响应 try (Response response client.newCall(request).execute()) { if (!response.isSuccessful()) { throw new IOException(Unexpected code response , body: response.body().string()); } String responseBody response.body().string(); // 4. 解析响应JSON提取生成的文本注释 JsonObject jsonResponse gson.fromJson(responseBody, JsonObject.class); // 这里需要根据实际的API响应结构来解析以下为示例 String generatedText jsonResponse.getAsJsonArray(choices) .get(0).getAsJsonObject() .get(text).getAsString().trim(); return generatedText; } } // 后续可以添加更多方法如 codeCompletion, suggestRefactoring 等 }注意你需要将API_BASE_URL和API_KEY替换成真实值。更安全的做法是从 IDEA 的配置PersistentStateComponent中读取这些配置而不是硬编码在代码里。为了教程简洁我们先这样写。另外OkHttp库需要添加依赖我们马上会做。3.3 添加项目依赖我们的插件需要用到OkHttp来发网络请求用Gson来解析 JSON。我们需要在项目的build.gradle.kts如果你创建的是 Gradle 项目或pom.xmlMaven 项目里声明这些依赖。这里以更常见的 Gradle 为例。打开项目根目录下的build.gradle.kts文件在dependencies块里添加dependencies { // ... 其他已有依赖 implementation(com.squareup.okhttp3:okhttp:4.12.0) implementation(com.google.code.gson:gson:2.10.1) }添加后IDEA 通常会提示同步Sync项目点击同步它就会自动下载这些库了。4. 实现核心功能动作与交互骨架搭好了信使ApiService也有了现在该让插件“动起来”了。我们将实现一个具体的功能右键菜单生成注释。4.1 创建“生成注释”动作Action在 IDEA 插件中用户的一个操作比如点击菜单、按钮通常对应一个AnAction类。我们来创建一个。在src/main/java下新建一个包com.berserk.assistant.actions然后创建类GenerateCommentAction。package com.berserk.assistant.actions; import com.berserk.assistant.service.BerserkApiService; import com.intellij.openapi.actionSystem.AnAction; import com.intellij.openapi.actionSystem.AnActionEvent; import com.intellij.openapi.actionSystem.CommonDataKeys; import com.intellij.openapi.editor.Editor; import com.intellij.openapi.project.Project; import com.intellij.openapi.ui.Messages; import org.jetbrains.annotations.NotNull; import java.io.IOException; public class GenerateCommentAction extends AnAction { Override public void update(NotNull AnActionEvent e) { // 这个方法决定动作菜单项是否可见/可用 final Project project e.getProject(); final Editor editor e.getData(CommonDataKeys.EDITOR); // 只在有项目、有编辑器、且编辑器中有文本被选中时该动作才可用 e.getPresentation().setEnabledAndVisible(project ! null editor ! null editor.getSelectionModel().hasSelection()); } Override public void actionPerformed(NotNull AnActionEvent e) { // 当用户点击菜单项时执行这个方法 final Editor editor e.getRequiredData(CommonDataKeys.EDITOR); final Project project e.getProject(); // 1. 获取选中的代码文本 String selectedText editor.getSelectionModel().getSelectedText(); if (selectedText null || selectedText.trim().isEmpty()) { Messages.showWarningDialog(project, Please select some code first., No Code Selected); return; } // 2. 获取文件语言用于构造更精准的提示词 // 这里简化处理实际可能需要从 PsiFile 获取 String language Java; // 示例实际应动态获取 // 3. 调用我们的 API 服务 BerserkApiService apiService new BerserkApiService(); try { // 显示一个“处理中”的提示是个好主意这里省略了 String generatedComment apiService.generateCommentForCode(selectedText, language); // 4. 将生成的注释插入到选中代码的前一行 int selectionStart editor.getSelectionModel().getSelectionStart(); int lineNumber editor.getDocument().getLineNumber(selectionStart); int lineStartOffset editor.getDocument().getLineStartOffset(lineNumber); editor.getDocument().insertString(lineStartOffset, // generatedComment \n); Messages.showInfoMessage(project, Comment generated and inserted above the selection., Success); } catch (IOException ex) { Messages.showErrorDialog(project, Failed to generate comment: ex.getMessage(), API Error); ex.printStackTrace(); } } }这个类做了几件事update方法控制右键菜单里这个项什么时候能点actionPerformed是点击后执行的逻辑——获取选中代码调用 UNIT-00然后把返回的注释插到代码前面。4.2 注册动作到菜单光有动作类还不行得告诉 IDEA 把这个动作放到哪里。我们需要回到plugin.xml文件在actions标签内注册它。actions !-- 将我们的动作添加到编辑器右键菜单PopupMenu -- action idBerserk.GenerateComment classcom.berserk.assistant.actions.GenerateCommentAction textGenerate Comment with Berserk descriptionGenerate AI-powered comment for selected code !-- 指定这个动作添加到“EditorPopupMenu”这个组里 -- add-to-group group-idEditorPopupMenu anchorfirst/ /action /actions这样配置后当你在编辑器里选中代码并点击右键时弹出的菜单最上面就会出现一个“Generate Comment with Berserk”的选项。5. 运行、调试与打包代码写完了让我们看看它能不能跑起来最后再把它打包成可以安装的插件。5.1 运行与调试插件IDEA 为插件开发提供了极其方便的调试方式你可以直接运行一个安装了当前插件的 IDEA 沙盒实例。在项目界面找到并打开src/main/resources/META-INF/plugin.xml。在文件内容区域右键选择Run ‘Plugin’。或者你也可以在右上角的运行配置下拉框中选择Plugin然后点击绿色的运行按钮。IDEA 会开始构建插件并启动一个新的 IDEA 实例称为 “IDE Instance” 或沙盒。在这个新打开的 IDEA 里新建一个 Java 项目或打开一个现有项目。在编辑器里写几行代码比如一个简单的方法。选中这几行代码右键点击你应该能看到我们添加的“Generate Comment with Berserk”菜单项。点击它如果一切配置正确特别是 API 地址和密钥它会调用 UNIT-00 并生成注释插入到代码上方。如果遇到错误检查运行窗口的日志。调试就和普通 Java 程序一样你可以在自己的动作类里打上断点然后以Debug模式运行Plugin配置。5.2 打包插件测试没问题后就可以打包了这样就能分享给其他人安装。点击 IDEA 顶部菜单栏的Build构建然后选择Prepare Plugin Module ‘你的模块名’ For Deployment。这个操作会在你的项目根目录下生成一个.jar文件比如BerserkCoderAssistant.jar。这个 JAR 文件就是你的插件包。其他人可以在他们的 IDEA 中通过File - Settings - Plugins - 齿轮图标 - Install Plugin from Disk...来选择这个 JAR 文件进行安装。6. 下一步可以做什么走到这里你已经成功创建了一个具备基础功能的 IDEA 插件。虽然它现在只实现了“生成注释”这一个功能但整个框架已经搭起来了。你可以用类似的模式去实现更酷的功能智能补全实现CompletionContributor类在用户输入时提供补全建议。这需要更精细地获取光标前后的代码上下文。重构建议创建一个工具窗口ToolWindow展示 UNIT-00 对当前文件或模块的分析结果和建议。配置界面使用Configurable接口提供一个设置页面让用户可以方便地输入和修改 UNIT-00 的 API 地址和密钥而不是硬编码。错误处理与用户体验添加网络超时、重试机制以及更美观的进度提示和结果展示。开发插件就像搭积木最重要的第一步已经完成。你可以基于这个项目不断添加新的“积木”让它变得越来越强大。多看看 IDEA 官方插件开发文档里面有很多现成的例子和扩展点说明能给你带来更多灵感。获取更多AI镜像想探索更多AI镜像和应用场景访问 CSDN星图镜像广场提供丰富的预置镜像覆盖大模型推理、图像生成、视频生成、模型微调等多个领域支持一键部署。
IntelliJ IDEA 集成 UNIT-00：Berserk Interface 插件开发指南
IntelliJ IDEA 集成 UNIT-00Berserk Interface 插件开发指南你是不是也想过要是写代码的时候IDE能像有个经验丰富的搭档在旁边随时帮你补全代码、解释逻辑、甚至重构优化那该多省心今天我们就来动手实现这个想法。我会带你一步步在 IntelliJ IDEA 里开发一个能集成 UNIT-00Berserk Interface 的智能插件让它成为你编码时的得力助手。这个插件能做什么呢简单来说就是让 UNIT-00 的能力直接嵌入到你的开发环境中。比如你写了一半的方法它能帮你智能补全你看不懂的复杂代码块它能生成清晰的注释甚至它还能分析你的代码结构给出重构建议。整个过程我们都会从零开始用最直白的方式讲清楚即使你之前没怎么接触过 IDEA 插件开发也能跟着做出来。1. 环境准备与项目创建工欲善其事必先利其器。在开始敲代码之前我们得先把“厨房”收拾好。1.1 安装 IntelliJ IDEA 与插件 SDK首先你需要一个 IntelliJ IDEA。如果你还没有可以去官网下载社区版Community Edition它对于插件开发来说完全够用而且是免费的。下载后按照提示安装即可这个过程和安装其他软件没什么两样。安装好 IDEA 后打开它我们还需要配置插件开发环境。IDEA 本身已经集成了插件开发所需的 SDK软件开发工具包我们只需要确认一下。打开 IDEA点击File文件菜单选择Project Structure...项目结构。在左侧找到Platform Settings平台设置下的SDKs。检查列表里是否有一个名为IntelliJ IDEA Plugin SDK的条目。通常IDEA 会自动检测并添加它。如果没有点击左上角的号选择Add IntelliJ Platform Plugin SDK...然后选择你 IDEA 的安装目录它会自动配置好。这就好比你要做木工IDEA 是你的工作台而这个 Plugin SDK 就是你工具箱里最核心的那套凿子和刨子。1.2 创建你的第一个插件项目环境好了我们来创建项目。这个过程 IDEA 提供了模板非常方便。在 IDEA 启动界面或通过File - New - Project...新建项目。在左侧选择IntelliJ Platform Plugin。右边会显示项目信息。给项目起个名字比如BerserkCoderAssistant。选择好项目存放的位置。最关键的一步是选择SDK。点击下拉框选择我们刚才确认过的IntelliJ IDEA Plugin SDK。点击Create创建。稍等片刻一个最基础的插件项目骨架就生成好了。你会看到项目里自动包含了一些文件其中最重要的是src/main/resources/META-INF/plugin.xml这是插件的“身份证”和“说明书”我们后面会详细配置它。2. 理解插件基础与连接 UNIT-00项目建好了但在动手连接 UNIT-00 之前我们得先花几分钟了解一下 IDEA 插件是怎么工作的以及我们打算让 UNIT-00 帮什么忙。2.1 IDEA 插件能干什么你可以把 IDEA 插件想象成给 IDEA 这个超级跑车加装的“定制配件”。通过插件我们可以监听事件比如用户敲击了键盘、点击了菜单、或者编辑器里的内容发生了变化。访问和操作代码读取当前文件的代码、分析语法结构、甚至直接插入或修改代码文本。扩展界面添加新的工具栏按钮、菜单项、侧边栏工具窗口等等。我们的智能插件核心就是监听用户在编辑器中的操作比如输入停顿、选中代码然后将相关的代码上下文发送给 UNIT-00 接口再把返回的智能建议补全的代码、生成的注释展示给用户或直接应用到代码中。2.2 规划插件功能与 API 调用为了让教程更聚焦我们主要实现三个核心功能你可以把它们看作插件的三个“技能”智能代码补全当用户输入时插件能调用 UNIT-00根据上下文预测并建议接下来的代码行。注释生成用户选中一段代码右键菜单可以选择“生成注释”插件会请求 UNIT-00 为这段代码生成解释性注释。重构建议在项目视图中右键点击文件或目录插件可以分析代码结构并给出 UNIT-00 提供的重构优化建议。要实现这些我们需要和 UNIT-00Berserk Interface 对话。这通常意味着通过 HTTP 请求调用其提供的 API。你需要准备 UNIT-00 的 API 访问地址Endpoint和认证密钥API Key。这部分信息你需要从 UNIT-00 的服务提供商那里获取。在代码中我们会把这些配置放在安全的地方。3. 搭建插件骨架与配置现在我们来给插件“装上轮子”和“方向盘”也就是配置核心文件并建立基础的代码结构。3.1 配置 plugin.xml打开plugin.xml文件它现在可能只有一些基础信息。我们需要对它进行增补告诉 IDEA 我们的插件是谁有什么能力。idea-plugin !-- 插件唯一ID通常用倒置域名格式 -- idcom.yourname.berserk.coder.assistant/id !-- 插件在插件市场显示的名字 -- nameBerserk Coder Assistant/name !-- 版本号 -- version1.0/version !-- 供应商信息 -- vendor emailyour.emailexample.com urlhttps://yourwebsite.comYour Name/vendor !-- 插件描述会显示在插件详情页 -- description![CDATA[ An intelligent coding assistant plugin that integrates UNIT-00: Berserk Interface into IntelliJ IDEA, providing AI-powered code completion, comment generation, and refactoring suggestions. ]]/description !-- 依赖声明我们的插件需要依赖哪些其他插件或平台功能 -- dependscom.intellij.modules.platform/depends !-- 如果你用到其他语言支持比如Python可能需要添加 dependscom.intellij.modules.python/depends -- !-- 扩展点声明这是我们插件扩展IDEA功能的地方 -- extensions defaultExtensionNscom.intellij !-- 例如我们可以在这里注册代码补全贡献器 -- !-- completion.contributor ... / 这个我们稍后在Java代码里用另一种方式实现 -- /extensions actions !-- 这里我们将定义具体的菜单动作比如“生成注释”的右键菜单项 -- !-- 下一节我们会具体添加 -- /actions /idea-plugin这个文件是插件的元数据核心先有个大概了解后面我们会一步步填充actions部分。3.2 创建 API 服务类我们需要一个专门负责和 UNIT-00 通信的“信使”。在src/main/java下你可能需要先创建这些目录新建一个包比如com.berserk.assistant.service然后在这个包里创建一个 Java 类BerserkApiService。这个类主要做两件事构建发送给 UNIT-00 的请求数据以及解析 UNIT-00 返回的结果。这里我们先写一个骨架并模拟一个简单的注释生成请求。package com.berserk.assistant.service; import com.google.gson.Gson; import com.google.gson.JsonObject; import okhttp3.*; import java.io.IOException; public class BerserkApiService { private static final String API_BASE_URL https://api.your-unit00-service.com/v1; // 替换为实际API地址 private static final String API_KEY your-api-key-here; // 替换为你的API密钥 private static final OkHttpClient client new OkHttpClient(); private static final Gson gson new Gson(); // 一个示例方法为选中的代码生成注释 public String generateCommentForCode(String selectedCode, String language) throws IOException { // 1. 构建请求体JSON格式根据UNIT-00的API文档来构造 JsonObject requestBody new JsonObject(); requestBody.addProperty(model, unit-00-codex); // 假设的模型名 requestBody.addProperty(prompt, Generate a concise inline comment for the following language code:\n selectedCode); requestBody.addProperty(max_tokens, 100); RequestBody body RequestBody.create( gson.toJson(requestBody), MediaType.parse(application/json; charsetutf-8) ); // 2. 构建HTTP请求 Request request new Request.Builder() .url(API_BASE_URL /completions) // 假设的补全端点 .post(body) .addHeader(Authorization, Bearer API_KEY) .addHeader(Content-Type, application/json) .build(); // 3. 发送请求并获取响应 try (Response response client.newCall(request).execute()) { if (!response.isSuccessful()) { throw new IOException(Unexpected code response , body: response.body().string()); } String responseBody response.body().string(); // 4. 解析响应JSON提取生成的文本注释 JsonObject jsonResponse gson.fromJson(responseBody, JsonObject.class); // 这里需要根据实际的API响应结构来解析以下为示例 String generatedText jsonResponse.getAsJsonArray(choices) .get(0).getAsJsonObject() .get(text).getAsString().trim(); return generatedText; } } // 后续可以添加更多方法如 codeCompletion, suggestRefactoring 等 }注意你需要将API_BASE_URL和API_KEY替换成真实值。更安全的做法是从 IDEA 的配置PersistentStateComponent中读取这些配置而不是硬编码在代码里。为了教程简洁我们先这样写。另外OkHttp库需要添加依赖我们马上会做。3.3 添加项目依赖我们的插件需要用到OkHttp来发网络请求用Gson来解析 JSON。我们需要在项目的build.gradle.kts如果你创建的是 Gradle 项目或pom.xmlMaven 项目里声明这些依赖。这里以更常见的 Gradle 为例。打开项目根目录下的build.gradle.kts文件在dependencies块里添加dependencies { // ... 其他已有依赖 implementation(com.squareup.okhttp3:okhttp:4.12.0) implementation(com.google.code.gson:gson:2.10.1) }添加后IDEA 通常会提示同步Sync项目点击同步它就会自动下载这些库了。4. 实现核心功能动作与交互骨架搭好了信使ApiService也有了现在该让插件“动起来”了。我们将实现一个具体的功能右键菜单生成注释。4.1 创建“生成注释”动作Action在 IDEA 插件中用户的一个操作比如点击菜单、按钮通常对应一个AnAction类。我们来创建一个。在src/main/java下新建一个包com.berserk.assistant.actions然后创建类GenerateCommentAction。package com.berserk.assistant.actions; import com.berserk.assistant.service.BerserkApiService; import com.intellij.openapi.actionSystem.AnAction; import com.intellij.openapi.actionSystem.AnActionEvent; import com.intellij.openapi.actionSystem.CommonDataKeys; import com.intellij.openapi.editor.Editor; import com.intellij.openapi.project.Project; import com.intellij.openapi.ui.Messages; import org.jetbrains.annotations.NotNull; import java.io.IOException; public class GenerateCommentAction extends AnAction { Override public void update(NotNull AnActionEvent e) { // 这个方法决定动作菜单项是否可见/可用 final Project project e.getProject(); final Editor editor e.getData(CommonDataKeys.EDITOR); // 只在有项目、有编辑器、且编辑器中有文本被选中时该动作才可用 e.getPresentation().setEnabledAndVisible(project ! null editor ! null editor.getSelectionModel().hasSelection()); } Override public void actionPerformed(NotNull AnActionEvent e) { // 当用户点击菜单项时执行这个方法 final Editor editor e.getRequiredData(CommonDataKeys.EDITOR); final Project project e.getProject(); // 1. 获取选中的代码文本 String selectedText editor.getSelectionModel().getSelectedText(); if (selectedText null || selectedText.trim().isEmpty()) { Messages.showWarningDialog(project, Please select some code first., No Code Selected); return; } // 2. 获取文件语言用于构造更精准的提示词 // 这里简化处理实际可能需要从 PsiFile 获取 String language Java; // 示例实际应动态获取 // 3. 调用我们的 API 服务 BerserkApiService apiService new BerserkApiService(); try { // 显示一个“处理中”的提示是个好主意这里省略了 String generatedComment apiService.generateCommentForCode(selectedText, language); // 4. 将生成的注释插入到选中代码的前一行 int selectionStart editor.getSelectionModel().getSelectionStart(); int lineNumber editor.getDocument().getLineNumber(selectionStart); int lineStartOffset editor.getDocument().getLineStartOffset(lineNumber); editor.getDocument().insertString(lineStartOffset, // generatedComment \n); Messages.showInfoMessage(project, Comment generated and inserted above the selection., Success); } catch (IOException ex) { Messages.showErrorDialog(project, Failed to generate comment: ex.getMessage(), API Error); ex.printStackTrace(); } } }这个类做了几件事update方法控制右键菜单里这个项什么时候能点actionPerformed是点击后执行的逻辑——获取选中代码调用 UNIT-00然后把返回的注释插到代码前面。4.2 注册动作到菜单光有动作类还不行得告诉 IDEA 把这个动作放到哪里。我们需要回到plugin.xml文件在actions标签内注册它。actions !-- 将我们的动作添加到编辑器右键菜单PopupMenu -- action idBerserk.GenerateComment classcom.berserk.assistant.actions.GenerateCommentAction textGenerate Comment with Berserk descriptionGenerate AI-powered comment for selected code !-- 指定这个动作添加到“EditorPopupMenu”这个组里 -- add-to-group group-idEditorPopupMenu anchorfirst/ /action /actions这样配置后当你在编辑器里选中代码并点击右键时弹出的菜单最上面就会出现一个“Generate Comment with Berserk”的选项。5. 运行、调试与打包代码写完了让我们看看它能不能跑起来最后再把它打包成可以安装的插件。5.1 运行与调试插件IDEA 为插件开发提供了极其方便的调试方式你可以直接运行一个安装了当前插件的 IDEA 沙盒实例。在项目界面找到并打开src/main/resources/META-INF/plugin.xml。在文件内容区域右键选择Run ‘Plugin’。或者你也可以在右上角的运行配置下拉框中选择Plugin然后点击绿色的运行按钮。IDEA 会开始构建插件并启动一个新的 IDEA 实例称为 “IDE Instance” 或沙盒。在这个新打开的 IDEA 里新建一个 Java 项目或打开一个现有项目。在编辑器里写几行代码比如一个简单的方法。选中这几行代码右键点击你应该能看到我们添加的“Generate Comment with Berserk”菜单项。点击它如果一切配置正确特别是 API 地址和密钥它会调用 UNIT-00 并生成注释插入到代码上方。如果遇到错误检查运行窗口的日志。调试就和普通 Java 程序一样你可以在自己的动作类里打上断点然后以Debug模式运行Plugin配置。5.2 打包插件测试没问题后就可以打包了这样就能分享给其他人安装。点击 IDEA 顶部菜单栏的Build构建然后选择Prepare Plugin Module ‘你的模块名’ For Deployment。这个操作会在你的项目根目录下生成一个.jar文件比如BerserkCoderAssistant.jar。这个 JAR 文件就是你的插件包。其他人可以在他们的 IDEA 中通过File - Settings - Plugins - 齿轮图标 - Install Plugin from Disk...来选择这个 JAR 文件进行安装。6. 下一步可以做什么走到这里你已经成功创建了一个具备基础功能的 IDEA 插件。虽然它现在只实现了“生成注释”这一个功能但整个框架已经搭起来了。你可以用类似的模式去实现更酷的功能智能补全实现CompletionContributor类在用户输入时提供补全建议。这需要更精细地获取光标前后的代码上下文。重构建议创建一个工具窗口ToolWindow展示 UNIT-00 对当前文件或模块的分析结果和建议。配置界面使用Configurable接口提供一个设置页面让用户可以方便地输入和修改 UNIT-00 的 API 地址和密钥而不是硬编码。错误处理与用户体验添加网络超时、重试机制以及更美观的进度提示和结果展示。开发插件就像搭积木最重要的第一步已经完成。你可以基于这个项目不断添加新的“积木”让它变得越来越强大。多看看 IDEA 官方插件开发文档里面有很多现成的例子和扩展点说明能给你带来更多灵感。获取更多AI镜像想探索更多AI镜像和应用场景访问 CSDN星图镜像广场提供丰富的预置镜像覆盖大模型推理、图像生成、视频生成、模型微调等多个领域支持一键部署。
