org.eclipse.jdt.ui/ui refactoring/org/eclipse/jdt/internal/ui/refactoring/UseSupertypeInputPage.java

/*******************************************************************************
 * Copyright (c) 2000, 2003 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials 
 * are made available under the terms of the Common Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/cpl-v10.html
 * 
 * Contributors:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.jdt.internal.ui.refactoring;

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

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.layout.GridLayout;
import org.eclipse.swt.widgets.Button;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Label;

import org.eclipse.jface.dialogs.Dialog;
import org.eclipse.jface.dialogs.DialogPage;
import org.eclipse.jface.viewers.ArrayContentProvider;
import org.eclipse.jface.viewers.ISelectionChangedListener;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.SelectionChangedEvent;
import org.eclipse.jface.viewers.StructuredSelection;
import org.eclipse.jface.viewers.TableViewer;
import org.eclipse.jface.wizard.IWizardPage;

import org.eclipse.jdt.core.IType;
import org.eclipse.jdt.core.JavaModelException;

import org.eclipse.jdt.ui.JavaElementLabelProvider;

import org.eclipse.jdt.internal.ui.JavaPlugin;
import org.eclipse.jdt.internal.ui.util.ExceptionHandler;

import org.eclipse.jdt.internal.corext.refactoring.base.IChange;
import org.eclipse.jdt.internal.corext.refactoring.base.ICompositeChange;
import org.eclipse.jdt.internal.corext.refactoring.structure.UseSupertypeWherePossibleRefactoring;

class UseSupertypeInputPage extends UserInputWizardPage{

	public static final String PAGE_NAME= "UseSupertypeInputPage";//$NON-NLS-1$
	private TableViewer fTableViewer; 
	private final Map fFileCount;  //IType -> Integer
	private final static String MESSAGE= RefactoringMessages.getString("UseSupertypeInputPage.Select_supertype"); //$NON-NLS-1$
	private JavaElementLabelProvider fTableLabelProvider;

	public UseSupertypeInputPage() {
		super(PAGE_NAME, true);
		fFileCount= new HashMap(2);
		setMessage(MESSAGE);
	}

	public void createControl(Composite parent) {
		Composite composite= new Composite(parent, SWT.NONE);
		setControl(composite);
		composite.setLayout(new GridLayout());
		
		final Button checkbox= new Button(composite, SWT.CHECK);
		checkbox.setText(RefactoringMessages.getString("UseSupertypeInputPage.Use_in_instanceof")); //$NON-NLS-1$
		checkbox.setLayoutData(new GridData());
		checkbox.setSelection(getUseSupertypeRefactoring().getUseSupertypeInInstanceOf());
		checkbox.addSelectionListener(new SelectionAdapter(){
			public void widgetSelected(SelectionEvent e) {
				getUseSupertypeRefactoring().setUseSupertypeInInstanceOf(checkbox.getSelection());
				setMessage(MESSAGE);
				setPageComplete(true);
				fFileCount.clear();
				fTableViewer.refresh();
			}
		});
		
		Label label= new Label(composite, SWT.NONE);
		label.setText(RefactoringMessages.getString("UseSupertypeInputPage.Select_supertype_to_use")); //$NON-NLS-1$
		label.setLayoutData(new GridData());
		
		addTableComponent(composite);
		Dialog.applyDialogFont(composite);
	}

	private void addTableComponent(Composite composite) {
		fTableViewer= new TableViewer(composite, SWT.SINGLE | SWT.BORDER | SWT.V_SCROLL | SWT.H_SCROLL);
		fTableViewer.getTable().setLayoutData(new GridData(GridData.FILL_BOTH));
		fTableLabelProvider= new UseSupertypeLabelProvider(fFileCount);
		fTableViewer.setLabelProvider(fTableLabelProvider);
		fTableViewer.setContentProvider(new ArrayContentProvider());
		fTableViewer.addSelectionChangedListener(new ISelectionChangedListener(){
			public void selectionChanged(SelectionChangedEvent event) {
				IStructuredSelection ss= (IStructuredSelection)event.getSelection();
				if (new Integer(0).equals(fFileCount.get(ss.getFirstElement()))){
					setMessage(RefactoringMessages.getString("UseSupertypeInputPage.No_updates"), DialogPage.INFORMATION); //$NON-NLS-1$
					setPageComplete(false);
				} else {
					setMessage(MESSAGE);
					setPageComplete(true);
				}
				fTableViewer.refresh();
			}
		});
		try {
			fTableViewer.setInput(getUseSupertypeRefactoring().getSuperTypes());
		} catch (JavaModelException e) {
			ExceptionHandler.handle(e, RefactoringMessages.getString("UseSupertypeInputPage.Use_Supertype"), RefactoringMessages.getString("UseSupertypeInputPage.Internal_Error")); //$NON-NLS-1$ //$NON-NLS-2$
			fTableViewer.setInput(new IType[0]);
		}
		fTableViewer.getTable().setSelection(0);
	}
	
	private UseSupertypeWherePossibleRefactoring getUseSupertypeRefactoring(){
		return (UseSupertypeWherePossibleRefactoring)getRefactoring();
	}
	/*
	 * @see org.eclipse.jface.wizard.IWizardPage#getNextPage()
	 */
	public IWizardPage getNextPage() {
		try {
			initializeRefactoring();
			IWizardPage nextPage= super.getNextPage();
			IStructuredSelection ss= (IStructuredSelection)fTableViewer.getSelection();
			IType selectedType= (IType)ss.getFirstElement();
			if (nextPage == this){
				setMessage(RefactoringMessages.getString("UseSupertypeInputPage.No_updates"), DialogPage.INFORMATION); //$NON-NLS-1$
				setPageComplete(false);
				fFileCount.put(selectedType, new Integer(0));
			} else if (nextPage instanceof IPreviewWizardPage){
				IChange change= getRefactoringWizard().getChange();
				if (change instanceof ICompositeChange){
					ICompositeChange cc= (ICompositeChange)change;
					fFileCount.put(selectedType, new Integer(cc.getChildren().length));
				}
			}
			fTableViewer.refresh();
			return nextPage;
		} catch (JavaModelException e) {
			JavaPlugin.log(e);
			return null;
		}
	}

	/*
	 * @see org.eclipse.jdt.internal.ui.refactoring.RefactoringWizardPage#performFinish()
	 */
	public boolean performFinish(){
		try {
			initializeRefactoring();
			return super.performFinish();
		} catch (JavaModelException e) {
			JavaPlugin.log(e);
			return false;
		}
	}

	private void initializeRefactoring() throws JavaModelException {
		StructuredSelection ss= (StructuredSelection)fTableViewer.getSelection();
		getUseSupertypeRefactoring().setSuperTypeToUse((IType)ss.getFirstElement());
	}
	
	/*
	 * @see org.eclipse.jface.dialogs.IDialogPage#dispose()
	 */
	public void dispose() {
		fTableViewer= null;
		fFileCount.clear();
		fTableLabelProvider= null;
		super.dispose();
	}
	
	private static class UseSupertypeLabelProvider extends JavaElementLabelProvider{
		private final Map fFileCount;
		private UseSupertypeLabelProvider(Map fileCount){
			fFileCount= fileCount;
		}
		public String getText(Object element) {
			String superText= super.getText(element);
			if  (! fFileCount.containsKey(element))
				return superText;
			int count= ((Integer)fFileCount.get(element)).intValue();
			if (count == 0){
				String[] keys= {superText};
				return RefactoringMessages.getFormattedString("UseSupertypeInputPage.no_possible_updates", keys); //$NON-NLS-1$
			} else if (count == 1){
				String [] keys= {superText};
				return RefactoringMessages.getFormattedString("UseSupertypeInputPage.updates_possible_in_file", keys); //$NON-NLS-1$
			}	else {
				String[] keys= {superText, String.valueOf(count)};
				return RefactoringMessages.getFormattedString("UseSupertypeInputPage.updates_possible_in_files", keys); //$NON-NLS-1$
			}	
		}
	}
	/* (non-Javadoc)
	 * @see org.eclipse.jface.dialogs.IDialogPage#setVisible(boolean)
	 */
	public void setVisible(boolean visible) {
		super.setVisible(visible);
		if (visible && fTableViewer != null)
			fTableViewer.getTable().setFocus();
	}

}