update/org.eclipse.update.ui/src/org/eclipse/update/internal/ui/forms/RevertSection.java

package org.eclipse.update.internal.ui.forms;

import java.lang.reflect.InvocationTargetException;

import org.eclipse.core.runtime.*;
import org.eclipse.jface.dialogs.ErrorDialog;
import org.eclipse.jface.dialogs.ProgressMonitorDialog;
import org.eclipse.jface.operation.IRunnableWithProgress;
import org.eclipse.jface.resource.JFaceResources;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.widgets.*;
import org.eclipse.update.configuration.IInstallConfiguration;
import org.eclipse.update.configuration.ILocalSite;
import org.eclipse.update.core.SiteManager;
import org.eclipse.update.internal.ui.UpdateUIPlugin;
import org.eclipse.update.internal.ui.pages.UpdateFormPage;
import org.eclipse.update.ui.forms.internal.*;

public class RevertSection extends UpdateSection {
	// NL keys
	private static final String KEY_TITLE =
		"InstallConfigurationPage.RevertSection.title";
	private static final String KEY_DESC =
		"InstallConfigurationPage.RevertSection.desc";
	private static final String KEY_CURRENT_TEXT =
		"InstallConfigurationPage.RevertSection.currentText";
	private static final String KEY_REVERT_TEXT =
		"InstallConfigurationPage.RevertSection.revertText";
	private static final String KEY_REVERT_BUTTON =
		"InstallConfigurationPage.RevertSection.revertButton";
	private static final String KEY_RESTORE_TEXT =
		"InstallConfigurationPage.RevertSection.restoreText";
	private static final String KEY_RESTORE_BUTTON =
		"InstallConfigurationPage.RevertSection.restoreButton";
	private static final String KEY_DIALOG_TITLE =
		"InstallConfigurationPage.RevertSection.dialog.title";
	private static final String KEY_DIALOG_MESSAGE =
		"InstallConfigurationPage.RevertSection.dialog.message";

	private Composite container;
	private FormWidgetFactory factory;
	private IInstallConfiguration config;
	private Label currentTextLabel;
	private Label textLabel;
	private Button revertButton;

	public RevertSection(UpdateFormPage page) {
		super(page);
		setAddSeparator(false);
		setHeaderText(UpdateUIPlugin.getResourceString(KEY_TITLE));
		setDescription(UpdateUIPlugin.getResourceString(KEY_DESC));
	}

	public Composite createClient(
		Composite parent,
		FormWidgetFactory factory) {
		HTMLTableLayout layout = new HTMLTableLayout();
		this.factory = factory;
		//header.setForeground(factory.getColor(factory.COLOR_COMPOSITE_SEPARATOR));
		updateHeaderColor();
		layout.leftMargin = layout.rightMargin = 0;
		layout.horizontalSpacing = 0;
		container = factory.createComposite(parent);
		container.setLayout(layout);
		layout.numColumns = 3;
		currentTextLabel = factory.createLabel(container, "");
		currentTextLabel.setFont(JFaceResources.getBannerFont());
		TableData td = new TableData();
		td.valign = TableData.MIDDLE;
		currentTextLabel.setLayoutData(td);
		textLabel = factory.createLabel(container, "", SWT.WRAP);
		td = new TableData();
		td.valign = TableData.MIDDLE;
		textLabel.setLayoutData(td);
		revertButton = factory.createButton(container, "", SWT.PUSH);
		revertButton.addSelectionListener(new SelectionAdapter() {
			public void widgetSelected(SelectionEvent e) {
				performRevert();
			}
		});
		td = new TableData();
		td.indent = 5;
		td.valign = TableData.MIDDLE;
		revertButton.setLayoutData(td);
		return container;
	}

	public void configurationChanged(IInstallConfiguration config) {
		this.config = config;
		boolean canRevert = false;
		try {
			ILocalSite localSite = SiteManager.getLocalSite();
			IInstallConfiguration[] history =
				localSite.getConfigurationHistory();
			int length = history.length;
			canRevert = length > 1;
		} catch (CoreException e) {
			UpdateUIPlugin.logException(e);
		}
		container.getParent().setVisible(canRevert);
		if (!canRevert)
			return;
		if (config.isCurrent()) {
			currentTextLabel.setText(
				UpdateUIPlugin.getResourceString(KEY_CURRENT_TEXT));
			textLabel.setText(
				UpdateUIPlugin.getResourceString(KEY_REVERT_TEXT));
			revertButton.setText(
				UpdateUIPlugin.getResourceString(KEY_REVERT_BUTTON));
		} else {
			currentTextLabel.setText("");
			textLabel.setText(
				UpdateUIPlugin.getResourceString(KEY_RESTORE_TEXT));
			revertButton.setText(
				UpdateUIPlugin.getResourceString(KEY_RESTORE_BUTTON));
		}
	}

	private void performRevert() {
		try {
			ILocalSite localSite = SiteManager.getLocalSite();
			IInstallConfiguration target = config;
			if (config.isCurrent()) {
				// take the previous one
				IInstallConfiguration[] history =
					localSite.getConfigurationHistory();
				target = history[history.length - 2];
			}
			performRevert(target);
		} catch (CoreException e) {
			UpdateUIPlugin.logException(e);
		}
	}

	public static void performRevert(final IInstallConfiguration target) {
		// make sure we can actually do the revert
		IStatus status = ActivityConstraints.validatePendingRevert(target);
		if (status != null) {
			ErrorDialog.openError(
				UpdateUIPlugin.getActiveWorkbenchShell(),
				null,
				null,
				status);
			return;
		}

		// ok to perform the operation
		IRunnableWithProgress operation = new IRunnableWithProgress() {

			public void run(IProgressMonitor monitor) {
				boolean success = false;
				try {
					ILocalSite localSite = SiteManager.getLocalSite();
					localSite.revertTo(target, monitor, new UIProblemHandler());
					saveLocalSite();
					success = true;
				} catch (CoreException e) {
					UpdateUIPlugin.logException(e);
				} finally {
					monitor.done();
					if (success)
						UpdateUIPlugin.informRestartNeeded();
				}
			}
		};
		try {
			ProgressMonitorDialog dialog =
				new ProgressMonitorDialog(
					UpdateUIPlugin.getActiveWorkbenchShell().getShell());
			dialog.run(false, true, operation);
		} catch (InvocationTargetException e) {
			UpdateUIPlugin.logException(e);
		} catch (InterruptedException e) {
		}
	}

	private static void saveLocalSite() throws CoreException {
		ILocalSite localSite = SiteManager.getLocalSite();
		localSite.save();
	}
}