IntelliJ IDE 插件开发 | （四）来查收你的 IDEA 使用报告吧

news/2024/7/16 9:28:59 标签: kotlin, idea, ide

系列文章

  • IntelliJ IDE 插件开发 |（一）快速入门
  • IntelliJ IDE 插件开发 |（二）UI 界面与数据持久化
  • IntelliJ IDE 插件开发 |（三）消息通知与事件监听
  • IntelliJ IDE 插件开发 |（四）来查收你的 IDEA 使用报告吧

前言

在上篇文章的结尾提到本文将参考 VS Code 中 TimerMaster 插件的实现效果来实现一个在 IDEA 中统计编码情况的插件，TimerMaster 的效果如下：

image-20231225092034655

image-20231225092049204

本文最终实现的插件效果如下（增加了 CV 操作和代码提交的统计😎）：

image-20231228112248383

Clip_20231225_092825

由于本文涉及到大部分知识都是在前几篇文章中介绍过的，因此本文只对关键功能点和实现方式进行介绍，一些基础知识和配置就不再讲解，该插件的完整代码已上传到GitHub。

实现思路

在开发前，正如前言中提到的，是参考 TimerMaster 的实现效果。因此首先确认了要实现的功能如下：

  1. 可以统计当天、昨天、过去七天和每天平均的代码活动。
  2. 统计项包括编辑器使用/活跃时间、添加/删除的代码行数、总的键入数、扩展增加了 CV(懂得都懂) 和代码提交活动的统计。
  3. 通过点击右键菜单项后在控制台进行展示。

对于第一点，主要通过本地持久化数据进行实现，当天数据和历史数据分别进行存储(选择了 JSON 格式进行存储)，然后按需取用进行统计即可。

对于第二点，涉及的功能较多，这里分开进行介绍：

  • 编辑器使用/活跃时间

    增加一个对项目生命周期的监听器，在启动事件中注册一个定时任务，每隔 n 秒(可配置)将使用时间增加 n 秒。增加一个对文档和游标位置的监听器，只要有文档操作或者游标的移动都认为是活跃状态，然后更新最新操作时间，在定时任务中会将当前时间和最新操作时间进行比较，如果不大于 m 秒(可配置)就认为编辑器处于活跃状态，并将活跃时间增加 n 秒。

  • 添加/删除的代码行数、总的键入数

    通过增加对文档的监听器，根据变更内容中的换行数目得到添加/删除的代码行数，同时根据变更内容长度是否为 1 或者为空白符来决定是否增加键入数。

  • CV 统计

    增加一个对复制粘贴事件的监听器，分别在复制和粘贴事件中增加相应的次数。

  • 代码提交统计

    监听自带 GIT 插件提供的 PUSH 事件监听器，在提交完成事件中增加对提交提交次数的统计。

对于第三点，只需要增加一个 action 并绑定到右键菜单，然后通过 ToolWindowFactory 获取到控制台视图来输出计算得到的结果即可。

根据个人需求，统计每天最早/最晚运行时间、活跃时间段、数据云存储等也很容易实现，本文就不进行拓展了。

代码实现及讲解

配置界面

在实现思路中提到了定时器间隔和代码活跃时间间隔是可配置的，由于界面比较简单，这里使用了 Kotlin UI DSL 的方式(参考第二篇文章使用 Swing UI 的拖拽方式也很容易实现)，对应代码及效果如下：

kotlin">// TimerMasterConfig 中部分代码

private var panel = panel { 
    row("更新间隔(秒): ") {
        intTextField()
            .bindIntText(model::updateInterval)
            .comment("<icon src='AllIcons.General.Information'>&nbsp;不设置或者小于 10, 最终都为 10.")
    }
    
    row("活跃间隔(秒): ") {
        intTextField()
            .bindIntText(model::activeInterval)
            .comment("<icon src='AllIcons.General.Information'>&nbsp;不设置或者小于 30, 最终都为 30.")
    }
}

image-20231228124547932

存储方式及格式

根据要存储的数据，定义了一个统计数据类，并增加了一个创建时间便于归档：

kotlin">data class StatisticsData(
    var runTime: Long = 0,
    var activeTime: Long = 0,
    var keyCount: Long = 0,
    var addLineCount: Long = 0,
    var removeLineCount: Long = 0,
    var copyCount: Long = 0,
    var pasteCount: Long = 0,
    var pushCount: Long = 0,
    var createDate: String = Utils.getTodayYmd()
)

