org.eclipse.mylyn.context.core/src/org/eclipse/mylyn/internal/core/search/MylarWorkingSetUpdater.java

/*******************************************************************************
 * Copyright (c) 2004 - 2006 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
 *******************************************************************************/

package org.eclipse.mylar.internal.core.search;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.mylar.provisional.core.IMylarContext;
import org.eclipse.mylar.provisional.core.IMylarContextListener;
import org.eclipse.mylar.provisional.core.IMylarElement;
import org.eclipse.mylar.provisional.core.IMylarStructureBridge;
import org.eclipse.mylar.provisional.core.MylarPlugin;
import org.eclipse.swt.widgets.Display;
import org.eclipse.ui.IWorkingSet;
import org.eclipse.ui.IWorkingSetUpdater;

/**
 * @author Shawn Minto
 */
public class MylarWorkingSetUpdater implements IWorkingSetUpdater, IMylarContextListener {

	/** Should only ever have 1 working set */
	private List<IWorkingSet> workingSets = new ArrayList<IWorkingSet>();

	public MylarWorkingSetUpdater() {
		MylarPlugin.getDefault().addWorkingSetUpdater(this);
	}

	public void add(IWorkingSet workingSet) {
		workingSets.add(workingSet);
	}

	public boolean remove(IWorkingSet workingSet) {
		return workingSets.remove(workingSet);

	}

	public boolean contains(IWorkingSet workingSet) {
		return workingSets.contains(workingSet);
	}

	public void dispose() {
		// nothing to do here
	}

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

	public void contextDeactivated(IMylarContext taskscape) {
		updateWorkingSet();
	}

	public void presentationSettingsChanging(UpdateKind kind) {
		// don't care about this event

	}

	public void presentationSettingsChanged(UpdateKind kind) {
		// don't care about this event

	}

	public void interestChanged(List<IMylarElement> nodes) {
		updateWorkingSet();

	}

	public void nodeDeleted(IMylarElement node) {
		updateWorkingSet();
	}

	public void landmarkAdded(IMylarElement node) {
		updateWorkingSet();

	}

	public void landmarkRemoved(IMylarElement node) {
		updateWorkingSet();

	}

	public void edgesChanged(IMylarElement node) {
		// don't care about this relationship

	}

	private void updateWorkingSet() {
		Display.getDefault().asyncExec(new Runnable() {
			public void run() {
				if (workingSets.size() <= 0)
					return;
				IWorkingSet set = workingSets.get(0);
				set.setElements(new IAdaptable[] {});
				List<IAdaptable> elements = new ArrayList<IAdaptable>();
				getElementsFromTaskscape(elements);
				set.setElements(elements.toArray(new IAdaptable[elements.size()]));
			}
		});
	}

	public static void getElementsFromTaskscape(List<IAdaptable> elements) {
		// IMylarContext t = MylarPlugin.getContextManager().getActiveContext();
		for (IMylarElement node : MylarPlugin.getContextManager().getInterestingDocuments()) {
			IMylarStructureBridge bridge = MylarPlugin.getDefault().getStructureBridge(node.getContentType());

			// HACK comparing extension to string
			// No need to add bugzilla resources to the taskscape
			// search...really slow and eclipese doesn't know about them
			if (bridge.getContentType().equals("bugzilla"))
				continue;

			Object o = bridge.getObjectForHandle(node.getHandleIdentifier());
			if (o instanceof IAdaptable) {
				elements.add((IAdaptable) o);
			}

		}
	}

	public IWorkingSet getWorkingSet() {
		return workingSets.get(0);
	}
}