2014-11-24 17:40:06 +01:00
|
|
|
/*
|
|
|
|
* Copyright 2014 dorkbox, llc
|
|
|
|
*
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
* you may not use this file except in compliance with the License.
|
|
|
|
* You may obtain a copy of the License at
|
|
|
|
*
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
*
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
* See the License for the specific language governing permissions and
|
|
|
|
* limitations under the License.
|
|
|
|
*/
|
2014-11-03 02:11:03 +01:00
|
|
|
package dorkbox.util.tray.swing;
|
|
|
|
|
2015-08-04 21:10:01 +02:00
|
|
|
import dorkbox.util.ScreenUtil;
|
2015-06-28 01:47:02 +02:00
|
|
|
import dorkbox.util.SwingUtil;
|
2015-11-09 16:13:57 +01:00
|
|
|
import dorkbox.util.tray.MenuEntry;
|
2015-06-28 01:47:02 +02:00
|
|
|
import dorkbox.util.tray.SystemTrayMenuAction;
|
|
|
|
import dorkbox.util.tray.SystemTrayMenuPopup;
|
|
|
|
|
2015-11-09 16:13:57 +01:00
|
|
|
import javax.swing.ImageIcon;
|
|
|
|
import javax.swing.JMenuItem;
|
|
|
|
import java.awt.AWTException;
|
|
|
|
import java.awt.Dimension;
|
|
|
|
import java.awt.Font;
|
|
|
|
import java.awt.Image;
|
|
|
|
import java.awt.Point;
|
|
|
|
import java.awt.Rectangle;
|
|
|
|
import java.awt.SystemTray;
|
|
|
|
import java.awt.TrayIcon;
|
2014-11-03 02:11:03 +01:00
|
|
|
import java.awt.event.MouseAdapter;
|
|
|
|
import java.awt.event.MouseEvent;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Class for handling all system tray interaction, via SWING
|
|
|
|
*/
|
2015-06-28 01:47:02 +02:00
|
|
|
public
|
|
|
|
class SwingSystemTray extends dorkbox.util.tray.SystemTray {
|
2015-11-09 16:13:57 +01:00
|
|
|
volatile SystemTrayMenuPopup menu;
|
2015-08-04 21:10:01 +02:00
|
|
|
volatile JMenuItem connectionStatusItem;
|
2014-11-24 17:40:06 +01:00
|
|
|
|
2015-08-04 21:10:01 +02:00
|
|
|
volatile SystemTray tray;
|
|
|
|
volatile TrayIcon trayIcon;
|
2014-11-24 17:40:06 +01:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Creates a new system tray handler class.
|
|
|
|
*/
|
2015-06-28 01:47:02 +02:00
|
|
|
public
|
2015-11-09 16:13:57 +01:00
|
|
|
SwingSystemTray(final String iconName) {
|
|
|
|
super();
|
2014-12-12 02:25:50 +01:00
|
|
|
SwingUtil.invokeAndWait(new Runnable() {
|
2014-11-03 02:11:03 +01:00
|
|
|
@Override
|
2015-06-28 01:47:02 +02:00
|
|
|
public
|
|
|
|
void run() {
|
2014-11-24 17:40:06 +01:00
|
|
|
SwingSystemTray.this.tray = SystemTray.getSystemTray();
|
|
|
|
if (SwingSystemTray.this.tray == null) {
|
2015-01-23 02:52:09 +01:00
|
|
|
logger.error("The system tray is not available");
|
2015-06-28 01:47:02 +02:00
|
|
|
}
|
|
|
|
else {
|
2015-11-09 16:13:57 +01:00
|
|
|
SwingSystemTray.this.menu = new SystemTrayMenuPopup();
|
2014-11-24 17:40:06 +01:00
|
|
|
|
2015-11-09 16:13:57 +01:00
|
|
|
String iconPath = iconPath(iconName);
|
|
|
|
Image trayImage = new ImageIcon(iconPath).getImage()
|
|
|
|
.getScaledInstance(TRAY_SIZE, TRAY_SIZE, Image.SCALE_SMOOTH);
|
|
|
|
trayImage.flush();
|
2015-08-04 21:10:01 +02:00
|
|
|
final TrayIcon trayIcon = new TrayIcon(trayImage);
|
|
|
|
SwingSystemTray.this.trayIcon = trayIcon;
|
2015-11-09 16:13:57 +01:00
|
|
|
|
|
|
|
// appindicators don't support this, so we cater to the lowest common denominator
|
|
|
|
// trayIcon.setToolTip(SwingSystemTray.this.appName);
|
2014-11-24 17:40:06 +01:00
|
|
|
|
2015-08-04 21:10:01 +02:00
|
|
|
trayIcon.addMouseListener(new MouseAdapter() {
|
2014-11-24 17:40:06 +01:00
|
|
|
@Override
|
2015-06-28 01:47:02 +02:00
|
|
|
public
|
|
|
|
void mousePressed(MouseEvent e) {
|
2015-11-09 16:13:57 +01:00
|
|
|
final SystemTrayMenuPopup menu = SwingSystemTray.this.menu;
|
|
|
|
Dimension size = menu.getPreferredSize();
|
2014-11-24 17:40:06 +01:00
|
|
|
|
|
|
|
Point point = e.getPoint();
|
2015-08-04 21:10:01 +02:00
|
|
|
Rectangle bounds = ScreenUtil.getScreenBoundsAt(point);
|
2014-11-24 17:40:06 +01:00
|
|
|
|
|
|
|
int x = point.x;
|
|
|
|
int y = point.y;
|
|
|
|
|
|
|
|
if (y < bounds.y) {
|
|
|
|
y = bounds.y;
|
2015-06-28 01:47:02 +02:00
|
|
|
}
|
|
|
|
else if (y + size.height > bounds.y + bounds.height) {
|
2015-01-23 00:25:35 +01:00
|
|
|
// our menu cannot have the top-edge snap to the mouse
|
|
|
|
// so we make the bottom-edge snap to the mouse
|
|
|
|
y -= size.height; // snap to edge of mouse
|
2014-11-24 17:40:06 +01:00
|
|
|
}
|
2015-01-23 00:25:35 +01:00
|
|
|
|
2014-11-24 17:40:06 +01:00
|
|
|
if (x < bounds.x) {
|
|
|
|
x = bounds.x;
|
2015-06-28 01:47:02 +02:00
|
|
|
}
|
|
|
|
else if (x + size.width > bounds.x + bounds.width) {
|
2015-01-23 00:25:35 +01:00
|
|
|
// our menu cannot have the left-edge snap to the mouse
|
|
|
|
// so we make the right-edge snap to the mouse
|
|
|
|
x -= size.width; // snap to edge of mouse
|
2014-11-24 17:40:06 +01:00
|
|
|
}
|
|
|
|
|
2015-11-09 16:13:57 +01:00
|
|
|
// weird voodoo to get this to popup with the correct parent
|
|
|
|
menu.setInvoker(menu);
|
|
|
|
menu.setLocation(x, y);
|
|
|
|
menu.setVisible(true);
|
|
|
|
menu.requestFocus();
|
2014-11-24 17:40:06 +01:00
|
|
|
}
|
|
|
|
});
|
|
|
|
|
|
|
|
try {
|
2015-08-04 21:10:01 +02:00
|
|
|
SwingSystemTray.this.tray.add(trayIcon);
|
2014-11-24 17:40:06 +01:00
|
|
|
} catch (AWTException e) {
|
|
|
|
logger.error("TrayIcon could not be added.", e);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2014-12-12 02:25:50 +01:00
|
|
|
});
|
2014-11-03 02:11:03 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-09 16:13:57 +01:00
|
|
|
@Override
|
|
|
|
public
|
|
|
|
void shutdown() {
|
|
|
|
SwingUtil.invokeAndWait(new Runnable() {
|
|
|
|
@Override
|
|
|
|
public
|
|
|
|
void run() {
|
2015-11-15 14:51:47 +01:00
|
|
|
SwingSystemTray tray = SwingSystemTray.this;
|
|
|
|
synchronized (tray) {
|
|
|
|
tray.tray.remove(tray.trayIcon);
|
2015-11-09 16:13:57 +01:00
|
|
|
|
2015-11-15 14:51:47 +01:00
|
|
|
for (MenuEntry menuEntry : tray.menuEntries) {
|
|
|
|
menuEntry.remove();
|
|
|
|
}
|
|
|
|
tray.menuEntries.clear();
|
2015-11-09 16:13:57 +01:00
|
|
|
|
2015-11-15 14:51:47 +01:00
|
|
|
tray.connectionStatusItem = null;
|
|
|
|
}
|
2015-11-09 16:13:57 +01:00
|
|
|
}
|
|
|
|
});
|
2014-11-24 17:40:06 +01:00
|
|
|
}
|
2014-11-03 02:11:03 +01:00
|
|
|
|
2014-11-24 17:40:06 +01:00
|
|
|
@Override
|
2015-06-28 01:47:02 +02:00
|
|
|
public
|
2015-11-09 16:13:57 +01:00
|
|
|
void setStatus(final String infoString) {
|
|
|
|
SwingUtil.invokeLater(new Runnable() {
|
2014-11-24 17:40:06 +01:00
|
|
|
@Override
|
2015-06-28 01:47:02 +02:00
|
|
|
public
|
|
|
|
void run() {
|
2015-11-15 14:51:47 +01:00
|
|
|
SwingSystemTray tray = SwingSystemTray.this;
|
|
|
|
synchronized (tray) {
|
|
|
|
if (tray.connectionStatusItem == null) {
|
|
|
|
final JMenuItem connectionStatusItem = new JMenuItem(infoString);
|
|
|
|
Font font = connectionStatusItem.getFont();
|
|
|
|
Font font1 = font.deriveFont(Font.BOLD);
|
|
|
|
connectionStatusItem.setFont(font1);
|
|
|
|
|
|
|
|
connectionStatusItem.setEnabled(false);
|
|
|
|
tray.menu.add(connectionStatusItem);
|
|
|
|
|
|
|
|
tray.connectionStatusItem = connectionStatusItem;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
tray.connectionStatusItem.setText(infoString);
|
|
|
|
}
|
2014-11-24 17:40:06 +01:00
|
|
|
}
|
|
|
|
}
|
2014-12-12 02:25:50 +01:00
|
|
|
});
|
2014-11-03 02:11:03 +01:00
|
|
|
}
|
|
|
|
|
2014-11-24 17:40:06 +01:00
|
|
|
@Override
|
2015-06-28 01:47:02 +02:00
|
|
|
public
|
2015-11-09 16:13:57 +01:00
|
|
|
void setIcon(final String iconName) {
|
|
|
|
SwingUtil.invokeLater(new Runnable() {
|
2014-11-24 17:40:06 +01:00
|
|
|
@Override
|
2015-06-28 01:47:02 +02:00
|
|
|
public
|
|
|
|
void run() {
|
2015-11-15 14:51:47 +01:00
|
|
|
SwingSystemTray tray = SwingSystemTray.this;
|
|
|
|
synchronized (tray) {
|
|
|
|
String iconPath = iconPath(iconName);
|
|
|
|
Image trayImage = new ImageIcon(iconPath).getImage()
|
|
|
|
.getScaledInstance(TRAY_SIZE, TRAY_SIZE, Image.SCALE_SMOOTH);
|
|
|
|
trayImage.flush();
|
|
|
|
tray.trayIcon.setImage(trayImage);
|
|
|
|
}
|
2014-11-24 17:40:06 +01:00
|
|
|
}
|
2014-12-12 02:25:50 +01:00
|
|
|
});
|
2014-11-03 02:11:03 +01:00
|
|
|
}
|
2014-11-24 17:40:06 +01:00
|
|
|
|
|
|
|
/**
|
2015-11-09 16:13:57 +01:00
|
|
|
* Will add a new menu entry, or update one if it already exists
|
2014-11-24 17:40:06 +01:00
|
|
|
*/
|
|
|
|
@Override
|
2015-06-28 01:47:02 +02:00
|
|
|
public
|
2015-11-09 16:13:57 +01:00
|
|
|
void addMenuEntry(final String menuText, final String imagePath, final SystemTrayMenuAction callback) {
|
|
|
|
if (menuText == null) {
|
|
|
|
throw new NullPointerException("Menu text cannot be null");
|
|
|
|
}
|
|
|
|
|
|
|
|
SwingUtil.invokeLater(new Runnable() {
|
2014-11-24 17:40:06 +01:00
|
|
|
@Override
|
2015-06-28 01:47:02 +02:00
|
|
|
public
|
|
|
|
void run() {
|
2015-11-15 14:51:47 +01:00
|
|
|
SwingSystemTray tray = SwingSystemTray.this;
|
|
|
|
synchronized (tray) {
|
2015-11-09 16:13:57 +01:00
|
|
|
MenuEntry menuEntry = getMenuEntry(menuText);
|
2014-11-24 17:40:06 +01:00
|
|
|
|
|
|
|
if (menuEntry != null) {
|
2015-11-09 16:13:57 +01:00
|
|
|
throw new IllegalArgumentException("Menu entry already exists for given label '" + menuText + "'");
|
2015-06-28 01:47:02 +02:00
|
|
|
}
|
|
|
|
else {
|
2015-11-15 14:51:47 +01:00
|
|
|
menuEntry = new SwingMenuEntry(menu, menuText, imagePath, callback, tray);
|
2015-11-09 16:13:57 +01:00
|
|
|
menuEntries.add(menuEntry);
|
2014-11-24 17:40:06 +01:00
|
|
|
}
|
|
|
|
}
|
2014-11-03 02:11:03 +01:00
|
|
|
}
|
2014-12-12 02:25:50 +01:00
|
|
|
});
|
2014-11-03 02:11:03 +01:00
|
|
|
}
|
|
|
|
}
|