同时为了便于解析和处理，在进行本地持久化的时候，使用了 GSON 库来进行数据的 JSON (反)序列化操作。

编辑器运行/活跃时间

在前文中提到这里是通过在项目启动事件中增加定时任务来实现的，先展示代码：

kotlin">class ProjectStartListener: ProjectActivity, Disposable {
    
    private val state = TimerMasterState.getInstance()

    override suspend fun execute(project: Project) {
        while (true) {
            delay(TimeUnit.SECONDS.toMillis(state.updateInterval.toLong()))
            run {
                // 避免多个项目运行时间统计多次, 增加满足以下规则才进行时间统计: 
                // 当前项目与配置信息一致, 或配置信息为空, 或配置信息内的项目不处于打开状态
                val projectPath = project.locationHash
                val firstOrNull = ProjectManager.getInstance().openProjects.firstOrNull { it.locationHash == state.runProjectPath }
                // 判断编辑器是否处于活跃状态
                val active = (System.currentTimeMillis() - state.activeTime) / 1000 <= state.activeInterval
                state.runProjectPath.takeIf { it == projectPath || it.isBlank() || firstOrNull == null }?.let { 
                    state.runProjectPath = projectPath
                    val data = Utils.parse(state.statisticsData, TypeToken.get(StatisticsData::class.java))
                    if (data.createDate == Utils.getTodayYmd()) {
                        // 计算运行和活跃时间
                        data.runTime += state.updateInterval
                        if (active) {
                            data.activeTime += state.updateInterval
                        }
                        state.statisticsData = Utils.stringify(data)
                    } else {
                        // 存储的如果不是当日的数据, 则将数据加入到历史数据, 然后再初始化数据
                        val arr = Utils.parse(state.historyData, object : TypeToken<MutableList<String>>() {})
                        arr.add(state.statisticsData)
                        state.historyData = Utils.stringify(arr)
                        val newData = StatisticsData()
                        newData.runTime = state.updateInterval.toLong()
                        if (active) {
                            newData.activeTime = state.updateInterval.toLong()
                        }
                        state.statisticsData = Utils.stringify(newData)
                    }
                }
            }
        }
    }

    override fun dispose() {
    }

}

增加以下配置：

<extensions defaultExtensionNs="com.intellij">
	<postStartupActivity
        implementation="cn.butterfly.timermaster.listener.ProjectStartListener"/>
</extensions>

这里需要注意的是我们可能会同时打开多个项目，因此为了避免运行和活跃时间被多次计算，这里通过使用 runProjectPath 来存储当前项目路径的 hash 值，只有该值为空（暂无运行的项目）或者该值不空但打开的项目中没有等于该值的项目（项目打开后又关闭）时才会更新该值为当前运行项目的路径 hash 值。相应地，也只有满足以上条件才会统计运行和活跃时间，同时根据 createDate 字段来决定是在当日数据上进行累积，还是归档该数据然后初始化今日数据。

添加/删除的代码行数、总的键入数

这里是在编辑器创建事件中，对相应的文档和光标移动增加监听器来统计添加/删除的代码行数、总的键入数，此外还会更新编辑器活跃时间（Utils.initData() 中进行记录），代码如下：

kotlin">class EditorListener: EditorFactoryListener, BulkAwareDocumentListener, CaretListener {
    
    private val state = TimerMasterState.getInstance()
    
    private val fileSet = mutableSetOf<String>()

    override fun editorCreated(event: EditorFactoryEvent) {
        // 避免重复增加监听器
        val file = FileDocumentManager.getInstance().getFile(event.editor.document) ?: return
        if (file.path in fileSet) {
            return
        }
        fileSet.add(file.path)
        // 监听编辑操作
        event.editor.document.addDocumentListener(this)
        // 监听光标移动事件
        event.editor.caretModel.addCaretListener(this)
    }

    override fun documentChangedNonBulk(event: DocumentEvent) {
        val data = Utils.initData()
        event.takeIf { (it.oldFragment.isNotEmpty() or it.newFragment.isNotEmpty()) or !it.isWholeTextReplaced }?.let {
            // 只对字符长度为 1 和非空空白符的情况进行统计
            if (it.newFragment.isNotEmpty() && (it.newFragment.length == 1 || it.newFragment.trim().isEmpty())) {
                ++data.keyCount
            }
            // 根据文档代码段变更信息判断是新增还是删除行
            if (it.oldFragment.contains('\n')) {
                data.removeLineCount += it.oldFragment.count { item -> item == '\n' }
            }
            if (it.newFragment.contains('\n')) {
                data.addLineCount += it.newFragment.count { item -> item == '\n' }
            }
        }
        state.statisticsData = Utils.stringify(data)
    }

    override fun caretPositionChanged(event: CaretEvent) {
        state.statisticsData = Utils.stringify(Utils.initData())
    }
    
}

