James Moger
2011-04-25 1e83907b001a2df3f14a35fc36c47ea8f49155c4
src/com/gitblit/utils/JGitUtils.java
@@ -4,7 +4,6 @@
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.RandomAccessFile;
import java.text.DateFormat;
import java.text.ParseException;
import java.text.SimpleDateFormat;
@@ -15,11 +14,13 @@
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.atomic.AtomicInteger;
import org.eclipse.jgit.api.Git;
import org.eclipse.jgit.diff.DiffEntry;
import org.eclipse.jgit.diff.DiffEntry.ChangeType;
import org.eclipse.jgit.diff.DiffFormatter;
import org.eclipse.jgit.diff.RawTextComparator;
import org.eclipse.jgit.diff.DiffEntry.ChangeType;
import org.eclipse.jgit.errors.ConfigInvalidException;
import org.eclipse.jgit.errors.IncorrectObjectTypeException;
import org.eclipse.jgit.errors.MissingObjectException;
@@ -36,6 +37,7 @@
import org.eclipse.jgit.revwalk.RevBlob;
import org.eclipse.jgit.revwalk.RevCommit;
import org.eclipse.jgit.revwalk.RevObject;
import org.eclipse.jgit.revwalk.RevSort;
import org.eclipse.jgit.revwalk.RevTree;
import org.eclipse.jgit.revwalk.RevWalk;
import org.eclipse.jgit.revwalk.filter.RevFilter;
@@ -64,6 +66,11 @@
   public static final String R_NOTES_COMMITS = R_NOTES + "commits";
   private final static Logger LOGGER = LoggerFactory.getLogger(JGitUtils.class);
   public static Repository createRepository(File repositoriesFolder, String name, boolean bare) {
      Git git = Git.init().setDirectory(new File(repositoriesFolder, name)).setBare(bare).call();
      return git.getRepository();
   }
   public static List<String> getRepositoryList(File repositoriesFolder, boolean exportAll, boolean readNested) {
      List<String> list = new ArrayList<String>();
@@ -78,8 +85,18 @@
      for (File file : folder.listFiles()) {
         if (file.isDirectory() && !file.getName().equalsIgnoreCase(Constants.DOT_GIT)) {
            // if this is a git repository add it to the list
            //
            // first look for standard folder/.git structure
            File gitFolder = new File(file, Constants.DOT_GIT);
            boolean isGitRepository = gitFolder.exists() && gitFolder.isDirectory();
            // then look for folder.git/HEAD
            if (!isGitRepository) {
               if (file.getName().endsWith(Constants.DOT_GIT_EXT) && new File(file, Constants.HEAD).exists()) {
                  gitFolder = file;
                  isGitRepository = true;
               }
            }
            boolean exportRepository = isGitRepository && (exportAll || new File(gitFolder, "git-daemon-export-ok").exists());
            if (exportRepository) {
@@ -101,8 +118,44 @@
      return list;
   }
   public static Date getLastChange(Repository r) {
   public static RevCommit getFirstCommit(Repository r, String branch) {
      if (StringUtils.isEmpty(branch)) {
         branch = Constants.HEAD;
      }
      try {
         RevWalk walk = new RevWalk(r);
         walk.sort(RevSort.REVERSE);
         RevCommit head = walk.parseCommit(r.resolve(branch));
         walk.markStart(head);
         RevCommit commit = walk.next();
         walk.dispose();
         return commit;
      } catch (Throwable t) {
         LOGGER.error("Failed to determine first commit", t);
      }
      return null;
   }
   public static Date getFirstChange(Repository r, String branch) {
      try {
         RevCommit commit = getFirstCommit(r, branch);
         if (commit == null) {
            // fresh repository
            return new Date(r.getDirectory().lastModified());
         }
         return getCommitDate(commit);
      } catch (Throwable t) {
         LOGGER.error("Failed to determine first change", t);
      }
      return null;
   }
   public static Date getLastChange(Repository r) {
      RevCommit commit = getCommit(r, Constants.HEAD);
      if (commit == null) {
         // fresh repository
         return new Date(r.getDirectory().lastModified());
      }
      return getCommitDate(commit);
   }
@@ -223,6 +276,9 @@
   public static List<PathModel> getFilesInPath(Repository r, String basePath, RevCommit commit) {
      List<PathModel> list = new ArrayList<PathModel>();
      if (commit == null) {
         return list;
      }
      final TreeWalk walk = new TreeWalk(r);
      try {
         walk.addTree(commit.getTree());
@@ -297,19 +353,43 @@
      return list;
   }
   public static String getCommitDiff(Repository r, RevCommit commit, boolean outputHtml) {
      return getCommitDiff(r, null, commit, null, outputHtml);
   public static Map<ChangeType, AtomicInteger> getChangedPathsStats(List<PathChangeModel> paths) {
      Map<ChangeType, AtomicInteger> stats = new HashMap<ChangeType, AtomicInteger>();
      for (PathChangeModel path : paths) {
         if (!stats.containsKey(path.changeType)) {
            stats.put(path.changeType, new AtomicInteger(0));
         }
         stats.get(path.changeType).incrementAndGet();
      }
      return stats;
   }
   public static String getCommitDiff(Repository r, RevCommit commit, String path, boolean outputHtml) {
      return getCommitDiff(r, null, commit, path, outputHtml);
   public static enum DiffOutputType {
      PLAIN, GITWEB, GITBLIT;
      public static DiffOutputType forName(String name) {
         for (DiffOutputType type : values()) {
            if (type.name().equalsIgnoreCase(name)) {
               return type;
            }
         }
         return null;
      }
   }
   public static String getCommitDiff(Repository r, RevCommit baseCommit, RevCommit commit, boolean outputHtml) {
      return getCommitDiff(r, baseCommit, commit, null, outputHtml);
   public static String getCommitDiff(Repository r, RevCommit commit, DiffOutputType outputType) {
      return getCommitDiff(r, null, commit, null, outputType);
   }
   public static String getCommitDiff(Repository r, RevCommit baseCommit, RevCommit commit, String path, boolean outputHtml) {
   public static String getCommitDiff(Repository r, RevCommit commit, String path, DiffOutputType outputType) {
      return getCommitDiff(r, null, commit, path, outputType);
   }
   public static String getCommitDiff(Repository r, RevCommit baseCommit, RevCommit commit, DiffOutputType outputType) {
      return getCommitDiff(r, baseCommit, commit, null, outputType);
   }
   public static String getCommitDiff(Repository r, RevCommit baseCommit, RevCommit commit, String path, DiffOutputType outputType) {
      try {
         RevTree baseTree;
         if (baseCommit == null) {
@@ -333,10 +413,17 @@
         final ByteArrayOutputStream os = new ByteArrayOutputStream();
         RawTextComparator cmp = RawTextComparator.DEFAULT;
         DiffFormatter df;
         if (outputHtml) {
            df = new HtmlDiffFormatter(os);
         } else {
         switch (outputType) {
         case GITWEB:
            df = new GitWebDiffFormatter(os);
            break;
         case GITBLIT:
            df = new GitBlitDiffFormatter(os);
            break;
         case PLAIN:
         default:
            df = new DiffFormatter(os);
            break;
         }
         df.setRepository(r);
         df.setDiffComparator(cmp);
@@ -353,9 +440,9 @@
            df.format(diffs);
         }
         String diff;
         if (outputHtml) {
         if (df instanceof GitWebDiffFormatter) {
            // workaround for complex private methods in DiffFormatter
            diff = ((HtmlDiffFormatter) df).getHtml();
            diff = ((GitWebDiffFormatter) df).getHtml();
         } else {
            diff = os.toString();
         }
@@ -525,7 +612,7 @@
         }
         return null;
      }
      public String toString() {
         return name().toLowerCase();
      }
@@ -552,7 +639,7 @@
               case AUTHOR:
                  return (commit.getAuthorIdent().getName().toLowerCase().indexOf(lcValue) > -1) || (commit.getAuthorIdent().getEmailAddress().toLowerCase().indexOf(lcValue) > -1);
               case COMMITTER:
                  return (commit.getCommitterIdent().getName().toLowerCase().indexOf(lcValue) > -1)|| (commit.getCommitterIdent().getEmailAddress().toLowerCase().indexOf(lcValue) > -1);
                  return (commit.getCommitterIdent().getName().toLowerCase().indexOf(lcValue) > -1) || (commit.getCommitterIdent().getEmailAddress().toLowerCase().indexOf(lcValue) > -1);
               case COMMIT:
                  return commit.getFullMessage().toLowerCase().indexOf(lcValue) > -1;
               }
@@ -647,35 +734,94 @@
   }
   public static String getRepositoryDescription(Repository r) {
      File dir = r.getDirectory();
      if (dir.exists()) {
         File description = new File(dir, "description");
         if (description.exists() && description.length() > 0) {
            RandomAccessFile raf = null;
            try {
               raf = new RandomAccessFile(description, "r");
               byte[] buffer = new byte[(int) description.length()];
               raf.readFully(buffer);
               return new String(buffer);
            } catch (Throwable t) {
            } finally {
               try {
                  raf.close();
               } catch (Throwable t) {
               }
            }
         }
      }
      return "";
      return getRepositoryConfigString(r, "description");
   }
   public static void setRepositoryDescription(Repository r, String value) {
      setRepositoryConfigString(r, "description", value);
   }
   public static String getRepositoryOwner(Repository r) {
      return getRepositoryConfigString(r, "owner");
   }
   public static void setRepositoryOwner(Repository r, String owner) {
      setRepositoryConfigString(r, "owner", owner);
   }
   public static String getRepositoryGroup(Repository r) {
      return getRepositoryConfigString(r, "group");
   }
   public static void setRepositoryGroup(Repository r, String group) {
      setRepositoryConfigString(r, "group", group);
   }
   public static boolean getRepositoryUseTickets(Repository r) {
      return getRepositoryConfigBoolean(r, "useTickets", false);
   }
   public static void setRepositoryUseTickets(Repository r, boolean value) {
      setRepositoryConfigBoolean(r, "useTickets", value);
   }
   public static boolean getRepositoryUseDocs(Repository r) {
      return getRepositoryConfigBoolean(r, "useDocs", false);
   }
   public static void setRepositoryUseDocs(Repository r, boolean value) {
      setRepositoryConfigBoolean(r, "useDocs", value);
   }
   public static boolean getRepositoryRestrictedAccess(Repository r) {
      return getRepositoryConfigBoolean(r, "restrictedAccess", false);
   }
   public static void setRepositoryRestrictedAccess(Repository r, boolean value) {
      setRepositoryConfigBoolean(r, "restrictedAccess", value);
   }
   public static String getRepositoryConfigString(Repository r, String field) {
      StoredConfig c = readConfig(r);
      if (c == null) {
         return "";
      }
      String o = c.getString("gitweb", null, "owner");
      String o = c.getString("gitblit", null, field);
      return o == null ? "" : o;
   }
   public static void setRepositoryConfigString(Repository r, String field, String value) {
      StoredConfig c = readConfig(r);
      if (c == null) {
         throw new RuntimeException("Can't find stored config for " + r);
      }
      c.setString("gitblit", null, field, value);
      try {
         c.save();
      } catch (IOException e) {
         LOGGER.error("Failed to save repository config field " + field, e);
      }
   }
   public static boolean getRepositoryConfigBoolean(Repository r, String field, boolean defaultValue) {
      StoredConfig c = readConfig(r);
      if (c == null) {
         return defaultValue;
      }
      return c.getBoolean("gitblit", null, field, defaultValue);
   }
   public static void setRepositoryConfigBoolean(Repository r, String field, boolean value) {
      StoredConfig c = readConfig(r);
      if (c == null) {
         throw new RuntimeException("Can't find stored config for " + r);
      }
      c.setBoolean("gitblit", null, field, value);
      try {
         c.save();
      } catch (IOException e) {
         LOGGER.error("Failed to save repository config field " + field, e);
      }
   }
   private static StoredConfig readConfig(Repository r) {
@@ -695,31 +841,63 @@
   public static List<Metric> getDateMetrics(Repository r) {
      final List<RefModel> tags = getTags(r, -1);
      final Map<String, Metric> map = new HashMap<String, Metric>();
      final Map<ObjectId, RefModel> tagMap = new HashMap<ObjectId, RefModel>();
      for (RefModel tag : tags) {
         tagMap.put(tag.getCommitId(), tag);
      }
      Metric total = new Metric("TOTAL");
      final Map<String, Metric> metricMap = new HashMap<String, Metric>();
      try {
         DateFormat df = new SimpleDateFormat("yyyy-MM");
         RevWalk walk = new RevWalk(r);
         ObjectId object = r.resolve(Constants.HEAD);
         walk.markStart(walk.parseCommit(object));
         RevCommit firstCommit = getFirstCommit(r, Constants.HEAD);
         RevCommit lastCommit = walk.parseCommit(object);
         int diffDays = (lastCommit.getCommitTime() - firstCommit.getCommitTime()) / (60 * 60 * 24);
         total.duration = diffDays;
         DateFormat df;
         if (diffDays <= 90) {
            // Days
            df = new SimpleDateFormat("yyyy-MM-dd");
         } else if (diffDays > 90 && diffDays < 365) {
            // Weeks
            df = new SimpleDateFormat("yyyy-MM (w)");
         } else {
            // Months
            df = new SimpleDateFormat("yyyy-MM");
         }
         walk.markStart(lastCommit);
         Iterable<RevCommit> revlog = walk;
         for (RevCommit rev : revlog) {
            Date d = getCommitDate(rev);
            String p = df.format(d);
            if (!map.containsKey(p))
               map.put(p, new Metric(p));
            map.get(p).count++;
            if (!metricMap.containsKey(p))
               metricMap.put(p, new Metric(p));
            Metric m = metricMap.get(p);
            m.count++;
            total.count++;
            if (tagMap.containsKey(rev.getId())) {
               m.tag++;
               total.tag++;
            }
         }
      } catch (Throwable t) {
         LOGGER.error("Failed to mine log history for metrics", t);
      }
      List<String> keys = new ArrayList<String>(map.keySet());
      List<String> keys = new ArrayList<String>(metricMap.keySet());
      Collections.sort(keys);
      List<Metric> metrics = new ArrayList<Metric>();
      for (String key : keys) {
         metrics.add(map.get(key));
         metrics.add(metricMap.get(key));
      }
      metrics.add(0, total);
      return metrics;
   }
   public static RefModel getDocumentsBranch(Repository r) {
      return getTicketsBranch(r);
   }
   public static RefModel getTicketsBranch(Repository r) {
      RefModel ticgitBranch = null;