|
10 | 10 | import com.intellij.openapi.wm.ToolWindow;
|
11 | 11 | import com.intellij.openapi.wm.ToolWindowId;
|
12 | 12 | import com.intellij.openapi.wm.ToolWindowManager;
|
13 |
| -import com.intellij.ui.BadgeIcon; |
14 | 13 | import io.flutter.run.common.RunMode;
|
15 | 14 | import io.flutter.run.daemon.FlutterApp;
|
16 | 15 |
|
17 |
| -import java.awt.Color; |
18 |
| -import java.awt.Paint; |
| 16 | +import javax.swing.*; |
| 17 | +import java.awt.*; |
19 | 18 | import java.util.Objects;
|
20 |
| -import javax.swing.Icon; |
21 | 19 |
|
22 | 20 | public class ToolWindowBadgeUpdater {
|
23 |
| - public static final Paint BADGE_PAINT = Color.decode("#5ca963"); |
| 21 | + public static final Color BADGE_PAINT = Color.decode("#5ca963"); |
24 | 22 |
|
25 | 23 | /**
|
26 | 24 | * Updates the tool window icons for RUN or DEBUG mode with a green badge.
|
27 | 25 | *
|
28 |
| - * @param app The FlutterApp instance running in a given mode. |
| 26 | + * @param app The FlutterApp instance running in a given mode. |
29 | 27 | * @param project The current IntelliJ project context.
|
30 | 28 | */
|
31 | 29 | public static void updateBadgedIcon(FlutterApp app, Project project) {
|
32 | 30 | final ToolWindowManager manager = ToolWindowManager.getInstance(Objects.requireNonNull(project));
|
33 | 31 | final ToolWindow runToolWindow = manager.getToolWindow(ToolWindowId.RUN);
|
34 | 32 | final ToolWindow debugToolWindow = manager.getToolWindow(ToolWindowId.DEBUG);
|
35 | 33 |
|
36 |
| - if(Objects.requireNonNull(app).getMode() == RunMode.RUN) { |
| 34 | + if (Objects.requireNonNull(app).getMode() == RunMode.RUN) { |
37 | 35 | if (runToolWindow != null) {
|
38 | 36 | manager.invokeLater(() -> {
|
39 | 37 | Icon baseIcon = AllIcons.Toolwindows.ToolWindowRun;
|
40 | 38 | BadgeIcon iconWithBadge = new BadgeIcon(baseIcon, BADGE_PAINT);
|
41 |
| - |
42 | 39 | runToolWindow.setIcon(iconWithBadge);
|
43 | 40 | });
|
44 | 41 | }
|
45 | 42 | }
|
46 |
| - else if(app.getMode() == RunMode.DEBUG) { |
| 43 | + else if (app.getMode() == RunMode.DEBUG) { |
47 | 44 | manager.invokeLater(() -> {
|
48 | 45 | Icon baseIcon = AllIcons.Toolwindows.ToolWindowDebugger;
|
49 | 46 | BadgeIcon iconWithBadge = new BadgeIcon(baseIcon, BADGE_PAINT);
|
50 |
| - |
51 | 47 | Objects.requireNonNull(debugToolWindow).setIcon(iconWithBadge);
|
52 | 48 | });
|
53 | 49 | }
|
| 50 | + } |
| 51 | + |
| 52 | + private static class BadgeIcon implements Icon { |
| 53 | + private final Icon baseIcon; |
| 54 | + private final Color overlayColor; |
| 55 | + private static final float alpha = 1.0F; |
| 56 | + |
| 57 | + public BadgeIcon(Icon baseIcon, Color overlayColor) { |
| 58 | + this.baseIcon = baseIcon; |
| 59 | + this.overlayColor = overlayColor; |
| 60 | + } |
| 61 | + |
| 62 | + @Override |
| 63 | + public void paintIcon(Component c, Graphics g, int x, int y) { |
| 64 | + baseIcon.paintIcon(c, g, x, y); |
| 65 | + |
| 66 | + Graphics2D g2d = (Graphics2D)g.create(); |
| 67 | + try { |
| 68 | + g2d.translate(x, y); |
| 69 | + |
| 70 | + g2d.setComposite(AlphaComposite.SrcOver.derive(alpha)); |
| 71 | + |
| 72 | + g2d.setColor(overlayColor); |
| 73 | + g2d.fillRect(0, 0, getIconWidth(), getIconHeight()); |
| 74 | + } |
| 75 | + finally { |
| 76 | + g2d.dispose(); |
| 77 | + } |
| 78 | + } |
| 79 | + |
| 80 | + @Override |
| 81 | + public int getIconWidth() { |
| 82 | + return baseIcon.getIconWidth(); |
| 83 | + } |
54 | 84 |
|
| 85 | + @Override |
| 86 | + public int getIconHeight() { |
| 87 | + return baseIcon.getIconHeight(); |
| 88 | + } |
55 | 89 | }
|
56 | 90 | }
|
0 commit comments