1 files changed, 162 insertions, 0 deletions
diff --git a/bundles/org.eclipse.equinox.p2.ui/src/org/eclipse/equinox/internal/p2/ui/viewers/ProvElementContentProvider.java b/bundles/org.eclipse.equinox.p2.ui/src/org/eclipse/equinox/internal/p2/ui/viewers/ProvElementContentProvider.java
new file mode 100644
index 000000000..5e0c4bd9b
--- /dev/null
+++ b/bundles/org.eclipse.equinox.p2.ui/src/org/eclipse/equinox/internal/p2/ui/viewers/ProvElementContentProvider.java
@@ -0,0 +1,162 @@
+/*******************************************************************************
+ *  Copyright (c) 2008, 2009 IBM Corporation 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:
+ *     IBM Corporation - initial API and implementation
+ *******************************************************************************/
+
+package org.eclipse.equinox.internal.p2.ui.viewers;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import org.eclipse.core.runtime.*;
+import org.eclipse.core.runtime.jobs.Job;
+import org.eclipse.equinox.internal.p2.ui.ProvUIMessages;
+import org.eclipse.equinox.internal.p2.ui.model.ProvElement;
+import org.eclipse.jface.viewers.*;
+import org.eclipse.swt.widgets.Display;
+import org.eclipse.ui.progress.*;
+
+/**
+ * Content provider that retrieves children of a ProvElement.
+ * 
+ * @since 3.5
+ * 
+ */
+public class ProvElementContentProvider implements ITreeContentProvider {
+
+	private Viewer viewer;
+	private Job fetchJob;
+	// family is used by test cases
+	Object fetchFamily = new Object();
+
+	/**
+	 * 
+	 */
+	public ProvElementContentProvider() {
+		// Default constructor
+	}
+
+	/*
+	 * (non-Javadoc)
+	 * @see org.eclipse.jface.viewers.IStructuredContentProvider#getElements(java.lang.Object)
+	 */
+	public Object[] getElements(final Object input) {
+		// Simple deferred fetch handling for table viewers
+		if (input instanceof IDeferredWorkbenchAdapter && viewer instanceof AbstractTableViewer) {
+			final Display display = viewer.getControl().getDisplay();
+			final Object pending = new PendingUpdateAdapter();
+			if (fetchJob != null)
+				fetchJob.cancel();
+			fetchJob = new Job(ProvUIMessages.ProvElementContentProvider_FetchJobTitle) {
+				protected IStatus run(IProgressMonitor monitor) {
+					IDeferredWorkbenchAdapter parent = (IDeferredWorkbenchAdapter) input;
+					final ArrayList<Object> children = new ArrayList<Object>();
+					parent.fetchDeferredChildren(parent, new IElementCollector() {
+						public void add(Object element, IProgressMonitor mon) {
+							if (mon.isCanceled())
+								return;
+							children.add(element);
+						}
+
+						public void add(Object[] elements, IProgressMonitor mon) {
+							if (mon.isCanceled())
+								return;
+							children.addAll(Arrays.asList(elements));
+						}
+
+						public void done() {
+							// nothing special to do
+						}
+
+					}, monitor);
+					display.asyncExec(new Runnable() {
+						public void run() {
+							AbstractTableViewer tableViewer = (AbstractTableViewer) viewer;
+							if (viewer == null || viewer.getControl().isDisposed())
+								return;
+							tableViewer.getControl().setRedraw(false);
+							tableViewer.remove(pending);
+							tableViewer.add(children.toArray());
+							finishedFetchingElements(input);
+							tableViewer.getControl().setRedraw(true);
+						}
+					});
+					return Status.OK_STATUS;
+				}
+
+				public boolean belongsTo(Object family) {
+					return family == fetchFamily;
+				}
+
+			};
+			fetchJob.schedule();
+			return new Object[] {pending};
+		}
+		Object[] elements = getChildren(input);
+		finishedFetchingElements(input);
+		return elements;
+	}
+
+	/*
+	 * (non-Javadoc)
+	 * @see org.eclipse.jface.viewers.ITreeContentProvider#getParent(java.lang.Object)
+	 */
+	public Object getParent(Object child) {
+		if (child instanceof ProvElement) {
+			return ((ProvElement) child).getParent(child);
+		}
+		return null;
+	}
+
+	/*
+	 * (non-Javadoc)
+	 * @see org.eclipse.jface.viewers.ITreeContentProvider#hasChildren(java.lang.Object)
+	 */
+	public boolean hasChildren(Object element) {
+		if (element instanceof ProvElement)
+			return ((ProvElement) element).hasChildren(element);
+		return false;
+	}
+
+	/*
+	 * (non-Javadoc)
+	 * @see org.eclipse.jface.viewers.ITreeContentProvider#getChildren(java.lang.Object)
+	 */
+	public Object[] getChildren(final Object parent) {
+		if (parent instanceof ProvElement) {
+			return ((ProvElement) parent).getChildren(parent);
+		}
+		return new Object[0];
+	}
+
+	/* (non-Javadoc)
+	 * @see org.eclipse.jface.viewers.IContentProvider#dispose()
+	 */
+	public void dispose() {
+		viewer = null;
+		if (fetchJob != null) {
+			fetchJob.cancel();
+			fetchJob = null;
+		}
+	}
+
+	/* (non-Javadoc)
+	 * @see org.eclipse.jface.viewers.IContentProvider#inputChanged(org.eclipse.jface.viewers.Viewer, java.lang.Object, java.lang.Object)
+	 */
+	public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
+		this.viewer = viewer;
+		if (fetchJob != null) {
+			fetchJob.cancel();
+			fetchJob = null;
+		}
+	}
+
+	protected void finishedFetchingElements(Object parent) {
+		// do nothing
+	}
+}