bundles/org.eclipse.equinox.p2.ui/src/org/eclipse/equinox/internal/p2/ui/dialogs/RevertProfileWizardPage.java - equinox/rt.equinox.p2 - Git at Google

 /*******************************************************************************
  * Copyright (c) 2007, 2008 IBM Corporation and others.
  * All rights reserved. This program and the accompanying materials
  * are 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
  *
  * Contributors:
  *     IBM Corporation - initial API and implementation
  *******************************************************************************/
 package org.eclipse.equinox.internal.p2.ui.dialogs;

 import java.lang.reflect.InvocationTargetException;
 import org.eclipse.core.runtime.IProgressMonitor;
 import org.eclipse.core.runtime.IStatus;
 import org.eclipse.equinox.internal.p2.ui.ProvUIMessages;
 import org.eclipse.equinox.internal.p2.ui.viewers.IUDetailsLabelProvider;
 import org.eclipse.equinox.internal.provisional.p2.core.ProvisionException;
 import org.eclipse.equinox.internal.provisional.p2.director.ProvisioningPlan;
 import org.eclipse.equinox.internal.provisional.p2.engine.IProfile;
 import org.eclipse.equinox.internal.provisional.p2.metadata.IInstallableUnit;
 import org.eclipse.equinox.internal.provisional.p2.ui.ProvUI;
 import org.eclipse.equinox.internal.provisional.p2.ui.ProvisioningOperationRunner;
 import org.eclipse.equinox.internal.provisional.p2.ui.model.RollbackProfileElement;
 import org.eclipse.equinox.internal.provisional.p2.ui.model.RollbackRepositoryElement;
 import org.eclipse.equinox.internal.provisional.p2.ui.operations.*;
 import org.eclipse.equinox.internal.provisional.p2.ui.policy.IQueryProvider;
 import org.eclipse.equinox.internal.provisional.p2.ui.viewers.*;
 import org.eclipse.jface.dialogs.*;
 import org.eclipse.jface.dialogs.Dialog;
 import org.eclipse.jface.operation.IRunnableWithProgress;
 import org.eclipse.jface.viewers.*;
 import org.eclipse.jface.wizard.WizardPage;
 import org.eclipse.swt.SWT;
 import org.eclipse.swt.custom.BusyIndicator;
 import org.eclipse.swt.custom.SashForm;
 import org.eclipse.swt.layout.GridData;
 import org.eclipse.swt.layout.GridLayout;
 import org.eclipse.swt.widgets.*;
 import org.eclipse.ui.PlatformUI;
 import org.eclipse.ui.statushandlers.StatusManager;

 /**
  * @since 3.4
  */
 public class RevertProfileWizardPage extends WizardPage {

 	private TableViewer configsViewer;
 	private TableViewer configContentsViewer;
 	String profileId;
 	IQueryProvider queryProvider;
 	private static final int DEFAULT_COLUMN_WIDTH = 150;

 	public RevertProfileWizardPage(String profileId, IQueryProvider queryProvider) {
 		super("RevertConfiguration"); //$NON-NLS-1$
 		setTitle(ProvUIMessages.RevertDialog_PageTitle);
 		setDescription(ProvUIMessages.RevertDialog_Description);
 		this.profileId = profileId;
 		this.queryProvider = queryProvider;

 	}

 	public void createControl(Composite parent) {
 		SashForm sashForm = new SashForm(parent, SWT.HORIZONTAL);
 		sashForm.setLayout(new GridLayout());
 		sashForm.setLayoutData(new GridData(GridData.FILL_BOTH));
 		initializeDialogUnits(sashForm);

 		createConfigurationsSection(sashForm);
 		createContentsSection(sashForm);
 		setControl(sashForm);

 		// prime the selection
 		Object element = configsViewer.getElementAt(0);
 		if (element != null)
 			configsViewer.setSelection(new StructuredSelection(element));
 		Dialog.applyDialogFont(sashForm);
 	}

 	private void createConfigurationsSection(Composite parent) {
 		Composite composite = new Composite(parent, SWT.NONE);
 		GridLayout layout = new GridLayout();
 		layout.marginWidth = 0;
 		layout.marginHeight = 0;
 		composite.setLayout(layout);
 		GridData gd = new GridData(GridData.FILL_BOTH);
 		composite.setLayoutData(gd);

 		Label label = new Label(composite, SWT.NONE);
 		label.setText(ProvUIMessages.RevertDialog_ConfigsLabel);
 		configsViewer = new TableViewer(composite, SWT.FULL_SELECTION | SWT.H_SCROLL | SWT.V_SCROLL | SWT.BORDER);
 		configsViewer.setContentProvider(new RepositoryContentProvider(queryProvider));
 		configsViewer.setLabelProvider(new ProvElementLabelProvider());
 		configsViewer.setComparator(new ViewerComparator() {
 			// We override the ViewerComparator so that we don't get the labels of the elements
 			// for comparison, but rather get the version numbers and compare them.
 			// Reverse sorting is used so that newest is first.
 			public int compare(Viewer viewer, Object o1, Object o2) {
 				IInstallableUnit iu1 = (IInstallableUnit) ProvUI.getAdapter(o1, IInstallableUnit.class);
 				IInstallableUnit iu2 = (IInstallableUnit) ProvUI.getAdapter(o2, IInstallableUnit.class);
 				if (iu1 == null || iu2 == null)
 					// this is naive (doesn't consult the label provider), but shouldn't happen
 					return o2.toString().compareTo(o1.toString());
 				return iu2.getVersion().compareTo(iu1.getVersion());
 			}
 		});
 		configsViewer.setInput(getInput());

 		configsViewer.addSelectionChangedListener(new ISelectionChangedListener() {
 			public void selectionChanged(SelectionChangedEvent event) {
 				handleSelectionChanged((IStructuredSelection) event.getSelection());
 			}

 		});
 		gd = new GridData(GridData.FILL_BOTH);
 		configsViewer.getControl().setLayoutData(gd);
 	}

 	private void createContentsSection(Composite parent) {
 		Composite composite = new Composite(parent, SWT.NONE);
 		GridLayout layout = new GridLayout();
 		layout.marginWidth = 0;
 		layout.marginHeight = 0;
 		composite.setLayout(layout);
 		GridData gd = new GridData(GridData.FILL_BOTH);
 		composite.setLayoutData(gd);

 		Label label = new Label(composite, SWT.NONE);
 		label.setText(ProvUIMessages.RevertDialog_ConfigContentsLabel);
 		configContentsViewer = new TableViewer(composite, SWT.SINGLE | SWT.FULL_SELECTION | SWT.H_SCROLL | SWT.V_SCROLL | SWT.BORDER);
 		configContentsViewer.setComparator(new IUComparator(IUComparator.IU_NAME));
 		configContentsViewer.setComparer(new ProvElementComparer());
 		configContentsViewer.setContentProvider(new DeferredQueryContentProvider(queryProvider));

 		// columns before labels or you get a blank table
 		setTableColumns(configContentsViewer.getTable());
 		configContentsViewer.setLabelProvider(new IUDetailsLabelProvider());

 		gd = new GridData(GridData.FILL_BOTH);
 		configContentsViewer.getControl().setLayoutData(gd);

 	}

 	private Object getInput() {
 		try {
 			RollbackRepositoryElement element = new RollbackRepositoryElement(ProvisioningUtil.getRollbackRepositoryURL(), profileId);
 			element.setQueryProvider(queryProvider);
 			return element;
 		} catch (ProvisionException e) {
 			ProvUI.handleException(e, ProvUIMessages.RevertProfileWizardPage_ErrorRetrievingHistory, StatusManager.BLOCK | StatusManager.LOG);
 			return null;
 		}
 	}

 	void handleSelectionChanged(IStructuredSelection selection) {
 		if (!selection.isEmpty()) {
 			final Object selected = selection.getFirstElement();
 			if (selected instanceof RollbackProfileElement) {
 				final IProfile[] snapshot = new IProfile[1];
 				BusyIndicator.showWhile(getShell().getDisplay(), new Runnable() {
 					public void run() {
 						try {
 							snapshot[0] = ((RollbackProfileElement) selected).getProfileSnapshot(null);
 						} catch (ProvisionException e) {
 							ProvUI.handleException(e, ProvUIMessages.RollbackProfileElement_InvalidSnapshot, StatusManager.LOG | StatusManager.SHOW);
 						}
 					}
 				});
 				if (snapshot[0] != null) {
 					configContentsViewer.setInput(selected);
 					setPageComplete(true);
 					return;
 				}
 			}
 		}
 		configContentsViewer.setInput(null);
 		setPageComplete(false);
 	}

 	private void setTableColumns(Table table) {
 		IUColumnConfig[] columns = ProvUI.getIUColumnConfig();

 		for (int i = 0; i < columns.length; i++) {
 			TableColumn tc = new TableColumn(table, SWT.NONE, i);
 			tc.setResizable(true);
 			tc.setText(columns[i].columnTitle);
 			tc.setWidth(convertHorizontalDLUsToPixels(DEFAULT_COLUMN_WIDTH));
 		}
 	}

 	private IInstallableUnit getSelectedIU() {
 		Object selected = ((IStructuredSelection) configsViewer.getSelection()).getFirstElement();
 		if (selected != null && selected instanceof RollbackProfileElement)
 			return ((RollbackProfileElement) selected).getIU();
 		return null;
 	}

 	public boolean performFinish() {
 		Shell shell = getContainer().getShell();
 		boolean result = MessageDialog.openQuestion(shell, shell.getText(), ProvUIMessages.RevertDialog_ConfirmRestartMessage);
 		if (!result)
 			return false;

 		boolean finish = revert();
 		if (finish) {
 			PlatformUI.getWorkbench().restart();
 		}
 		return finish;
 	}

 	private boolean revert() {
 		final IInstallableUnit iu = getSelectedIU();
 		if (iu == null)
 			return false;
 		final ProvisioningPlan[] plan = new ProvisioningPlan[1];
 		IRunnableWithProgress runnable = new IRunnableWithProgress() {
 			public void run(IProgressMonitor monitor) {
 				try {
 					plan[0] = ProvisioningUtil.getRevertPlan(iu, monitor);
 				} catch (ProvisionException e) {
 					plan[0] = null;
 					ProvUI.handleException(e.getCause(), ProvUIMessages.ProfileModificationWizardPage_UnexpectedError, StatusManager.LOG);
 					setMessage(ProvUIMessages.ProfileModificationWizardPage_UnexpectedError, IMessageProvider.ERROR);
 				}
 			}
 		};
 		try {
 			getContainer().run(true, true, runnable);
 			if (plan[0] != null) {
 				if (plan[0].getStatus().isOK()) {
 					ProvisioningOperation op = new ProfileModificationOperation(ProvUIMessages.RevertDialog_RevertOperationLabel, profileId, plan[0]);
 					ProvisioningOperationRunner.run(op, getShell(), StatusManager.SHOW | StatusManager.LOG);
 					return true;
 				}
 				// If user cancelled, do not report an error
 				if (plan[0].getStatus().getSeverity() == IStatus.CANCEL)
 					return false;
 				ProvUI.reportStatus(plan[0].getStatus(), StatusManager.LOG);
 				setMessage(ProvUIMessages.ProfileModificationWizardPage_UnexpectedError, IMessageProvider.ERROR);
 			}
 		} catch (InterruptedException e) {
 			// don't report thread interruption
 		} catch (InvocationTargetException e) {
 			ProvUI.handleException(e.getCause(), ProvUIMessages.RevertDialog_RevertError, StatusManager.LOG);
 			setMessage(ProvUIMessages.RevertDialog_RevertError, IMessageProvider.ERROR);
 		}
 		return false;
 	}
 }
	/*******************************************************************************
	* Copyright (c) 2007, 2008 IBM Corporation and others.
	* All rights reserved. This program and the accompanying materials
	* are 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
	*
	* Contributors:
	* IBM Corporation - initial API and implementation
	*******************************************************************************/
	package org.eclipse.equinox.internal.p2.ui.dialogs;

	import java.lang.reflect.InvocationTargetException;
	import org.eclipse.core.runtime.IProgressMonitor;
	import org.eclipse.core.runtime.IStatus;
	import org.eclipse.equinox.internal.p2.ui.ProvUIMessages;
	import org.eclipse.equinox.internal.p2.ui.viewers.IUDetailsLabelProvider;
	import org.eclipse.equinox.internal.provisional.p2.core.ProvisionException;
	import org.eclipse.equinox.internal.provisional.p2.director.ProvisioningPlan;
	import org.eclipse.equinox.internal.provisional.p2.engine.IProfile;
	import org.eclipse.equinox.internal.provisional.p2.metadata.IInstallableUnit;
	import org.eclipse.equinox.internal.provisional.p2.ui.ProvUI;
	import org.eclipse.equinox.internal.provisional.p2.ui.ProvisioningOperationRunner;
	import org.eclipse.equinox.internal.provisional.p2.ui.model.RollbackProfileElement;
	import org.eclipse.equinox.internal.provisional.p2.ui.model.RollbackRepositoryElement;
	import org.eclipse.equinox.internal.provisional.p2.ui.operations.*;
	import org.eclipse.equinox.internal.provisional.p2.ui.policy.IQueryProvider;
	import org.eclipse.equinox.internal.provisional.p2.ui.viewers.*;
	import org.eclipse.jface.dialogs.*;
	import org.eclipse.jface.dialogs.Dialog;
	import org.eclipse.jface.operation.IRunnableWithProgress;
	import org.eclipse.jface.viewers.*;
	import org.eclipse.jface.wizard.WizardPage;
	import org.eclipse.swt.SWT;
	import org.eclipse.swt.custom.BusyIndicator;
	import org.eclipse.swt.custom.SashForm;
	import org.eclipse.swt.layout.GridData;
	import org.eclipse.swt.layout.GridLayout;
	import org.eclipse.swt.widgets.*;
	import org.eclipse.ui.PlatformUI;
	import org.eclipse.ui.statushandlers.StatusManager;

	/**
	* @since 3.4
	*/
	public class RevertProfileWizardPage extends WizardPage {

	private TableViewer configsViewer;
	private TableViewer configContentsViewer;
	String profileId;
	IQueryProvider queryProvider;
	private static final int DEFAULT_COLUMN_WIDTH = 150;

	public RevertProfileWizardPage(String profileId, IQueryProvider queryProvider) {
	super("RevertConfiguration"); //$NON-NLS-1$
	setTitle(ProvUIMessages.RevertDialog_PageTitle);
	setDescription(ProvUIMessages.RevertDialog_Description);
	this.profileId = profileId;
	this.queryProvider = queryProvider;

	}

	public void createControl(Composite parent) {
	SashForm sashForm = new SashForm(parent, SWT.HORIZONTAL);
	sashForm.setLayout(new GridLayout());
	sashForm.setLayoutData(new GridData(GridData.FILL_BOTH));
	initializeDialogUnits(sashForm);

	createConfigurationsSection(sashForm);
	createContentsSection(sashForm);
	setControl(sashForm);

	// prime the selection
	Object element = configsViewer.getElementAt(0);
	if (element != null)
	configsViewer.setSelection(new StructuredSelection(element));
	Dialog.applyDialogFont(sashForm);
	}

	private void createConfigurationsSection(Composite parent) {
	Composite composite = new Composite(parent, SWT.NONE);
	GridLayout layout = new GridLayout();
	layout.marginWidth = 0;
	layout.marginHeight = 0;
	composite.setLayout(layout);
	GridData gd = new GridData(GridData.FILL_BOTH);
	composite.setLayoutData(gd);

	Label label = new Label(composite, SWT.NONE);
	label.setText(ProvUIMessages.RevertDialog_ConfigsLabel);
	configsViewer = new TableViewer(composite, SWT.FULL_SELECTION \| SWT.H_SCROLL \| SWT.V_SCROLL \| SWT.BORDER);
	configsViewer.setContentProvider(new RepositoryContentProvider(queryProvider));
	configsViewer.setLabelProvider(new ProvElementLabelProvider());
	configsViewer.setComparator(new ViewerComparator() {
	// We override the ViewerComparator so that we don't get the labels of the elements
	// for comparison, but rather get the version numbers and compare them.
	// Reverse sorting is used so that newest is first.
	public int compare(Viewer viewer, Object o1, Object o2) {
	IInstallableUnit iu1 = (IInstallableUnit) ProvUI.getAdapter(o1, IInstallableUnit.class);
	IInstallableUnit iu2 = (IInstallableUnit) ProvUI.getAdapter(o2, IInstallableUnit.class);
	if (iu1 == null \|\| iu2 == null)
	// this is naive (doesn't consult the label provider), but shouldn't happen
	return o2.toString().compareTo(o1.toString());
	return iu2.getVersion().compareTo(iu1.getVersion());
	}
	});
	configsViewer.setInput(getInput());

	configsViewer.addSelectionChangedListener(new ISelectionChangedListener() {
	public void selectionChanged(SelectionChangedEvent event) {
	handleSelectionChanged((IStructuredSelection) event.getSelection());
	}

	});
	gd = new GridData(GridData.FILL_BOTH);
	configsViewer.getControl().setLayoutData(gd);
	}

	private void createContentsSection(Composite parent) {
	Composite composite = new Composite(parent, SWT.NONE);
	GridLayout layout = new GridLayout();
	layout.marginWidth = 0;
	layout.marginHeight = 0;
	composite.setLayout(layout);
	GridData gd = new GridData(GridData.FILL_BOTH);
	composite.setLayoutData(gd);

	Label label = new Label(composite, SWT.NONE);
	label.setText(ProvUIMessages.RevertDialog_ConfigContentsLabel);
	configContentsViewer = new TableViewer(composite, SWT.SINGLE \| SWT.FULL_SELECTION \| SWT.H_SCROLL \| SWT.V_SCROLL \| SWT.BORDER);
	configContentsViewer.setComparator(new IUComparator(IUComparator.IU_NAME));
	configContentsViewer.setComparer(new ProvElementComparer());
	configContentsViewer.setContentProvider(new DeferredQueryContentProvider(queryProvider));

	// columns before labels or you get a blank table
	setTableColumns(configContentsViewer.getTable());
	configContentsViewer.setLabelProvider(new IUDetailsLabelProvider());

	gd = new GridData(GridData.FILL_BOTH);
	configContentsViewer.getControl().setLayoutData(gd);

	}

	private Object getInput() {
	try {
	RollbackRepositoryElement element = new RollbackRepositoryElement(ProvisioningUtil.getRollbackRepositoryURL(), profileId);
	element.setQueryProvider(queryProvider);
	return element;
	} catch (ProvisionException e) {
	ProvUI.handleException(e, ProvUIMessages.RevertProfileWizardPage_ErrorRetrievingHistory, StatusManager.BLOCK \| StatusManager.LOG);
	return null;
	}
	}

	void handleSelectionChanged(IStructuredSelection selection) {
	if (!selection.isEmpty()) {
	final Object selected = selection.getFirstElement();
	if (selected instanceof RollbackProfileElement) {
	final IProfile[] snapshot = new IProfile[1];
	BusyIndicator.showWhile(getShell().getDisplay(), new Runnable() {
	public void run() {
	try {
	snapshot[0] = ((RollbackProfileElement) selected).getProfileSnapshot(null);
	} catch (ProvisionException e) {
	ProvUI.handleException(e, ProvUIMessages.RollbackProfileElement_InvalidSnapshot, StatusManager.LOG \| StatusManager.SHOW);
	}
	}
	});
	if (snapshot[0] != null) {
	configContentsViewer.setInput(selected);
	setPageComplete(true);
	return;
	}
	}
	}
	configContentsViewer.setInput(null);
	setPageComplete(false);
	}

	private void setTableColumns(Table table) {
	IUColumnConfig[] columns = ProvUI.getIUColumnConfig();

	for (int i = 0; i < columns.length; i++) {
	TableColumn tc = new TableColumn(table, SWT.NONE, i);
	tc.setResizable(true);
	tc.setText(columns[i].columnTitle);
	tc.setWidth(convertHorizontalDLUsToPixels(DEFAULT_COLUMN_WIDTH));
	}
	}

	private IInstallableUnit getSelectedIU() {
	Object selected = ((IStructuredSelection) configsViewer.getSelection()).getFirstElement();
	if (selected != null && selected instanceof RollbackProfileElement)
	return ((RollbackProfileElement) selected).getIU();
	return null;
	}

	public boolean performFinish() {
	Shell shell = getContainer().getShell();
	boolean result = MessageDialog.openQuestion(shell, shell.getText(), ProvUIMessages.RevertDialog_ConfirmRestartMessage);
	if (!result)
	return false;

	boolean finish = revert();
	if (finish) {
	PlatformUI.getWorkbench().restart();
	}
	return finish;
	}

	private boolean revert() {
	final IInstallableUnit iu = getSelectedIU();
	if (iu == null)
	return false;
	final ProvisioningPlan[] plan = new ProvisioningPlan[1];
	IRunnableWithProgress runnable = new IRunnableWithProgress() {
	public void run(IProgressMonitor monitor) {
	try {
	plan[0] = ProvisioningUtil.getRevertPlan(iu, monitor);
	} catch (ProvisionException e) {
	plan[0] = null;
	ProvUI.handleException(e.getCause(), ProvUIMessages.ProfileModificationWizardPage_UnexpectedError, StatusManager.LOG);
	setMessage(ProvUIMessages.ProfileModificationWizardPage_UnexpectedError, IMessageProvider.ERROR);
	}
	}
	};
	try {
	getContainer().run(true, true, runnable);
	if (plan[0] != null) {
	if (plan[0].getStatus().isOK()) {
	ProvisioningOperation op = new ProfileModificationOperation(ProvUIMessages.RevertDialog_RevertOperationLabel, profileId, plan[0]);
	ProvisioningOperationRunner.run(op, getShell(), StatusManager.SHOW \| StatusManager.LOG);
	return true;
	}
	// If user cancelled, do not report an error
	if (plan[0].getStatus().getSeverity() == IStatus.CANCEL)
	return false;
	ProvUI.reportStatus(plan[0].getStatus(), StatusManager.LOG);
	setMessage(ProvUIMessages.ProfileModificationWizardPage_UnexpectedError, IMessageProvider.ERROR);
	}
	} catch (InterruptedException e) {
	// don't report thread interruption
	} catch (InvocationTargetException e) {
	ProvUI.handleException(e.getCause(), ProvUIMessages.RevertDialog_RevertError, StatusManager.LOG);
	setMessage(ProvUIMessages.RevertDialog_RevertError, IMessageProvider.ERROR);
	}
	return false;
	}
	}