123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430 |
- /*
- * Copyright (C) 2010, Chris Aniszczyk <caniszczyk@gmail.com>
- * Copyright (C) 2011, Matthias Sohn <matthias.sohn@sap.com>
- * and other copyright owners as documented in the project's IP log.
- *
- * This program and the accompanying materials are made available
- * under the terms of the Eclipse Distribution License v1.0 which
- * accompanies this distribution, is reproduced below, and is
- * available at http://www.eclipse.org/org/documents/edl-v10.php
- *
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or
- * without modification, are permitted provided that the following
- * conditions are met:
- *
- * - Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- *
- * - Redistributions in binary form must reproduce the above
- * copyright notice, this list of conditions and the following
- * disclaimer in the documentation and/or other materials provided
- * with the distribution.
- *
- * - Neither the name of the Eclipse Foundation, Inc. nor the
- * names of its contributors may be used to endorse or promote
- * products derived from this software without specific prior
- * written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
- * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
- * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
- * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
- * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
- * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
- * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
- * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
- * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
- * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
- package org.eclipse.jgit.api;
-
- import java.io.File;
- import java.io.IOException;
- import java.text.MessageFormat;
- import java.util.LinkedList;
- import java.util.List;
-
- import org.eclipse.jgit.api.CheckoutResult.Status;
- import org.eclipse.jgit.api.errors.CheckoutConflictException;
- import org.eclipse.jgit.api.errors.InvalidRefNameException;
- import org.eclipse.jgit.api.errors.JGitInternalException;
- import org.eclipse.jgit.api.errors.RefAlreadyExistsException;
- import org.eclipse.jgit.api.errors.RefNotFoundException;
- import org.eclipse.jgit.dircache.DirCache;
- import org.eclipse.jgit.dircache.DirCacheCheckout;
- import org.eclipse.jgit.dircache.DirCacheEditor;
- import org.eclipse.jgit.dircache.DirCacheEditor.PathEdit;
- import org.eclipse.jgit.dircache.DirCacheEntry;
- import org.eclipse.jgit.dircache.DirCacheIterator;
- import org.eclipse.jgit.errors.AmbiguousObjectException;
- import org.eclipse.jgit.internal.JGitText;
- import org.eclipse.jgit.lib.AnyObjectId;
- import org.eclipse.jgit.lib.Constants;
- import org.eclipse.jgit.lib.FileMode;
- import org.eclipse.jgit.lib.ObjectId;
- import org.eclipse.jgit.lib.ObjectReader;
- import org.eclipse.jgit.lib.Ref;
- import org.eclipse.jgit.lib.RefUpdate;
- import org.eclipse.jgit.lib.RefUpdate.Result;
- import org.eclipse.jgit.lib.Repository;
- import org.eclipse.jgit.revwalk.RevCommit;
- import org.eclipse.jgit.revwalk.RevTree;
- import org.eclipse.jgit.revwalk.RevWalk;
- import org.eclipse.jgit.treewalk.TreeWalk;
- import org.eclipse.jgit.treewalk.filter.PathFilterGroup;
-
- /**
- * Checkout a branch to the working tree
- *
- * @see <a
- * href="http://www.kernel.org/pub/software/scm/git/docs/git-checkout.html"
- * >Git documentation about Checkout</a>
- */
- public class CheckoutCommand extends GitCommand<Ref> {
- private String name;
-
- private boolean force = false;
-
- private boolean createBranch = false;
-
- private CreateBranchCommand.SetupUpstreamMode upstreamMode;
-
- private String startPoint = null;
-
- private RevCommit startCommit;
-
- private CheckoutResult status;
-
- private List<String> paths;
-
- private boolean checkoutAllPaths;
-
- /**
- * @param repo
- */
- protected CheckoutCommand(Repository repo) {
- super(repo);
- this.paths = new LinkedList<String>();
- }
-
- /**
- * @throws RefAlreadyExistsException
- * when trying to create (without force) a branch with a name
- * that already exists
- * @throws RefNotFoundException
- * if the start point or branch can not be found
- * @throws InvalidRefNameException
- * if the provided name is <code>null</code> or otherwise
- * invalid
- * @return the newly created branch
- */
- public Ref call() throws JGitInternalException, RefAlreadyExistsException,
- RefNotFoundException, InvalidRefNameException,
- CheckoutConflictException {
- checkCallable();
- processOptions();
- try {
- if (checkoutAllPaths || !paths.isEmpty()) {
- checkoutPaths();
- status = CheckoutResult.OK_RESULT;
- setCallable(false);
- return null;
- }
-
- if (createBranch) {
- Git git = new Git(repo);
- CreateBranchCommand command = git.branchCreate();
- command.setName(name);
- command.setStartPoint(getStartPoint().name());
- if (upstreamMode != null)
- command.setUpstreamMode(upstreamMode);
- command.call();
- }
-
- Ref headRef = repo.getRef(Constants.HEAD);
- String shortHeadRef = getShortBranchName(headRef);
- String refLogMessage = "checkout: moving from " + shortHeadRef;
- ObjectId branch = repo.resolve(name);
- if (branch == null)
- throw new RefNotFoundException(MessageFormat.format(JGitText
- .get().refNotResolved, name));
-
- RevWalk revWalk = new RevWalk(repo);
- AnyObjectId headId = headRef.getObjectId();
- RevCommit headCommit = headId == null ? null : revWalk
- .parseCommit(headId);
- RevCommit newCommit = revWalk.parseCommit(branch);
- RevTree headTree = headCommit == null ? null : headCommit.getTree();
- DirCacheCheckout dco;
- DirCache dc = repo.lockDirCache();
- try {
- dco = new DirCacheCheckout(repo, headTree, dc,
- newCommit.getTree());
- dco.setFailOnConflict(true);
- try {
- dco.checkout();
- } catch (org.eclipse.jgit.errors.CheckoutConflictException e) {
- status = new CheckoutResult(Status.CONFLICTS,
- dco.getConflicts());
- throw new CheckoutConflictException(dco.getConflicts(), e);
- }
- } finally {
- dc.unlock();
- }
- Ref ref = repo.getRef(name);
- if (ref != null && !ref.getName().startsWith(Constants.R_HEADS))
- ref = null;
- String toName = Repository.shortenRefName(name);
- RefUpdate refUpdate = repo.updateRef(Constants.HEAD, ref == null);
- refUpdate.setForceUpdate(force);
- refUpdate.setRefLogMessage(refLogMessage + " to " + toName, false);
- Result updateResult;
- if (ref != null)
- updateResult = refUpdate.link(ref.getName());
- else {
- refUpdate.setNewObjectId(newCommit);
- updateResult = refUpdate.forceUpdate();
- }
-
- setCallable(false);
-
- boolean ok = false;
- switch (updateResult) {
- case NEW:
- ok = true;
- break;
- case NO_CHANGE:
- case FAST_FORWARD:
- case FORCED:
- ok = true;
- break;
- default:
- break;
- }
-
- if (!ok)
- throw new JGitInternalException(MessageFormat.format(JGitText
- .get().checkoutUnexpectedResult, updateResult.name()));
-
- if (!dco.getToBeDeleted().isEmpty()) {
- status = new CheckoutResult(Status.NONDELETED, dco
- .getToBeDeleted());
- }
- else
- status = CheckoutResult.OK_RESULT;
- return ref;
- } catch (IOException ioe) {
- throw new JGitInternalException(ioe.getMessage(), ioe);
- } finally {
- if (status == null)
- status = CheckoutResult.ERROR_RESULT;
- }
- }
-
- private String getShortBranchName(Ref headRef) {
- if (headRef.getTarget().getName().equals(headRef.getName()))
- return headRef.getTarget().getObjectId().getName();
- return Repository.shortenRefName(headRef.getTarget().getName());
- }
-
- /**
- * @param path
- * Path to update in the working tree and index.
- * @return {@code this}
- */
- public CheckoutCommand addPath(String path) {
- checkCallable();
- this.paths.add(path);
- return this;
- }
-
- /**
- * Set whether to checkout all paths
- * <p>
- * This options should be used when you want to do a path checkout on the
- * entire repository and so calling {@link #addPath(String)} is not possible
- * since empty paths are not allowed.
- *
- * @param all
- * true to checkout all paths, false otherwise
- * @return {@code this}
- */
- public CheckoutCommand setAllPaths(boolean all) {
- checkoutAllPaths = all;
- return this;
- }
-
- /**
- * Checkout paths into index and working directory
- *
- * @return this instance
- * @throws IOException
- * @throws RefNotFoundException
- */
- protected CheckoutCommand checkoutPaths() throws IOException,
- RefNotFoundException {
- RevWalk revWalk = new RevWalk(repo);
- DirCache dc = repo.lockDirCache();
- try {
- DirCacheEditor editor = dc.editor();
- TreeWalk startWalk = new TreeWalk(revWalk.getObjectReader());
- startWalk.setRecursive(true);
- if (!checkoutAllPaths)
- startWalk.setFilter(PathFilterGroup.createFromStrings(paths));
- boolean checkoutIndex = startCommit == null && startPoint == null;
- if (!checkoutIndex)
- startWalk.addTree(revWalk.parseCommit(getStartPoint())
- .getTree());
- else
- startWalk.addTree(new DirCacheIterator(dc));
-
- final File workTree = repo.getWorkTree();
- final ObjectReader r = repo.getObjectDatabase().newReader();
- try {
- while (startWalk.next()) {
- final ObjectId blobId = startWalk.getObjectId(0);
- final FileMode mode = startWalk.getFileMode(0);
- editor.add(new PathEdit(startWalk.getPathString()) {
- public void apply(DirCacheEntry ent) {
- ent.setObjectId(blobId);
- ent.setFileMode(mode);
- try {
- DirCacheCheckout.checkoutEntry(repo, new File(
- workTree, ent.getPathString()), ent, r);
- } catch (IOException e) {
- throw new JGitInternalException(
- MessageFormat.format(
- JGitText.get().checkoutConflictWithFile,
- ent.getPathString()), e);
- }
- }
- });
- }
- editor.commit();
- } finally {
- startWalk.release();
- r.release();
- }
- } finally {
- dc.unlock();
- revWalk.release();
- }
- return this;
- }
-
- private ObjectId getStartPoint() throws AmbiguousObjectException,
- RefNotFoundException, IOException {
- if (startCommit != null)
- return startCommit.getId();
- ObjectId result = null;
- try {
- result = repo.resolve((startPoint == null) ? Constants.HEAD
- : startPoint);
- } catch (AmbiguousObjectException e) {
- throw e;
- }
- if (result == null)
- throw new RefNotFoundException(MessageFormat.format(
- JGitText.get().refNotResolved,
- startPoint != null ? startPoint : Constants.HEAD));
- return result;
- }
-
- private void processOptions() throws InvalidRefNameException {
- if ((!checkoutAllPaths && paths.isEmpty())
- && (name == null || !Repository
- .isValidRefName(Constants.R_HEADS + name)))
- throw new InvalidRefNameException(MessageFormat.format(JGitText
- .get().branchNameInvalid, name == null ? "<null>" : name));
- }
-
- /**
- * @param name
- * the name of the new branch
- * @return this instance
- */
- public CheckoutCommand setName(String name) {
- checkCallable();
- this.name = name;
- return this;
- }
-
- /**
- * @param createBranch
- * if <code>true</code> a branch will be created as part of the
- * checkout and set to the specified start point
- * @return this instance
- */
- public CheckoutCommand setCreateBranch(boolean createBranch) {
- checkCallable();
- this.createBranch = createBranch;
- return this;
- }
-
- /**
- * @param force
- * if <code>true</code> and the branch with the given name
- * already exists, the start-point of an existing branch will be
- * set to a new start-point; if false, the existing branch will
- * not be changed
- * @return this instance
- */
- public CheckoutCommand setForce(boolean force) {
- checkCallable();
- this.force = force;
- return this;
- }
-
- /**
- * @param startPoint
- * corresponds to the start-point option; if <code>null</code>,
- * the current HEAD will be used
- * @return this instance
- */
- public CheckoutCommand setStartPoint(String startPoint) {
- checkCallable();
- this.startPoint = startPoint;
- this.startCommit = null;
- return this;
- }
-
- /**
- * @param startCommit
- * corresponds to the start-point option; if <code>null</code>,
- * the current HEAD will be used
- * @return this instance
- */
- public CheckoutCommand setStartPoint(RevCommit startCommit) {
- checkCallable();
- this.startCommit = startCommit;
- this.startPoint = null;
- return this;
- }
-
- /**
- * @param mode
- * corresponds to the --track/--no-track options; may be
- * <code>null</code>
- * @return this instance
- */
- public CheckoutCommand setUpstreamMode(
- CreateBranchCommand.SetupUpstreamMode mode) {
- checkCallable();
- this.upstreamMode = mode;
- return this;
- }
-
- /**
- * @return the result
- */
- public CheckoutResult getResult() {
- if (status == null)
- return CheckoutResult.NOT_TRIED_RESULT;
- return status;
- }
- }
|