From 30f9d25d77ccb5cd978d4cf8fa389ec819e90e95 Mon Sep 17 00:00:00 2001
From: Philip L. McMahon <philip.l.mcmahon@gmail.com>
Date: Fri, 27 Jan 2012 02:02:19 -0500
Subject: [PATCH] Correct update of HEAD symbolic reference when target is a tag.

---
 src/com/gitblit/utils/JGitUtils.java |   60 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 60 insertions(+), 0 deletions(-)

diff --git a/src/com/gitblit/utils/JGitUtils.java b/src/com/gitblit/utils/JGitUtils.java
index a540c2a..05c0852 100644
--- a/src/com/gitblit/utils/JGitUtils.java
+++ b/src/com/gitblit/utils/JGitUtils.java
@@ -1156,6 +1156,66 @@
 	}
 
 	/**
+	 * Returns the default HEAD for a repository. Normally returns the ref HEAD points to, but if HEAD points to nothing
+	 * it returns null.
+	 *
+	 * @param repository
+	 * @return the refmodel for HEAD or null
+	 */
+	public static RefModel getDefaultHead(Repository repository) {
+		RefModel ref = null;
+		try {
+			Ref head = repository.getRef(Constants.HEAD);
+			if (head != null) {
+				Ref target = head.getTarget();
+				RevWalk rw = new RevWalk(repository);
+				ObjectId targetId = target.getObjectId();
+				if (targetId != null) {
+					RevObject object = rw.parseAny(targetId);
+					ref = new RefModel(target.getName(), target, object);
+				}
+				rw.dispose();
+			}
+		} catch (Throwable t) {
+			LOGGER.error("Failed to get default head!", t);
+		}
+		return ref;
+	}
+
+	/**
+	 * Sets the default HEAD symbolic ref for a repository.
+	 *
+	 * @param repository
+	 * @param ref
+	 */
+	public static void setDefaultHead(Repository repository, Ref ref) {
+		try {
+			boolean detach = !ref.getName().startsWith(Constants.R_HEADS); // detach if not a branch
+			RefUpdate.Result result;
+			RefUpdate head = repository.updateRef(Constants.HEAD, detach);
+			if (detach) { // Tag
+				RevCommit commit = getCommit(repository, ref.getObjectId().getName());
+				head.setNewObjectId(commit.getId());
+				result = head.forceUpdate();
+			} else {
+				result = head.link(ref.getName());
+			}
+			switch (result) {
+			case NEW:
+			case FORCED:
+			case NO_CHANGE:
+			case FAST_FORWARD:
+				break;
+			default:
+				LOGGER.error(MessageFormat.format("{0} failed to set default head to {1} ({2})",
+						repository.getDirectory().getAbsolutePath(), ref.getName(), result));
+			}
+		} catch (Throwable t) {
+			error(t, repository, "{0} failed to set default head to {1}", ref.getName());
+		}
+	}
+
+	/**
 	 * Returns all refs grouped by their associated object id.
 	 * 
 	 * @param repository

--
Gitblit v1.9.1