Owner editing. Frozen status. Grouped repositories. Documentation.

James Moger

2011-05-19 00afd77a2182ece3d9522d41b86b4ddd7e132288

 src/com/gitblit/utils/JGitUtils.java

@@ -4,22 +4,25 @@
import java.io.File;

import java.io.IOException;

import java.io.InputStream;

import java.io.RandomAccessFile;

import java.nio.charset.Charset;

import java.text.DateFormat;

import java.text.ParseException;

import java.text.SimpleDateFormat;

import java.util.ArrayList;

import java.util.Collection;

import java.util.Collections;

import java.util.Date;

import java.util.HashMap;

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;

@@ -42,8 +45,10 @@
import org.eclipse.jgit.revwalk.filter.RevFilter;

import org.eclipse.jgit.treewalk.TreeWalk;

import org.eclipse.jgit.treewalk.filter.AndTreeFilter;

import org.eclipse.jgit.treewalk.filter.OrTreeFilter;

import org.eclipse.jgit.treewalk.filter.PathFilter;

import org.eclipse.jgit.treewalk.filter.PathFilterGroup;

import org.eclipse.jgit.treewalk.filter.PathSuffixFilter;

import org.eclipse.jgit.treewalk.filter.TreeFilter;

import org.eclipse.jgit.util.io.DisabledOutputStream;

import org.slf4j.Logger;

@@ -58,13 +63,12 @@


public class JGitUtils {



   /** Prefix for notes refs */

   public static final String R_NOTES = "refs/notes/";



   /** Standard notes ref */

   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>();

@@ -76,11 +80,25 @@
   public static List<String> getNestedRepositories(File repositoriesFolder, File folder, boolean exportAll, boolean readNested) {

      String basefile = repositoriesFolder.getAbsolutePath();

      List<String> list = new ArrayList<String>();

      if (folder == null || !folder.exists()) {

         return list;

      }

      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 or folder/HEAD and

            // folder/config

            if (!isGitRepository) {

               if ((file.getName().endsWith(Constants.DOT_GIT_EXT) && new File(file, Constants.HEAD).exists()) || (new File(file, "config").exists() && new File(file, Constants.HEAD).exists())) {

                  gitFolder = file;

                  isGitRepository = true;

               }

            }

            boolean exportRepository = isGitRepository && (exportAll || new File(gitFolder, "git-daemon-export-ok").exists());



            if (exportRepository) {

@@ -103,6 +121,9 @@
   }



   public static RevCommit getFirstCommit(Repository r, String branch) {

      if (!hasCommits(r)) {

         return null;

      }

      if (StringUtils.isEmpty(branch)) {

         branch = Constants.HEAD;

      }

@@ -123,6 +144,10 @@
   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);

@@ -130,13 +155,24 @@
      return null;

   }



   public static boolean hasCommits(Repository r) {

      return new File(r.getDirectory(), Constants.R_HEADS).list().length > 0;

   }



   public static Date getLastChange(Repository r) {

      if (!hasCommits(r)) {

         // fresh repository

         return new Date(r.getDirectory().lastModified());

      }

      RevCommit commit = getCommit(r, Constants.HEAD);

      return getCommitDate(commit);

   }



   public static RevCommit getCommit(Repository r, String objectId) {

      RevCommit commit = null;

      if (!hasCommits(r)) {

         return null;

      }

      try {

         if (objectId == null || objectId.trim().length() == 0) {

            objectId = Constants.HEAD;

@@ -147,7 +183,7 @@
         commit = rev;

         walk.dispose();

      } catch (Throwable t) {

         LOGGER.error("Failed to determine last change", t);

         LOGGER.error("Failed to get commit " + objectId, t);

      }

      return commit;

   }

@@ -237,12 +273,14 @@
   }



   public static String getRawContentAsString(Repository r, RevBlob blob) {

      return new String(getRawContent(r, blob));

      byte[] content = getRawContent(r, blob);

      return new String(content, Charset.forName(Constants.CHARACTER_ENCODING));

   }



   public static String getRawContentAsString(Repository r, RevCommit commit, String blobPath) {

      RevObject obj = getRevObject(r, commit.getTree(), blobPath);

      return new String(getRawContent(r, (RevBlob) obj));

      byte[] content = getRawContent(r, (RevBlob) obj);

      return new String(content, Charset.forName(Constants.CHARACTER_ENCODING));

   }



