Jeroen Baten
2014-03-08 b01c943a26ba0eba27cc83a5740a6387a27a2151
src/main/java/com/gitblit/utils/RefLogUtils.java
@@ -66,19 +66,19 @@
/**
 * Utility class for maintaining a reflog within a git repository on an
 * orphan branch.
 *
 *
 * @author James Moger
 *
 */
public class RefLogUtils {
   private static final String GB_REFLOG = "refs/gitblit/reflog";
   private static final Logger LOGGER = LoggerFactory.getLogger(RefLogUtils.class);
   /**
    * Log an error message and exception.
    *
    *
    * @param t
    * @param repository
    *            if repository is not null it MUST be the {0} parameter in the
@@ -98,10 +98,10 @@
      }
      LOGGER.error(MessageFormat.format(pattern, parameters.toArray()), t);
   }
   /**
    * Returns true if the repository has a reflog branch.
    *
    *
    * @param repository
    * @return true if the repository has a reflog branch
    */
@@ -117,7 +117,7 @@
   /**
    * Returns a RefModel for the reflog branch in the repository. If the
    * branch can not be found, null is returned.
    *
    *
    * @param repository
    * @return a refmodel for the reflog branch or null
    */
@@ -152,7 +152,7 @@
      }
      return null;
   }
   private static UserModel newUserModelFrom(PersonIdent ident) {
      String name = ident.getName();
      String username;
@@ -165,47 +165,47 @@
         displayname = name;
         username = ident.getEmailAddress();
      }
      UserModel user = new UserModel(username);
      user.displayName = displayname;
      user.emailAddress = ident.getEmailAddress();
      return user;
   }
   /**
    * Logs a ref deletion.
    *
    *
    * @param user
    * @param repository
    * @param ref
    * @return true, if the update was successful
    */
   public static boolean deleteRef(UserModel user, Repository repository, String ref) {
   public static boolean deleteRef(UserModel user, Repository repository, Ref ref) {
      try {
         Ref refObj = repository.getRef(ref);
         if (refObj == null && !ref.startsWith(Constants.R_HEADS) && ref.startsWith(Constants.R_TAGS)) {
            // find fully qualified ref
            refObj = repository.getRef(Constants.R_HEADS + ref);
            if (refObj == null) {
               refObj = repository.getRef(Constants.R_TAGS + ref);
            }
         }
         if (refObj == null) {
         if (ref == null) {
            return false;
         }
         ReceiveCommand cmd = new ReceiveCommand(refObj.getObjectId(), ObjectId.zeroId(), refObj.getName());
         RefModel reflogBranch = getRefLogBranch(repository);
         if (reflogBranch == null) {
            return false;
         }
         List<RevCommit> log = JGitUtils.getRevLog(repository, reflogBranch.getName(), ref.getName(), 0, 1);
         if (log.isEmpty()) {
            // this ref is not in the reflog branch
            return false;
         }
         ReceiveCommand cmd = new ReceiveCommand(ref.getObjectId(), ObjectId.zeroId(), ref.getName());
         return updateRefLog(user, repository, Arrays.asList(cmd));
      } catch (Throwable t) {
         error(t, repository, "Failed to commit reflog entry to {0}");
      }
      return false;
   }
   /**
    * Updates the reflog with the received commands.
    *
    *
    * @param user
    * @param repository
    * @param commands
@@ -213,14 +213,30 @@
    */
   public static boolean updateRefLog(UserModel user, Repository repository,
         Collection<ReceiveCommand> commands) {
      // only track branches and tags
      List<ReceiveCommand> filteredCommands = new ArrayList<ReceiveCommand>();
      for (ReceiveCommand cmd : commands) {
         if (!cmd.getRefName().startsWith(Constants.R_HEADS)
               && !cmd.getRefName().startsWith(Constants.R_TAGS)) {
            continue;
         }
         filteredCommands.add(cmd);
      }
      if (filteredCommands.isEmpty()) {
         // nothing to log
         return true;
      }
      RefModel reflogBranch = getRefLogBranch(repository);
      if (reflogBranch == null) {
         JGitUtils.createOrphanBranch(repository, GB_REFLOG, null);
      }
      boolean success = false;
      String message = "push";
      try {
         ObjectId headId = repository.resolve(GB_REFLOG + "^{commit}");
         ObjectInserter odi = repository.newObjectInserter();
@@ -286,17 +302,17 @@
      }
      return success;
   }
   /**
    * Creates an in-memory index of the push log entry.
    *
    *
    * @param repo
    * @param headId
    * @param commands
    * @return an in-memory index
    * @throws IOException
    */
   private static DirCache createIndex(Repository repo, ObjectId headId,
   private static DirCache createIndex(Repository repo, ObjectId headId,
         Collection<ReceiveCommand> commands) throws IOException {
      DirCache inCoreIndex = DirCache.newInCore();
@@ -335,7 +351,7 @@
               continue;
            }
            String content = change.toString();
            // create an index entry for this attachment
            final DirCacheEntry dcEntry = new DirCacheEntry(path);
            dcEntry.setLength(content.length());
@@ -386,7 +402,7 @@
      }
      return inCoreIndex;
   }
   public static List<RefLogEntry> getRefLog(String repositoryName, Repository repository) {
      return getRefLog(repositoryName, repository, null, 0, -1);
   }
@@ -402,11 +418,11 @@
   public static List<RefLogEntry> getRefLog(String repositoryName, Repository repository, Date minimumDate) {
      return getRefLog(repositoryName, repository, minimumDate, 0, -1);
   }
   /**
    * Returns the list of reflog entries as they were recorded by Gitblit.
    * Each RefLogEntry may represent multiple ref updates.
    *
    *
    * @param repositoryName
    * @param repository
    * @param minimumDate
@@ -425,7 +441,7 @@
      if (maxCount == 0) {
         return list;
      }
      Map<ObjectId, List<RefModel>> allRefs = JGitUtils.getAllRefs(repository);
      List<RevCommit> pushes;
      if (minimumDate == null) {
@@ -441,29 +457,47 @@
         UserModel user = newUserModelFrom(push.getAuthorIdent());
         Date date = push.getAuthorIdent().getWhen();
         RefLogEntry log = new RefLogEntry(repositoryName, date, user);
         // only report HEADS and TAGS for now
         List<PathChangeModel> changedRefs = new ArrayList<PathChangeModel>();
         for (PathChangeModel refChange : JGitUtils.getFilesInCommit(repository, push)) {
            if (refChange.path.startsWith(Constants.R_HEADS)
                  || refChange.path.startsWith(Constants.R_TAGS)) {
               changedRefs.add(refChange);
            }
         }
         if (changedRefs.isEmpty()) {
            // skip empty commits
            continue;
         }
         list.add(log);
         List<PathChangeModel> changedRefs = JGitUtils.getFilesInCommit(repository, push);
         for (PathChangeModel change : changedRefs) {
            switch (change.changeType) {
            case DELETE:
               log.updateRef(change.path, ReceiveCommand.Type.DELETE);
               break;
            case ADD:
               log.updateRef(change.path, ReceiveCommand.Type.CREATE);
            default:
               String content = JGitUtils.getStringContent(repository, push.getTree(), change.path);
               String [] fields = content.split(" ");
               String oldId = fields[1];
               String newId = fields[2];
               log.updateRef(change.path, ReceiveCommand.Type.valueOf(fields[0]), oldId, newId);
               List<RevCommit> pushedCommits = JGitUtils.getRevLog(repository, oldId, newId);
               for (RevCommit pushedCommit : pushedCommits) {
                  RepositoryCommit repoCommit = log.addCommit(change.path, pushedCommit);
                  if (repoCommit != null) {
                     repoCommit.setRefs(allRefs.get(pushedCommit.getId()));
               if (ObjectId.zeroId().getName().equals(newId)) {
                  // ref deletion
                  continue;
               }
               try {
                  List<RevCommit> pushedCommits = JGitUtils.getRevLog(repository, oldId, newId);
                  for (RevCommit pushedCommit : pushedCommits) {
                     RepositoryCommit repoCommit = log.addCommit(change.path, pushedCommit);
                     if (repoCommit != null) {
                        repoCommit.setRefs(allRefs.get(pushedCommit.getId()));
                     }
                  }
               } catch (Exception e) {
               }
            }
         }
@@ -475,7 +509,7 @@
   /**
    * Returns the list of pushes separated by ref (e.g. each ref has it's own
    * PushLogEntry object).
    *
    *
    * @param repositoryName
    * @param repository
    * @param maxCount
@@ -484,11 +518,11 @@
   public static List<RefLogEntry> getLogByRef(String repositoryName, Repository repository, int maxCount) {
      return getLogByRef(repositoryName, repository, 0, maxCount);
   }
   /**
    * Returns the list of pushes separated by ref (e.g. each ref has it's own
    * PushLogEntry object).
    *
    *
    * @param repositoryName
    * @param repository
    * @param offset
@@ -505,7 +539,7 @@
            if (!refMap.containsKey(ref)) {
               refMap.put(ref, new ArrayList<RefLogEntry>());
            }
            // construct new ref-specific ref change entry
            RefLogEntry refChange;
            if (entry instanceof DailyLogEntry) {
@@ -520,23 +554,23 @@
            refMap.get(ref).add(refChange);
         }
      }
      // merge individual ref changes into master list
      List<RefLogEntry> mergedRefLog = new ArrayList<RefLogEntry>();
      for (List<RefLogEntry> refPush : refMap.values()) {
         mergedRefLog.addAll(refPush);
      }
      // sort ref log
      Collections.sort(mergedRefLog);
      return mergedRefLog;
   }
   /**
    * Returns the list of ref changes separated by ref (e.g. each ref has it's own
    * RefLogEntry object).
    *
    *
    * @param repositoryName
    * @param repository
    * @param minimumDate
@@ -559,16 +593,16 @@
            refMap.get(ref).add(refPush);
         }
      }
      // merge individual ref pushes into master list
      List<RefLogEntry> refPushLog = new ArrayList<RefLogEntry>();
      for (List<RefLogEntry> refPush : refMap.values()) {
         refPushLog.addAll(refPush);
      }
      // sort ref push log
      Collections.sort(refPushLog);
      return refPushLog;
   }
@@ -623,18 +657,18 @@
                   linearParent = commit.getParents()[0].getId().getName();
                   digest.updateRef(branch, ReceiveCommand.Type.UPDATE, linearParent, commit.getName());
                }
                RepositoryCommit repoCommit = digest.addCommit(commit);
                if (repoCommit != null) {
                   List<RefModel> matchedRefs = allRefs.get(commit.getId());
                    repoCommit.setRefs(matchedRefs);
                    if (!ArrayUtils.isEmpty(matchedRefs)) {
                        for (RefModel ref : matchedRefs) {
                            if (ref.getName().startsWith(Constants.R_TAGS)) {
                                // treat tags as special events in the log
                                if (!tags.containsKey(dateStr)) {
                                 UserModel tagUser = newUserModelFrom(commit.getAuthorIdent());
                                 UserModel tagUser = newUserModelFrom(ref.getAuthorIdent());
                                 Date tagDate = commit.getAuthorIdent().getWhen();
                                 tags.put(dateStr, new DailyLogEntry(repositoryName, tagDate, tagUser));
                                }
@@ -645,7 +679,7 @@
                            } else if (ref.getName().startsWith(Constants.R_PULL)) {
                                // treat pull requests as special events in the log
                                if (!pulls.containsKey(dateStr)) {
                                 UserModel commitUser = newUserModelFrom(commit.getAuthorIdent());
                                 UserModel commitUser = newUserModelFrom(ref.getAuthorIdent());
                                 Date commitDate = commit.getAuthorIdent().getWhen();
                                 pulls.put(dateStr, new DailyLogEntry(repositoryName, commitDate, commitUser));
                                }