/*
 * Copyright 2000-2013 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 git4idea.history.wholeTree;

import com.intellij.codeInsight.completion.CompletionResultSet;
import com.intellij.codeInsight.lookup.LookupElementBuilder;
import com.intellij.openapi.actionSystem.AnAction;
import com.intellij.openapi.actionSystem.AnActionEvent;
import com.intellij.openapi.actionSystem.CommonShortcuts;
import com.intellij.openapi.components.ServiceManager;
import com.intellij.openapi.fileTypes.FileTypes;
import com.intellij.openapi.keymap.KeymapUtil;
import com.intellij.openapi.project.DumbAwareAction;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.ui.popup.ComponentPopupBuilder;
import com.intellij.openapi.ui.popup.JBPopup;
import com.intellij.openapi.ui.popup.JBPopupFactory;
import com.intellij.openapi.util.Comparing;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.spellchecker.ui.SpellCheckingEditorCustomization;
import com.intellij.ui.EditorCustomization;
import com.intellij.ui.EditorTextField;
import com.intellij.ui.EditorTextFieldProvider;
import com.intellij.ui.SoftWrapsEditorCustomization;
import com.intellij.util.Consumer;
import com.intellij.util.TextFieldCompletionProvider;
import com.intellij.util.TextFieldCompletionProviderDumbAware;
import git4idea.history.NewGitUsersComponent;
import org.jetbrains.annotations.NotNull;

import javax.swing.*;
import javax.swing.border.CompoundBorder;
import java.awt.*;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * @author irengrig
 *         Date: 1/20/11
 *         Time: 7:35 PM
 */
public class UsersFilterAction extends BasePopupAction {
  public static final String ALL = "All";
  public static final String USER = "User:";
  private final UserFilterI myUserFilterI;
  private final DumbAwareAction mySelectCurrent;
  private AnAction myAllAction;
  private AnAction mySelectMe;
  private AnAction mySelect;
  private String myCurrentText;
  private final NewGitUsersComponent myUsers;
  private EditorTextField myEditorField;
  private JBPopup myPopup;
  private ComponentPopupBuilder myComponentPopupBuilder;
  private AnAction mySelectOkAction;
  private TextFieldCompletionProvider myTextFieldCompletionProvider;
  private String myPreselectedUser;
  private DumbAwareAction myPresetUserAction;

  public UsersFilterAction(final Project project, final UserFilterI userFilterI) {
    super(project, USER, "User");
    myUserFilterI = userFilterI;
    myCurrentText = "";
    myUsers = NewGitUsersComponent.getInstance(myProject);
    myAllAction = new DumbAwareAction(ALL) {
      @Override
      public void actionPerformed(AnActionEvent e) {
        myLabel.setText(ALL);
        myUserFilterI.allSelected();
        myCurrentText = "";
        myPanel.setToolTipText(USER + " " + ALL);
        myPreselectedUser = null;
      }
    };
    mySelectMe = new DumbAwareAction() {
      @Override
      public void actionPerformed(AnActionEvent e) {
        final String meText = getMeText();
        myLabel.setText(meText);
        myPanel.setToolTipText(USER + " " + meText);
        myUserFilterI.meSelected();
        myCurrentText = "";
        myPreselectedUser = null;
      }

      @Override
      public void update(AnActionEvent e) {
        super.update(e);
        e.getPresentation().setVisible(myUserFilterI.isMeKnown());
        e.getPresentation().setText(getMeText());
      }
    };
    mySelectCurrent = new DumbAwareAction() {
      @Override
      public void actionPerformed(AnActionEvent e) {
        final String user = myUserFilterI.getUserIfOne();
        if (user != null) {
          final String newText = StringUtil.shortenTextWithEllipsis(user, 20, 0);
          myLabel.setText(newText);
          myPanel.setToolTipText(USER + " " + user);
          myUserFilterI.filter(user);
          myCurrentText = user;
          myPreselectedUser = null;
        }
      }

      @Override
      public void update(AnActionEvent e) {
        final String user = myUserFilterI.getUserIfOne();
        final boolean enabled = user != null && !user.equals(myUserFilterI.getMe()) && !user.equals(myCurrentText);
        e.getPresentation().setEnabledAndVisible(enabled);
        if (enabled) {
          e.getPresentation().setText(StringUtil.shortenTextWithEllipsis(user, 20, 0));
        }
      }
    };
    myPresetUserAction = new DumbAwareAction() {
      @Override
      public void actionPerformed(AnActionEvent e) {
        myLabel.setText(myPreselectedUser);
        myPanel.setToolTipText(USER + " " + myPreselectedUser);
        myUserFilterI.filter(myPreselectedUser);
        myCurrentText = myPreselectedUser;
        myPreselectedUser = null;
      }

      @Override
      public void update(AnActionEvent e) {
        super.update(e);
        e.getPresentation().setVisible((! StringUtil.isEmptyOrSpaces(myPreselectedUser)) &&
                                       (! Comparing.equal(myPreselectedUser, myUserFilterI.getMe())));
        e.getPresentation().setText(myPreselectedUser);
      }
    };
    createPopup(project);
    mySelect = new DumbAwareAction("Select..") {
      @Override
      public void actionPerformed(AnActionEvent e) {
        if (myPopup != null) {
          mySelectOkAction.unregisterCustomShortcutSet(myPopup.getContent());
        }
        myComponentPopupBuilder.setMayBeParent(true);
        myPopup = myComponentPopupBuilder.createPopup();
        myTextFieldCompletionProvider.apply(myEditorField);
        myEditorField.setText(myCurrentText);
        final JComponent content = myPopup.getContent();
        mySelectOkAction.registerCustomShortcutSet(CommonShortcuts.CTRL_ENTER, content);
        final Point point = new Point(0, 0);
        SwingUtilities.convertPointToScreen(point, myLabel);
        myPopup.setMinimumSize(new Dimension(200, 90));
        myPopup.showInScreenCoordinates(myLabel, point);
        myPreselectedUser = null;
      }
    };
    myLabel.setText(ALL);
  }

