SystemTray/src/dorkbox/util/tray/SystemTrayMenuPopup.java

package dorkbox.util.tray;

import java.awt.Dimension;
import java.awt.MouseInfo;
import java.awt.Point;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;

import javax.swing.JPopupMenu;

import dorkbox.util.DelayTimer;
import dorkbox.util.SwingUtil;

public class SystemTrayMenuPopup extends JPopupMenu {
    private static final long serialVersionUID = 1L;

    private DelayTimer timer;

    public SystemTrayMenuPopup() {
        super();

        this.timer = new DelayTimer("PopupMenuHider", true, new DelayTimer.Callback() {
            @Override
            public void execute() {
                SwingUtil.invokeLater(new Runnable() {
                    @Override
                    public void run() {
                        Point location = MouseInfo.getPointerInfo().getLocation();
                        Point locationOnScreen = getLocationOnScreen();
                        Dimension size = getSize();

                        if (location.x >= locationOnScreen.x && location.x < locationOnScreen.x + size.width
                            && location.y >= locationOnScreen.y && location.y < locationOnScreen.y + size.height) {

                            SystemTrayMenuPopup.this.timer.delay(SystemTrayMenuPopup.this.timer.getDelay());
                        } else {
                            setVisible(false);
                        }
                    }
                });
            }
        });

        addMouseListener(new MouseAdapter() {
            @Override
            public void mouseExited(MouseEvent event) {
                // wait before checking if mouse is still on the menu
                SystemTrayMenuPopup.this.timer.delay(SystemTrayMenuPopup.this.timer.getDelay());
            }
        });
    }


    @Override
    public void setVisible(boolean b) {
        this.timer.cancel();

        if (b) {
            // if the mouse isn't inside the popup in 5 seconds, close the popup
            this.timer.delay(5000L);
        }

        super.setVisible(b);
    }
}