src/main/kotlin/com/chylex/intellij/inspectionlens/EditorInlayLensManager.kt

@@ -4,7 +4,7 @@ import com.intellij.codeInsight.daemon.impl.HighlightInfo
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.editor.Inlay
 import com.intellij.openapi.editor.InlayProperties
-import com.intellij.openapi.editor.markup.RangeHighlighter
+import com.intellij.openapi.editor.ex.RangeHighlighterEx
 import com.intellij.openapi.util.Key
 
 /**
@@ -18,32 +18,28 @@ class EditorInlayLensManager private constructor(private val editor: Editor) {
 			return editor.getUserData(KEY) ?: EditorInlayLensManager(editor).also { editor.putUserData(KEY, it) }
 		}
 		
-		fun remove(editor: Editor) {
-			val manager = editor.getUserData(KEY)
-			if (manager != null) {
-				manager.hideAll()
-				editor.putUserData(KEY, null)
-			}
+		private fun updateRenderer(renderer: LensRenderer, info: HighlightInfo) {
+			renderer.text = info.description.takeIf(String::isNotBlank)?.let(::addMissingPeriod) ?: " "
+			renderer.severity = LensSeverity.from(info.severity)
 		}
 		
-		private fun getInlayHintOffset(info: HighlightInfo): Int {
-			// Ensures a highlight at the end of a line does not overflow to the next line.
-			return info.actualEndOffset - 1
+		private fun addMissingPeriod(text: String): String {
+			return if (text.endsWith('.')) text else "$text."
 		}
 	}
 	
-	private val inlays = mutableMapOf<RangeHighlighter, Inlay<LensRenderer>>()
+	private val inlays = mutableMapOf<RangeHighlighterEx, Inlay<LensRenderer>>()
 	
-	fun show(highlighter: RangeHighlighter, info: HighlightInfo) {
+	fun show(highlighter: RangeHighlighterEx, info: HighlightInfo) {
 		val currentInlay = inlays[highlighter]
 		if (currentInlay != null && currentInlay.isValid) {
-			currentInlay.renderer.setPropertiesFrom(info)
+			updateRenderer(currentInlay.renderer, info)
 			currentInlay.update()
 		}
 		else {
-			val offset = getInlayHintOffset(info)
-			val renderer = LensRenderer(info)
-			val properties = InlayProperties().relatesToPrecedingText(true).disableSoftWrapping(true).priority(-offset)
+			val offset = info.actualEndOffset - 1
+			val renderer = LensRenderer().also { updateRenderer(it, info) }
+			val properties = InlayProperties().relatesToPrecedingText(true).priority(-offset)
 			
 			editor.inlayModel.addAfterLineEndElement(offset, properties, renderer)?.let {
 				inlays[highlighter] = it
@@ -51,12 +47,7 @@ class EditorInlayLensManager private constructor(private val editor: Editor) {
 		}
 	}
 	
-	fun hide(highlighter: RangeHighlighter) {
+	fun hide(highlighter: RangeHighlighterEx) {
 		inlays.remove(highlighter)?.dispose()
 	}
-	
-	fun hideAll() {
-		inlays.values.forEach(Inlay<*>::dispose)
-		inlays.clear()
-	}
 }

src/main/kotlin/com/chylex/intellij/inspectionlens/InspectionLensPluginDisposableService.kt

@@ -1,10 +0,0 @@
-package com.chylex.intellij.inspectionlens
-
-import com.intellij.openapi.Disposable
-
-/**
- * Gets automatically disposed when the plugin is unloaded.
- */
-class InspectionLensPluginDisposableService : Disposable {
-	override fun dispose() {}
-}

src/main/kotlin/com/chylex/intellij/inspectionlens/InspectionLensPluginListener.kt

@@ -1,43 +0,0 @@
-package com.chylex.intellij.inspectionlens
-
-import com.intellij.ide.plugins.DynamicPluginListener
-import com.intellij.ide.plugins.IdeaPluginDescriptor
-import com.intellij.openapi.fileEditor.FileEditorManager
-import com.intellij.openapi.fileEditor.TextEditor
-import com.intellij.openapi.project.ProjectManager
-
-/**
- * Handles dynamic plugin loading.
- * 
- * On load, it installs the [LensMarkupModelListener] to all open editors.
- * On unload, it removes all lenses from all open editors.
- */
-class InspectionLensPluginListener : DynamicPluginListener {
-	companion object {
-		private const val PLUGIN_ID = "com.chylex.intellij.inspectionlens"
-		
-		private inline fun ProjectManager.forEachEditor(action: (TextEditor) -> Unit) {
-			for (project in this.openProjects.filterNot { it.isDisposed }) {
-				val fileEditorManager = FileEditorManager.getInstance(project)
-				
-				for (editor in fileEditorManager.allEditors.filterIsInstance<TextEditor>()) {
-					action(editor)
-				}
-			}
-		}
-	}
-	
-	override fun pluginLoaded(pluginDescriptor: IdeaPluginDescriptor) {
-		if (pluginDescriptor.pluginId.idString == PLUGIN_ID) {
-			ProjectManager.getInstanceIfCreated()?.forEachEditor(LensMarkupModelListener.Companion::install)
-		}
-	}
-	
-	override fun beforePluginUnload(pluginDescriptor: IdeaPluginDescriptor, isUpdate: Boolean) {
-		if (pluginDescriptor.pluginId.idString == PLUGIN_ID) {
-			ProjectManager.getInstanceIfCreated()?.forEachEditor {
-				EditorInlayLensManager.remove(it.editor)
-			}
-		}
-	}
-}

