ProxyAI/src/main/java/ee/carlrobert/codegpt/actions/TrackableAction.java

package ee.carlrobert.codegpt.actions;

import com.intellij.openapi.actionSystem.AnAction;
import com.intellij.openapi.actionSystem.AnActionEvent;
import com.intellij.openapi.editor.Editor;
import ee.carlrobert.codegpt.telemetry.TelemetryAction;
import javax.swing.Icon;
import org.jetbrains.annotations.NotNull;

public abstract class TrackableAction extends AnAction {

  private final ActionType actionType;
  protected final Editor editor;

  public TrackableAction(
      @NotNull Editor editor,
      String text,
      String description,
      Icon icon,
      ActionType actionType) {
    super(text, description, icon);
    this.editor = editor;
    this.actionType = actionType;
  }

  public abstract void handleAction(@NotNull AnActionEvent e);

  @Override
  public void actionPerformed(@NotNull AnActionEvent e) {
    try {
      handleAction(e);
    } catch (Exception ex) {
      TelemetryAction.IDE_ACTION_ERROR
          .createActionMessage()
          .error(ex)
          .send();
      throw ex;
    } finally {
      TelemetryAction.IDE_ACTION
          .createActionMessage()
          .property("group", null)
          .property("action", actionType.name())
          .send();
    }
  }
}