增加以下配置：

<extensions defaultExtensionNs="com.intellij">
	<editorFactoryListener implementation="cn.butterfly.timermaster.listener.EditorListener"/>
</extensions>

结合注释，代码不难理解，这里不再介绍。

CV 统计

关于复制粘贴的统计也很简单，官方给我们提供了相应的扩展点，我们增加相应的实现即可：

kotlin">class CopyPasteListener: CopyPastePreProcessor {
    
    private val state = TimerMasterState.getInstance()
    
    override fun preprocessOnCopy(p0: PsiFile?, p1: IntArray?, p2: IntArray?, p3: String?): String? {
        val data = Utils.initData()
        ++data.copyCount
        state.statisticsData = Utils.stringify(data)
        return null
    }
    override fun preprocessOnPaste(p0: Project?, p1: PsiFile?, p2: Editor?, text: String?, p4: RawText?): String {
        val data = Utils.initData()
        ++data.pasteCount
        state.statisticsData = Utils.stringify(data)
        return text ?: ""
    }

}

增加以下配置：

<extensions defaultExtensionNs="com.intellij">
    <copyPastePreProcessor
        implementation="cn.butterfly.timermaster.listener.CopyPasteListener"/>
</extensions>

而如何知道官方给我们提供了哪些扩展点，在上一篇文章中也说明查看官方文档即可，例如复制粘贴的扩展点：

image-20231228132128049

代码提交统计

这里使用的GitPushListener也是在上述官方文档中提供的：

image-20231228132534460

使用方式也很简单：

kotlin">class GitListener: GitPushListener {
    
    private val state = TimerMasterState.getInstance()
    
    override fun onCompleted(repository: GitRepository, pushResult: GitPushRepoResult) {
        val data = Utils.initData()
        ++data.pushCount
        state.statisticsData = Utils.stringify(data)
    }
    
}

增加以下配置：

<depends>Git4Idea</depends>

<projectListeners>
    <listener class="cn.butterfly.timermaster.listener.GitListener"
              topic="git4idea>idea.push.GitPushListener"/>
</projectListeners>

build.gradle.kts中也需要增加以下配置：

kotlin">intellij {
    // 用到的插件
    plugins.set(listOf("Git4Idea"))
}

由于GitPushListener是自带 GIT 插件所提供的监听器，因此还增加了第一行，其中Git4Idea是 GIT 的插件 id：

image-20231228132835555

展示报告

有了以上获取和存储数据的基础，这里就很简单了，关于得到展示数据的方式这里就不介绍了，大家可以去查看TimerMasterOutputAction的源码，这里主要介绍如何在控制台进行报告的输出。

其实控制台视图也可以看作是第二篇文章中提到的侧边栏，因此也需要继承实现ToolWindowFactory：

kotlin">class ConsoleWindow: ToolWindowFactory {
    
    override fun createToolWindowContent(project: Project, toolWindow: ToolWindow) {
        if (Utils.getConsoleViews()[project] == null) {
            Utils.createToolWindow(project, toolWindow)
        }
        Utils.toolWindows[project] = toolWindow
    }

}

// Utils 中的方法
fun createToolWindow(project: Project, toolWindow: ToolWindow) {
    val consoleView = TextConsoleBuilderFactory.getInstance().createBuilder(project).console
    consoleViews[project] = consoleView
    val content = toolWindow.contentManager.factory.createContent(consoleView.component, "TimerMaster Output", false)
    content.component.isVisible = true
    content.isCloseable = true
    toolWindow.contentManager.addContent(content)
}

// 在控制台中显示信息
fun consoleInfo(project: Project, msg: String) {
    if (consoleViews[project] == null) {
        ToolWindowManager.getInstance(project).getToolWindow("TimerMaster Console")
            ?.let { createToolWindow(project, it) }
    }
    consoleViews[project]?.clear()
    consoleViews[project]?.print(msg, ConsoleViewContentType.NORMAL_OUTPUT)
    // 显示控制台窗口, 减去手动点击侧边栏按钮的操作
    toolWindows[project]?.activate(null, false)
}

