/**
 * 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.
 *
 * Copyright 2012-2015 the original author or authors.
 */
package org.assertj.swing.fixture;

import static java.awt.BorderLayout.CENTER;
import static java.awt.BorderLayout.SOUTH;
import static org.assertj.core.util.Strings.concat;
import static org.assertj.core.util.Strings.quote;
import static org.assertj.swing.edt.GuiActionRunner.execute;

import java.awt.BorderLayout;
import java.awt.Dialog;
import java.awt.event.ActionEvent;

import javax.annotation.Nonnull;
import javax.swing.AbstractAction;
import javax.swing.JButton;
import javax.swing.JDialog;
import javax.swing.JLabel;

import org.assertj.swing.annotation.RunsInEDT;
import org.assertj.swing.core.GenericTypeMatcher;
import org.assertj.swing.finder.WindowFinder;
import org.assertj.swing.test.core.RobotBasedTestCase;
import org.junit.Test;

/**
 * Tests for <a href="http://code.google.com/p/fest/issues/detail?id=195" target="_blank">Bug 195</a>.
 * 
 * @author Bryan Shannon
 * @author Alex Ruiz
 */
public class Bug195_findDialogMultipleTimes_Test extends RobotBasedTestCase {
  @Test
  public void should_Find_Dialog_Multiple_Times() {
    JDialog dialog = TestDialog.createNew("title 1");
    robot.showWindow(dialog);
    DialogFixture fixtureOne = WindowFinder.findDialog(new DialogTitleStartsWithMatcher("title")).using(robot);
    fixtureOne.button("OK").click();

    dialog = TestDialog.createNew("title 2");
    robot.showWindow(dialog);
    DialogFixture fixtureTwo = WindowFinder.findDialog(new DialogTitleStartsWithMatcher("title")).using(robot);
    fixtureTwo.button("OK").click();
  }

  private static class DialogTitleStartsWithMatcher extends GenericTypeMatcher<Dialog> {
    private final String matchString;

    public DialogTitleStartsWithMatcher(String s) {
      super(Dialog.class, false);
      this.matchString = s != null ? s.toUpperCase() : null;
    }

    @Override
    protected boolean isMatching(@Nonnull Dialog dialog) {
      String title = dialog.getTitle();
      return title != null && title.toUpperCase().startsWith(matchString) && isShowing(dialog);
    }

    @RunsInEDT
    private static boolean isShowing(final Dialog dialog) {
      return execute(() -> dialog.isShowing());
    }

    @Override
    public String toString() {
      return concat(DialogTitleStartsWithMatcher.class.getSimpleName(), "(", quote(matchString), ")");
    }
  }

  private static class TestDialog extends JDialog {
    JButton okayButton = new JButton();

    @RunsInEDT
    static TestDialog createNew(final String title) {
      return execute(() -> new TestDialog(title));
    }

    private TestDialog(String title) {
      super();
      setName(title);
      setLayout(new BorderLayout());
      add(new JLabel("Test Dialog"), CENTER);
      okayButton.setAction(new OKAction(this));
      okayButton.setName("OK");
      getContentPane().add(okayButton, SOUTH);
      setTitle(title);
    }
  }

  // Simple action for the dialog that closes the dialog.
  private static class OKAction extends AbstractAction {
    private final JDialog dialog;

    public OKAction(JDialog dialog) {
      super("OK");
      this.dialog = dialog;
    }

    @Override
    public void actionPerformed(ActionEvent e) {
      dialog.setVisible(false);
      dialog.dispose();
    }
  }
}