Do not use internal API

vaadin · Dec 10, 2024 · 0bc3bcb · 0bc3bcb
1 parent 5608e01
commit 0bc3bcb
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/main/kotlin/com/vaadin/plugin/ui/VaadinStatusBarWidget.kt b/src/main/kotlin/com/vaadin/plugin/ui/VaadinStatusBarWidget.kt
@@ -4,9 +4,10 @@ import com.intellij.openapi.project.Project
 import com.intellij.openapi.ui.popup.JBPopupFactory
 import com.intellij.openapi.wm.StatusBarWidget
 import com.intellij.openapi.wm.WindowManager
-import com.intellij.ui.BadgeIconSupplier
+import com.intellij.ui.IconManager
 import com.intellij.ui.awt.RelativePoint
 import com.intellij.util.Consumer
+import com.intellij.util.ui.JBUI.CurrentTheme.IconBadge
 import com.vaadin.plugin.copilot.CopilotPluginUtil
 import com.vaadin.plugin.utils.VaadinIcons
 import com.vaadin.plugin.utils.hasEndpoints
@@ -23,8 +24,6 @@ class VaadinStatusBarWidget(private val project: Project) : StatusBarWidget, Sta
         }
     }
 
-    private val iconSupplier: BadgeIconSupplier = BadgeIconSupplier(VaadinIcons.VAADIN)
-
     override fun ID(): String {
         return ID
     }
@@ -57,9 +56,9 @@ class VaadinStatusBarWidget(private val project: Project) : StatusBarWidget, Sta
 
     override fun getIcon(): Icon {
         if (!CopilotPluginUtil.isActive(project) || !hasEndpoints()) {
-            return iconSupplier.warningIcon
+            return IconManager.getInstance().withIconBadge(VaadinIcons.VAADIN, IconBadge.WARNING)
         }
 
-        return iconSupplier.originalIcon
+        return VaadinIcons.VAADIN
     }
 }