ShortcutPromoterManager.java

package com.intellij.promoter;

import com.intellij.openapi.Disposable;
import com.intellij.openapi.actionSystem.ActionManager;
import com.intellij.openapi.actionSystem.AnAction;
import com.intellij.openapi.actionSystem.AnActionEvent;
import com.intellij.openapi.actionSystem.DataContext;
import com.intellij.openapi.actionSystem.ex.AnActionListener;
import com.intellij.openapi.components.*;
import com.intellij.openapi.util.text.StringUtil;
import gnu.trove.THashMap;
import org.jdom.Element;
import org.jetbrains.annotations.Nullable;

import java.awt.event.InputEvent;
import java.awt.event.MouseEvent;
import java.util.LinkedHashMap;
import java.util.Map;

/**
 * @author Konstantin Bulenkov
 */
@State(
  name = "ShortcutPromoterManager",
  storages = @Storage(value = "promoter.xml", roamingType = RoamingType.PER_OS)
)
public class ShortcutPromoterManager implements Disposable, AnActionListener, PersistentStateComponent<Element>, ApplicationComponent {
  private final Map<String, PromoterState> myState = new LinkedHashMap<>();
  private final Map<String, ShortcutPromoterEP> myExtensions = new THashMap<>();

  @Override
  public void initComponent() {
    myExtensions.clear();
    myState.clear();

    for (ShortcutPromoterEP ep : ShortcutPromoterEP.EP_NAME.getExtensions()) {
      myExtensions.put(ep.actionId, ep);
    }
    ActionManager.getInstance().addAnActionListener(this);
  }

  @Override
  public void dispose() {
    ActionManager.getInstance().removeAnActionListener(this);
  }

  @Override
  public void beforeActionPerformed(AnAction action, DataContext dataContext, AnActionEvent event) {
    final InputEvent input = event.getInputEvent();
    if (input instanceof MouseEvent) {
      final String id = ActionManager.getInstance().getId(action);
      final ShortcutPromoterEP ep = myExtensions.get(id);
      if (ep != null) {
        PromoterState state = myState.get(id);
        if (state == null) {
          state = new PromoterState();
          myState.put(id, state);
        }
        state.incClicks();
      }
    }
  }

  @Override
  public void afterActionPerformed(AnAction action, DataContext dataContext, AnActionEvent event) {
  }

  @Override
  public void beforeEditorTyping(char c, DataContext dataContext) {
  }

  @Nullable
  @Override
  public Element getState() {
    final Element actions = new Element("actions");
    for (String id : myState.keySet()) {
      final Element action = new Element("action");
      action.setAttribute("actionId", id);
      action.setAttribute("clicks", String.valueOf(myState.get(id).getClicks()));
      actions.addContent(action);
    }
    return actions;
  }

  @Override
  public void loadState(Element state) {
    myState.clear();
    for (Element action : state.getChildren("action")) {
      try {
        PromoterState info = new PromoterState();
        info.setClicks(StringUtil.parseInt(action.getAttributeValue("clicks"), 0));
        myState.put(action.getAttributeValue("actionId"), info);
      }
      catch (Exception ignore) {
      }
    }
  }
}