forked from dorkbox/SystemTray
358 lines
13 KiB
Java
358 lines
13 KiB
Java
|
package dorkbox.util.tray.swing;
|
||
|
|
||
|
|
||
|
import java.awt.AWTException;
|
||
|
import java.awt.Dimension;
|
||
|
import java.awt.GraphicsConfiguration;
|
||
|
import java.awt.GraphicsDevice;
|
||
|
import java.awt.GraphicsEnvironment;
|
||
|
import java.awt.Image;
|
||
|
import java.awt.Point;
|
||
|
import java.awt.Rectangle;
|
||
|
import java.awt.SystemTray;
|
||
|
import java.awt.TrayIcon;
|
||
|
import java.awt.event.ActionEvent;
|
||
|
import java.awt.event.ActionListener;
|
||
|
import java.awt.event.MouseAdapter;
|
||
|
import java.awt.event.MouseEvent;
|
||
|
import java.lang.reflect.InvocationTargetException;
|
||
|
import java.util.ArrayList;
|
||
|
import java.util.HashMap;
|
||
|
import java.util.Map;
|
||
|
|
||
|
import javax.swing.ImageIcon;
|
||
|
import javax.swing.JMenuItem;
|
||
|
import javax.swing.SwingUtilities;
|
||
|
|
||
|
import dorkbox.util.tray.SystemTrayMenuAction;
|
||
|
import dorkbox.util.tray.SystemTrayMenuPopup;
|
||
|
|
||
|
|
||
|
|
||
|
/**
|
||
|
* Class for handling all system tray interaction, via SWING
|
||
|
*/
|
||
|
public class SwingSystemTray extends dorkbox.util.tray.SystemTray {
|
||
|
private final Map<String, JMenuItem> menuEntries = new HashMap<String, JMenuItem>(2);
|
||
|
|
||
|
private volatile SystemTrayMenuPopup jmenu;
|
||
|
private volatile JMenuItem connectionStatusItem;
|
||
|
|
||
|
private volatile SystemTray tray;
|
||
|
private volatile TrayIcon trayIcon;
|
||
|
|
||
|
/**
|
||
|
* Creates a new system tray handler class.
|
||
|
*/
|
||
|
public SwingSystemTray() {
|
||
|
}
|
||
|
|
||
|
|
||
|
@Override
|
||
|
public void removeTray() {
|
||
|
Runnable doRun = new Runnable() {
|
||
|
@Override
|
||
|
public void run() {
|
||
|
SwingSystemTray.this.tray.remove(SwingSystemTray.this.trayIcon);
|
||
|
SwingSystemTray.this.menuEntries.clear();
|
||
|
}
|
||
|
};
|
||
|
|
||
|
if (SwingUtilities.isEventDispatchThread()) {
|
||
|
doRun.run();
|
||
|
} else {
|
||
|
try {
|
||
|
SwingUtilities.invokeAndWait(doRun);
|
||
|
} catch (InvocationTargetException e) {
|
||
|
logger.error("Error updating tray menu", e);
|
||
|
} catch (InterruptedException e) {
|
||
|
logger.error("Error updating tray menu", e);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
@Override
|
||
|
public void createTray(final String iconName) {
|
||
|
Runnable doRun = new Runnable() {
|
||
|
@Override
|
||
|
public void run() {
|
||
|
SwingSystemTray.this.tray = SystemTray.getSystemTray();
|
||
|
if (SwingSystemTray.this.tray == null) {
|
||
|
logger.warn("The system tray is not available");
|
||
|
} else {
|
||
|
SwingSystemTray.this.jmenu = new SystemTrayMenuPopup();
|
||
|
|
||
|
Image trayImage = newImage(iconName);
|
||
|
SwingSystemTray.this.trayIcon = new TrayIcon(trayImage);
|
||
|
SwingSystemTray.this.trayIcon.setToolTip(SwingSystemTray.this.appName);
|
||
|
|
||
|
SwingSystemTray.this.trayIcon.addMouseListener(new MouseAdapter() {
|
||
|
@Override
|
||
|
public void mousePressed(MouseEvent e) {
|
||
|
Dimension size = SwingSystemTray.this.jmenu.getPreferredSize();
|
||
|
|
||
|
Point point = e.getPoint();
|
||
|
Rectangle bounds = getScreenBoundsAt(point);
|
||
|
|
||
|
|
||
|
int x = point.x;
|
||
|
int y = point.y;
|
||
|
|
||
|
if (y < bounds.y) {
|
||
|
y = bounds.y;
|
||
|
} else if (y > bounds.y + bounds.height) {
|
||
|
y = bounds.y + bounds.height + ICON_SIZE + 4; // 4 for padding
|
||
|
}
|
||
|
if (x < bounds.x) {
|
||
|
x = bounds.x;
|
||
|
} else if (x > bounds.x + bounds.width) {
|
||
|
x = bounds.x + bounds.width;
|
||
|
}
|
||
|
|
||
|
if (x + size.width > bounds.x + bounds.width) {
|
||
|
// always put the menu in the middle
|
||
|
x = bounds.x + bounds.width - size.width;
|
||
|
}
|
||
|
if (y + size.height > bounds.y + bounds.height) {
|
||
|
y = bounds.y + bounds.height - size.height - ICON_SIZE - 4; // 4 for padding
|
||
|
}
|
||
|
|
||
|
// do we open at top-right or top-left?
|
||
|
// we ASSUME monitor size is greater than 640x480 AND that our tray icon is IN THE CORNER SOMEWHERE
|
||
|
|
||
|
// always put the menu in the middle
|
||
|
x -= size.width/4;
|
||
|
|
||
|
|
||
|
SwingSystemTray.this.jmenu.setInvoker(SwingSystemTray.this.jmenu);
|
||
|
SwingSystemTray.this.jmenu.setLocation(x, y);
|
||
|
SwingSystemTray.this.jmenu.setVisible(true);
|
||
|
}
|
||
|
});
|
||
|
|
||
|
try {
|
||
|
SwingSystemTray.this.tray.add(SwingSystemTray.this.trayIcon);
|
||
|
SwingSystemTray.this.active = true;
|
||
|
} catch (AWTException e) {
|
||
|
logger.error("TrayIcon could not be added.", e);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
|
||
|
if (SwingUtilities.isEventDispatchThread()) {
|
||
|
doRun.run();
|
||
|
} else {
|
||
|
try {
|
||
|
SwingUtilities.invokeAndWait(doRun);
|
||
|
} catch (InvocationTargetException e) {
|
||
|
logger.error("Error creating tray menu", e);
|
||
|
} catch (InterruptedException e) {
|
||
|
logger.error("Error creating tray menu", e);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
protected Image newImage(String name) {
|
||
|
String iconPath = iconPath(name);
|
||
|
|
||
|
Image scaledInstance = new ImageIcon(iconPath).getImage().getScaledInstance(ICON_SIZE, ICON_SIZE, Image.SCALE_SMOOTH);
|
||
|
return scaledInstance;
|
||
|
}
|
||
|
|
||
|
|
||
|
// public static Rectangle getSafeScreenBounds(Point pos) {
|
||
|
// Rectangle bounds = getScreenBoundsAt(pos);
|
||
|
// Insets insets = getScreenInsetsAt(pos);
|
||
|
//
|
||
|
// bounds.x += insets.left;
|
||
|
// bounds.y += insets.top;
|
||
|
// bounds.width -= insets.left + insets.right;
|
||
|
// bounds.height -= insets.top + insets.bottom;
|
||
|
//
|
||
|
// return bounds;
|
||
|
// }
|
||
|
|
||
|
// public static Insets getScreenInsetsAt(Point pos) {
|
||
|
// GraphicsDevice gd = getGraphicsDeviceAt(pos);
|
||
|
// Insets insets = null;
|
||
|
// if (gd != null) {
|
||
|
// insets = Toolkit.getDefaultToolkit().getScreenInsets(gd.getDefaultConfiguration());
|
||
|
// }
|
||
|
//
|
||
|
// return insets;
|
||
|
// }
|
||
|
|
||
|
public static Rectangle getScreenBoundsAt(Point pos) {
|
||
|
GraphicsDevice gd = getGraphicsDeviceAt(pos);
|
||
|
Rectangle bounds = null;
|
||
|
if (gd != null) {
|
||
|
bounds = gd.getDefaultConfiguration().getBounds();
|
||
|
}
|
||
|
|
||
|
return bounds;
|
||
|
}
|
||
|
|
||
|
public static GraphicsDevice getGraphicsDeviceAt(Point pos) {
|
||
|
GraphicsDevice device = null;
|
||
|
|
||
|
GraphicsEnvironment ge = GraphicsEnvironment.getLocalGraphicsEnvironment();
|
||
|
GraphicsDevice lstGDs[] = ge.getScreenDevices();
|
||
|
|
||
|
ArrayList<GraphicsDevice> lstDevices = new ArrayList<GraphicsDevice>(lstGDs.length);
|
||
|
|
||
|
for (GraphicsDevice gd : lstGDs) {
|
||
|
|
||
|
GraphicsConfiguration gc = gd.getDefaultConfiguration();
|
||
|
Rectangle screenBounds = gc.getBounds();
|
||
|
|
||
|
if (screenBounds.contains(pos)) {
|
||
|
lstDevices.add(gd);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (lstDevices.size() > 0) {
|
||
|
device = lstDevices.get(0);
|
||
|
} else {
|
||
|
device = ge.getDefaultScreenDevice();
|
||
|
}
|
||
|
|
||
|
return device;
|
||
|
}
|
||
|
|
||
|
@Override
|
||
|
public void setStatus(final String infoString, final String iconName) {
|
||
|
Runnable doRun = new Runnable() {
|
||
|
@Override
|
||
|
public void run() {
|
||
|
if (SwingSystemTray.this.connectionStatusItem == null) {
|
||
|
SwingSystemTray.this.connectionStatusItem = new JMenuItem(infoString);
|
||
|
SwingSystemTray.this.connectionStatusItem.setEnabled(false);
|
||
|
SwingSystemTray.this.jmenu.add(SwingSystemTray.this.connectionStatusItem);
|
||
|
} else {
|
||
|
SwingSystemTray.this.connectionStatusItem.setText(infoString);
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
if (SwingUtilities.isEventDispatchThread()) {
|
||
|
doRun.run();
|
||
|
} else {
|
||
|
try {
|
||
|
SwingUtilities.invokeAndWait(doRun);
|
||
|
} catch (InvocationTargetException e) {
|
||
|
logger.error("Error updating tray menu", e);
|
||
|
} catch (InterruptedException e) {
|
||
|
logger.error("Error updating tray menu", e);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
Image trayImage = newImage(iconName);
|
||
|
SwingSystemTray.this.trayIcon.setImage(trayImage);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Will add a new menu entry, or update one if it already exists
|
||
|
*/
|
||
|
@Override
|
||
|
public void addMenuEntry(final String menuText, final SystemTrayMenuAction callback) {
|
||
|
Runnable doRun = new Runnable() {
|
||
|
@Override
|
||
|
public void run() {
|
||
|
Map<String, JMenuItem> menuEntries2 = SwingSystemTray.this.menuEntries;
|
||
|
|
||
|
synchronized (menuEntries2) {
|
||
|
JMenuItem menuEntry = menuEntries2.get(menuText);
|
||
|
|
||
|
if (menuEntry == null) {
|
||
|
SystemTrayMenuPopup menu = SwingSystemTray.this.jmenu;
|
||
|
|
||
|
menuEntry = new JMenuItem(menuText);
|
||
|
menuEntry.addActionListener(new ActionListener() {
|
||
|
@Override
|
||
|
public void actionPerformed(ActionEvent e) {
|
||
|
SwingSystemTray.this.callbackExecutor.execute(new Runnable() {
|
||
|
@Override
|
||
|
public void run() {
|
||
|
callback.onClick(SwingSystemTray.this);
|
||
|
}
|
||
|
});
|
||
|
}
|
||
|
});
|
||
|
menu.add(menuEntry);
|
||
|
|
||
|
menuEntries2.put(menuText, menuEntry);
|
||
|
} else {
|
||
|
updateMenuEntry(menuText, menuText, callback);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
if (SwingUtilities.isEventDispatchThread()) {
|
||
|
doRun.run();
|
||
|
} else {
|
||
|
try {
|
||
|
SwingUtilities.invokeAndWait(doRun);
|
||
|
} catch (InvocationTargetException e) {
|
||
|
logger.error("Error updating tray menu", e);
|
||
|
} catch (InterruptedException e) {
|
||
|
logger.error("Error updating tray menu", e);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Will update an already existing menu entry (or add a new one, if it doesn't exist)
|
||
|
*/
|
||
|
@Override
|
||
|
public void updateMenuEntry(final String origMenuText, final String newMenuText, final SystemTrayMenuAction newCallback) {
|
||
|
Runnable doRun = new Runnable() {
|
||
|
@Override
|
||
|
public void run() {
|
||
|
Map<String, JMenuItem> menuEntries2 = SwingSystemTray.this.menuEntries;
|
||
|
|
||
|
synchronized (menuEntries2) {
|
||
|
JMenuItem menuEntry = menuEntries2.get(origMenuText);
|
||
|
|
||
|
if (menuEntry != null) {
|
||
|
ActionListener[] actionListeners = menuEntry.getActionListeners();
|
||
|
for (ActionListener l : actionListeners) {
|
||
|
menuEntry.removeActionListener(l);
|
||
|
}
|
||
|
|
||
|
menuEntry.addActionListener(new ActionListener() {
|
||
|
@Override
|
||
|
public void actionPerformed(ActionEvent e) {
|
||
|
SwingSystemTray.this.callbackExecutor.execute(new Runnable() {
|
||
|
@Override
|
||
|
public void run() {
|
||
|
newCallback.onClick(SwingSystemTray.this);
|
||
|
}
|
||
|
});
|
||
|
}
|
||
|
});
|
||
|
menuEntry.setText(newMenuText);
|
||
|
menuEntry.revalidate();
|
||
|
} else {
|
||
|
addMenuEntry(origMenuText, newCallback);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
if (SwingUtilities.isEventDispatchThread()) {
|
||
|
doRun.run();
|
||
|
} else {
|
||
|
try {
|
||
|
SwingUtilities.invokeAndWait(doRun);
|
||
|
} catch (InvocationTargetException e) {
|
||
|
logger.error("Error updating tray menu", e);
|
||
|
} catch (InterruptedException e) {
|
||
|
logger.error("Error updating tray menu", e);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|