src/main/kotlin/com/chylex/intellij/inspectionlens/LensFileEditorListener.kt

@@ -1,5 +1,7 @@
 package com.chylex.intellij.inspectionlens
 
+import com.intellij.openapi.editor.ex.MarkupModelEx
+import com.intellij.openapi.editor.impl.DocumentMarkupModel
 import com.intellij.openapi.fileEditor.FileEditorManager
 import com.intellij.openapi.fileEditor.FileEditorManagerListener
 import com.intellij.openapi.fileEditor.TextEditor
@@ -7,14 +9,18 @@ import com.intellij.openapi.fileEditor.ex.FileEditorWithProvider
 import com.intellij.openapi.vfs.VirtualFile
 
 /**
- * Listens for newly opened editors, and installs a [LensMarkupModelListener] on them.
+ * Listens for newly opened editors, and attaches a [LensMarkupModelListener] to their document model.
  */
 class LensFileEditorListener : FileEditorManagerListener {
 	override fun fileOpenedSync(source: FileEditorManager, file: VirtualFile, editorsWithProviders: MutableList<FileEditorWithProvider>) {
 		for (editorWrapper in editorsWithProviders) {
 			val fileEditor = editorWrapper.fileEditor
 			if (fileEditor is TextEditor) {
-				LensMarkupModelListener.install(fileEditor)
+				val editor = fileEditor.editor
+				val markupModel = DocumentMarkupModel.forDocument(editor.document, editor.project, false)
+				if (markupModel is MarkupModelEx) {
+					markupModel.addMarkupModelListener(fileEditor, LensMarkupModelListener(editor))
+				}
 			}
 		}
 	}

src/main/kotlin/com/chylex/intellij/inspectionlens/LensMarkupModelListener.kt

@@ -1,22 +1,16 @@
 package com.chylex.intellij.inspectionlens
 
-import com.chylex.intellij.inspectionlens.util.MultiParentDisposable
 import com.intellij.codeInsight.daemon.impl.AsyncDescriptionSupplier
 import com.intellij.codeInsight.daemon.impl.HighlightInfo
 import com.intellij.lang.annotation.HighlightSeverity
-import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.editor.Editor
-import com.intellij.openapi.editor.ex.MarkupModelEx
 import com.intellij.openapi.editor.ex.RangeHighlighterEx
-import com.intellij.openapi.editor.impl.DocumentMarkupModel
 import com.intellij.openapi.editor.impl.event.MarkupModelListener
-import com.intellij.openapi.editor.markup.RangeHighlighter
-import com.intellij.openapi.fileEditor.TextEditor
 
 /**
  * Listens for inspection highlights and reports them to [EditorInlayLensManager].
  */
-class LensMarkupModelListener private constructor(editor: Editor) : MarkupModelListener {
+class LensMarkupModelListener(editor: Editor) : MarkupModelListener {
 	private val lens = EditorInlayLensManager.getOrCreate(editor)
 	
 	override fun afterAdded(highlighter: RangeHighlighterEx) {
@@ -31,7 +25,7 @@ class LensMarkupModelListener private constructor(editor: Editor) : MarkupModelL
 		lens.hide(highlighter)
 	}
 	
-	private fun showIfValid(highlighter: RangeHighlighter) {
+	private fun showIfValid(highlighter: RangeHighlighterEx) {
 		if (!highlighter.isValid) {
 			return
 		}
@@ -53,35 +47,9 @@ class LensMarkupModelListener private constructor(editor: Editor) : MarkupModelL
 		}
 	}
 	
-	private fun showIfNonNullDescription(highlighter: RangeHighlighter, info: HighlightInfo) {
+	private fun showIfNonNullDescription(highlighter: RangeHighlighterEx, info: HighlightInfo) {
 		if (info.description != null) {
 			lens.show(highlighter, info)
 		}
 	}
-	
-	companion object {
-		/**
-		 * Attaches a new [LensMarkupModelListener] to the document model of the provided [TextEditor], and reports all existing inspection highlights to [EditorInlayLensManager].
-		 * 
-		 * The [LensMarkupModelListener] will be disposed when either the [TextEditor] is disposed, or via [InspectionLensPluginDisposableService] when the plugin is unloaded.
-		 */
-		fun install(textEditor: TextEditor) {
-			val editor = textEditor.editor
-			val markupModel = DocumentMarkupModel.forDocument(editor.document, editor.project, false)
-			if (markupModel is MarkupModelEx) {
-				val pluginDisposable = ApplicationManager.getApplication().getService(InspectionLensPluginDisposableService::class.java)
-				
-				val listenerDisposable = MultiParentDisposable()
-				listenerDisposable.registerWithParent(textEditor)
-				listenerDisposable.registerWithParent(pluginDisposable)
-				
-				val listener = LensMarkupModelListener(editor)
-				markupModel.addMarkupModelListener(listenerDisposable.self, listener)
-				
-				for (highlighter in markupModel.allHighlighters) {
-					listener.showIfValid(highlighter)
-				}
-			}
-		}
-	}
 }

src/main/kotlin/com/chylex/intellij/inspectionlens/LensRenderer.kt

@@ -1,6 +1,5 @@
 package com.chylex.intellij.inspectionlens
 
-import com.intellij.codeInsight.daemon.impl.HighlightInfo
 import com.intellij.codeInsight.daemon.impl.HintRenderer
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.editor.Inlay
@@ -12,44 +11,23 @@ import java.awt.Rectangle
 /**
  * Renders the text of an inspection lens.
  */
-class LensRenderer(info: HighlightInfo) : HintRenderer(null) {
-	private lateinit var severity: LensSeverity
-	
-	init {
-		setPropertiesFrom(info)
+class LensRenderer : HintRenderer(null) {
+	private companion object {
+		private val ATTRIBUTES = TextAttributes(null, null, null, null, Font.ITALIC)
 	}
 	
-	fun setPropertiesFrom(info: HighlightInfo) {
-		text = getValidDescriptionText(info.description)
-		severity = LensSeverity.from(info.severity)
-	}
+	var severity = LensSeverity.OTHER
 	
 	override fun paint(inlay: Inlay<*>, g: Graphics, r: Rectangle, textAttributes: TextAttributes) {
-		fixBaselineForTextRendering(r)
+		r.y += 1
 		super.paint(inlay, g, r, textAttributes)
 	}
 	
 	override fun getTextAttributes(editor: Editor): TextAttributes {
-		return ATTRIBUTES_SINGLETON.also { it.foregroundColor = severity.getColor(editor) }
+		return ATTRIBUTES.also { it.foregroundColor = severity.getColor(editor) }
 	}
 	
 	override fun useEditorFont(): Boolean {
 		return true
 	}
-	
-	private companion object {
-		private val ATTRIBUTES_SINGLETON = TextAttributes(null, null, null, null, Font.ITALIC)
-		
-		private fun getValidDescriptionText(text: String?): String {
-			return if (text.isNullOrBlank()) " " else addMissingPeriod(text)
-		}
-		
-		private fun addMissingPeriod(text: String): String {
-			return if (text.endsWith('.')) text else "$text."
-		}
-		
-		private fun fixBaselineForTextRendering(rect: Rectangle) {
-			rect.y += 1
-		}
-	}
 }

src/main/kotlin/com/chylex/intellij/inspectionlens/util/MultiParentDisposable.kt

@@ -1,18 +0,0 @@
-package com.chylex.intellij.inspectionlens.util
-
-import com.intellij.openapi.Disposable
-import com.intellij.openapi.util.Disposer
-import java.lang.ref.WeakReference
-
-/**
- * A [Disposable] that can have multiple parents, and will be disposed when any parent is disposed.
- * A [WeakReference] and a lambda will remain in memory for every parent that is not disposed.
- */
-class MultiParentDisposable {
-	val self = Disposer.newDisposable()
-	
-	fun registerWithParent(parent: Disposable) {
-		val weakSelfReference = WeakReference(self)
-		Disposer.register(parent) { weakSelfReference.get()?.let(Disposer::dispose) }
-	}
-}

src/main/resources/META-INF/plugin.xml

@@ -11,14 +11,6 @@
   
   <depends>com.intellij.modules.platform</depends>
   
-  <extensions defaultExtensionNs="com.intellij">
-    <applicationService serviceImplementation="com.chylex.intellij.inspectionlens.InspectionLensPluginDisposableService" />
-  </extensions>
-  
-  <applicationListeners>
-    <listener class="com.chylex.intellij.inspectionlens.InspectionLensPluginListener" topic="com.intellij.ide.plugins.DynamicPluginListener" />
-  </applicationListeners>
-  
   <projectListeners>
     <listener class="com.chylex.intellij.inspectionlens.LensFileEditorListener" topic="com.intellij.openapi.fileEditor.FileEditorManagerListener" />
   </projectListeners>