  private void createPopup(Project project) {
    final JPanel panel = new JPanel(new BorderLayout());
    final EditorTextFieldProvider service = ServiceManager.getService(project, EditorTextFieldProvider.class);
    Set<EditorCustomization> features = new HashSet<EditorCustomization>();
    features.add(SoftWrapsEditorCustomization.ENABLED);
    features.add(SpellCheckingEditorCustomization.DISABLED);
    myEditorField = service.getEditorField(FileTypes.PLAIN_TEXT.getLanguage(), project, features);
    myEditorField.setBorder(new CompoundBorder(BorderFactory.createEmptyBorder(2, 2, 2, 2), myEditorField.getBorder()));
    myEditorField.setText("s");
    myEditorField.setText(myCurrentText);
    myEditorField.setOneLineMode(false);
    panel.add(myEditorField, BorderLayout.CENTER);

    myTextFieldCompletionProvider = new TextFieldCompletionProviderDumbAware(true) {
      @NotNull
      @Override
      protected String getPrefix(@NotNull String currentTextPrefix) {
        final int text = currentTextPrefix.lastIndexOf(',');
        return text == -1 ? currentTextPrefix : currentTextPrefix.substring(text + 1).trim();
      }

      @Override
      protected void addCompletionVariants(@NotNull String text,
                                           int offset,
                                           @NotNull String prefix,
                                           @NotNull CompletionResultSet result) {
        final List<String> list = myUsers.get();
        if (list != null) {
          for (String completionVariant : list) {
            final LookupElementBuilder element = LookupElementBuilder.create(completionVariant);
            result.addElement(element.withLookupString(completionVariant.toLowerCase()));
          }
        }
      }
    };

    myComponentPopupBuilder = JBPopupFactory.getInstance().createComponentPopupBuilder(panel, myEditorField)
      .setCancelOnClickOutside(true)
      .setAdText(KeymapUtil.getShortcutsText(CommonShortcuts.CTRL_ENTER.getShortcuts()) + " to finish")
      .setTitle("Specify user names, comma separated")
      .setMovable(true)
      .setRequestFocus(true).setResizable(true);
    mySelectOkAction = new DumbAwareAction() {
      @Override
      public void actionPerformed(AnActionEvent e) {
        myPopup.closeOk(e.getInputEvent());
        final String newText = myEditorField.getText();
        if (Comparing.equal(newText.trim(), myCurrentText.trim())) return;
        myCurrentText = newText;
        setText(myCurrentText.trim());
        myPanel.setToolTipText(USER + " " + myCurrentText);
        myUserFilterI.filter(myCurrentText);
      }
    };
  }

  private void setText(final String text) {
    final String[] pieces = text.split(",");
    if (pieces.length == 0) {
      myLabel.setText(ALL);
    } else if (pieces.length == 1) {
      myLabel.setText(StringUtil.shortenTextWithEllipsis(pieces[0].trim(), 20, 0));
    } else {
      myLabel.setText(StringUtil.shortenTextWithEllipsis(pieces[0].trim(), 20, 0) + "+");
    }
  }

  public void setSelectedPresets(final String selected, final boolean meSelected) {
    myCurrentText = selected == null ? "" : selected;
    if (selected == null) {
      myLabel.setText(ALL);
    } else if (meSelected) {
      myLabel.setText(getMeText(selected));
      myPanel.setToolTipText(USER + " " + selected);
    } else {
      setText(selected);
      myPanel.setToolTipText(USER + " " + selected);
    }
  }

  private String getMeText() {
    return getMeText(myUserFilterI.getMe());
  }
  
  private String getMeText(final String name) {
    return "me ( " + StringUtil.shortenTextWithEllipsis(name, 20, 0) + " )";
  }

  @Override
  protected void createActions(Consumer<AnAction> actionConsumer) {
    actionConsumer.consume(myPresetUserAction);
    actionConsumer.consume(myAllAction);
    actionConsumer.consume(mySelectMe);
    actionConsumer.consume(mySelect);
    actionConsumer.consume(mySelectCurrent);
  }

  public void setPreselectedUser(String preselectedUser) {
    myPreselectedUser = StringUtil.shortenTextWithEllipsis(preselectedUser, 20, 0);
  }
}