core/com.nokia.carbide.discovery.ui/src/com/nokia/carbide/internal/discovery/ui/wizard/P2Utils.java
changeset 1818 bb528d06abcd
parent 1810 2ccd3660a736
child 1824 7e34f628583a
equal deleted inserted replaced
1817:4eae2726ff31 1818:bb528d06abcd
       
     1 /*
       
     2 * Copyright (c) 2010 Nokia Corporation and/or its subsidiary(-ies).
       
     3 * All rights reserved.
       
     4 * This component and the accompanying materials are made available
       
     5 * under the terms of the License "Eclipse Public License v1.0"
       
     6 * which accompanies this distribution, and is available
       
     7 * at the URL "http://www.eclipse.org/legal/epl-v10.html".
       
     8 *
       
     9 * Initial Contributors:
       
    10 * Nokia Corporation - initial contribution.
       
    11 *
       
    12 * Contributors:
       
    13 *
       
    14 * Description: 
       
    15 *
       
    16 */
       
    17 package com.nokia.carbide.internal.discovery.ui.wizard;
       
    18 
       
    19 import java.net.URI;
       
    20 import java.util.ArrayList;
       
    21 import java.util.Arrays;
       
    22 import java.util.Collection;
       
    23 import java.util.HashSet;
       
    24 import java.util.Iterator;
       
    25 import java.util.List;
       
    26 import java.util.Set;
       
    27 
       
    28 import org.eclipse.core.runtime.CoreException;
       
    29 import org.eclipse.core.runtime.IProgressMonitor;
       
    30 import org.eclipse.core.runtime.IStatus;
       
    31 import org.eclipse.core.runtime.OperationCanceledException;
       
    32 import org.eclipse.core.runtime.SubMonitor;
       
    33 import org.eclipse.equinox.p2.core.IProvisioningAgent;
       
    34 import org.eclipse.equinox.p2.core.ProvisionException;
       
    35 import org.eclipse.equinox.p2.engine.IProfile;
       
    36 import org.eclipse.equinox.p2.engine.IProfileRegistry;
       
    37 import org.eclipse.equinox.p2.metadata.IInstallableUnit;
       
    38 import org.eclipse.equinox.p2.metadata.Version;
       
    39 import org.eclipse.equinox.p2.operations.InstallOperation;
       
    40 import org.eclipse.equinox.p2.operations.ProvisioningSession;
       
    41 import org.eclipse.equinox.p2.operations.RepositoryTracker;
       
    42 import org.eclipse.equinox.p2.query.IQueryResult;
       
    43 import org.eclipse.equinox.p2.query.QueryUtil;
       
    44 import org.eclipse.equinox.p2.repository.metadata.IMetadataRepository;
       
    45 import org.eclipse.equinox.p2.repository.metadata.IMetadataRepositoryManager;
       
    46 import org.eclipse.equinox.p2.ui.ProvisioningUI;
       
    47 import org.eclipse.swt.widgets.Display;
       
    48 
       
    49 class P2Utils {
       
    50 	
       
    51 	public static Collection<URI> getKnownRepositories() {
       
    52 		ProvisioningUI provisioningUI = ProvisioningUI.getDefaultUI();
       
    53 		ProvisioningSession session = provisioningUI.getSession();
       
    54 		URI[] uris = provisioningUI.getRepositoryTracker().getKnownRepositories(session);
       
    55 		return Arrays.asList(uris);
       
    56 	}
       
    57 
       
    58 	public static Collection<FeatureInfo> getInstalledFeatures(IProgressMonitor monitor) {
       
    59 		Set<FeatureInfo> infos = new HashSet<FeatureInfo>();
       
    60 		ProvisioningUI provisioningUI = ProvisioningUI.getDefaultUI();
       
    61 		ProvisioningSession session = provisioningUI.getSession();
       
    62 		IProvisioningAgent agent = session.getProvisioningAgent();
       
    63 		IProfileRegistry profileRegistry = (IProfileRegistry) agent.getService(IProfileRegistry.SERVICE_NAME);
       
    64 		IProfile profile = null;
       
    65 		if (profileRegistry != null) {
       
    66 			profile = profileRegistry.getProfile(provisioningUI.getProfileId());
       
    67 			if (profile != null) {
       
    68 				IQueryResult<IInstallableUnit> result = 
       
    69 					profile.query(QueryUtil.createIUGroupQuery(), monitor);
       
    70 				for (Iterator<IInstallableUnit> iterator = result.iterator(); iterator.hasNext();) {
       
    71 					infos.add(new FeatureInfo((IInstallableUnit) iterator.next()));
       
    72 				}
       
    73 			}
       
    74 		}
       
    75 		return infos;
       
    76 	}
       
    77 
       
    78 	public static void doInstall(Collection<URI> uris, Collection<FeatureInfo> featureInfos, 
       
    79 			boolean wantVersions, IProgressMonitor monitor) throws OperationCanceledException, CoreException {
       
    80 		SubMonitor subMonitor = SubMonitor.convert(monitor, "Configuring install operation", 100);
       
    81 		Collection<IMetadataRepository> repositories = 
       
    82 			getRepositories(uris, wantVersions, subMonitor.newChild(30));
       
    83 		final Collection<IInstallableUnit> ius = 
       
    84 			findInstallableUnits(repositories, featureInfos, wantVersions, subMonitor.newChild(40));
       
    85 		final InstallOperation operation = resolve(ius, (URI[]) uris.toArray(new URI[uris.size()]), subMonitor.newChild(30));
       
    86 		Display.getDefault().asyncExec(new Runnable() {
       
    87 			@Override
       
    88 			public void run() {
       
    89 				ProvisioningUI.getDefaultUI().openInstallWizard(ius, operation, null);
       
    90 			}
       
    91 		});
       
    92 	}
       
    93 
       
    94 	private static Collection<IMetadataRepository> getRepositories(Collection<URI> uris, 
       
    95 			boolean wantVersions, SubMonitor monitor) throws ProvisionException, OperationCanceledException {
       
    96 		monitor.setWorkRemaining(uris.size());
       
    97 		List<IMetadataRepository> repositories = new ArrayList<IMetadataRepository>();
       
    98 		ProvisioningUI provisioningUI = ProvisioningUI.getDefaultUI();
       
    99 		ProvisioningSession session = provisioningUI.getSession();
       
   100 		RepositoryTracker repositoryTracker = provisioningUI.getRepositoryTracker();
       
   101 		IMetadataRepositoryManager manager = 
       
   102 			(IMetadataRepositoryManager) session.getProvisioningAgent().getService(IMetadataRepositoryManager.SERVICE_NAME);
       
   103 		for (URI uri : uris) {
       
   104 			checkIfCanceled(monitor);
       
   105 			repositoryTracker.addRepository(uri, null, session);
       
   106 			repositories.add(manager.loadRepository(uri, monitor.newChild(1)));
       
   107 		}
       
   108 
       
   109 		return repositories;
       
   110 	}
       
   111 
       
   112 	private static Collection<IInstallableUnit> findInstallableUnits(Collection<IMetadataRepository> repositories, 
       
   113 			Collection<FeatureInfo> featureInfos, boolean wantVersions, SubMonitor monitor) throws ProvisionException {
       
   114 		monitor.setWorkRemaining(repositories.size());
       
   115 		List<IInstallableUnit> ius = new ArrayList<IInstallableUnit>();
       
   116 		Set<FeatureInfo> remainingInfos = new HashSet<FeatureInfo>(featureInfos);
       
   117 		for (IMetadataRepository repository : repositories) {
       
   118 			checkIfCanceled(monitor);
       
   119 			IQueryResult<IInstallableUnit> result = repository.query(QueryUtil.createIUGroupQuery(), monitor.newChild(1));
       
   120 			for (Iterator<IInstallableUnit> iter = result.iterator(); iter.hasNext();) {
       
   121 				IInstallableUnit iu = iter.next();
       
   122 				String id = iu.getId();
       
   123 				Version version = iu.getVersion();
       
   124 				FeatureInfo featureInfo = findInfo(remainingInfos, id, wantVersions ? version : null);
       
   125 				if (featureInfo != null) {
       
   126 					ius.add(iu);
       
   127 					remainingInfos.remove(featureInfo);
       
   128 				}
       
   129 			}
       
   130 		}
       
   131 		if (!remainingInfos.isEmpty()) {
       
   132 			StringBuilder sb = new StringBuilder();
       
   133 			sb.append("Install failed. Could not find some features: \n");
       
   134 			for (FeatureInfo featureInfo : remainingInfos) {
       
   135 				sb.append("id=");
       
   136 				sb.append(featureInfo.getId());
       
   137 				if (wantVersions) {
       
   138 					sb.append("version=");
       
   139 					sb.append(featureInfo.getVersion().toString());
       
   140 				}
       
   141 				sb.append("\n");
       
   142 			}
       
   143 			throw new ProvisionException(sb.toString());
       
   144 		}
       
   145 		return ius;
       
   146 	}
       
   147 	
       
   148 	private static FeatureInfo findInfo(Collection<FeatureInfo> featureInfos, String id, Version version) {
       
   149 		for (FeatureInfo featureInfo : featureInfos) {
       
   150 			if (featureInfo.getId().equals(id) && 
       
   151 					(version == null || version.toString().equals(featureInfo.getVersion().toString())))
       
   152 				return featureInfo;
       
   153 		}
       
   154 		return null;
       
   155 	}
       
   156 	
       
   157 	private static InstallOperation resolve(Collection<IInstallableUnit> ius, URI[] repositories, 
       
   158 			SubMonitor monitor) throws CoreException {
       
   159 		checkIfCanceled(monitor);
       
   160 		ProvisioningUI provisioningUI = ProvisioningUI.getDefaultUI();
       
   161 		InstallOperation installOperation = provisioningUI.getInstallOperation(ius, repositories);
       
   162 		IStatus operationStatus = installOperation.resolveModal(monitor);
       
   163 		if (operationStatus.getSeverity() > IStatus.WARNING) {
       
   164 			throw new CoreException(operationStatus);
       
   165 		}
       
   166 		return installOperation;
       
   167 	}
       
   168 
       
   169 	private static void checkIfCanceled(IProgressMonitor monitor) {
       
   170 		if (monitor.isCanceled()) {
       
   171 			throw new OperationCanceledException();
       
   172 		}
       
   173 	}
       
   174 }