EditorTracker.java

/*******************************************************************************
 * Copyright (c) 2006, 2016 Mountainminds GmbH & Co. KG and Contributors
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Marc R. Hoffmann - initial API and implementation
 *    
 ******************************************************************************/
package com.mountainminds.eclemma.internal.ui.annotation;

import org.eclipse.ui.IEditorReference;
import org.eclipse.ui.IPartListener2;
import org.eclipse.ui.IWindowListener;
import org.eclipse.ui.IWorkbench;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.IWorkbenchPart;
import org.eclipse.ui.IWorkbenchPartReference;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.texteditor.ITextEditor;

/**
 * Tracks the workbench editors and to attach coverage annotation models.
 */
public class EditorTracker {

  private final IWorkbench workbench;

  private IWindowListener windowListener = new IWindowListener() {
    public void windowOpened(IWorkbenchWindow window) {
      window.getPartService().addPartListener(partListener);
    }

    public void windowClosed(IWorkbenchWindow window) {
      window.getPartService().removePartListener(partListener);
    }

    public void windowActivated(IWorkbenchWindow window) {
    }

    public void windowDeactivated(IWorkbenchWindow window) {
    }
  };

  private IPartListener2 partListener = new IPartListener2() {
    public void partOpened(IWorkbenchPartReference partref) {
      annotateEditor(partref);
    }

    public void partActivated(IWorkbenchPartReference partref) {
    }

    public void partBroughtToTop(IWorkbenchPartReference partref) {
    }

    public void partVisible(IWorkbenchPartReference partref) {
    }

    public void partInputChanged(IWorkbenchPartReference partref) {
    }

    public void partClosed(IWorkbenchPartReference partref) {
    }

    public void partDeactivated(IWorkbenchPartReference partref) {
    }

    public void partHidden(IWorkbenchPartReference partref) {
    }
  };

  public EditorTracker(IWorkbench workbench) {
    this.workbench = workbench;
    for (final IWorkbenchWindow w : workbench.getWorkbenchWindows()) {
      w.getPartService().addPartListener(partListener);
    }
    workbench.addWindowListener(windowListener);
    annotateAllEditors();
  }

  public void dispose() {
    workbench.removeWindowListener(windowListener);
    for (final IWorkbenchWindow w : workbench.getWorkbenchWindows()) {
      w.getPartService().removePartListener(partListener);
    }
  }

  private void annotateAllEditors() {
    for (final IWorkbenchWindow w : workbench.getWorkbenchWindows()) {
      for (final IWorkbenchPage p : w.getPages()) {
        for (final IEditorReference e : p.getEditorReferences()) {
          annotateEditor(e);
        }
      }
    }
  }

  private void annotateEditor(IWorkbenchPartReference partref) {
    IWorkbenchPart part = partref.getPart(false);
    if (part instanceof ITextEditor) {
      CoverageAnnotationModel.attach((ITextEditor) part);
    }
  }

}