Idea插件-自定义提示插件部分 Java版 第二部分

最新推荐文章于 2024-08-30 18:40:43 发布
最新推荐文章于 2024-08-30 18:40:43 发布
阅读量347 收藏 9
点赞数 6
文章标签： java intellij-idea 开发语言
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/li1760/article/details/140419653
版权

第一部分是来用来监听用户行为，这一部分是监听后，渲染到idea页面的部分。

渲染部分，主要是逻辑就是，根据监听触发渲染后，获取到用户当前的光标位置和光标上下文，将上下文获取到发整合后发送给AI算法部分，获取到AI模型返回的结果后，渲染回显到idea的页面。

首先，获取光标位置：

	Caret primaryCaret = editor.getCaretModel().getPrimaryCaret();
	//这就是光标的偏移量
	 int selectionStart = primaryCaret.getSelectionStart();

获取到光标位置后来获取上下文（这个是idea插件获取最简单的方式，暂时先用这个最简单的，后续对代码上下文的拆分会有单独的一部分会单独介绍）：

	 public static String[] getCaretContext(Caret caret, int contextLength) {
        Editor editor = caret.getEditor();
        Document document = editor.getDocument();
        int caretOffset = caret.getOffset();

        // 检查是否超出了文档的范围
        int start = Math.max(0, caretOffset - contextLength);
        int end = Math.min(document.getTextLength(), caretOffset + contextLength);

        // 获取光标前后的文本
        String beforeText = document.getText(new TextRange(start, caretOffset)).trim();
        String caretText = String.valueOf(document.getCharsSequence().charAt(caretOffset)); // 假设我们只关心光标位置的单个字符
        String afterText = document.getText(new TextRange(caretOffset, end)).trim();

        return new String[]{beforeText, caretText, afterText};
    }

这个获取上下文代码中，其实

        Editor editor = caret.getEditor();
        Document document = editor.getDocument();
        int caretOffset = caret.getOffset();

这部分和上文获取光标位置是一致的，都是获取偏移量。

获取到的光标位置和上下文后，整合后发给AI模型，但我们中间包装一次API，统计用量和一些用户验证相关的部分，这个我就先省略掉了，而且大伙还是根据自己项目使用的续写模型自己来整合自己的出入参。

Caret primaryCaret = editor.getCaretModel().getPrimaryCaret();
//当初是简单的获取上下文的2000个偏移量
String[] caretContext = getCaretContext(primaryCaret,2000);
String url = config.autoCompleteUrl;
AutoCompleteVo vo = new AutoCompleteVo();
vo.setBeginCode(stringAtomicReferenceArray.get(0));
vo.setEndCode(stringAtomicReferenceArray.get(2));
String params = JSONObject.toJSONString(vo);
String deviceId = 设备的唯一标识;//我们主要是为了控制每一个用户只能使用一个idea插件，如果换设备，可以重新申请
String json = HttpUtils.sendPostJson(url, params, deviceId);

附上HttpUtils.sendPostJson方法的代码

    public static String sendPostJson(String uri, String json, String deviceId)  {
        String response = "";
        try {
            URL url = new URL(uri);
            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
            conn.setRequestMethod("POST");
            conn.setRequestProperty("Accept", "application/json");
            conn.setRequestProperty("Content-Type", "application/json; charset=utf-8");
//            conn.setReadTimeout(1200);
             conn.setRequestProperty("deviceId", deviceId);
            conn.setConnectTimeout(200);
            conn.setDoOutput(true);
            String jsonInputString = json;
            try (OutputStream os = conn.getOutputStream()) {
                byte[] input = jsonInputString.getBytes("utf-8");
                os.write(input, 0, input.length);
            }
            int responseCode = conn.getResponseCode();
            log.info("sendPostJson responseCode=" + responseCode);
            if (responseCode == HttpURLConnection.HTTP_OK) { // success
                try (InputStream is = conn.getInputStream()) {
                    BufferedReader reader = new BufferedReader(new InputStreamReader(is, "utf-8"));
                    String line;
                    while ((line = reader.readLine()) != null) {
                        response += line;
                    }
                }
            }
            conn.disconnect();
        } catch (Exception e) {
            log.error("sendPostJson error", e);
        } finally {
            return response;
        }
    }

json是AI模型返回的内容，根据自身返回的内容进行拆解，获取AI模型返回的内容“text”。再接下来就是根据text去渲染到idea插件上。

