UpdateHandler.java

package com.example.e4.rcp.todo.handlers;

import java.net.URI;
import java.net.URISyntaxException;

import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.jobs.IJobChangeEvent;
import org.eclipse.core.runtime.jobs.Job;
import org.eclipse.core.runtime.jobs.JobChangeAdapter;
import org.eclipse.e4.core.di.annotations.Execute;
import org.eclipse.e4.core.services.log.Logger;
import org.eclipse.e4.ui.di.UISynchronize;
import org.eclipse.e4.ui.workbench.IWorkbench;
import org.eclipse.equinox.p2.core.IProvisioningAgent;
import org.eclipse.equinox.p2.operations.ProvisioningJob;
import org.eclipse.equinox.p2.operations.ProvisioningSession;
import org.eclipse.equinox.p2.operations.UpdateOperation;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.swt.widgets.Shell;

// Require-Bundle: org.eclipse.equinox.p2.core|engine|operation|metadata.repository
// Feature: org.eclipse.equinox.p2.core.feature
//
// !!! Do not run from within IDE. Update only works in an exported product !!!
//
public class UpdateHandler {
	private static final String REPOSITORY_LOC = System.getProperty(
			"UpdateHandler.Repo", "http://localhost/repository");

	@Execute
	public void execute(final IProvisioningAgent agent, final Shell shell,
			final UISynchronize sync, final IWorkbench workbench,
			final Logger logger) {
		Job j = new Job("Update Job") {
			@Override
			protected IStatus run(final IProgressMonitor monitor) {
				return checkForUpdates(agent, shell, sync, workbench, monitor,
						logger);
			}
		};
		j.schedule();
	}

	private IStatus checkForUpdates(final IProvisioningAgent agent,
			final Shell shell, final UISynchronize sync,
			final IWorkbench workbench, IProgressMonitor monitor, Logger logger) {

		/* 1. configure update operation */
		final ProvisioningSession session = new ProvisioningSession(agent);
		final UpdateOperation operation = new UpdateOperation(session);
		configureUpdate(operation, logger);

		/* 2. Check for updates */

		// run check if updates are available (causing I/O)
		final IStatus status = operation.resolveModal(monitor);

		// Failed to find updates (inform user and exit)
		if (status.getCode() == UpdateOperation.STATUS_NOTHING_TO_UPDATE) {
			showMessage(shell, sync);
			return Status.CANCEL_STATUS;
		}

		/* 3. run installation */
		final ProvisioningJob provisioningJob = operation
				.getProvisioningJob(monitor);

		// updates cannot run from within Eclipse IDE!!!
		if (provisioningJob == null) {
			logger.error("Maybe you are trying to update from the Eclipse IDE? This won't work!!!");
			return Status.CANCEL_STATUS;
		}
		configureProvisioningJob(provisioningJob, shell, sync, workbench);

		provisioningJob.schedule();
		return Status.OK_STATUS;

	}

	private void configureProvisioningJob(ProvisioningJob provisioningJob,
			final Shell shell, final UISynchronize sync,
			final IWorkbench workbench) {

		// Register a job change listener to track
		// installation progress and notify user upon success
		provisioningJob.addJobChangeListener(new JobChangeAdapter() {
			@Override
			public void done(IJobChangeEvent event) {
				if (event.getResult().isOK()) {
					sync.syncExec(new Runnable() {

						@Override
						public void run() {
							boolean restart = MessageDialog
									.openQuestion(shell,
											"Updates installed, restart?",
											"Updates have been installed successfully, do you want to restart?");
							if (restart) {
								workbench.restart();
							}
						}
					});

				}
				super.done(event);
			}
		});

	}

	private void showMessage(final Shell parent, final UISynchronize sync) {
		sync.syncExec(new Runnable() {

			@Override
			public void run() {
				MessageDialog
						.openWarning(parent, "No update",
								"No updates for the current installation have been found");
			}
		});
	}

	private UpdateOperation configureUpdate(final UpdateOperation operation,
			Logger logger) {
		// create uri and check for validity
		URI uri = null;
		try {
			uri = new URI(REPOSITORY_LOC);
		} catch (final URISyntaxException e) {
			logger.error(e);
			return null;
		}

		// set location of artifact and metadata repo
		operation.getProvisioningContext().setArtifactRepositories(
				new URI[] { uri });
		operation.getProvisioningContext().setMetadataRepositories(
				new URI[] { uri });
		return operation;
	}
}