/*******************************************************************************
* Copyright (c) 2008, 2011 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
* Sonatype, Inc. - ongoing development
******************************************************************************/
package org.eclipse.equinox.internal.p2.ui.model;
import java.net.URI;
import java.util.*;
import org.eclipse.core.runtime.URIUtil;
import org.eclipse.equinox.internal.p2.ui.*;
import org.eclipse.equinox.p2.metadata.IInstallableUnit;
import org.eclipse.equinox.p2.operations.Remedy;
import org.eclipse.equinox.p2.operations.RemedyIUDetail;
import org.eclipse.equinox.p2.repository.IRepository;
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.ProvisioningUI;
/**
* Utility methods for manipulating model elements.
*
* @since 3.4
*
*/
public class ElementUtils {
public static void updateRepositoryUsingElements(final ProvisioningUI ui, final MetadataRepositoryElement[] elements) {
ui.signalRepositoryOperationStart();
IMetadataRepositoryManager metaManager = ProvUI.getMetadataRepositoryManager(ui.getSession());
IArtifactRepositoryManager artManager = ProvUI.getArtifactRepositoryManager(ui.getSession());
try {
int visibilityFlags = ui.getRepositoryTracker().getMetadataRepositoryFlags();
URI[] currentlyEnabled = metaManager.getKnownRepositories(visibilityFlags);
URI[] currentlyDisabled = metaManager.getKnownRepositories(IRepositoryManager.REPOSITORIES_DISABLED | visibilityFlags);
for (int i = 0; i < elements.length; i++) {
URI location = elements[i].getLocation();
if (elements[i].isEnabled()) {
if (containsURI(currentlyDisabled, location))
// It should be enabled and is not currently
setColocatedRepositoryEnablement(ui, location, true);
else if (!containsURI(currentlyEnabled, location)) {
// It is not known as enabled or disabled. Add it.
metaManager.addRepository(location);
artManager.addRepository(location);
}
} else {
if (containsURI(currentlyEnabled, location))
// It should be disabled, and is currently enabled
setColocatedRepositoryEnablement(ui, location, false);
else if (!containsURI(currentlyDisabled, location)) {
// It is not known as enabled or disabled. Add it and then disable it.
metaManager.addRepository(location);
artManager.addRepository(location);
setColocatedRepositoryEnablement(ui, location, false);
}
}
String name = elements[i].getName();
if (name != null && name.length() > 0) {
metaManager.setRepositoryProperty(location, IRepository.PROP_NICKNAME, name);
artManager.setRepositoryProperty(location, IRepository.PROP_NICKNAME, name);
}
}
// Are there any elements that need to be deleted? Go over the original state
// and remove any elements that weren't in the elements we were given
Set<String> nowKnown = new HashSet<>();
for (int i = 0; i < elements.length; i++)
nowKnown.add(URIUtil.toUnencodedString(elements[i].getLocation()));
for (int i = 0; i < currentlyEnabled.length; i++) {
if (!nowKnown.contains(URIUtil.toUnencodedString(currentlyEnabled[i]))) {
metaManager.removeRepository(currentlyEnabled[i]);
artManager.removeRepository(currentlyEnabled[i]);
}
}
for (int i = 0; i < currentlyDisabled.length; i++) {
if (!nowKnown.contains(URIUtil.toUnencodedString(currentlyDisabled[i]))) {
metaManager.removeRepository(currentlyDisabled[i]);
artManager.removeRepository(currentlyDisabled[i]);
}
}
} finally {
ui.signalRepositoryOperationComplete(null, true);
}
}
private static void setColocatedRepositoryEnablement(ProvisioningUI ui, URI location, boolean enable) {
ProvUI.getArtifactRepositoryManager(ui.getSession()).setEnabled(location, enable);
ProvUI.getMetadataRepositoryManager(ui.getSession()).setEnabled(location, enable);
}
public static IInstallableUnit getIU(Object element) {
if (element instanceof IInstallableUnit)
return (IInstallableUnit) element;
if (element instanceof IIUElement)
return ((IIUElement) element).getIU();
return ProvUI.getAdapter(element, IInstallableUnit.class);
}
public static List<IInstallableUnit> elementsToIUs(Object[] elements) {
ArrayList<IInstallableUnit> theIUs = new ArrayList<>(elements.length);
for (int i = 0; i < elements.length; i++) {
IInstallableUnit iu = ProvUI.getAdapter(elements[i], IInstallableUnit.class);
if (iu != null)
theIUs.add(iu);
}
return theIUs;
}
public static IInstallableUnit elementToIU(Object selectedElement) {
return ProvUI.getAdapter(selectedElement, IInstallableUnit.class);
}
static boolean containsURI(URI[] locations, URI url) {
for (int i = 0; i < locations.length; i++)
if (locations[i].equals(url))
return true;
return false;
}
public static AvailableIUElement[] requestToElement(Remedy remedy, boolean installMode) {
if (remedy == null)
return new AvailableIUElement[0];
ArrayList<AvailableIUElement> temp = new ArrayList<>();
ProvisioningUI ui = ProvisioningUI.getDefaultUI();
IUElementListRoot root = new IUElementListRoot(ui);
for (Iterator<RemedyIUDetail> iterator = remedy.getIusDetails().iterator(); iterator.hasNext();) {
RemedyIUDetail iuDetail = iterator.next();
if (iuDetail.getStatus() == RemedyIUDetail.STATUS_NOT_ADDED)
continue;
AvailableIUElement element = new AvailableIUElement(root, iuDetail.getIu(), ui.getProfileId(), true);
if (iuDetail.getBeingInstalledVersion() != null && iuDetail.getRequestedVersion() != null && iuDetail.getBeingInstalledVersion().compareTo(iuDetail.getRequestedVersion()) < 0 && !installMode)
element.setImageOverlayId(ProvUIImages.IMG_INFO);
else if (iuDetail.getStatus() == RemedyIUDetail.STATUS_REMOVED)
element.setImageId(ProvUIImages.IMG_REMOVED);
temp.add(element);
}
return temp.toArray(new AvailableIUElement[temp.size()]);
}
public static RemedyElementCategory[] requestToRemedyElementsCategories(Remedy remedy) {
List<RemedyElementCategory> categories = new ArrayList<>();
RemedyElementCategory categoryAdded = new RemedyElementCategory(ProvUIMessages.RemedyCategoryAdded);
RemedyElementCategory cateogyRemoved = new RemedyElementCategory(ProvUIMessages.RemedyCategoryRemoved);
RemedyElementCategory categoryNotAdded = new RemedyElementCategory(ProvUIMessages.RemedyCategoryNotAdded);
RemedyElementCategory categoryChanged = new RemedyElementCategory(ProvUIMessages.RemedyCategoryChanged);
for (Iterator<RemedyIUDetail> iterator = remedy.getIusDetails().iterator(); iterator.hasNext();) {
RemedyIUDetail remedyIUVersions = iterator.next();
if (remedyIUVersions.getStatus() == RemedyIUDetail.STATUS_ADDED)
categoryAdded.add(remedyIUVersions);
else if (remedyIUVersions.getStatus() == RemedyIUDetail.STATUS_CHANGED)
categoryChanged.add(remedyIUVersions);
else if (remedyIUVersions.getStatus() == RemedyIUDetail.STATUS_REMOVED)
cateogyRemoved.add(remedyIUVersions);
else if (remedyIUVersions.getStatus() == RemedyIUDetail.STATUS_NOT_ADDED)
categoryNotAdded.add(remedyIUVersions);
}
if (cateogyRemoved.getElements().size() > 0)
categories.add(cateogyRemoved);
if (categoryChanged.getElements().size() > 0)
categories.add(categoryChanged);
if (categoryNotAdded.getElements().size() > 0)
categories.add(categoryNotAdded);
if (categoryAdded.getElements().size() > 0)
categories.add(categoryAdded);
return categories.toArray(new RemedyElementCategory[categories.size()]);
}
}
