Merge pull request #96 from mallowigi/counter

counter

Author: mallowigi

src/main/java/krasa/editorGroups/EditorGroupPanel.kt

@@ -37,10 +37,7 @@ import com.intellij.util.BitUtil
 import com.intellij.util.concurrency.AppExecutorUtil
 import com.intellij.util.ui.JBUI
 import com.intellij.util.ui.UIUtil
-import krasa.editorGroups.actions.PopupMenu
-import krasa.editorGroups.actions.RefreshAction
-import krasa.editorGroups.actions.RemoveFromCurrentBookmarksAction
-import krasa.editorGroups.actions.SwitchGroupAction
+import krasa.editorGroups.actions.*
 import krasa.editorGroups.events.EditorGroupChangeListener
 import krasa.editorGroups.index.IndexNotReady
 import krasa.editorGroups.language.EditorGroupsLanguage.isEditorGroupsLanguage
@@ -330,12 +327,14 @@ class EditorGroupPanel(
 
   /** Add the action buttons at the left of the panel. */
   private fun createToolbar() {
+    val actionManager = ActionManager.getInstance()
     val actionGroup = DefaultActionGroup().apply {
-      add(ActionManager.getInstance().getAction(RefreshAction.ID))
-      add(ActionManager.getInstance().getAction(SwitchGroupAction.ID))
+      add(actionManager.getAction(RefreshAction.ID))
+      add(actionManager.getAction(SwitchGroupAction.ID))
+      add(GroupLinksCountAction())
     }
 
-    toolbar = ActionManager.getInstance().createActionToolbar(
+    toolbar = actionManager.createActionToolbar(
       TOOLBAR_PLACE,
       actionGroup,
       true

src/main/java/krasa/editorGroups/actions/GroupLinksCountAction.kt

@@ -0,0 +1,60 @@
+package krasa.editorGroups.actions
+
+import com.intellij.openapi.actionSystem.ActionUpdateThread
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.project.DumbAwareAction
+import com.intellij.ui.JBColor
+import com.intellij.util.ui.UIUtil
+import com.intellij.util.ui.UIUtil.drawCenteredString
+import krasa.editorGroups.EditorGroupManager
+import java.awt.*
+import javax.swing.Icon
+
+class GroupLinksCountAction : DumbAwareAction() {
+  override fun isDumbAware(): Boolean = false
+
+  override fun getActionUpdateThread(): ActionUpdateThread = ActionUpdateThread.BGT
+
+  override fun actionPerformed(e: AnActionEvent): Unit = Unit
+
+  override fun update(e: AnActionEvent) {
+    val project = e.project ?: return
+    val currentGroup = EditorGroupManager.getInstance(project).lastGroup
+    val count = currentGroup.size(project)
+
+    e.presentation.text = count.toString()
+    e.presentation.icon = CompositeIcon(count)
+  }
+
+  internal class CompositeIcon(private val counter: Int) : Icon {
+    val backgroundColor: Color
+      get() = JBColor.namedColor("Counter.background", Color(-0x33655850, true))
+
+    override fun paintIcon(c: Component?, g: Graphics, x: Int, y: Int) {
+      val counterText = counter.toString()
+      val g2d = g as Graphics2D
+      val fontSize = if (counter < 100) FONT_SIZE else SMALL_FONT_SIZE
+
+      g2d.color = backgroundColor
+      g2d.fillOval(x, y, ICON_SIZE, ICON_SIZE)
+
+      g2d.color = UIUtil.getPanelBackground()
+      g2d.font = g2d.font.deriveFont(fontSize)
+      drawCenteredString(
+        g2d,
+        Rectangle(x, y, ICON_SIZE, ICON_SIZE),
+        counterText,
+      )
+    }
+
+    override fun getIconWidth(): Int = ICON_SIZE
+
+    override fun getIconHeight(): Int = ICON_SIZE
+  }
+
+  companion object {
+    private const val FONT_SIZE = 10.0f
+    private const val SMALL_FONT_SIZE = 8.0f
+    private const val ICON_SIZE = 16
+  }
+}

src/main/java/krasa/editorGroups/extensions/CustomFileIconLayerProvider.kt

@@ -43,11 +43,12 @@ internal class EditorGroupsToolWindowFactory : ToolWindowFactory {
     panel.add(tabList, BorderLayout.CENTER)
 
     // Add action toolbar with RefreshAction and SwitchGroupAction
+    val actionManager = ActionManager.getInstance()
     val actionGroup = DefaultActionGroup().apply {
-      add(ActionManager.getInstance().getAction(RefreshAction.ID))
-      add(ActionManager.getInstance().getAction(SwitchGroupAction.ID))
+      add(actionManager.getAction(RefreshAction.ID))
+      add(actionManager.getAction(SwitchGroupAction.ID))
     }
-    val actionToolbar: ActionToolbar = ActionManager.getInstance().createActionToolbar("EditorGroupsToolbar", actionGroup, true)
+    val actionToolbar: ActionToolbar = actionManager.createActionToolbar("EditorGroupsToolbar", actionGroup, true)
     actionToolbar.targetComponent = panel
     panel.add(actionToolbar.component, BorderLayout.NORTH)
 

src/main/java/krasa/editorGroups/toolwindow/EditorGroupsToolWindowFactory.kt

@@ -60,9 +60,6 @@
         <editorTabColorProvider implementation="krasa.editorGroups.extensions.CustomEditorGroupsTabColorProvider" order="first"/>
         <editorTabColorProvider implementation="krasa.editorGroups.extensions.CurrentEditorGroupColorProvider" order="last"/>
 
-        <!-- TODO: Custom size (WIP) -->
-        <!--<iconLayerProvider implementation="krasa.editorGroups.extensions.CustomFileIconLayerProvider"/>-->
-
         <!-- Indices -->
         <fileBasedIndex implementation="krasa.editorGroups.index.EditorGroupIndex"/>
         <fileBasedIndex implementation="krasa.editorGroups.index.FilenameWithoutExtensionIndex"/>