/*

 * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/*

 @test

 @bug 8147841

 @summary Updating Tray Icon popup menu does not update menu items on Mac OS X

 @run main/manual UpdatePopupMenu

 */

import java.awt.SystemTray;

import java.awt.TrayIcon;

import java.awt.PopupMenu;

import java.awt.MenuItem;

import java.awt.Image;

import java.awt.Graphics2D;

import java.awt.Color;

import java.awt.image.BufferedImage;

import java.awt.RenderingHints;

import java.awt.AWTException;

import java.awt.Button;

import java.awt.Frame;

import java.awt.GridBagConstraints;

import java.awt.GridBagLayout;

import java.awt.Panel;

import java.awt.TextArea;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

public class UpdatePopupMenu implements ActionListener {

    private static final int imageSize = 32;

    private static final int imageInset = 4;

    private static GridBagLayout layout;

    private static Panel mainControlPanel;

    private static Panel resultButtonPanel;

    private static TextArea instructionTextArea;

    private static Button passButton;

    private static Button failButton;

    private static Frame mainFrame;

    private static Thread mainThread = null;

    private static boolean testPassed = false;

    private static boolean isInterrupted = false;

    private static final int testTimeOut = 300000;

    private Image createSystemTrayIconImage() {

        final BufferedImage trayImage = new BufferedImage(

                imageSize,

                imageSize,

                BufferedImage.TYPE_INT_ARGB);

        final Graphics2D imageGraphics = (Graphics2D) trayImage.getGraphics();

        imageGraphics.setRenderingHint(RenderingHints.KEY_ANTIALIASING,

                RenderingHints.VALUE_ANTIALIAS_ON);

        imageGraphics.setColor(new Color(255, 255, 255, 0));

        imageGraphics.fillRect(0, 0, trayImage.getWidth(),

                trayImage.getHeight());

        imageGraphics.setColor(Color.green);

        int imageWidth = trayImage.getWidth() - 2 * imageInset;

        int imageHeight = trayImage.getHeight() - 2 * imageInset;

        imageGraphics.fillOval(imageInset, imageInset, imageWidth, imageHeight);

        imageGraphics.setColor(Color.darkGray);

        imageGraphics.drawOval(imageInset, imageInset, imageWidth, imageHeight);

        return trayImage;

    }

    private PopupMenu createPopupMenu(final TrayIcon trayIcon,

            final int menuCount) {

        final PopupMenu trayIconPopupMenu = new PopupMenu();

        for (int i = 1; i <= menuCount; ++i) {

            final MenuItem popupMenuItem = new MenuItem("MenuItem_" + i);

            popupMenuItem.addActionListener(new ActionListener() {

                @Override

                public void actionPerformed(final ActionEvent ae) {

                    trayIcon.setPopupMenu(createPopupMenu(trayIcon,

                            menuCount + 1));

                }

            });

            trayIconPopupMenu.add(popupMenuItem);

        }

        return trayIconPopupMenu;

    }

    private void createSystemTrayIcons() {

        final TrayIcon trayIcon = new TrayIcon(createSystemTrayIconImage());

        trayIcon.setImageAutoSize(true);

        trayIcon.setToolTip("Update Popup Menu items");

        try {

            trayIcon.setPopupMenu(createPopupMenu(trayIcon, 2));

            SystemTray.getSystemTray().add(trayIcon);

        } catch (AWTException ex) {

            throw new RuntimeException("System Tray cration failed");

        }

    }

    private void createInstructionUI() {

        mainFrame = new Frame("Updating TrayIcon Popup Menu Item Test");

        layout = new GridBagLayout();

        mainControlPanel = new Panel(layout);

        resultButtonPanel = new Panel(layout);

        GridBagConstraints gbc = new GridBagConstraints();

        String instructions

                = "INSTRUCTIONS:"

                + "\n   1. Click on the System Tray Icon"

                + "\n   2. Click on any of the displayed Menu items"

                + "\n   3. Repeat step 1 and count the number of items in the "

                + "Menu"

                + "\n   4. The number of items in the Menu should increase by 1"

                + "\n   5. Repeating steps 1, 2 and 3 should not break 4th step"

                + "\n   6. Click Fail if the 4th step is broken, Otherwise "

                + "click Pass ";

        instructionTextArea = new TextArea();

        instructionTextArea.setText(instructions);

        instructionTextArea.setEnabled(false);

        instructionTextArea.setBackground(Color.white);

        gbc.gridx = 0;

        gbc.gridy = 0;

        gbc.fill = GridBagConstraints.HORIZONTAL;

        mainControlPanel.add(instructionTextArea, gbc);

        passButton = new Button("Pass");

        passButton.setName("Pass");

        passButton.addActionListener(this);

        failButton = new Button("Fail");

        failButton.setName("Fail");

        failButton.addActionListener(this);

        gbc.gridx = 0;

        gbc.gridy = 0;

        resultButtonPanel.add(passButton, gbc);

        gbc.gridx = 1;

        gbc.gridy = 0;

        resultButtonPanel.add(failButton, gbc);

        gbc.gridx = 0;

        gbc.gridy = 1;

        mainControlPanel.add(resultButtonPanel, gbc);

        mainFrame.add(mainControlPanel);

        mainFrame.pack();

        mainFrame.setVisible(true);

    }

    @Override

    public void actionPerformed(ActionEvent ae) {

        if (ae.getSource() instanceof Button) {

            Button btn = (Button) ae.getSource();

            switch (btn.getName()) {

                case "Pass":

                    testPassed = true;

                    isInterrupted = true;

                    mainThread.interrupt();

                    break;

                case "Fail":

                    testPassed = false;

                    isInterrupted = true;

                    mainThread.interrupt();

                    break;

            }

        }

    }

    private static void cleanUp() {

        mainFrame.dispose();

    }

    public static void main(final String[] args) throws Exception {

        if (SystemTray.isSupported()) {

            UpdatePopupMenu updatePopupMenu = new UpdatePopupMenu();

            updatePopupMenu.createInstructionUI();

            updatePopupMenu.createSystemTrayIcons();

            mainThread = Thread.currentThread();

            try {

                mainThread.sleep(testTimeOut);

            } catch (InterruptedException ex) {

                if (!testPassed) {

                    throw new RuntimeException("Updating TrayIcon popup menu"

                            + " items FAILED");

                }

            } finally {

                cleanUp();

            }

            if (!isInterrupted) {

                throw new RuntimeException("Test Timed out after "

                        + testTimeOut / 1000 + " seconds");

            }

        } else {

            System.out.println("System Tray is not supported on this platform");

        }

    }

}