InlayProperties properties = new InlayProperties();
properties.relatesToPrecedingText(true);
properties.disableSoftWrapping(true);
if (text.split("\n").length > 1) {
  Inlay<ContinueMultilineCustomElementRenderer> continueMultilineCustomElementRendererInlay = editor.getInlayModel().addBlockElement(caretOffset , properties, new ContinueMultilineCustomElementRenderer(editor, text));

} else {
  Inlay<ContinueCustomElementRenderer> continueCustomElementRendererInlay = editor.getInlayModel().addInlineElement(caretOffset, properties, new ContinueCustomElementRenderer(editor, text));
}

这两个会根据返回的内容来判断是否是多行，如果是多行，使用多行渲染。如果是单行进行单行渲染。渲染方法也是idea插件自己提供的。其中caretOffset是光标位置。还有ContinueMultilineCustomElementRenderer和ContinueCustomElementRenderer，这个类就是渲染内容的类，一个是多行渲染和单行渲染。
ContinueCustomElementRenderer（单行渲染）：

package com.xxx.action.autocomplete;

import com.intellij.openapi.actionSystem.ActionGroup;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.editor.EditorCustomElementRenderer;
import com.intellij.openapi.editor.Inlay;
import com.intellij.openapi.editor.colors.EditorFontType;
import com.intellij.openapi.editor.impl.EditorImpl;
import com.intellij.openapi.editor.impl.FontInfo;
import com.intellij.openapi.editor.markup.GutterIconRenderer;
import com.intellij.openapi.editor.markup.TextAttributes;
import com.intellij.ui.JBColor;
import org.jetbrains.annotations.NonNls;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.awt.*;
import java.awt.font.FontRenderContext;
import java.awt.font.GlyphVector;
import java.awt.geom.Rectangle2D;

/**
 * 自定义的EditorCustomElementRenderer
 * 主要用于编辑器单行续写内容的渲染
 * @Date: 2024年06月13日 11:16
 */
public class ContinueCustomElementRenderer implements EditorCustomElementRenderer{
    private final Editor editor;
    private final String text;

    public ContinueCustomElementRenderer(Editor editor, String text) {
        this.editor = editor;
        this.text = text;
    }

    private Font getFont() {
        Font editorFont = editor.getColorsScheme().getFont(EditorFontType.PLAIN);
        Font derivedFont = editorFont.deriveFont(Font.PLAIN);
        return derivedFont != null ? derivedFont : editorFont;
    }

    @Override
    public int calcWidthInPixels(@NotNull Inlay inlay) {
        EditorImpl editorImpl = (EditorImpl) inlay.getEditor();

        return editorImpl.getFontMetrics(Font.PLAIN).stringWidth(text);
    }

    @Override
    public int calcHeightInPixels(@NotNull Inlay inlay) {
        return EditorCustomElementRenderer.super.calcHeightInPixels(inlay);
    }

    @Override
    public void paint(@NotNull Inlay inlay, @NotNull Graphics g, @NotNull Rectangle targetRegion, @NotNull TextAttributes textAttributes) {
        Graphics2D g2d = (Graphics2D) g.create();
        g2d.setColor(JBColor.GRAY);
        g2d.setFont(getFont());
        g2d.drawString(text, targetRegion.x, targetRegion.y + offsetY());
        g2d.dispose();
    }

    private int offsetY() {
        FontRenderContext fontRenderContext = FontInfo.getFontRenderContext(editor.getContentComponent());
        GlyphVector glyphVector = getFont().createGlyphVector(fontRenderContext, text);
        Rectangle2D visualBounds = glyphVector.getVisualBounds();
        double fontHeight = visualBounds.getHeight();
        double height = (editor.getLineHeight() + fontHeight) / 2;
        return (int) Math.ceil(height);
    }

    @Override
    public void paint(@NotNull Inlay inlay, @NotNull Graphics2D g, @NotNull Rectangle2D targetRegion, @NotNull TextAttributes textAttributes) {
        EditorCustomElementRenderer.super.paint(inlay, g, targetRegion, textAttributes);
    }

    @Override
    public @Nullable @NonNls String getContextMenuGroupId(@NotNull Inlay inlay) {
        return EditorCustomElementRenderer.super.getContextMenuGroupId(inlay);
    }

