CherryPickWizard.java

package com.proalpha.pds.gitutils.cherrypick;
 
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.util.Arrays;
import java.util.List;
 
import org.eclipse.core.runtime.jobs.ISchedulingRule;
import org.eclipse.core.runtime.jobs.Job;
import org.eclipse.egit.core.internal.job.RuleUtil;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.wizard.Wizard;
import org.eclipse.jgit.api.Git;
import org.eclipse.jgit.api.errors.GitAPIException;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.revwalk.RevCommit;
import org.eclipse.swt.custom.BusyIndicator;
import org.eclipse.swt.widgets.Display;
 
import com.proalpha.git.PaGit;
 
@SuppressWarnings("restriction")
public class CherryPickWizard extends Wizard {
 
    private final CherryPickPage cpPage;
    private final CherryPickCommitPage comPage;
    private Repository targetRepository;
    private List<RevCommit> sourceCommits;
    private CherryPickSettings cherryPickSettings = new CherryPickSettings();
 
    public List<RevCommit> getSourceCommits() {
        return sourceCommits;
    }
 
    public CherryPickSettings getSettings() {
        return cherryPickSettings;
    }
 
    public CherryPickWizard(Repository repository) {
        super();
        cherryPickSettings.setTargetRepository(repository.getDirectory().toString());
 
        this.targetRepository = repository;
        this.cpPage = new CherryPickPage(repository);
        this.comPage = new CherryPickCommitPage();
        cherryPickSettings.setTargetRepository(repository.getDirectory().toString());
        PaGit.init(new Git(targetRepository));
    }
 
    /*
     * (non-Javadoc)
     * 
     * @see org.eclipse.jface.wizard.Wizard#performFinish()
     */
    @Override
    public boolean performFinish() {
 
        try {
            CherryPick.getInstance().createCherryPickCommand(cherryPickSettings);
        } catch (IOException | GitAPIException e) {
            MessageDialog.openError(getShell(), e.getCause().getMessage(),
                    Arrays.toString(e.getCause().getStackTrace()));
        }
 
        Job cherryPickJob = new CherryPickJob(this.sourceCommits.size());
        cherryPickJob.setUser(true);
        cherryPickJob.setRule(RuleUtil.getRule(CherryPick.getInstance().getCherryPickCommand().getRepository()));
        cherryPickJob.schedule();
 
        return true;
    }
 
    /*
     * (non-Javadoc)
     * 
     * @see org.eclipse.jface.wizard.Wizard#addPages()
     */
    @Override
    public void addPages() {
        addPage(cpPage);
        addPage(comPage);
    }
 
    public boolean gatherCommits() {
 
        GatherCommitsOperation op = new GatherCommitsOperation(cherryPickSettings.getSourceRepoString(),
                cherryPickSettings.getSourceBranch(), cherryPickSettings.getSourceIssue());
 
        Runnable r = () -> {
            ISchedulingRule rule = op.getSchedulingRule();
            try {
                Job.getJobManager().beginRule(rule, null);
                op.run();
            } catch (InvocationTargetException e) {
                MessageDialog.openError(getShell(), e.getCause().getMessage(),
                        Arrays.toString(e.getCause().getStackTrace()));
            } finally {
                Job.getJobManager().endRule(rule);
            }
        };
        BusyIndicator.showWhile(Display.getCurrent(), r);
        this.sourceCommits = op.getOperationResult();
        comPage.setCommitsfound(this.sourceCommits);
        return !this.sourceCommits.isEmpty();
    }
 
}