Add recursion depth control for repository search (issue 103)

James Moger

2012-08-01 65f55eee1c41a9cdfec96dc4310efcc6e57df97f

 src/com/gitblit/utils/JGitUtils.java

@@ -35,6 +35,7 @@
import org.eclipse.jgit.api.CloneCommand;

import org.eclipse.jgit.api.FetchCommand;

import org.eclipse.jgit.api.Git;

import org.eclipse.jgit.api.errors.GitAPIException;

import org.eclipse.jgit.diff.DiffEntry;

import org.eclipse.jgit.diff.DiffEntry.ChangeType;

import org.eclipse.jgit.diff.DiffFormatter;

@@ -257,8 +258,12 @@
    * @return Repository

    */

   public static Repository createRepository(File repositoriesFolder, String name) {

      Git git = Git.init().setDirectory(new File(repositoriesFolder, name)).setBare(true).call();

      return git.getRepository();

      try {

         Git git = Git.init().setDirectory(new File(repositoriesFolder, name)).setBare(true).call();

         return git.getRepository();

      } catch (GitAPIException e) {

         throw new RuntimeException(e);

      }

   }



   /**

@@ -270,16 +275,18 @@
    *            false all repositories are included.

    * @param searchSubfolders

    *            recurse into subfolders to find grouped repositories

    * @param depth

    *            optional recursion depth, -1 = infinite recursion

    * @return list of repository names

    */

   public static List<String> getRepositoryList(File repositoriesFolder, boolean onlyBare,

         boolean searchSubfolders) {

         boolean searchSubfolders, int depth) {

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

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

         return list;

      }

      list.addAll(getRepositoryList(repositoriesFolder.getAbsolutePath(), repositoriesFolder,

            onlyBare, searchSubfolders));

            onlyBare, searchSubfolders, depth));

      StringUtils.sortRepositorynames(list);

      return list;

   }

@@ -296,12 +303,18 @@
    *            repositories are included.

    * @param searchSubfolders

    *            recurse into subfolders to find grouped repositories

    * @param depth

    *            recursion depth, -1 = infinite recursion

    * @return

    */

   private static List<String> getRepositoryList(String basePath, File searchFolder,

         boolean onlyBare, boolean searchSubfolders) {

         boolean onlyBare, boolean searchSubfolders, int depth) {

      File baseFile = new File(basePath);

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

      if (depth == 0) {

         return list;

      }

      int nextDepth = (depth == -1) ? -1 : depth - 1;

      for (File file : searchFolder.listFiles()) {

         if (file.isDirectory()) {

            File gitDir = FileKey.resolve(new File(searchFolder, file.getName()), FS.DETECTED);

@@ -309,12 +322,17 @@
               if (onlyBare && gitDir.getName().equals(".git")) {

                  continue;

               }

               // determine repository name relative to base path

               String repository = FileUtils.getRelativePath(baseFile, file);

               list.add(repository);

               if (gitDir.equals(file) || gitDir.getParentFile().equals(file)) {

                  // determine repository name relative to base path

                  String repository = FileUtils.getRelativePath(baseFile, file);

                  list.add(repository);

               } else if (searchSubfolders && file.canRead()) {

                  // look for repositories in subfolders

                  list.addAll(getRepositoryList(basePath, file, onlyBare, searchSubfolders, nextDepth));

               }

            } else if (searchSubfolders && file.canRead()) {

               // look for repositories in subfolders

               list.addAll(getRepositoryList(basePath, file, onlyBare, searchSubfolders));

               list.addAll(getRepositoryList(basePath, file, onlyBare, searchSubfolders, nextDepth));

            }

         }

      }