EclipseGitProgressTransformer.java

package org.bndtools.templating.jgit;

import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.SubProgressMonitor;
import org.eclipse.jgit.lib.ProgressMonitor;

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

    private final IProgressMonitor root;

    private IProgressMonitor task;

    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;
    }

    @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);
    }

    @Override
    public void update(final int work) {
        if (task == null)
            return;

        final int cmp = lastWorked + work;
        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());
        }
        lastWorked = cmp;
        task.worked(work);
    }

    @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();
    }
}