AddRemoveTableComposite.java

/*******************************************************************************
 * Copyright (c) 2009 Red Hat, Inc.
 * Distributed under license by Red Hat, Inc. All rights reserved.
 * This program is 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
 *
 * Contributor:
 *     Red Hat, Inc. - initial API and implementation
 ******************************************************************************/

package org.hibernate.eclipse.jdt.ui.wizards;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.IPackageFragment;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jdt.core.search.IJavaSearchScope;
import org.eclipse.jdt.core.search.SearchEngine;
import org.eclipse.jdt.internal.core.JavaModelManager;
import org.eclipse.jdt.internal.ui.dialogs.PackageSelectionDialog;
import org.eclipse.jdt.ui.IJavaElementSearchConstants;
import org.eclipse.jdt.ui.JavaElementLabelProvider;
import org.eclipse.jdt.ui.JavaUI;
import org.eclipse.jface.dialogs.IDialogConstants;
import org.eclipse.jface.viewers.ArrayContentProvider;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.widgets.Button;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Event;
import org.eclipse.swt.widgets.Listener;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.swt.widgets.Table;
import org.eclipse.swt.widgets.TableColumn;
import org.eclipse.swt.widgets.TableItem;
import org.eclipse.swt.widgets.Text;
import org.eclipse.swt.widgets.Widget;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.dialogs.SelectionDialog;
import org.hibernate.eclipse.console.HibernateConsoleMessages;
import org.hibernate.eclipse.console.wizards.UpDownListComposite;
import org.hibernate.eclipse.jdt.ui.internal.JdtUiMessages;

/**
 * @author Dmitry Geraskov
 *
 */
@SuppressWarnings("restriction")
public class AddRemoveTableComposite extends UpDownListComposite {

	protected Button depthControl;
	protected Text depthNumbers;
	
	public AddRemoveTableComposite(Composite parent, int style) {
		super(parent, style, "", //$NON-NLS-1$
				false, new JavaElementLabelProvider(),
				ArrayContentProvider.getInstance());
		setUpDownVisible(false);
	}
	
	@Override
	protected String[] getAddButtonLabels() {
		return new String[] { JdtUiMessages.AddRemoveTableComposite_add_class,
							  JdtUiMessages.AddRemoveTableComposite_add_package};
	}

	@Override
	protected void createAdditionalOptions(Composite parent) {
		GridData gridData = new org.eclipse.swt.layout.GridData();
		gridData.horizontalAlignment = org.eclipse.swt.layout.GridData.FILL;
		gridData.grabExcessHorizontalSpace = false;
		gridData.grabExcessVerticalSpace = false;
		GridData gridData2 = new org.eclipse.swt.layout.GridData();
		gridData2.horizontalAlignment = org.eclipse.swt.layout.GridData.FILL;
		gridData2.grabExcessHorizontalSpace = false;
		gridData2.grabExcessVerticalSpace = false;
		depthControl = new Button(parent, SWT.CHECK);
		//depthControl.setText(HibernateConsoleMessages.AddRemoveTableComposite_no_dependencies);
		depthControl.setText(HibernateConsoleMessages.AddRemoveTableComposite_control_depth_level);
		depthControl.setLayoutData(gridData);
		depthNumbers = new Text(parent, SWT.SINGLE | SWT.BORDER | SWT.TRAIL);
		depthNumbers.setText("0"); //$NON-NLS-1$
		depthNumbers.setLayoutData(gridData2);
		depthNumbers.setEnabled(false);
		depthNumbers.addListener(SWT.Verify, new Listener() {
			public void handleEvent(Event e) {
				String string = e.text;
				char [] chars = new char[string.length()];
				string.getChars (0, chars.length, chars, 0);
				for (int i = 0; i < chars.length; i++) {
					if (!('0' <= chars [i] && chars [i] <= '9')) {
						e.doit = false;
						return;
					}
				}
			}
		});
		depthControl.addSelectionListener(new SelectionAdapter() {
			public void widgetSelected(SelectionEvent e) {
				depthNumbers.setEnabled(depthControl.getSelection());
			}
		});
	}

	public int getProcessDepth() {
		String num = depthNumbers.getText();
		return depthControl.getSelection() ? Integer.parseInt(num) : Integer.MAX_VALUE;
	}
	
	protected void createColumns(Table table) {
		TableColumn column = new TableColumn(table, SWT.NULL);
		column.setWidth(350);
	}

	public void add(Object[] o, boolean notify) {
		ArrayList<Object> arr = new ArrayList<Object>();
		ArrayList<Object> arrSelItems = new ArrayList<Object>();
		for (int i = 0; i < o.length; i++) {
			Widget w = getTableViewer().testFindItem(o[i]);
			if (w == null) {
				arr.add(o[i]);
			} else {
				arrSelItems.add(w);
			}
		}
		super.add(arr.toArray(), notify);
		for (int i = 0; i < arr.size(); i++) {
			Widget w = getTableViewer().testFindItem(arr.get(i));
			if (w != null) {
				arrSelItems.add(w);
			}
		}
		getTableViewer().getTable().setSelection(arrSelItems.toArray(new TableItem[0]));
		getTableViewer().setSelection(getTableViewer().getSelection(), false);
	}
	
	protected Object[] handleAdd(int i) {
		
			SelectionDialog dialog= null;
			try {
				IJavaProject[] projects = JavaModelManager.getJavaModelManager().getJavaModel().getJavaProjects();
				
				int includeMask = IJavaSearchScope.SOURCES | IJavaSearchScope.REFERENCED_PROJECTS;
				IJavaSearchScope scope = SearchEngine.createJavaSearchScope(projects, includeMask);
				
				if (i == 0){
					dialog=
					JavaUI.createTypeDialog(
						getShell(),
						PlatformUI.getWorkbench().getProgressService(),
						scope,
						IJavaElementSearchConstants.CONSIDER_CLASSES_AND_INTERFACES,
						true);
					dialog.setTitle(JdtUiMessages.AddRemoveTableComposite_java_types_title); 
					dialog.setMessage(JdtUiMessages.AddRemoveTableComposite_java_select_types);
				} else if (i == 1){
					dialog = new JavaPackageSelectionDialog(getShell(), scope);
					dialog.setTitle(JdtUiMessages.AddRemoveTableComposite_java_packages_title); 
					dialog.setMessage(JdtUiMessages.AddRemoveTableComposite_java_select_packages);
				} else {
					return null;
				}					

				if (dialog.open() == IDialogConstants.CANCEL_ID)
					return null;

				return dialog.getResult();
			} catch (JavaModelException jme) {
				return null;
			}
	}

}

@SuppressWarnings("restriction")
class JavaPackageSelectionDialog extends PackageSelectionDialog {
	
	public JavaPackageSelectionDialog(Shell parent, IJavaSearchScope scope) {
		super(parent,
				PlatformUI.getWorkbench().getProgressService(),
				PackageSelectionDialog.F_HIDE_EMPTY_INNER,
				scope);
		setIgnoreCase(false);
		setMultipleSelection(true);
	}
	
	@Override
	public void setElements(Object[] elements) {
		List<IPackageFragment> javaPackages = new ArrayList<IPackageFragment>();
		for (Object element : elements) {
			if (element instanceof IPackageFragment) {
				IPackageFragment pkg = (IPackageFragment) element;
				try {
					if (pkg.containsJavaResources()) javaPackages.add(pkg);
				} catch (JavaModelException e) {
					e.printStackTrace();
				}				
			}
		}
		super.setElements(javaPackages.toArray());
	}
	
}