QD-12662: Allow supplying dynamic tools with custom inspection wrappers

(cherry picked from commit e7a0f49f0139231b43c7878d428845077df1d34c)

IJ-MR-186129

GitOrigin-RevId: 77dcb836560de92cba90cf6e1344dc1c0f8fe254

Author: hybloid

platform/analysis-impl/src/com/intellij/codeInspection/ex/dynamic-inspections.kt

@@ -33,18 +33,37 @@ sealed class DynamicInspectionDescriptor {
         else -> error("Got ${tool}, expected ${LocalInspectionTool::class.java} or ${GlobalInspectionTool::class.java}")
       }
     }
+
+    fun fromLocalToolWithWrapper(
+      tool: LocalInspectionTool,
+      wrapperFactory: (LocalInspectionTool) -> InspectionToolWrapper<*,*>
+    ): DynamicInspectionDescriptor {
+      return LocalWithCustomWrapper(tool, wrapperFactory)
+    }
   }
 
-  val toolWrapper: InspectionToolWrapper<*, *> by lazy {
-    when(this) {
-      is Local -> DynamicLocalInspectionToolWrapper(tool)
-      is Global -> DynamicGlobalInspectionToolWrapper(tool)
+  abstract val toolWrapper: InspectionToolWrapper<*, *>
+
+  class Local(val tool: LocalInspectionTool) : DynamicInspectionDescriptor() {
+    override val toolWrapper: InspectionToolWrapper<*, *> by lazy {
+      DynamicLocalInspectionToolWrapper(tool)
     }
   }
 
-  class Local(val tool: LocalInspectionTool) : DynamicInspectionDescriptor()
+  class Global(val tool: GlobalInspectionTool) : DynamicInspectionDescriptor() {
+    override val toolWrapper: InspectionToolWrapper<*, *> by lazy {
+      DynamicGlobalInspectionToolWrapper(tool)
+    }
+  }
 
-  class Global(val tool: GlobalInspectionTool) : DynamicInspectionDescriptor()
+  class LocalWithCustomWrapper(
+    val tool: LocalInspectionTool,
+    private val wrapperFactory: (LocalInspectionTool) -> InspectionToolWrapper<*,*>
+  ) : DynamicInspectionDescriptor() {
+    override val toolWrapper: InspectionToolWrapper<*, *> by lazy {
+      wrapperFactory(tool)
+    }
+  }
 
   private class DynamicLocalInspectionToolWrapper(tool: LocalInspectionTool) : LocalInspectionToolWrapper(tool) {
     override fun createCopy(): LocalInspectionToolWrapper = DynamicLocalInspectionToolWrapper(tool)