   public static List<PathModel> getFilesInPath(Repository r, String basePath, String objectId) {

@@ -252,6 +290,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());

@@ -294,6 +335,10 @@


   public static List<PathChangeModel> getFilesInCommit(Repository r, RevCommit commit) {

      List<PathChangeModel> list = new ArrayList<PathChangeModel>();

      if (commit == null) {

         LOGGER.warn("getFilesInCommit for NULL commit");

         return list;

      }

      try {

         final RevWalk rw = new RevWalk(r);

         RevCommit parent = rw.parseCommit(commit.getParent(0).getId());

@@ -324,6 +369,53 @@
         LOGGER.error("failed to determine files in commit!", t);

      }

      return list;

   }



   public static List<PathModel> getDocuments(Repository r, List<String> extensions) {

      List<PathModel> list = new ArrayList<PathModel>();

      RevCommit commit = getCommit(r, Constants.HEAD);

      final TreeWalk walk = new TreeWalk(r);

      try {

         walk.addTree(commit.getTree());

         if (extensions != null && extensions.size() > 0) {

            Collection<TreeFilter> suffixFilters = new ArrayList<TreeFilter>();

            for (String extension : extensions) {

               if (extension.charAt(0) == '.') {

                  suffixFilters.add(PathSuffixFilter.create(extension));

               } else {

                  // escape the . since this is a regexp filter

                  suffixFilters.add(PathSuffixFilter.create("\\." + extension));

               }

            }

            TreeFilter filter = OrTreeFilter.create(suffixFilters);

            walk.setFilter(filter);

            walk.setRecursive(true);

            while (walk.next()) {

               list.add(getPathModel(walk, null, commit));

            }

         } else {

            while (walk.next()) {

               list.add(getPathModel(walk, null, commit));

            }

         }

      } catch (IOException e) {

         LOGGER.error("Failed to get files for commit " + commit.getName(), e);

      } finally {

         walk.release();

      }

      Collections.sort(list);

      return list;

   }



   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 enum DiffOutputType {

@@ -525,6 +617,9 @@


   public static List<RevCommit> getRevLog(Repository r, String objectId, String path, int offset, int maxCount) {

      List<RevCommit> list = new ArrayList<RevCommit>();

      if (!hasCommits(r)) {

         return list;

      }

      try {

         if (objectId == null || objectId.trim().length() == 0) {

            objectId = Constants.HEAD;

@@ -583,6 +678,9 @@
   public static List<RevCommit> searchRevlogs(Repository r, String objectId, String value, final SearchType type, int offset, int maxCount) {

      final String lcValue = value.toLowerCase();

      List<RevCommit> list = new ArrayList<RevCommit>();

      if (!hasCommits(r)) {

         return list;

      }

      try {

         if (objectId == null || objectId.trim().length() == 0) {

            objectId = Constants.HEAD;

@@ -695,39 +793,7 @@
      return r.toString();

   }



   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 "";

   }



   public static String getRepositoryOwner(Repository r) {

      StoredConfig c = readConfig(r);

      if (c == null) {

         return "";

      }

      String o = c.getString("gitweb", null, "owner");

      return o == null ? "" : o;

   }



   private static StoredConfig readConfig(Repository r) {

   public static StoredConfig readConfig(Repository r) {

      StoredConfig c = r.getConfig();

      if (c != null) {

         try {

@@ -743,50 +809,53 @@
   }



   public static List<Metric> getDateMetrics(Repository r) {

      final List<RefModel> tags = getTags(r, -1);

      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 {

         RevWalk walk = new RevWalk(r);

         ObjectId object = r.resolve(Constants.HEAD);



         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");

		
      if (hasCommits(r)) {

         final List<RefModel> tags = getTags(r, -1);

         final Map<ObjectId, RefModel> tagMap = new HashMap<ObjectId, RefModel>();

         for (RefModel tag : tags) {

            tagMap.put(tag.getCommitId(), tag);

         }

         walk.markStart(lastCommit);

         try {

            RevWalk walk = new RevWalk(r);

            ObjectId object = r.resolve(Constants.HEAD);



         Iterable<RevCommit> revlog = walk;

         for (RevCommit rev : revlog) {

            Date d = getCommitDate(rev);

            String p = df.format(d);

            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++;

            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 (!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);

         }

      } catch (Throwable t) {

         LOGGER.error("Failed to mine log history for metrics", t);

      }

      List<String> keys = new ArrayList<String>(metricMap.keySet());

      Collections.sort(keys);