add icon to error hotspot (#149)

Signed-off-by: shalom <[email protected]>

Signed-off-by: shalom <[email protected]>

Author: shalom938

rider/Digma.Rider.Plugin/Digma.Rider/Highlighting/CodeObjectsHighlighter.cs

@@ -4,8 +4,11 @@
 using JetBrains.ProjectModel;
 using JetBrains.ReSharper.Daemon.CodeInsights;
 using JetBrains.ReSharper.Feature.Services.Daemon;
+using JetBrains.ReSharper.Features.Internal.Resources;
 using JetBrains.ReSharper.Psi.CSharp.Tree;
 using JetBrains.ReSharper.Psi.Tree;
+using JetBrains.Rider.Backend.Platform.Icons;
+using JetBrains.UI.Icons;
 using JetBrains.Util;
 using static Digma.Rider.Logging.Logger;
 
@@ -18,17 +21,19 @@ public class CodeObjectsHighlighter : ElementProblemAnalyzer<ICSharpFunctionDecl
         private readonly CodeObjectsHost _codeObjectsHost;
         private readonly ErrorHotspotMethodInsightsProvider _errorHotspotMethodInsightsProvider;
         private readonly UsageMethodInsightsProvider _usageMethodInsightsProvider;
+        private readonly IconHost _iconHost;
 
         public CodeObjectsHighlighter(ILogger logger,
             CodeObjectsHost codeObjectsHost,
             ISolution solution,
             ErrorHotspotMethodInsightsProvider errorHotspotMethodInsightsProvider,
-            UsageMethodInsightsProvider usageMethodInsightsProvider)
+            UsageMethodInsightsProvider usageMethodInsightsProvider,IconHost iconHost)
         {
             _logger = logger;
             _codeObjectsHost = codeObjectsHost;
             _errorHotspotMethodInsightsProvider = errorHotspotMethodInsightsProvider;
             _usageMethodInsightsProvider = usageMethodInsightsProvider;
+            _iconHost = iconHost;
         }
 
         protected override void Run(ICSharpFunctionDeclaration functionDeclaration,
@@ -51,14 +56,19 @@ protected override void Run(ICSharpFunctionDeclaration functionDeclaration,
                             ? _errorHotspotMethodInsightsProvider
                             : _usageMethodInsightsProvider;
 
+                    var icon =
+                        riderCodeLensInfo.Type.Equals(CodeLensType.ErrorHotspot)
+                            ? _iconHost.Transform(Icons.ExclMark)
+                            : null;
+                    
                     highlightingConsumer.AddHighlighting(
                         new CodeInsightsHighlighting(
                             functionDeclaration.GetNameDocumentRange(),
                             riderCodeLensInfo.LensText ?? throw new InvalidOperationException("LensText must not be null"),
                             riderCodeLensInfo.LensTooltip ?? string.Empty, //todo: can be null
                             riderCodeLensInfo.MoreText ?? string.Empty, //todo: can be null
                             codeInsightsProvider,
-                            functionDeclaration.DeclaredElement,null)
+                            functionDeclaration.DeclaredElement,icon)
                     );
                 }
 

rider/Digma.Rider.Plugin/Digma.Rider/Highlighting/Icons.cs

@@ -0,0 +1,9 @@
+using JetBrains.RdBackend.Common.Features.Icons;
+
+namespace Digma.Rider.Highlighting
+{
+    public class Icons
+    {
+        public static FrontendIconId ExclMark = new FrontendIconId("icons/exclMark.svg");
+    }
+}