1 files changed, 194 insertions, 0 deletions
diff --git a/bundles/org.eclipse.equinox.p2.ui.sdk.scheduler/src/org/eclipse/equinox/internal/p2/ui/sdk/scheduler/migration/MigrationWizard.java b/bundles/org.eclipse.equinox.p2.ui.sdk.scheduler/src/org/eclipse/equinox/internal/p2/ui/sdk/scheduler/migration/MigrationWizard.java
new file mode 100644
index 000000000..5d6af1c6b
--- /dev/null
+++ b/bundles/org.eclipse.equinox.p2.ui.sdk.scheduler/src/org/eclipse/equinox/internal/p2/ui/sdk/scheduler/migration/MigrationWizard.java
@@ -0,0 +1,194 @@
+/*******************************************************************************
+ * Copyright (c) 2011 WindRiver 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:
+ *     WindRiver Corporation - initial API and implementation
+ *     Ericsson AB - Ongoing development
+ *     Ericsson AB (Pascal Rapicault)
+ *     Ericsson AB (Hamdan Msheik) 
+ *******************************************************************************/
+package org.eclipse.equinox.internal.p2.ui.sdk.scheduler.migration;
+
+import java.net.URI;
+import java.util.*;
+import org.eclipse.core.runtime.OperationCanceledException;
+import org.eclipse.core.runtime.Platform;
+import org.eclipse.equinox.internal.p2.core.helpers.ServiceHelper;
+import org.eclipse.equinox.internal.p2.ui.ProvUIActivator;
+import org.eclipse.equinox.internal.p2.ui.dialogs.ISelectableIUsPage;
+import org.eclipse.equinox.internal.p2.ui.dialogs.InstallWizard;
+import org.eclipse.equinox.internal.p2.ui.model.IUElementListRoot;
+import org.eclipse.equinox.internal.p2.ui.sdk.scheduler.AutomaticUpdatePlugin;
+import org.eclipse.equinox.p2.core.IProvisioningAgent;
+import org.eclipse.equinox.p2.core.ProvisionException;
+import org.eclipse.equinox.p2.engine.*;
+import org.eclipse.equinox.p2.metadata.IInstallableUnit;
+import org.eclipse.equinox.p2.operations.InstallOperation;
+import org.eclipse.equinox.p2.repository.IRepositoryManager;
+import org.eclipse.equinox.p2.repository.artifact.IArtifactRepositoryManager;
+import org.eclipse.equinox.p2.repository.metadata.IMetadataRepositoryManager;
+import org.eclipse.equinox.p2.ui.LoadMetadataRepositoryJob;
+import org.eclipse.equinox.p2.ui.ProvisioningUI;
+import org.eclipse.jface.dialogs.*;
+import org.eclipse.jface.resource.ImageDescriptor;
+import org.eclipse.jface.viewers.IStructuredSelection;
+import org.eclipse.jface.wizard.IWizardPage;
+import org.eclipse.ui.IImportWizard;
+import org.eclipse.ui.IWorkbench;
+
+public class MigrationWizard extends InstallWizard implements IImportWizard {
+
+	private IProfile toImportFrom;
+	Collection<IInstallableUnit> unitsToMigrate;
+	private URI[] reposToMigrate;
+	private List<URI> addedRepos = new ArrayList<URI>();
+	private boolean firstTime = false;
+
+	public MigrationWizard() {
+		this(ProvisioningUI.getDefaultUI(), null, null, null);
+	}
+
+	public MigrationWizard(IProfile toImportFrom, Collection<IInstallableUnit> unitsToMigrate, URI[] reposToMigrate, boolean firstTime) {
+		this(ProvisioningUI.getDefaultUI(), null, null, null);
+		this.toImportFrom = toImportFrom;
+		this.unitsToMigrate = unitsToMigrate;
+		this.reposToMigrate = reposToMigrate;
+		this.firstTime = firstTime;
+		addRepos();
+	}
+
+	public MigrationWizard(ProvisioningUI ui, InstallOperation operation, Collection<IInstallableUnit> initialSelections, LoadMetadataRepositoryJob preloadJob) {
+		super(ui, operation, initialSelections, preloadJob);
+		IDialogSettings workbenchSettings = ProvUIActivator.getDefault().getDialogSettings();
+		String sectionName = "MigrationWizard"; //$NON-NLS-1$
+		IDialogSettings section = workbenchSettings.getSection(sectionName);
+		if (section == null) {
+			section = workbenchSettings.addNewSection(sectionName);
+		}
+		setDialogSettings(section);
+	}
+
+	public void init(IWorkbench workbench, IStructuredSelection selection) {
+		setWindowTitle(firstTime ? ProvUIMessages.MigrationWizard_WINDOWTITLE_FIRSTRUN : ProvUIMessages.MigrationWizard_WINDOWTITLE);
+		setDefaultPageImageDescriptor(ImageDescriptor.createFromURL(Platform.getBundle(ProvUIActivator.PLUGIN_ID).getEntry("icons/install_wiz.gif"))); //$NON-NLS-1$
+		setNeedsProgressMonitor(true);
+	}
+
+	@Override
+	protected ISelectableIUsPage createMainPage(IUElementListRoot input, Object[] selections) {
+		if (unitsToMigrate != null)
+			return new MigrationPage(ui, this, toImportFrom, unitsToMigrate, firstTime);
+		return new MigrationPage(ui, this, firstTime);
+	}
+
+	@Override
+	protected ProvisioningContext getProvisioningContext() {
+		return ((MigrationPage) mainPage).getProvisioningContext();
+	}
+
+	@Override
+	public boolean performFinish() {
+		cleanupProfileRegistry();
+		boolean finished = super.performFinish();
+		if (finished) {
+			rememberMigrationCompleted();
+		}
+		return finished;
+	}
+
+	private void addRepos() {
+		IProvisioningAgent agent = (IProvisioningAgent) ServiceHelper.getService(AutomaticUpdatePlugin.getContext(), IProvisioningAgent.SERVICE_NAME);
+		IMetadataRepositoryManager metaManager = (IMetadataRepositoryManager) agent.getService(IMetadataRepositoryManager.SERVICE_NAME);
+		IArtifactRepositoryManager artifactManager = (IArtifactRepositoryManager) agent.getService(IArtifactRepositoryManager.SERVICE_NAME);
+		List<URI> currentMetaRepos = Arrays.asList(metaManager.getKnownRepositories(IRepositoryManager.REPOSITORIES_ALL));
+
+		if (reposToMigrate != null && metaManager != null && artifactManager != null) {
+			for (int i = 0; i < reposToMigrate.length; i++) {
+				if (!currentMetaRepos.contains(reposToMigrate[i])) {
+					metaManager.addRepository(reposToMigrate[i]);
+					artifactManager.addRepository(reposToMigrate[i]);
+					addedRepos.add(reposToMigrate[i]);
+				}
+			}
+		}
+	}
+
+	private void removeRepos() {
+		IProvisioningAgent agent = (IProvisioningAgent) ServiceHelper.getService(AutomaticUpdatePlugin.getContext(), IProvisioningAgent.SERVICE_NAME);
+		IMetadataRepositoryManager metaManager = (IMetadataRepositoryManager) agent.getService(IMetadataRepositoryManager.SERVICE_NAME);
+		IArtifactRepositoryManager artifactManager = (IArtifactRepositoryManager) agent.getService(IArtifactRepositoryManager.SERVICE_NAME);
+
+		if (metaManager != null && artifactManager != null) {
+			for (int i = 0; i < addedRepos.size(); i++) {
+				metaManager.removeRepository(reposToMigrate[i]);
+				artifactManager.removeRepository(reposToMigrate[i]);
+			}
+		}
+	}
+
+	//Remember that we completed the migration
+	private void rememberMigrationCompleted() {
+		new MigrationSupport().rememberMigrationCompleted();
+	}
+
+	//Purge the profile registry from all the entries that are no longer relevant
+	//We keep the base we import from on purpose to help with debugging
+	private void cleanupProfileRegistry() {
+		IProfileRegistry registry = (IProfileRegistry) ProvisioningUI.getDefaultUI().getSession().getProvisioningAgent().getService(IProfileRegistry.SERVICE_NAME);
+		long[] history = registry.listProfileTimestamps(toImportFrom.getProfileId());
+		for (int i = 0; i < history.length; i++) {
+			if (history[i] < toImportFrom.getTimestamp())
+				try {
+					registry.removeProfile(toImportFrom.getProfileId(), history[i]);
+				} catch (ProvisionException e) {
+					//Can't happen
+				}
+		}
+	}
+
+	public boolean performCancel() {
+		String[] buttons = new String[] {IDialogConstants.YES_LABEL, ProvUIMessages.MigrationPage_LATER_BUTTON, IDialogConstants.NO_LABEL};
+		MessageDialog dialog = new MessageDialog(getShell(), ProvUIMessages.MigrationPage_CONFIRMATION_TITLE, null, ProvUIMessages.MigrationPage_CONFIRMATION_DIALOG, MessageDialog.QUESTION, buttons, 2);
+
+		return rememberCancellationDecision(dialog.open());
+	}
+
+	//Method public for test
+	public boolean rememberCancellationDecision(int answer) {
+		boolean result = false;
+		switch (answer) {
+			case -1 : // if the user closes the dialog without clicking any button.
+				break;
+			case 0 :
+				result = true;
+				removeRepos();
+				rememberMigrationCompleted();
+				break;
+			case 1 :
+				result = true;
+				removeRepos();
+				break;
+			case 2 :
+				result = false;
+				break;
+		}
+		return result;
+	}
+
+	public IWizardPage getNextPage(IWizardPage page) {
+		IWizardPage toReturn = page;
+
+		try {
+			toReturn = super.getNextPage(page);
+		} catch (OperationCanceledException oce) {
+			// swallow and stay on the same page
+		}
+
+		return toReturn;
+	}
+
+}