org.apache.click.eclipse/src/org/apache/click/eclipse/ui/editor/forms/ClickTreeLabelProvider.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.click.eclipse.ui.editor.forms;

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;


import org.apache.click.eclipse.ClickPlugin;
import org.eclipse.swt.graphics.Image;
import org.eclipse.wst.xml.core.internal.provisional.document.IDOMElement;
import org.eclipse.wst.xml.ui.internal.contentoutline.JFaceNodeLabelProvider;
import org.w3c.dom.Attr;
import org.w3c.dom.Element;
import org.w3c.dom.NamedNodeMap;

/**
 *
 * @author Naoki Takezoe
 */
public class ClickTreeLabelProvider extends JFaceNodeLabelProvider {

	private Map<String, Image> images = new HashMap<String, Image>();

	public ClickTreeLabelProvider(){
		images.put(ClickPlugin.TAG_CLICK_APP,
				ClickPlugin.getImageDescriptor("icons/click.gif").createImage());
		images.put(ClickPlugin.TAG_HEADER,
				ClickPlugin.getImageDescriptor("icons/header.gif").createImage());
		images.put(ClickPlugin.TAG_HEADERS,
				ClickPlugin.getImageDescriptor("icons/folder.gif").createImage());
		images.put(ClickPlugin.TAG_PAGE,
				ClickPlugin.getImageDescriptor("icons/page.gif").createImage());
		images.put(ClickPlugin.TAG_PAGES,
				ClickPlugin.getImageDescriptor("icons/folder.gif").createImage());
		images.put(ClickPlugin.TAG_CONTROL,
				ClickPlugin.getImageDescriptor("icons/control.gif").createImage());
		images.put(ClickPlugin.TAG_CONTROL_SET,
				ClickPlugin.getImageDescriptor("icons/control.gif").createImage());
		images.put(ClickPlugin.TAG_CONTROLS,
				ClickPlugin.getImageDescriptor("icons/folder.gif").createImage());
		images.put(ClickPlugin.TAG_EXCLUDES,
				ClickPlugin.getImageDescriptor("icons/page.gif").createImage());
		images.put(ClickPlugin.TAG_PROPERTY,
				ClickPlugin.getImageDescriptor("icons/property.gif").createImage());
		images.put(ClickPlugin.TAG_FILE_UPLOAD_SERVICE,
				ClickPlugin.getImageDescriptor("icons/class.gif").createImage());
		images.put(ClickPlugin.TAG_LOG_SERVICE,
				ClickPlugin.getImageDescriptor("icons/class.gif").createImage());
		images.put(ClickPlugin.TAG_TEMPLATE_SERVICE,
				ClickPlugin.getImageDescriptor("icons/class.gif").createImage());
		images.put(ClickPlugin.TAG_RESOURCE_SERVICE,
				ClickPlugin.getImageDescriptor("icons/class.gif").createImage());
		images.put(ClickPlugin.TAG_PAGE_INTERCEPTOR,
				ClickPlugin.getImageDescriptor("icons/class.gif").createImage());
	}

	public void dispose(){
		for(Iterator<Image> ite = images.values().iterator(); ite.hasNext();){
			Image image = ite.next();
			image.dispose();
		}
		super.dispose();
	}

	public String getText(Object object) {
		if(object instanceof Element){
			Element element = (Element)object;
			StringBuffer sb = new StringBuffer();
			sb.append(element.getNodeName());
			sb.append(" ");
			NamedNodeMap attrs = element.getAttributes();
			for(int i=0;i<attrs.getLength();i++){
				if(i!=0){
					sb.append(", ");
				}
				Attr attr = (Attr)attrs.item(i);
				sb.append(attr.getName());
				sb.append("=");
				sb.append(attr.getValue());
			}
			return sb.toString();
		}
		return super.getText(object);
	}

	public Image getImage(Object object) {
		if(object instanceof IDOMElement){
			String name = ((IDOMElement)object).getNodeName();
			Image image = (Image)images.get(name);
			if(image != null){
				return image;
			}
		}
		return super.getImage(object);
	}
}