/*

 * Copyright (c) 2007, 2014, 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.

 */

import java.awt.*;

public class FocusTransferDialogsTest {

    class CustomDialog1 extends TestDialog {

        public CustomDialog1(Frame f) {

            super(f);

        }

        @Override

        public void doOpenAction() {

            if (dialog2 != null) {

                dialog2.setVisible(true);

            }

        }

        @Override

        public void doCloseAction() {

            this.dispose();

        }

    }

    class CustomDialog2 extends TestDialog {

        public CustomDialog2(Dialog d) {

            super(d);

        }

        @Override

        public void doOpenAction() {

            if (dialog3 != null) {

                dialog3.setVisible(true);

            }

        }

        @Override

        public void doCloseAction() {

            this.dispose();

        }

    }

    class CustomDialog3 extends TestDialog {

        public CustomDialog3(Frame f) {

            super(f);

        }

        public CustomDialog3(Dialog d) {

            super(d);

        }

        @Override

        public void doCloseAction() {

            this.dispose();

        }

    }

    private TestDialog dialog1, dialog2, dialog3;

    private Frame parentFrame;

    private static final int delay = 1000;

    private final ExtendedRobot robot;

    private Dialog.ModalityType modalityType;

    FocusTransferDialogsTest(Dialog.ModalityType modType) throws Exception {

        modalityType = modType;

        robot = new ExtendedRobot();

        EventQueue.invokeLater(this::createGUI);

    }

    private void createGUI() {

        dialog1 = new CustomDialog1((Frame) null);

        dialog1.setTitle("Dialog1");

        dialog1.setLocation(50, 50);

        if (modalityType != null) {

            dialog1.setModalityType(modalityType);

        } else {

            modalityType = Dialog.ModalityType.MODELESS;

        }

        dialog2 = new CustomDialog2(dialog1);

        dialog2.setTitle("Dialog2");

        dialog2.setLocation(250, 50);

        parentFrame = new Frame();

        dialog3 = new CustomDialog3(parentFrame);

        dialog3.setTitle("Dialog3");

        dialog3.setLocation(450, 50);

        dialog1.setVisible(true);

    }

    private void closeAll() {

        if (dialog1 != null) { dialog1.dispose(); }

        if (dialog2 != null) { dialog2.dispose(); }

        if (dialog3 != null) { dialog3.dispose(); }

        if (parentFrame != null) { parentFrame.dispose(); }

    }

    public void doTest() throws Exception {

        robot.waitForIdle(delay);

        try {

            dialog1.checkCloseButtonFocusGained(true);

            dialog1.clickOpenButton(robot);

            robot.waitForIdle(delay);

            dialog2.checkCloseButtonFocusGained(true);

            dialog1.checkOpenButtonFocusLost(true);

            dialog1.openGained.reset();

            dialog2.clickOpenButton(robot);

            robot.waitForIdle(delay);

            switch (modalityType) {

                case APPLICATION_MODAL:

                    dialog3.checkCloseButtonFocusGained(false, 10);

                    dialog2.checkOpenButtonFocusLost(true);

                    dialog1.checkCloseButtonFocusGained(true);

                    dialog3.closeGained.reset();

                    dialog1.clickCloseButton(robot);

                    robot.waitForIdle(delay);

                    dialog3.checkCloseButtonFocusGained(true);

                    break;

                case DOCUMENT_MODAL:

                case MODELESS:

                    dialog3.checkCloseButtonFocusGained(true);

                    dialog2.checkOpenButtonFocusLost(true);

                    dialog1.openGained.reset();

                    dialog2.clickCloseButton(robot);

                    robot.waitForIdle(delay);

                    dialog1.checkOpenButtonFocusGained(true);

                    break;

            }

        } catch (Exception e) {

            // make screenshot before exit

            Rectangle rect = new Rectangle(0, 0, 650, 250);

            java.awt.image.BufferedImage img = robot.createScreenCapture(rect);

            javax.imageio.ImageIO.write(img, "jpg", new java.io.File("NOK.jpg"));

            throw e;

        }

        robot.waitForIdle(delay);

        EventQueue.invokeAndWait(this::closeAll);

    }

}