EclipseGitProgressTransformer.java

/*******************************************************************************
 * Copyright (C) 2008, 2015 Shawn O. Pearce <spearce@spearce.org> 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
 *******************************************************************************/
package org.eclipse.orion.server.git.jobs;

import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.SubProgressMonitor;
import org.eclipse.jgit.lib.ProgressMonitor;
import org.eclipse.orion.server.core.tasks.TaskJob;

/**
 * Create a new Git to Eclipse progress monitor. From org.eclipse.egit.core.
 */
public class EclipseGitProgressTransformer implements ProgressMonitor {
	private static final String EMPTY_STRING = ""; //$NON-NLS-1$

	private final IProgressMonitor root;

	private IProgressMonitor task;

	private TaskJob job;

	private String msg;

	private int lastWorked;

	private int totalWork;

	/**
	 * Create a new progress monitor.
	 *
	 * @param eclipseMonitor
	 *            the Eclipse monitor we update.
	 */
	public EclipseGitProgressTransformer(final IProgressMonitor eclipseMonitor) {
		root = eclipseMonitor;
	}

	/**
	 * Create a new progress monitor.
	 *
	 * @param eclipseMonitor
	 *            the Eclipse monitor we update.
	 * @param job
	 *            the Orion task job
	 */
	public EclipseGitProgressTransformer(final IProgressMonitor eclipseMonitor, TaskJob job) {
		root = eclipseMonitor;
		this.job = job;
	}

	@Override
	public void start(final int totalTasks) {
		root.beginTask(EMPTY_STRING, totalTasks * 1000);
	}

	@Override
	public void beginTask(final String name, final int total) {
		endTask();
		msg = name;
		lastWorked = 0;
		totalWork = total;
		task = new SubProgressMonitor(root, 1000);
		if (totalWork == UNKNOWN)
			task.beginTask(EMPTY_STRING, IProgressMonitor.UNKNOWN);
		else
			task.beginTask(EMPTY_STRING, totalWork);
		task.subTask(msg);

		if (job != null) {
			job.setTaskTotal(total);
			job.setTaskMessage(msg);
		}
	}

	@Override
	public void update(final int work) {
		final int cmp = lastWorked + work;
		if (task != null) {
			if (totalWork == UNKNOWN && cmp > 0) {
				if (lastWorked != cmp)
					task.subTask(msg + ", " + cmp); //$NON-NLS-1$
			} else if (totalWork <= 0) {
				// Do nothing to update the task.
			} else if (cmp * 100 / totalWork != lastWorked * 100 / totalWork) {
				final StringBuilder m = new StringBuilder();
				m.append(msg);
				m.append(": "); //$NON-NLS-1$
				while (m.length() < 25)
					m.append(' ');

				final String twstr = String.valueOf(totalWork);
				String cmpstr = String.valueOf(cmp);
				while (cmpstr.length() < twstr.length())
					cmpstr = " " + cmpstr; //$NON-NLS-1$
				final int pcnt = (cmp * 100 / totalWork);
				if (pcnt < 100)
					m.append(' ');
				if (pcnt < 10)
					m.append(' ');
				m.append(pcnt);
				m.append("% ("); //$NON-NLS-1$
				m.append(cmpstr);
				m.append("/"); //$NON-NLS-1$
				m.append(twstr);
				m.append(")"); //$NON-NLS-1$

				task.subTask(m.toString());
				if (job != null) {
					if (work < totalWork) {
						job.setTaskLoaded(cmp);
					}
				}
			}
			task.worked(work);
		}
		lastWorked = cmp;
	}

	@Override
	public void endTask() {
		if (task != null) {
			try {
				task.done();
			} finally {
				task = null;
			}
		}
	}

	@Override
	public boolean isCancelled() {
		if (task != null)
			return task.isCanceled();
		return root.isCanceled();
	}
}