viewlabelprovider.java

J2EE eclipse 下开发数据库一个插件

/*
 * Created on 2003-5-16
 *
 */
package com.tanghan.plugin.dbviews.elements;

import org.eclipse.swt.graphics.Image;

import org.eclipse.jface.viewers.LabelProvider;
import org.eclipse.ui.ISharedImages;
import org.eclipse.ui.PlatformUI;

import com.tanghan.db.Field;
import com.tanghan.db.util.DBConnection;
import com.tanghan.db.util.Driver;
import com.tanghan.plugin.IPublicImages;
import com.tanghan.plugin.TanghanPlugin;

/**用来为视图提供内容
 * @author Jerry Tang
 * @version v0.1.0
 * @copyright  (C) 2003 Tanghan工作组
 *  */
public class ViewLabelProvider extends LabelProvider {

	private static IPublicImages pubImages =
		TanghanPlugin.getDefault().getPublicImages();

	public String getText(Object obj) {
		return obj.toString();
	}
	public Image getImage(Object obj) {
		Image dfImage =
			PlatformUI.getWorkbench().getSharedImages().getImage(
				ISharedImages.IMG_OBJ_ELEMENT);
		if (obj instanceof TreeObject) {
			TreeObject toObj = (TreeObject) obj;
			switch (toObj.getType()) {
				case TreeObject.DATABASE_NOTE_ITEM :
					dfImage = pubImages.getImage(IPublicImages.IMG_DB_DATABASE);
					break;
				case TreeObject.CONNECTION_ITEM :
					if (toObj.getObject() != null
						&& toObj.getObject() instanceof DBConnection) {
						DBConnection dbobj = (DBConnection) toObj.getObject();
						if (dbobj.isConnected()) {
							dfImage =
								pubImages.getImage(
									IPublicImages.IMG_DB_CONNECTION);
						} else {
							dfImage =
								pubImages.getImage(
									IPublicImages.IMG_DB_DISCONNECTION);
						}
					} else {
						dfImage =
							pubImages.getImage(IPublicImages.IMG_DB_CONNECTION);
					}
					break;
				case TreeObject.TABLE_NOTE_ITEM :
					dfImage = pubImages.getImage(IPublicImages.IMG_DB_TABLES);
					break;
				case TreeObject.TABLE_ITEM :
					dfImage = pubImages.getImage(IPublicImages.IMG_DB_TABLE);
					break;
				case TreeObject.VIEW_NOTE_ITEM :
					dfImage = pubImages.getImage(IPublicImages.IMG_DB_VIEWS);
					break;
				case TreeObject.VIEW_ITEM :
					dfImage = pubImages.getImage(IPublicImages.IMG_DB_VIEW);
					break;
				case TreeObject.DRIVERS_NOTE_ITEM :
					dfImage =
						pubImages.getImage(IPublicImages.IMG_DRIVER_DRIVERS);
					break;
				case TreeObject.DRIVER_NOTE_ITEM :
					if (toObj.getObject() != null
						&& toObj.getObject() instanceof Driver) {
						Driver driver = (Driver) toObj.getObject();
						if (driver.isActive()) {
							dfImage =
								pubImages.getImage(
									IPublicImages.IMG_DRIVER_DRIVER_ACTIVE);
						} else {
							dfImage =
								pubImages.getImage(
									IPublicImages.IMG_DRIVER_DRIVER_INACTIVE);
						}
					} else {				
						dfImage = pubImages.getImage(IPublicImages.IMG_DRIVER_DRIVER_INACTIVE);
					}
					break;
				case TreeObject.TABLEFIELD_ITEM:
					if (toObj.getObject() != null
						&& toObj.getObject() instanceof Field) {

						Field field = (Field) toObj.getObject();
						if (field.isPrimaryKey()) {
							dfImage =
								pubImages.getImage(
									IPublicImages.IMG_DB_COLUMN_PRIMARYKEY);
						}else if(field.isForeignKey())  {
							dfImage =
								pubImages.getImage(
									IPublicImages.IMG_DB_COLUMN_FOREIGNKEY);
						}else {
							dfImage =
								pubImages.getImage(
									IPublicImages.IMG_DB_FIELD);
						}
					} else {				
						dfImage = pubImages.getImage(IPublicImages.IMG_DB_FIELD);
					}					
					break;
				default :
					break;
			}
		}
		return dfImage;
	}
}