/*
 * Copyright 2000-2012 JetBrains s.r.o.
 *
 * 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.
 */
package com.intellij.tasks;

import com.intellij.ide.ui.customization.CustomActionsSchema;
import com.intellij.openapi.actionSystem.*;
import com.intellij.openapi.actionSystem.ex.CustomComponentAction;
import com.intellij.openapi.actionSystem.impl.ActionToolbarImpl;
import com.intellij.openapi.actionSystem.impl.PresentationFactory;
import com.intellij.openapi.actionSystem.impl.Utils;
import com.intellij.tasks.actions.SwitchTaskAction;
import com.intellij.tasks.config.TaskSettings;
import com.intellij.tasks.impl.LocalTaskImpl;
import com.intellij.testFramework.TestActionEvent;
import com.intellij.testFramework.fixtures.CodeInsightFixtureTestCase;

import javax.swing.*;
import java.util.List;

/**
 * @author Dmitry Avdeev
 */
public class TaskUiTest extends CodeInsightFixtureTestCase {

  public void testTaskComboVisible() {

    SwitchTaskAction combo = null;
    ActionGroup group = (ActionGroup)CustomActionsSchema.getInstance().getCorrectedAction(IdeActions.GROUP_MAIN_TOOLBAR);
    ActionToolbarImpl toolbar = (ActionToolbarImpl)ActionManager.getInstance().createActionToolbar(ActionPlaces.MAIN_TOOLBAR, group, true);
    AnAction[] children = group.getChildren(TestActionEvent.createTestEvent());
    for (AnAction child : children) {
      if (child instanceof ActionGroup) {
        AnAction[] actions = ((ActionGroup)child).getChildren(TestActionEvent.createTestEvent());
        for (AnAction action : actions) {
          if (action instanceof SwitchTaskAction) {
            combo = (SwitchTaskAction)action;
          }
        }
      }
    }

    List<AnAction> actions = Utils.expandActionGroup(
      group, new PresentationFactory(), DataContext.EMPTY_CONTEXT, ActionPlaces.MAIN_TOOLBAR);
    assertFalse(actions.contains(combo));

    TaskManager manager = TaskManager.getManager(getProject());
    LocalTask defaultTask = manager.getActiveTask();
    assertTrue(defaultTask.isDefault());
    assertEquals(defaultTask.getCreated(), defaultTask.getUpdated());

    toolbar.updateActionsImmediately();
    Presentation presentation = doTest(combo, toolbar);
    assertFalse(presentation.isVisible());
    assertNull(presentation.getClientProperty(CustomComponentAction.COMPONENT_KEY));

    try {
      TaskSettings.getInstance().ALWAYS_DISPLAY_COMBO = true;
      presentation = doTest(combo, toolbar);
      assertTrue(presentation.isVisible());
    }
    finally {
      TaskSettings.getInstance().ALWAYS_DISPLAY_COMBO = false;
    }

    LocalTask task = manager.createLocalTask("test");
    manager.activateTask(task, false);
    TaskSettings.getInstance().ALWAYS_DISPLAY_COMBO = true;

    presentation = doTest(combo, toolbar);
    assertTrue(presentation.isVisible());
    assertTrue(presentation.isEnabled());

    toolbar.updateActionsImmediately();
    JComponent component = presentation.getClientProperty(CustomComponentAction.COMPONENT_KEY);
    assertNotNull(component);
    assertTrue(component.isVisible());
    assertTrue(component.isEnabled());

    manager.activateTask(defaultTask, false);
    task = manager.getActiveTask();
    assertTrue(task.isDefault());

    presentation = doTest(combo, toolbar);
    if (!presentation.isVisible()) {
      LocalTask activeTask = manager.getActiveTask();
      System.out.println(activeTask);
      System.out.println(activeTask.getCreated());
      System.out.println(activeTask.getUpdated());
      fail();
    }
  }

  public void testUnderscore() {
    String summary = "foo_bar";
    TaskManager.getManager(getProject()).activateTask(new LocalTaskImpl("", summary), false);
    TaskSettings.getInstance().ALWAYS_DISPLAY_COMBO = true;
    AnActionEvent event = TestActionEvent.createTestToolbarEvent(null);
    new SwitchTaskAction().update(event);
    assertEquals(summary, event.getPresentation().getText());
  }

  @Override
  protected void tearDown() throws Exception {
    super.tearDown();
    TaskSettings.getInstance().ALWAYS_DISPLAY_COMBO = false;
  }

  private static Presentation doTest(AnAction action, ActionToolbarImpl toolbar) {
    AnActionEvent event = TestActionEvent.createTestToolbarEvent(toolbar.getPresentation(action));
    action.update(event);
    return event.getPresentation();
  }
}
