package com.vectrace.MercurialEclipse.ui;

import java.util.Set;

import org.eclipse.core.resources.IContainer;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.ui.model.BaseWorkbenchContentProvider;

import com.vectrace.MercurialEclipse.MercurialEclipsePlugin;

public class ResourcesTreeContentProvider extends BaseWorkbenchContentProvider {

	//the pseudo-root
	public static final Object ROOT = new Object();
	//the real roots
	private final Set<? extends IResource> roots;

	public ResourcesTreeContentProvider(Set<? extends IResource> roots) {
		super();
		this.roots = roots;
	}

	@Override
	public Object[] getChildren(Object parentElement) {
		if(parentElement == ROOT) {
			return roots.toArray(new IResource[0]);
		} else if(parentElement instanceof IContainer){
			try {
				return  ((IContainer) parentElement).members();
			} catch (CoreException e) {
				MercurialEclipsePlugin.logWarning(Messages.getString("ResourcesTreeContentProvider.failedToGetChildrenOf")+parentElement, e); //$NON-NLS-1$
				return null;
			}
		} else {
			return null;
		}
	}

	@Override
	public Object getParent(Object element) {
		return ((IResource) element).getParent();
	}

	@Override
	public boolean hasChildren(Object element) {
		return element == ROOT || element instanceof IContainer;
	}

	@Override
	public Object[] getElements(Object inputElement) {
		return getChildren(inputElement);
	}

	@Override
	public void dispose() {
	}

	@Override
	public void inputChanged(Viewer viewer, Object oldInput,
			Object newInput) {
	}

}

/*******************************************************************************
 * Copyright (c) 2005-2008 VecTrace (Zingo Andersen) 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:
 *     Andrei Loskutov         - bug fixes
 *******************************************************************************/
package com.vectrace.MercurialEclipse.ui;

import java.util.Set;

import org.eclipse.core.resources.IContainer;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.ui.model.BaseWorkbenchContentProvider;

import com.vectrace.MercurialEclipse.MercurialEclipsePlugin;

public class ResourcesTreeContentProvider extends BaseWorkbenchContentProvider {

	//the pseudo-root
	public static final Object ROOT = new Object();
	//the real roots
	private final Set<? extends IResource> roots;

	public ResourcesTreeContentProvider(Set<? extends IResource> roots) {
		super();
		this.roots = roots;
	}

	@Override
	public Object[] getChildren(Object parentElement) {
		if(parentElement == ROOT) {
			return roots.toArray(new IResource[0]);
		} else if(parentElement instanceof IContainer){
			try {
				return  ((IContainer) parentElement).members();
			} catch (CoreException e) {
				MercurialEclipsePlugin.logWarning(Messages.getString("ResourcesTreeContentProvider.failedToGetChildrenOf")+parentElement, e); //$NON-NLS-1$
				return new Object[0];
			}
		} else {
			return new Object[0];
		}
	}

	@Override
	public Object getParent(Object element) {
		return ((IResource) element).getParent();
	}

	@Override
	public boolean hasChildren(Object element) {
		return element == ROOT || element instanceof IContainer;
	}

	@Override
	public Object[] getElements(Object inputElement) {
		return getChildren(inputElement);
	}

	@Override
	public void dispose() {
	}

	@Override
	public void inputChanged(Viewer viewer, Object oldInput,
			Object newInput) {
	}

}

Return empty arrays instead of null's