    @Override
    public @Nullable ActionGroup getContextMenuGroup(@NotNull Inlay inlay) {
        return EditorCustomElementRenderer.super.getContextMenuGroup(inlay);
    }

    @Override
    public @Nullable GutterIconRenderer calcGutterIconRenderer(@NotNull Inlay inlay) {
        return EditorCustomElementRenderer.super.calcGutterIconRenderer(inlay);
    }
}

ContinueMultilineCustomElementRenderer（多行渲染）：

package com.xxx.action.autocomplete;

import com.intellij.openapi.actionSystem.ActionGroup;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.editor.EditorCustomElementRenderer;
import com.intellij.openapi.editor.Inlay;
import com.intellij.openapi.editor.colors.EditorFontType;
import com.intellij.openapi.editor.impl.EditorImpl;
import com.intellij.openapi.editor.impl.FontInfo;
import com.intellij.openapi.editor.markup.GutterIconRenderer;
import com.intellij.openapi.editor.markup.TextAttributes;
import com.intellij.ui.JBColor;
import org.jetbrains.annotations.NonNls;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.awt.*;
import java.awt.font.FontRenderContext;
import java.awt.font.GlyphVector;
import java.awt.geom.Rectangle2D;

/**
 * 自定义的EditorCustomElementRenderer
 * 主要用于编辑器多行续写内容的渲染
 * @Date: 2024年06月13日 11:29
 */
public class ContinueMultilineCustomElementRenderer implements EditorCustomElementRenderer{

    private final Editor editor;
    private final String text;

    public ContinueMultilineCustomElementRenderer(Editor editor, String text) {
        this.editor = editor;
        this.text = text;
    }

    private Font getFont() {
        Font editorFont = editor.getColorsScheme().getFont(EditorFontType.PLAIN);
        Font derivedFont = editorFont.deriveFont(Font.PLAIN);
        return derivedFont != null ? derivedFont : editorFont;
    }

    private int offsetY() {
        FontRenderContext fontRenderContext = FontInfo.getFontRenderContext(editor.getContentComponent());
        GlyphVector glyphVector = getFont().createGlyphVector(fontRenderContext, text);
        Rectangle2D visualBounds = glyphVector.getVisualBounds();
        double fontHeight = visualBounds.getHeight();
        double height = (editor.getLineHeight() + fontHeight) / 2;
        return (int) Math.ceil(height);
    }

    private int offsetX() {
        int currentColumn = editor.getCaretModel().getPrimaryCaret().getLogicalPosition().column;
        FontRenderContext fontRenderContext = FontInfo.getFontRenderContext(editor.getContentComponent());
        GlyphVector glyphVector = getFont().createGlyphVector(fontRenderContext, text);
        Rectangle2D visualBounds = glyphVector.getVisualBounds();
        double fontWidth = visualBounds.getWidth();
        int widthBeforeCaret = ((EditorImpl) editor).getFontMetrics(Font.PLAIN).stringWidth(text.substring(0, currentColumn));
        int horizontalScrollOffset = ((EditorImpl) editor).getScrollingModel().getHorizontalScrollOffset();
        return Math.max(0, widthBeforeCaret - horizontalScrollOffset);
    }


    @Override
    public int calcWidthInPixels(@NotNull Inlay inlay) {
        String[] lines = text.split("\\r?\\n");
        int maxLen = 0;
        for (String line : lines) {
            int len = ((EditorImpl) inlay.getEditor()).getFontMetrics(Font.PLAIN).stringWidth(line);
            maxLen = Math.max(maxLen, len);
        }
        return maxLen;
    }

    @Override
    public int calcHeightInPixels(@NotNull Inlay inlay) {
        String[] lines = text.split("\\r?\\n");
        return ((EditorImpl) inlay.getEditor()).getLineHeight() * lines.length;
    }

    @Override
    public void paint(@NotNull Inlay inlay, @NotNull Graphics g, @NotNull Rectangle targetRegion, @NotNull TextAttributes textAttributes) {

        Graphics2D g2d = (Graphics2D) g.create();
        g2d.setColor(JBColor.GRAY);
        g2d.setFont(getFont());
        int additionalYOffset = -editor.getLineHeight();
        boolean isFirstLine = true;
        String[] lines = text.split("\\r?\\n");
        for (String line : lines) {
            int x = isFirstLine ? targetRegion.x + offsetX() : targetRegion.x;
            int y = targetRegion.y + offsetY() + additionalYOffset;
            g2d.drawString(line, x, y);
            additionalYOffset += editor.getLineHeight();
            isFirstLine = false;
        }
    }