这里需要注意的是由于可能多个项目都打开了该控制台视图，因此需要使用一个 map 继续保存，避免内容错乱。

最后以一个动图结尾：

动画

总结

本文通过实现一个对代码活动统计的插件算是对前几篇文章的一个总结，目前只在本地运行测试了几天，可能有一些不妥的实现方式和 BUG，欢迎一起交流讨论。另外截止到本篇，关于 IntelliJ 平台插件开发的基础知识也算告一段落，后续文章则优先讲解关于虚拟文件、PSI和编辑器操作相关的知识，而关于国际化、主题插件开发等则看情况夹杂在其中进行讲解。


http://www.niftyadmin.cn/n/5293047.html

相关文章

ALSA学习(4)——Control设备的创建

ALSA学习(4)——Control设备的创建

参考博客&#xff1a; https://blog.csdn.net/DroidPhone/article/details/6409983 &#xff08;下面的内容基本是原博主的内容&#xff0c;我只是修改了一些格式之类的&#xff09; 文章目录 一、Control接口二、Controls的定义三、Control的名字四、访问标志&#xff08;ACC…
阅读更多...
linux iptables简介

linux iptables简介

表与链 iptables是4表五链 4表&#xff1a;filter表&#xff08;过滤表&#xff09; nat表 raw表 mangle表 五链&#xff1a;INPUT OUTPUT FORWARD PREROUTING POSTROUTING &#xff08;所有链的名字要大写&#xff09; pre…之前 post…之后 filter表 iptables默认操作…
阅读更多...
排查格式化符与待格式化参数不一致导致的程序崩溃问题

排查格式化符与待格式化参数不一致导致的程序崩溃问题

目录 1、问题说明 2、初步分析 3、进一步分析 3.1、查看崩溃时的函数调用堆栈
阅读更多...
如何本地搭建FastDFS文件服务器并实现远程访问【内网穿透】

如何本地搭建FastDFS文件服务器并实现远程访问【内网穿透】

文章目录 前言1. 本地搭建FastDFS文件系统1.1 环境安装1.2 安装libfastcommon1.3 安装FastDFS1.4 配置Tracker1.5 配置Storage1.6 测试上传下载1.7 与Nginx整合1.8 安装Nginx1.9 配置Nginx 2. 局域网测试访问FastDFS3. 安装cpolar内网穿透4. 配置公网访问地址5. 固定公网地址5.…
阅读更多...
信息泄露总结

信息泄露总结

文章目录 一、备份文件下载1.1 网站源码1.2 bak文件泄露1.3 vim缓存1.4 .DS_Store 二、Git泄露2.1 git知识点2.1 log2.2 stash 三、SVN泄露3.1 SVN简介3.2 SVN的文件3.3 SVN利用 四、Hg泄露 一、备份文件下载 1.1 网站源码 常见的网站源码备份文件后缀&#xff1a; tartar.gz…
阅读更多...
关于k8s

关于k8s

Kubernetes&#xff08;通常被称为K8s&#xff09;是一个开源的容器编排平台&#xff0c;用于自动化部署、扩展和管理容器化应用程序。它最初由Google开发&#xff0c;并于2014年发布为开源项目&#xff0c;现在由云原生计算基金会&#xff08;CNCF&#xff09;负责维护。 Kub…
阅读更多...
基于Java车间工时管理系统（源码+部署文档）

基于Java车间工时管理系统（源码+部署文档）

博主介绍&#xff1a; ✌至今服务客户已经1000、专注于Java技术领域、项目定制、技术答疑、开发工具、毕业项目实战 ✌ &#x1f345; 文末获取源码联系 &#x1f345; &#x1f447;&#x1f3fb; 精彩专栏 推荐订阅 &#x1f447;&#x1f3fb; 不然下次找不到 Java项目精品实…
阅读更多...
算法题Python常用内置函数、方法、技巧汇总（其八：推导式）

算法题Python常用内置函数、方法、技巧汇总（其八：推导式）

文章目录 推导式列表推导式元组推导式集合推导式字典推导式 华为OD算法/大厂面试高频题算法练习冲刺训练 推导式 python中的推导式是一种独特的数据处理方式&#xff0c;可以从一个数据序列构建另一个新的数据序列。可以简单理解为for循环语句(if条件语句)的简写版本&#xff…
阅读更多...
最新文章

Copyright @ 2022~2023