From 61300d807349c23b6635575531fafba3cffa2e8d Mon Sep 17 00:00:00 2001
From: Konstantin Bulenkov <kb@jetbrains.com>
Date: Mon, 20 Nov 2017 23:16:01 +0100
Subject: [PATCH] Intentions collector (Quick fixes + Intentions)

---
 .../actions/IntentionsCollector.java          | 110 ++++++++++++++++++
 .../src/META-INF/PlatformExtensions.xml       |   2 +
 2 files changed, 112 insertions(+)
 create mode 100644 platform/platform-impl/src/com/intellij/internal/statistic/customUsageCollectors/actions/IntentionsCollector.java

diff --git a/platform/platform-impl/src/com/intellij/internal/statistic/customUsageCollectors/actions/IntentionsCollector.java b/platform/platform-impl/src/com/intellij/internal/statistic/customUsageCollectors/actions/IntentionsCollector.java
new file mode 100644
index 000000000000..fb6aebcd8ff8
--- /dev/null
+++ b/platform/platform-impl/src/com/intellij/internal/statistic/customUsageCollectors/actions/IntentionsCollector.java
@@ -0,0 +1,110 @@
+// Copyright 2000-2017 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
+package com.intellij.internal.statistic.customUsageCollectors.actions;
+
+import com.intellij.codeInsight.intention.IntentionAction;
+import com.intellij.codeInsight.intention.IntentionActionDelegate;
+import com.intellij.codeInspection.LocalQuickFix;
+import com.intellij.codeInspection.ex.QuickFixWrapper;
+import com.intellij.internal.statistic.UsagesCollector;
+import com.intellij.internal.statistic.beans.GroupDescriptor;
+import com.intellij.internal.statistic.beans.UsageDescriptor;
+import com.intellij.lang.Language;
+import com.intellij.openapi.components.*;
+import com.intellij.openapi.util.text.StringUtil;
+import com.intellij.util.containers.ContainerUtil;
+import com.intellij.util.xmlb.annotations.MapAnnotation;
+import com.intellij.util.xmlb.annotations.Tag;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.*;
+
+/**
+ * @author Konstantin Bulenkov
+ */
+@State(
+  name = "IntentionsCollector",
+  storages = @Storage(value = "statistics.intentions.xml", roamingType = RoamingType.DISABLED)
+)
+public class IntentionsCollector implements PersistentStateComponent<IntentionsCollector.State> {
+  private State myState = new State();
+
+  @Nullable
+  @Override
+  public State getState() {
+    return myState;
+  }
+
+  @Override
+  public void loadState(State state) {
+    myState = state;
+  }
+
+  private static final List<String> PREFIXES_TO_STRIP = Arrays.asList("com.intellij.codeInsight.",
+                                                                      "com.intellij.");
+  public void record(@NotNull IntentionAction action, @NotNull Language language) {
+    State state = getState();
+    if (state == null) return;
+
+    String id = getIntentionId(action);
+
+    String key = language.getID() + ": " + id;
+    final Integer count = state.myIntentions.get(key);
+    int value = count == null ? 1 : count + 1;
+    state.myIntentions.put(key, value);
+  }
+
+  @NotNull
+  private static String getIntentionId(@NotNull IntentionAction action) {
+    Object handler = action;
+    if (action instanceof IntentionActionDelegate) {
+      IntentionAction delegate = ((IntentionActionDelegate)action).getDelegate();
+      if (delegate != action) {
+        return getIntentionId(delegate);
+      }
+    } else if (action instanceof QuickFixWrapper) {
+      LocalQuickFix fix = ((QuickFixWrapper)action).getFix();
+      if (fix != action) {
+        handler = fix;
+      }
+    }
+
+    String fqn = handler.getClass().getName();
+    for (String prefix : PREFIXES_TO_STRIP) {
+      fqn = StringUtil.trimStart(fqn, prefix);
+    }
+
+    if (UsagesCollector.isNotBundledPluginClass(handler.getClass())) {
+      fqn = "[!]" + fqn;
+    }
+
+    return fqn;
+  }
+
+  public static IntentionsCollector getInstance() {
+    return ServiceManager.getService(IntentionsCollector.class);
+  }
+
+  final static class State {
+    @Tag("Intentions")
+    @MapAnnotation(surroundWithTag = false, keyAttributeName = "name", valueAttributeName = "count")
+    public Map<String, Integer> myIntentions = new HashMap<>();
+  }
+
+  final static class IntentionUsagesCollector extends UsagesCollector {
+    private static final GroupDescriptor GROUP = GroupDescriptor.create("Intentions");
+
+    @NotNull
+    public Set<UsageDescriptor> getUsages() {
+      State state = getInstance().getState();
+      assert state != null;
+      return ContainerUtil.map2Set(state.myIntentions.entrySet(), e -> new UsageDescriptor(e.getKey(), e.getValue()));
+    }
+
+    @NotNull
+    public GroupDescriptor getGroupId() {
+      return GROUP;
+    }
+  }
+}
+
diff --git a/platform/platform-resources/src/META-INF/PlatformExtensions.xml b/platform/platform-resources/src/META-INF/PlatformExtensions.xml
index d0ded38ff6bc..3ac9d337338c 100644
--- a/platform/platform-resources/src/META-INF/PlatformExtensions.xml
+++ b/platform/platform-resources/src/META-INF/PlatformExtensions.xml
@@ -207,6 +207,7 @@
     <applicationService serviceImplementation="com.intellij.internal.statistic.customUsageCollectors.ui.ShortcutsCollector"/>
     <applicationService serviceImplementation="com.intellij.internal.statistic.customUsageCollectors.actions.ActionsCollector"/>
     <applicationService serviceImplementation="com.intellij.internal.statistic.customUsageCollectors.actions.MainMenuCollector"/>
+    <applicationService serviceImplementation="com.intellij.internal.statistic.customUsageCollectors.actions.IntentionsCollector"/>
 
     <applicationService serviceImplementation="com.intellij.openapi.util.DimensionService"/>
     <applicationService serviceInterface="com.intellij.openapi.util.WindowStateService"
@@ -389,6 +390,7 @@
     <statistics.usagesCollector implementation="com.intellij.internal.statistic.customUsageCollectors.ui.ToolbarClicksCollector$ToolbarClicksUsagesCollector"/>
     <statistics.usagesCollector implementation="com.intellij.internal.statistic.customUsageCollectors.ui.ShortcutsCollector$ShortcutUsagesCollector"/>
     <statistics.usagesCollector implementation="com.intellij.internal.statistic.customUsageCollectors.actions.ActionsCollector$ActionUsagesCollector"/>
+    <statistics.usagesCollector implementation="com.intellij.internal.statistic.customUsageCollectors.actions.IntentionsCollector$IntentionUsagesCollector"/>
     <statistics.usagesCollector implementation="com.intellij.internal.statistic.customUsageCollectors.ExperimentalFeaturesUsageCollector"/>
     <statistics.usagesCollector implementation="com.intellij.internal.statistic.customUsageCollectors.actions.MainMenuCollector$MainMenuUsagesCollector"/>
     <statistics.usagesCollector implementation="com.intellij.internal.statistic.customUsageCollectors.ui.LaFUsagesCollector"/>