Simplified links some more. Added TicGit browser.

James Moger

2011-04-05 232890f03476b8bb231d17883eb0faff93ec5049

 src/com/gitblit/utils/JGitUtils.java

@@ -5,6 +5,9 @@
import java.io.IOException;

import java.io.InputStream;

import java.io.RandomAccessFile;

import java.text.DateFormat;

import java.text.ParseException;

import java.text.SimpleDateFormat;

import java.util.ArrayList;

import java.util.Collections;

import java.util.Date;

@@ -35,9 +38,10 @@
import org.slf4j.Logger;

import org.slf4j.LoggerFactory;



import com.gitblit.utils.TicGitTicket.Comment;

import com.gitblit.wicket.models.Metric;

import com.gitblit.wicket.models.PathModel;

import com.gitblit.wicket.models.RefModel;





public class JGitUtils {



@@ -204,7 +208,7 @@


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

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

      final TreeWalk walk = new TreeWalk(r);		
      final TreeWalk walk = new TreeWalk(r);

      try {

         walk.addTree(commit.getTree());

         if (basePath != null && basePath.length() > 0) {

@@ -214,7 +218,7 @@
            boolean foundFolder = false;

            while (walk.next()) {

               if (!foundFolder && walk.isSubtree()) {

                  walk.enterSubtree();						
                  walk.enterSubtree();

               }

               if (walk.getPathString().equals(basePath)) {

                  foundFolder = true;

@@ -264,7 +268,7 @@
      return list;

   }



   private static PathModel getPathModel(TreeWalk walk, String basePath, RevCommit commit) {		
   private static PathModel getPathModel(TreeWalk walk, String basePath, RevCommit commit) {

      String name;

      long size = 0;

      if (basePath == null) {

@@ -281,7 +285,7 @@
            size = walk.getObjectReader().getObjectSize(walk.getObjectId(0), Constants.OBJ_BLOB);

         }

      } catch (Throwable t) {

         LOGGER.error("Failed to retrieve blobl size", t);

         LOGGER.error("Failed to retrieve blob size", t);

      }

      return new PathModel(name, walk.getPathString(), size, walk.getFileMode(0).getBits(), commit.getName());

   }

@@ -310,7 +314,6 @@
      return FileMode.TREE.equals(mode);

   }



	
   public static List<RevCommit> getRevLog(Repository r, int maxCount) {

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

      try {

@@ -332,8 +335,12 @@
      return getRefs(r, Constants.R_TAGS, maxCount);

   }



   public static List<RefModel> getHeads(Repository r, int maxCount) {

   public static List<RefModel> getLocalBranches(Repository r, int maxCount) {

      return getRefs(r, Constants.R_HEADS, maxCount);

   }



   public static List<RefModel> getRemoteBranches(Repository r, int maxCount) {

      return getRefs(r, Constants.R_REMOTES, maxCount);

   }



   public static List<RefModel> getRefs(Repository r, String refs, int maxCount) {

@@ -348,7 +355,7 @@
         Collections.sort(list);

         Collections.reverse(list);

         if (maxCount > 0 && list.size() > maxCount) {

            list = list.subList(0, maxCount);

            list = new ArrayList<RefModel>(list.subList(0, maxCount));

         }

      } catch (IOException e) {

         LOGGER.error("Failed to retrieve " + refs, e);

@@ -427,4 +434,130 @@
      }

      return null;

   }



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

      final Map<String, Metric> map = new HashMap<String, Metric>();

      try {

         DateFormat df = new SimpleDateFormat("yyyy-MM");

         Git git = new Git(r);

         Iterable<RevCommit> revlog = git.log().call();

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

         }

      } catch (Throwable t) {

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

      }

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

      Collections.sort(keys);

      List<Metric> metrics = new ArrayList<Metric>();

      for (String key : keys) {

         metrics.add(map.get(key));

      }

      return metrics;

   }



   public static RefModel getTicGitBranch(Repository r) {

      RefModel ticgitBranch = null;

      try {

         // search for ticgit branch in local heads

         for (RefModel ref : getLocalBranches(r, -1)) {

            if (ref.getDisplayName().endsWith("ticgit") || ref.getDisplayName().endsWith("ticgit-ng")) {

               ticgitBranch = ref;

               break;

            }

         }



         // search for ticgit branch in remote heads

         if (ticgitBranch == null) {

            for (RefModel ref : getRemoteBranches(r, -1)) {

               if (ref.getDisplayName().endsWith("ticgit") || ref.getDisplayName().endsWith("ticgit-ng")) {

                  ticgitBranch = ref;

                  break;

               }

            }

         }

      } catch (Throwable t) {

         LOGGER.error("Failed to find ticgit branch!", t);

      }

      return ticgitBranch;

   }



   public static List<TicGitTicket> getTicGitTickets(Repository r) {

      RefModel ticgitBranch = getTicGitBranch(r);

      List<PathModel> paths = getFilesInPath(r, null, ticgitBranch.getCommit());

      List<TicGitTicket> tickets = new ArrayList<TicGitTicket>();

      for (PathModel ticketFolder : paths) {

         if (ticketFolder.isTree()) {

            try {

               TicGitTicket t = new TicGitTicket(ticketFolder.name);

               readTicketContents(r, ticgitBranch, t);

               tickets.add(t);

            } catch (Throwable t) {

               LOGGER.error("Failed to get a ticgit ticket!", t);

            }

         }

      }

      Collections.sort(tickets);

      Collections.reverse(tickets);

      return tickets;

   }



   public static TicGitTicket getTicGitTicket(Repository r, String ticketFolder) {

      RefModel ticgitBranch = getTicGitBranch(r);

      if (ticgitBranch != null) {

         try {

            TicGitTicket ticket = new TicGitTicket(ticketFolder);

            readTicketContents(r, ticgitBranch, ticket);

            return ticket;

         } catch (Throwable t) {

            LOGGER.error("Failed to get ticgit ticket " + ticketFolder, t);

         }

      }

      return null;

   }

	
   private static void readTicketContents(Repository r, RefModel ticgitBranch, TicGitTicket ticket) {

      List<PathModel> ticketFiles = getFilesInPath(r, ticket.name, ticgitBranch.getCommit());

      for (PathModel file : ticketFiles) {

         String content = getRawContentAsString(r, ticgitBranch.getCommit(), file.path).trim();

         if (file.name.equals("TICKET_ID")) {

            ticket.id = content;

         } else if (file.name.equals("TITLE")) {

            ticket.title = content;

         } else {

            String[] chunks = file.name.split("_");

            if (chunks[0].equals("ASSIGNED")) {

               ticket.handler = content;

            } else if (chunks[0].equals("COMMENT")) {

               try {

                  Comment c = new Comment(file.name, content);

                  ticket.comments.add(c);

               } catch (ParseException e) {

                  e.printStackTrace();

               }

            } else if (chunks[0].equals("TAG")) {

               if (content.startsWith("TAG_")) {

                  ticket.tags.add(content.substring(4));

               } else {

                  ticket.tags.add(content);

               }

            } else if (chunks[0].equals("STATE")) {

               ticket.state = content;

            }

         }

      }

      Collections.sort(ticket.comments);

   }



   public static String getTicGitContent(Repository r, String filePath) {

      RefModel ticgitBranch = getTicGitBranch(r);

      if (ticgitBranch != null) {

         return getRawContentAsString(r, ticgitBranch.getCommit(), filePath);

      }

      return "";

   }

}