    @Override
    public void paint(@NotNull Inlay inlay, @NotNull Graphics2D g, @NotNull Rectangle2D targetRegion, @NotNull TextAttributes textAttributes) {
        EditorCustomElementRenderer.super.paint(inlay, g, targetRegion, textAttributes);
    }

    @Override
    public @Nullable @NonNls String getContextMenuGroupId(@NotNull Inlay inlay) {
        return EditorCustomElementRenderer.super.getContextMenuGroupId(inlay);
    }

    @Override
    public @Nullable ActionGroup getContextMenuGroup(@NotNull Inlay inlay) {
        return EditorCustomElementRenderer.super.getContextMenuGroup(inlay);
    }

    @Override
    public @Nullable GutterIconRenderer calcGutterIconRenderer(@NotNull Inlay inlay) {
        return EditorCustomElementRenderer.super.calcGutterIconRenderer(inlay);
    }
}

以上就是获取光标位置，然后获取上下文，在请求到AI模型后渲染到idea编辑器中。

万千之喜
关注
  • 6
    点赞
  • 踩
  • 9
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
自定义IDEA代码提示插件开发
tianyouououou的博客
11-09 2601
通过开发IDEA插件实现一个自定义代码提示工具。
（三）IDEA插件开发教程-UI篇-创建自定义弹框
qq_39017064的博客
04-11 4337
这里以笔者的插件MyCode展示自定义弹框效果 核心实现类 1.JBPopupFactory JBPopupFactory是IntelliJ 平台创建弹框的实现类，它支持根据不同需求显示不同类型的组件弹框。弹框可以设置标题、调整大小、复现上次打开弹窗的位置以及嵌套显示弹窗等，JBPopupFactory常用的方法如下所示： 函数 功能 createComponentPopupBuilder() 使用最多，允许您在弹出窗口中显示任何 Swing 组件 Createpupchooserbu
Idea插件-自定义提示插件部分 Java版 第一部分
li1760的博客
07-13 171
前段时间搞AI代码提示的Idea插件。我整理了一下idea插件部分的代码，包括监听用户触发调用代码提示、代码提示回显、使用tab采纳，使用esc取消提示等功能。
IntelliJ IDEA 插件开发中监听用户的保存事件
liudong200618的博客
03-30 438
在上面的示例中，我们创建了一个实现了 `VirtualFileListener` 接口的监听器类 `MySaveListener`，并在其中实现了各种文件事件的处理方法。然后，在插件初始化时，通过 `VirtualFileManager.getInstance().addVirtualFileListener()` 方法注册了这个监听器。通过这种方式，你的插件就能够监听到用户在 IntelliJ IDEA 中保存文件的事件，并做出相应的处理。
IntelliJ插件开发 - 常用监听器
liuzhenghua66的博客
12-01 604
IntelliJ插件开发 - 消息文章中介绍了消息的使用方法，除了自定义消息外，开发插件过程中需要经常与系统内置消息打交道，例如：插件被加载，卸载，用户输入了一些命令等。
IntelliJ IDE 插件开发 | （五）VFS 与编辑器
庄周de蝴蝶的博客
01-24 1936
在前几篇文章中主要介绍了关于 IntelliJ IDE 插件开发的基础知识，这部分内容对开发一些小功能的插件的开发已经足够。不过，如果想要开发一些与具体编程语言相关的、提升开发效率的插件（例如MybatisX），那么前几篇的内容就不足以支撑了。而从本篇开始，则会介绍实现相关功能所需要的知识： VFS、编辑器、PSI、自定义语言等，最后再以两三个插件的实战开发（例如老生常谈的代码生成）进行结尾，本文涉及到的完整代码已上传到GitHub。
关于Editor和Renderer的一点认识
执天使
12-17 1663
在Sun的官方网站上对Editor和Renderer的解释是如下：Renderer：Instead, a single cell renderer is generally used to draw all of the cells that contain the same type of data. You can think of the renderer as a configurable
idea插件开发-自定义语言7-Code Completion
编程是一门实践技术
01-24 1500
Intellij idea plugin程序开发-代码自动完成
ideaIU-2021.1.3.zip
03-23
描述中的"ideaIU-2021.1.3"进一步确认了这是IntelliJ IDEA的某个发行版，可能包含了新功能、性能优化和错误修复。通常，版本号的格式为“年.月.小版本”，其中“2021”代表发布年份，“1”代表该年度内的主要版本，...
Intellij platform plugin开发指导手册，java集成开发环境Intellij idea插件开发
10-11
学习本手册时，初学者可以从第一部分开始，了解插件开发的基本概念和流程，然后根据实际需求，选择性地深入第二或第三部分。对于有经验的开发者，可以直接进入第三部分，探索更高级的插件开发技术。无论你是想要编写...
intellij-idea php插件-18134203.513
05-21
IntelliJ IDEA是一款备受开发者喜爱的集成开发环境（IDE），尤其在Java开发领域中占据了重要地位。然而，它也提供了对其他编程语言的支持，包括PHP。"intellij-idea php插件-18134203.513"就是这样一个专为IntelliJ ...
IntelliJ IDE 插件开发 | （十一）解析 Mybatis 的控制台 SQL 日志
庄周de蝴蝶的博客
06-29 1130
在前面的十个章节中主要介绍了关于插件开发的基础知识，从本节开始则会通过一个一个实战的 demo 样例来展示这些内容的综合运用，同时会偶尔夹杂一些进阶的理论内容。本文则是开发一个解析控制台中 MyBatis 的 SQL 日志，通过点击图标即可实现获取到完整 SQL 日志并复制到粘贴板中，效果如下图所示，另外本文所涉及到的完整代码也已上传到GitHub。/*** 用于判断鼠标是否在渲染的图标上停留*/// 处理 SQL 日志和参数信息并将拼接结果保存到粘贴板中value", it)
扩展HT for Web之HTML5表格组件的Renderer和Editor
HT for Web 3D
05-26 201
在HT for Web提供了一下几种常用的Editor，分别是： slider：拉条 color picker：颜色选择器 enum：枚举类型 boolean：真假编辑器 string：普通的文本编辑器 除了这几种常用编辑器之外，用户还可以通过继承ht.widget.BaseItemEditor类来实现自定义编辑器。 而渲染器，在HT for Web提供常用的Renderer有...
IDEA使用技巧，无鼠标操作又进了一步，20181217
yethyeth的专栏
03-26 3214
转自：http://hackday.cn/?p=702 直接上干货 演出模式 使用【Presentation Mode】，将IDEA弄到最大，可以让你只关注一个类里面的代码，进行毫无干扰的coding。 可以使用Alt+V快捷键，弹出View视图，然后选择Enter Presentation Mode。 这个模式的好处就是，可以让你更加专注，因为你只能看到特定某个类的代码。...
idea插件开发-Editor
编程是一门实践技术
05-30 1602
idea中操作文件有两种方式一种是通过Document另一种是PSI方式，其中Editor就是操作Document的载体-编辑器。Ediotr更的是如何打开、定位文档等功能。本章将通过一个例子来完成以下功能：1、使用文本；2、编辑坐标系；3、编辑器事件；4、文本选择；5、文本编辑。 上述代码中可能过以下两种方式取得Editor中的数据模型DataContext： Editor可选择的数据模型：CaretModel、FoldingModel、Inden
idea插件 监听编辑板块切换
qq_15248459的博客
09-17 1900
监听编辑板块切换开发工具：IntellJ Idea 开发语言：Kotlin功能描述：实现监听编辑板块切换效果图： 打印日志： 下面是实现该功能的代码：<idea-plugin> <id>com.your.company.unique.plugin.id</id> <name>Djy</name> <version>1.0</version> <vendor email="sup
[IDEA插件从零开始系列]-2.创建带GUI的对话框并修改文件
cuixx的专栏
12-22 1933
教你如何创建一个带自定义GUI的对话框, 并根据对话框输入的或者配置的内容在文件中插入数据或者替换选择的数据
Java基础（6）- Java代码笔记3
最新发布
weixin_47062560的博客
08-30 617
数据类型[][] 数组名 = new 数据类型[m][n]数据类型 数组名[][] = new 数据类型[m][n]数据类型[] 数组名[] = new 数据类型[m][n]m：代表二维数组长度n：代表二维数组中每个一维数组的长度。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

万千之喜

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值