IdeaDebuggerPositionHighlighter.java

/*
 * Copyright 2003-2016 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 jetbrains.mps.idea.java.debugger;

import com.intellij.debugger.DebuggerManagerEx;
import com.intellij.debugger.SourcePosition;
import com.intellij.debugger.engine.DebugProcess;
import com.intellij.debugger.impl.DebuggerContextImpl;
import com.intellij.debugger.impl.DebuggerContextListener;
import com.intellij.debugger.impl.DebuggerManagerListener;
import com.intellij.debugger.impl.DebuggerSession;
import com.intellij.debugger.impl.DebuggerSession.Event;
import com.intellij.execution.ExecutionManager;
import com.intellij.execution.Executor;
import com.intellij.execution.ui.RunContentDescriptor;
import com.intellij.execution.ui.RunContentManager;
import com.intellij.execution.ui.RunContentWithExecutorListener;
import com.intellij.openapi.components.ProjectComponent;
import com.intellij.openapi.fileEditor.FileEditorManager;
import com.intellij.openapi.project.Project;
import jetbrains.mps.debugger.core.CurrentLinePositionComponentEx;
import jetbrains.mps.idea.java.trace.MpsSourcePosition;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.mps.openapi.model.SNode;
import org.jetbrains.mps.openapi.model.SNodeReference;

import java.util.Collection;

public class IdeaDebuggerPositionHighlighter extends CurrentLinePositionComponentEx<DebuggerSession> implements ProjectComponent {
  private DebuggerManagerEx myDebuggerManager;
  private final DebuggerManagerListener myDebuggerManagerListener = new MyDebuggerManagerListener();
  private final DebuggerContextListener mySessionListener = new MyDebuggerContextListener();
  private final RunContentWithExecutorListener myRunContentListener = new MyRunContentListener();
  private final ExecutionManager myExecutionManager;

  public IdeaDebuggerPositionHighlighter(Project project, FileEditorManager fileEditorManager) {
    super(project, fileEditorManager);
    myExecutionManager = ExecutionManager.getInstance(myProject);
  }

  @Override
  protected SNodeReference getNode(DebuggerSession session) {
    SourcePosition sourcePosition = session.getContextManager().getContext().getSourcePosition();
    if (sourcePosition instanceof MpsSourcePosition) {
      return ((MpsSourcePosition) sourcePosition).getNode();
    }
    return null;
  }

  @Override
  public void initComponent() {
  }

  @Override
  public void disposeComponent() {
  }

  @NotNull
  @Override
  public String getComponentName() {
    return "Idea Debugger Position Highlighter";
  }

  @Override
  public void projectOpened() {
    myDebuggerManager = DebuggerManagerEx.getInstanceEx(myProject);
    myDebuggerManager.addDebuggerManagerListener(myDebuggerManagerListener);
    myProject.getMessageBus().connect(myProject).subscribe(RunContentManager.TOPIC, myRunContentListener);
  }

  @Override
  public void projectClosed() {
    myDebuggerManager.removeDebuggerManagerListener(myDebuggerManagerListener);
  }

  @Override
  protected DebuggerSession getCurrentSession() {
    RunContentDescriptor selectedContent = myExecutionManager.getContentManager().getSelectedContent();
    if (selectedContent == null) return null;
    return myDebuggerManager.getSession(myDebuggerManager.getDebugProcess(selectedContent.getProcessHandler()));
  }

  @Override
  protected Collection<? extends DebuggerSession> getAllSessions() {
    return myDebuggerManager.getSessions();
  }

  private class MyDebuggerManagerListener implements DebuggerManagerListener {
    @Override
    public void sessionCreated(DebuggerSession session) {
      session.getContextManager().addListener(mySessionListener);
    }

    public void sessionAttached(DebuggerSession session) {
    }

    public void sessionDetached(DebuggerSession session) {
    }

    @Override
    public void sessionRemoved(DebuggerSession session) {
      detachPainter(session);
      session.getContextManager().removeListener(mySessionListener);
    }
  }

  private class MyDebuggerContextListener implements DebuggerContextListener {
    @Override
    public void changeEvent(@NotNull DebuggerContextImpl newContext, Event event) {
      if (event != Event.REFRESH_WITH_STACK && event != Event.THREADS_REFRESH) {
        reAttachPainter(newContext.getDebuggerSession(), true);
      }
    }
  }

  private class MyRunContentListener implements RunContentWithExecutorListener {
    @Override
    public void contentSelected(RunContentDescriptor runContentDescriptor, @NotNull Executor executor) {
      if (runContentDescriptor != null) {
        DebugProcess debugProcess = myDebuggerManager.getDebugProcess(runContentDescriptor.getProcessHandler());
        if (debugProcess != null) {
          DebuggerSession debuggerSession = myDebuggerManager.getSession(debugProcess);
          currentSessionChanged(debuggerSession);
        }
      }
    }

    @Override
    public void contentRemoved(RunContentDescriptor runContentDescriptor, @NotNull Executor executor) {

    }
  }
}