org.eclipse.mylyn.java.ui/src/org/eclipse/mylyn/java/ui/LandmarkMarkerManager.java

/*******************************************************************************
 * Copyright (c) 2004 - 2005 University Of British Columbia and others.
 * 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:
 *     University Of British Columbia - initial API and implementation
 *******************************************************************************/
/*
 * Created on Feb 16, 2005
  */
package org.eclipse.mylar.java.ui;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.eclipse.core.resources.IMarker;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.IWorkspaceRunnable;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.jdt.core.IJavaElement;
import org.eclipse.jdt.core.IMember;
import org.eclipse.jdt.core.ISourceRange;
import org.eclipse.jdt.core.ISourceReference;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.mylar.core.IMylarContext;
import org.eclipse.mylar.core.IMylarContextListener;
import org.eclipse.mylar.core.IMylarElement;
import org.eclipse.mylar.core.MylarPlugin;
import org.eclipse.mylar.java.JavaStructureBridge;


/**
 * @author Mik Kersten
 */
public class LandmarkMarkerManager implements IMylarContextListener {

    private Map<IMylarElement, Long> markerMap = new HashMap<IMylarElement, Long>();
    
    public LandmarkMarkerManager() {
        super();
    }

    public void contextActivated(IMylarContext taskscape) {
        modelUpdated();
    }

    public void contextDeactivated(IMylarContext taskscape) {
        modelUpdated();
    }
    
    private void modelUpdated() {
    	try {
	        for (IMylarElement node : markerMap.keySet()) {
	            landmarkRemoved(node);
	        }
	        markerMap.clear();
	        for (IMylarElement node : MylarPlugin.getContextManager().getActiveLandmarks()) {
	            landmarkAdded(node);
	        }
        } catch (Throwable t) {
        	MylarPlugin.fail(t, "Could not update landmark markers", false);
        }
    } 

    public void interestChanged(IMylarElement element) {
    	// don't care when the interest changes
    }
    
    public void interestChanged(List<IMylarElement> nodes) {
    	// don't care when the interest changes
    }

    public void landmarkAdded(final IMylarElement node) {
        if (node == null || node.getContentType() == null) return;
        if (node.getContentType().equals(JavaStructureBridge.CONTENT_TYPE)) {
            final IJavaElement element = JavaCore.create(node.getHandleIdentifier());
            if (!element.exists()) return;
            if (element instanceof IMember) {
                try {
                    final ISourceRange range = ((IMember)element).getNameRange();
                    final IResource resource = element.getUnderlyingResource();
                    if (resource == null) return;
                    IWorkspaceRunnable runnable = new IWorkspaceRunnable() {
                        public void run(IProgressMonitor monitor) throws CoreException {
                            IMarker marker = resource.createMarker("org.eclipse.mylar.ui.landmark");//MylarUiPlugin.MARKER_LANDMARK);
                            if (marker != null) {
	                            marker.setAttribute(IMarker.CHAR_START, range.getOffset());
	                            marker.setAttribute(IMarker.CHAR_END, range.getOffset() + range.getLength());
	                            marker.setAttribute(IMarker.MESSAGE, "Mylar Landmark"); 
	                            marker.setAttribute(IMarker.SEVERITY, IMarker.SEVERITY_INFO); 
	                            markerMap.put(node, marker.getId());
                            }
                        }
                    };
                    if (resource != null) resource.getWorkspace().run(runnable, null);
                } catch (JavaModelException e) {
                    MylarPlugin.fail(e, "couldn't update marker", false);
                }catch (CoreException e) {
                    MylarPlugin.fail(e, "couldn't update marker", false);
                }
            }
        }
    }
    
    public void landmarkRemoved(final IMylarElement node) {
        if (node == null) return;
        if (node.getContentType().equals(JavaStructureBridge.CONTENT_TYPE)) {
            IJavaElement element = JavaCore.create(node.getHandleIdentifier());
            if (!element.exists()) return;
            if (element.getAncestor(IJavaElement.COMPILATION_UNIT) != null  // stuff from .class files
               && element instanceof ISourceReference) {
                try { 
                    final IResource resource = element.getUnderlyingResource();
                    IWorkspaceRunnable runnable = new IWorkspaceRunnable() {
                        public void run(IProgressMonitor monitor) throws CoreException {
                            if (resource != null) {
                                try {
                                	if (markerMap.containsKey(node)) {
                                		long id = markerMap.get(node);	
                                		IMarker marker = resource.getMarker(id); 
                                		if (marker != null) marker.delete();
                                	}
                                } catch (NullPointerException e) {
                                	MylarPlugin.log(e, "could not update markers");
                                }
                            }
                        }
                    };
                    resource.getWorkspace().run(runnable, null);
                } catch (JavaModelException e) {
                    MylarPlugin.fail(e, "couldn't update landmark marker", false);
                } catch (CoreException e) {
                    MylarPlugin.fail(e, "couldn't update landmark marker", false);
                }
            }
        }
    }

    public void edgesChanged(IMylarElement node) {
    	// don't care when the relationships changed
    }

    public void presentationSettingsChanging(UpdateKind kind) {
    	// don't care when there is a presentations setting change
    }

    public void presentationSettingsChanged(UpdateKind kind) {
    	// don't care when there is a presentation setting change
    }

    public void nodeDeleted(IMylarElement node) {
    	// don't care when a node is deleted
    }

}