Java tutorial
/******************************************************************************* * Copyright (c) 2013, 2016 Robin Stocker <robin@nibor.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.egit.ui.internal.push; import java.io.IOException; import java.net.URISyntaxException; import java.text.MessageFormat; import java.util.Arrays; import java.util.List; import java.util.Set; import org.eclipse.egit.core.op.PushOperationResult; import org.eclipse.egit.core.op.PushOperationSpecification; import org.eclipse.egit.ui.internal.SecureStoreUtils; import org.eclipse.egit.ui.internal.UIIcons; import org.eclipse.egit.ui.internal.UIText; import org.eclipse.egit.ui.internal.components.RepositorySelection; import org.eclipse.egit.ui.internal.credentials.EGitCredentialsProvider; import org.eclipse.jface.wizard.Wizard; import org.eclipse.jgit.lib.BranchConfig.BranchRebaseMode; import org.eclipse.jgit.lib.ConfigConstants; import org.eclipse.jgit.lib.Constants; import org.eclipse.jgit.lib.ObjectId; import org.eclipse.jgit.lib.Ref; import org.eclipse.jgit.lib.Repository; import org.eclipse.jgit.lib.StoredConfig; import org.eclipse.jgit.transport.RefSpec; import org.eclipse.jgit.transport.RemoteConfig; import org.eclipse.jgit.transport.URIish; /** * A wizard dedicated to pushing a commit. */ public class PushBranchWizard extends Wizard { private final Repository repository; private final ObjectId commitToPush; /** * In case of detached HEAD, reference is null. */ private final Ref ref; private AddRemotePage addRemotePage; private PushBranchPage pushBranchPage; private ConfirmationPage confirmationPage; /** * @param repository * the repository the ref belongs to * @param ref */ public PushBranchWizard(final Repository repository, Ref ref) { this(repository, ref.getObjectId(), ref); } /** * @param repository * the repository commit belongs to * @param commitToPush */ public PushBranchWizard(final Repository repository, ObjectId commitToPush) { this(repository, commitToPush, null); } private PushBranchWizard(final Repository repository, ObjectId commitToPush, Ref ref) { this.repository = repository; this.commitToPush = commitToPush; this.ref = ref; assert (this.repository != null); assert (this.commitToPush != null); Set<String> remoteNames = repository.getConfig().getSubsections(ConfigConstants.CONFIG_REMOTE_SECTION); if (remoteNames.isEmpty()) addRemotePage = new AddRemotePage(repository); pushBranchPage = new PushBranchPage(repository, commitToPush, ref) { @Override public void setVisible(boolean visible) { if (visible && addRemotePage != null) { setSelectedRemote(addRemotePage.getRemoteName(), addRemotePage.getSelection().getURI()); } super.setVisible(visible); } }; // Don't show button if we're configuring a remote in the first step pushBranchPage.setShowNewRemoteButton(addRemotePage == null); confirmationPage = new ConfirmationPage(repository) { @Override public void setVisible(boolean visible) { setSelection(getRepositorySelection(), getRefSpecs()); AddRemotePage remotePage = getAddRemotePage(); if (remotePage != null) setCredentials(remotePage.getCredentials()); super.setVisible(visible); } }; setDefaultPageImageDescriptor(UIIcons.WIZBAN_PUSH); } @Override public void addPages() { if (addRemotePage != null) addPage(addRemotePage); addPage(pushBranchPage); addPage(confirmationPage); } @Override public String getWindowTitle() { if (ref != null) return MessageFormat.format(UIText.PushBranchWizard_WindowTitle, Repository.shortenRefName(this.ref.getName())); else return UIText.PushCommitHandler_pushCommitTitle; } @Override public boolean canFinish() { return getContainer().getCurrentPage() == confirmationPage && confirmationPage.isPageComplete(); } @Override public boolean performFinish() { try { AddRemotePage remotePage = getAddRemotePage(); if (remotePage != null) { storeCredentials(remotePage); URIish uri = remotePage.getSelection().getURI(); configureNewRemote(uri); } if (pushBranchPage.getUpstreamConfig() != null) { configureUpstream(); } startPush(); } catch (IOException e) { confirmationPage.setErrorMessage(e.getMessage()); return false; } catch (URISyntaxException e) { confirmationPage.setErrorMessage(e.getMessage()); return false; } return true; } private AddRemotePage getAddRemotePage() { if (addRemotePage != null) return addRemotePage; else return pushBranchPage.getAddRemotePage(); } private RepositorySelection getRepositorySelection() { AddRemotePage remotePage = getAddRemotePage(); if (remotePage != null) return remotePage.getSelection(); else return new RepositorySelection(null, pushBranchPage.getRemoteConfig()); } private List<RefSpec> getRefSpecs() { String src = this.ref != null ? this.ref.getName() : this.commitToPush.getName(); String dst = pushBranchPage.getFullRemoteReference(); RefSpec refSpec = new RefSpec().setSourceDestination(src, dst) .setForceUpdate(pushBranchPage.isForceUpdateSelected()); return Arrays.asList(refSpec); } private void storeCredentials(AddRemotePage remotePage) { if (remotePage.getStoreInSecureStore()) { URIish uri = remotePage.getSelection().getURI(); if (uri != null) SecureStoreUtils.storeCredentials(remotePage.getCredentials(), uri); } } private void configureNewRemote(URIish uri) throws URISyntaxException, IOException { StoredConfig config = repository.getConfig(); String remoteName = getRemoteName(); RemoteConfig remoteConfig = new RemoteConfig(config, remoteName); remoteConfig.addURI(uri); RefSpec defaultFetchSpec = new RefSpec().setForceUpdate(true).setSourceDestination(Constants.R_HEADS + "*", //$NON-NLS-1$ Constants.R_REMOTES + remoteName + "/*"); //$NON-NLS-1$ remoteConfig.addFetchRefSpec(defaultFetchSpec); remoteConfig.update(config); config.save(); } private void configureUpstream() throws IOException { if (this.ref == null) { // Don't configure upstream for detached HEAD return; } String remoteName = getRemoteName(); String fullRemoteBranchName = pushBranchPage.getFullRemoteReference(); String localBranchName = Repository.shortenRefName(this.ref.getName()); StoredConfig config = repository.getConfig(); config.setString(ConfigConstants.CONFIG_BRANCH_SECTION, localBranchName, ConfigConstants.CONFIG_KEY_REMOTE, remoteName); config.setString(ConfigConstants.CONFIG_BRANCH_SECTION, localBranchName, ConfigConstants.CONFIG_KEY_MERGE, fullRemoteBranchName); BranchRebaseMode rebaseMode = pushBranchPage.getUpstreamConfig(); if (rebaseMode != null) { config.setEnum(ConfigConstants.CONFIG_BRANCH_SECTION, localBranchName, ConfigConstants.CONFIG_KEY_REBASE, rebaseMode); } config.save(); } private void startPush() throws IOException { PushOperationResult result = confirmationPage.getConfirmedResult(); PushOperationSpecification pushSpec = result .deriveSpecification(confirmationPage.isRequireUnchangedSelected()); PushOperationUI pushOperationUI = new PushOperationUI(repository, pushSpec, false); pushOperationUI.setCredentialsProvider(new EGitCredentialsProvider()); pushOperationUI.setShowConfigureButton(false); if (confirmationPage.isShowOnlyIfChangedSelected()) pushOperationUI.setExpectedResult(result); pushOperationUI.start(); } private String getRemoteName() { return pushBranchPage.getRemoteConfig().getName(); } }