From 4fcac9d2cbdafb51e3ee9ca3b3da64fd86103174 Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Tue, 26 Nov 2013 15:58:15 -0500
Subject: [PATCH] Remove artifact setting from manager dialog

---
 src/main/java/com/gitblit/AddIndexedBranch.java |   43 ++++++++++++++++++++++++++++++-------------
 1 files changed, 30 insertions(+), 13 deletions(-)

diff --git a/src/main/java/com/gitblit/AddIndexedBranch.java b/src/main/java/com/gitblit/AddIndexedBranch.java
index e2f3dac..4b668c4 100644
--- a/src/main/java/com/gitblit/AddIndexedBranch.java
+++ b/src/main/java/com/gitblit/AddIndexedBranch.java
@@ -33,15 +33,16 @@
 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 {
 
@@ -55,17 +56,17 @@
 			jc.usage();
 			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) {
@@ -76,24 +77,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);
 				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) {
@@ -108,11 +122,11 @@
 				e.printStackTrace();
 			}
 		}
-		
+
 		System.out.println(MessageFormat.format("updated {0} repository configurations, skipped {1}", modCount, skipCount));
 	}
 
-	
+
 
 	/**
 	 * JCommander Parameters class for AddIndexedBranch.
@@ -123,10 +137,13 @@
 		@Parameter(names = { "--repositoriesFolder" }, description = "The root repositories folder ", required = true)
 		public String folder;
 
-		@Parameter(names = { "--branch" }, description = "The branch to index", required = true)
+		@Parameter(names = { "--branch" }, description = "The branch to index", required = false)
 		public String branch = "default";
 
 		@Parameter(names = { "--skip" }, description = "Skip the named repository (simple fizzy matching is supported)", required = false)
 		public List<String> exclusions = new ArrayList<String>();
+
+		@Parameter(names = { "--all-local-branches" }, description = "Add all local branches to index. If specified, the --branch parameter is not considered.", required = false)
+		public boolean addAllLocalBranches = false;
 	}
 }

--
Gitblit v1.9.1