NavigationTreeLabelDecorator.java

/*******************************************************************************
 * Copyright (c) 2006-2011 Gluster, Inc. <http://www.gluster.com>
 * This file is part of Gluster Management Console.
 *
 * Gluster Management Console is free software; you can redistribute
 * it and/or modify it under the terms of the GNU General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * Gluster Management Console is distributed in the hope that it
 * will be useful, but WITHOUT ANY WARRANTY; without even the
 * implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
 * PURPOSE.  See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see
 * <http://www.gnu.org/licenses/>.
 *******************************************************************************/
package org.gluster.storage.management.console.views;

import org.eclipse.jface.viewers.IDecoration;
import org.eclipse.jface.viewers.ILabelProviderListener;
import org.eclipse.jface.viewers.ILightweightLabelDecorator;
import org.eclipse.ui.plugin.AbstractUIPlugin;
import org.gluster.storage.management.console.Application;
import org.gluster.storage.management.console.IImageKeys;
import org.gluster.storage.management.core.model.EntityGroup;
import org.gluster.storage.management.core.model.GlusterServer;
import org.gluster.storage.management.core.model.Server;
import org.gluster.storage.management.core.model.Volume;


public class NavigationTreeLabelDecorator implements ILightweightLabelDecorator {

	@Override
	public void addListener(ILabelProviderListener listener) {
	}

	@Override
	public void dispose() {
	}

	@Override
	public boolean isLabelProperty(Object element, String property) {
		return false;
	}

	@Override
	public void removeListener(ILabelProviderListener listener) {
	}

	@SuppressWarnings("rawtypes")
	@Override
	public void decorate(Object element, IDecoration decoration) {
		if (element instanceof Volume) {
			Volume volume = (Volume) element;
			if (volume.getStatus() == Volume.VOLUME_STATUS.OFFLINE) {
				decoration.addOverlay(AbstractUIPlugin.imageDescriptorFromPlugin(Application.PLUGIN_ID,
						IImageKeys.OVERLAY_OFFLINE_8x8));
			} else {
				decoration.addOverlay(AbstractUIPlugin.imageDescriptorFromPlugin(Application.PLUGIN_ID,
						IImageKeys.OVERLAY_ONLINE_8x8));
			}
		}

		if (element instanceof GlusterServer) {
			GlusterServer server = (GlusterServer) element;
			if (server.getStatus() == GlusterServer.SERVER_STATUS.OFFLINE) {
				decoration.addOverlay(AbstractUIPlugin.imageDescriptorFromPlugin(Application.PLUGIN_ID,
						IImageKeys.OVERLAY_OFFLINE_8x8));
			} else {
				decoration.addOverlay(AbstractUIPlugin.imageDescriptorFromPlugin(Application.PLUGIN_ID,
						IImageKeys.OVERLAY_ONLINE_8x8));
			}
		}

		if (element instanceof Server) {
			decoration.addOverlay(AbstractUIPlugin.imageDescriptorFromPlugin(Application.PLUGIN_ID,
					IImageKeys.OVERLAY_STAR_8x8));
		}
		
		if(element instanceof EntityGroup && ((EntityGroup)element).getEntityType() == Server.class) {
			decoration.addOverlay(AbstractUIPlugin.imageDescriptorFromPlugin(Application.PLUGIN_ID,
					IImageKeys.OVERLAY_STAR_8x8));
		}
	}
}