ManagerContentProvider.java

/*******************************************************************************
 * Copyright (c) 2013 hangum.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser Public License v2.1
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * 
 * Contributors:
 *     hangum - initial API and implementation
 ******************************************************************************/
package com.hangum.tadpole.rdb.core.viewers.connections;

import org.apache.log4j.Logger;
import org.eclipse.jface.viewers.ArrayContentProvider;
import org.eclipse.jface.viewers.ITreeContentProvider;

import com.hangum.tadpole.engine.query.dao.ManagerListDTO;
import com.hangum.tadpole.engine.query.dao.system.UserDBDAO;
import com.hangum.tadpole.engine.query.dao.system.userdb.ResourcesDAO;

/**
 * manager view의 content provider
 * @author hangum
 *
 */
public class ManagerContentProvider extends ArrayContentProvider implements ITreeContentProvider {
	private static final Logger logger = Logger.getLogger(ManagerContentProvider.class);
	@Override
	public Object[] getChildren(Object parentElement) {
		
		if(parentElement instanceof ManagerListDTO) {
			ManagerListDTO dao = (ManagerListDTO)parentElement;
			return dao.getManagerList().toArray();
		} else if(parentElement instanceof UserDBDAO) {
			UserDBDAO dao = (UserDBDAO)parentElement;
			return dao.getListResource().toArray();
		} else if(parentElement instanceof ResourcesDAO) {
			ResourcesDAO dao = (ResourcesDAO)parentElement;
			return dao.getListResource().toArray();
		}
		
		return null;
	}

	@Override
	public Object getParent(Object element) {
		
//		if(element instanceof UserDBDAO) {
//			UserDBDAO dto = (UserDBDAO)element;
//			return dto.getParent();
//		} else if(element instanceof UserDBResourceDAO) {
//			UserDBResourceDAO dao = (UserDBResourceDAO)element;
//			return dao.getParent();
//		}
		
		return null;
	}

	@Override
	public boolean hasChildren(Object element) {
		if(element instanceof ManagerListDTO) {
			ManagerListDTO dto = (ManagerListDTO)element;
			return dto.getManagerList().size() > 0;
		} else if(element instanceof UserDBDAO) {
			UserDBDAO dto = (UserDBDAO)element;
			if(dto.getListResource() == null) return false;
			else return dto.getListResource().size() > 0;
		} else if(element instanceof ResourcesDAO) {
			ResourcesDAO dto = (ResourcesDAO)element;
			return dto.getListResource().size() > 0;
		}

		return false;
	}
	
}