| /******************************************************************************* |
| * Copyright (c) 2011, 2017 WindRiver Corporation and others. |
| * |
| * This program and the accompanying materials |
| * are made available under the terms of the Eclipse Public License 2.0 |
| * which accompanies this distribution, and is available at |
| * https://www.eclipse.org/legal/epl-2.0/ |
| * |
| * SPDX-License-Identifier: EPL-2.0 |
| * |
| * Contributors: |
| * WindRiver Corporation - initial API and implementation |
| * Ericsson AB - Ongoing development |
| * Ericsson AB (Pascal Rapicault) |
| * Ericsson AB (Hamdan Msheik) |
| * Red Hat Inc. - Bug 460967 |
| *******************************************************************************/ |
| 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.IDialogConstants; |
| import org.eclipse.jface.dialogs.MessageDialog; |
| 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<>(); |
| 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); |
| } |
| |
| @Override |
| 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.png"))); //$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 = ServiceHelper.getService(AutomaticUpdatePlugin.getContext(), IProvisioningAgent.class); |
| 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 = ServiceHelper.getService(AutomaticUpdatePlugin.getContext(), IProvisioningAgent.class); |
| 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 |
| } |
| } |
| } |
| |
| @Override |
| 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; |
| } |
| |
| @Override |
| 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; |
| } |
| |
| } |