James Moger
2014-05-24 8f50853cf767c869d63f39c23073b6beb7e0593d
src/main/java/com/gitblit/AddIndexedBranch.java
@@ -23,48 +23,49 @@
import java.util.Set;
import java.util.TreeSet;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.lib.RepositoryCache.FileKey;
import org.eclipse.jgit.storage.file.FileBasedConfig;
import org.eclipse.jgit.storage.file.FileRepository;
import org.eclipse.jgit.lib.StoredConfig;
import org.eclipse.jgit.storage.file.FileRepositoryBuilder;
import org.eclipse.jgit.util.FS;
import org.kohsuke.args4j.CmdLineException;
import org.kohsuke.args4j.CmdLineParser;
import org.kohsuke.args4j.Option;
import com.beust.jcommander.JCommander;
import com.beust.jcommander.Parameter;
import com.beust.jcommander.ParameterException;
import com.beust.jcommander.Parameters;
import com.gitblit.models.RefModel;
import com.gitblit.utils.ArrayUtils;
import com.gitblit.utils.JGitUtils;
import com.gitblit.utils.StringUtils;
/**
 * Utility class to add an indexBranch setting to matching repositories.
 *
 *
 * @author James Moger
 *
 *
 */
public class AddIndexedBranch {
   public static void main(String... args) {
      Params params = new Params();
      JCommander jc = new JCommander(params);
      CmdLineParser parser = new CmdLineParser(params);
      try {
         jc.parse(args);
      } catch (ParameterException t) {
         parser.parseArgument(args);
      } catch (CmdLineException t) {
         System.err.println(t.getMessage());
         jc.usage();
         parser.printUsage(System.out);
         return;
      }
      // create a lowercase set of excluded repositories
      Set<String> exclusions = new TreeSet<String>();
      for (String exclude : params.exclusions) {
         exclusions.add(exclude.toLowerCase());
      }
      // determine available repositories
      File folder = new File(params.folder);
      List<String> repoList = JGitUtils.getRepositoryList(folder, false, true, -1, null);
      int modCount = 0;
      int skipCount = 0;
      for (String repo : repoList) {
@@ -75,24 +76,37 @@
               break;
            }
         }
         if (skip) {
            System.out.println("skipping " + repo);
            skipCount++;
            continue;
         }
         System.out.println(MessageFormat.format("adding [gitblit] indexBranch={0} for {1}", params.branch, repo));
         try {
            // load repository config
            File gitDir = FileKey.resolve(new File(folder, repo), FS.DETECTED);
            FileRepository repository = new FileRepository(gitDir);
            FileBasedConfig config = repository.getConfig();
            Repository repository = new FileRepositoryBuilder().setGitDir(gitDir).build();
            StoredConfig config = repository.getConfig();
            config.load();
            Set<String> indexedBranches = new LinkedHashSet<String>();
            indexedBranches.add(Constants.DEFAULT_BRANCH);
            // add all local branches to index
            if (params.addAllLocalBranches) {
               List<RefModel> list = JGitUtils.getLocalBranches(repository, true, -1);
               for (RefModel refModel : list) {
                  System.out.println(MessageFormat.format("adding [gitblit] indexBranch={0} for {1}", refModel.getName(), repo));
                  indexedBranches.add(refModel.getName());
               }
            }
            else {
               // add only one branch to index ('default' if not specified)
               System.out.println(MessageFormat.format("adding [gitblit] indexBranch={0} for {1}", params.branch, repo));
               indexedBranches.add(params.branch);
            }
            String [] branches = config.getStringList("gitblit", null, "indexBranch");
            if (!ArrayUtils.isEmpty(branches)) {
               for (String branch : branches) {
@@ -107,25 +121,27 @@
            e.printStackTrace();
         }
      }
      System.out.println(MessageFormat.format("updated {0} repository configurations, skipped {1}", modCount, skipCount));
   }
   /**
    * JCommander Parameters class for AddIndexedBranch.
    * Parameters class for AddIndexedBranch.
    */
   @Parameters(separators = " ")
   private static class Params {
      @Parameter(names = { "--repositoriesFolder" }, description = "The root repositories folder ", required = true)
      @Option(name =  "--repositoriesFolder", usage = "The root repositories folder ", required = true, metaVar = "PATH")
      public String folder;
      @Parameter(names = { "--branch" }, description = "The branch to index", required = true)
      @Option(name = "--branch", usage = "The branch to index", metaVar = "BRANCH")
      public String branch = "default";
      @Parameter(names = { "--skip" }, description = "Skip the named repository (simple fizzy matching is supported)", required = false)
      @Option(name = "--skip", usage = "Skip the named repository (simple fizzy matching is supported)")
      public List<String> exclusions = new ArrayList<String>();
      @Option(name = "--all-local-branches", usage = "Add all local branches to index. If specified, the --branch parameter is not considered.")
      public boolean addAllLocalBranches;
   }
}