From cdbbda53e245621065bf142e48737faf88a5fee7 Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Sat, 17 Mar 2012 11:00:17 -0400
Subject: [PATCH] Incremental commit indexing was missing branch field

---
 src/com/gitblit/wicket/pages/LucenePage.java |   60 ++++++++++++++++++++++++++++++++++++------------------------
 1 files changed, 36 insertions(+), 24 deletions(-)

diff --git a/src/com/gitblit/wicket/pages/LucenePage.java b/src/com/gitblit/wicket/pages/LucenePage.java
index dc74012..099471a 100644
--- a/src/com/gitblit/wicket/pages/LucenePage.java
+++ b/src/com/gitblit/wicket/pages/LucenePage.java
@@ -28,14 +28,16 @@
 import org.apache.wicket.markup.repeater.data.ListDataProvider;
 import org.apache.wicket.model.Model;
 import org.eclipse.jgit.lib.Constants;
-import org.eclipse.jgit.lib.Repository;
 
 import com.gitblit.Constants.SearchType;
 import com.gitblit.GitBlit;
+import com.gitblit.models.RepositoryModel;
 import com.gitblit.models.SearchResult;
+import com.gitblit.models.UserModel;
 import com.gitblit.utils.ArrayUtils;
-import com.gitblit.utils.LuceneUtils;
 import com.gitblit.utils.StringUtils;
+import com.gitblit.wicket.GitBlitWebSession;
+import com.gitblit.wicket.StringChoiceRenderer;
 import com.gitblit.wicket.WicketUtils;
 import com.gitblit.wicket.panels.LinkPanel;
 
@@ -86,9 +88,30 @@
 			}
 		}
 		
+		// display user-accessible selections
+		UserModel user = GitBlitWebSession.get().getUser();
+		List<String> availableRepositories = new ArrayList<String>();
+		for (RepositoryModel model : GitBlit.self().getRepositoryModels(user)) {
+			if (model.hasCommits && !ArrayUtils.isEmpty(model.indexedBranches)) {
+				availableRepositories.add(model.name);
+			}
+		}
+		
+		if (availableRepositories.size() == 0) {
+			info(getString("gb.noIndexedRepositoriesWarning"));
+		}
+
+		// enforce user-accessible repository selections
+		ArrayList<String> searchRepositories = new ArrayList<String>();
+		for (String selectedRepository : repositories) {
+			if (availableRepositories.contains(selectedRepository)) {
+				searchRepositories.add(selectedRepository);
+			}
+		}
+		
 		// search form
 		final Model<String> queryModel = new Model<String>(query);
-		final Model<ArrayList<String>> repositoriesModel = new Model<ArrayList<String>>(repositories);
+		final Model<ArrayList<String>> repositoriesModel = new Model<ArrayList<String>>(searchRepositories);
 		StatelessForm<Void> form = new StatelessForm<Void>("searchForm") {
 			
 			private static final long serialVersionUID = 1L;
@@ -97,11 +120,11 @@
 			public void onSubmit() {
 				String q = queryModel.getObject();
 				if (StringUtils.isEmpty(q)) {
-					error("Query is empty!");
+					error(getString("gb.undefinedQueryWarning"));
 					return;
 				}				
 				if (repositoriesModel.getObject().size() == 0) {
-					error("Please select one or more repositories!");
+					error(getString("gb.noSelectedRepositoriesWarning"));
 					return;
 				}
 				PageParameters params = new PageParameters();
@@ -110,7 +133,9 @@
 				setResponsePage(LucenePage.class, params);
 			}
 		};
-		ListMultipleChoice<String> selections = new ListMultipleChoice<String>("repositories", repositoriesModel, GitBlit.self().getRepositoryList());
+				
+		ListMultipleChoice<String> selections = new ListMultipleChoice<String>("repositories", 
+				repositoriesModel, availableRepositories, new StringChoiceRenderer());
 		selections.setMaxRows(10);
 		form.add(selections);
 		form.add(new TextField<String>("query", queryModel));
@@ -118,7 +143,9 @@
 				
 		// execute search
 		final List<SearchResult> results = new ArrayList<SearchResult>();
-		results.addAll(search(repositories, query));
+		if (!ArrayUtils.isEmpty(searchRepositories) && !StringUtils.isEmpty(query)) {
+			results.addAll(GitBlit.self().search(query, 100, searchRepositories));
+		}
 		
 		// search results view
 		ListDataProvider<SearchResult> resultsDp = new ListDataProvider<SearchResult>(results);
@@ -153,24 +180,9 @@
 				item.add(new LinkPanel("repository", null, sr.repository, SummaryPage.class, WicketUtils.newRepositoryParameter(sr.repository)));
 				item.add(new LinkPanel("branch", "branch", StringUtils.getRelativePath(Constants.R_HEADS, sr.branch), LogPage.class, WicketUtils.newObjectParameter(sr.repository, sr.branch)));
 				item.add(new Label("author", sr.author));
-				item.add(WicketUtils.createTimestampLabel("date", sr.date, getTimeZone()));
+				item.add(WicketUtils.createDatestampLabel("date", sr.date, getTimeZone()));
 			}
 		};
 		add(resultsView.setVisible(results.size() > 0));
-	}
-	
-	private List<SearchResult> search(List<String> repositories, String query) {
-		if (ArrayUtils.isEmpty(repositories) || StringUtils.isEmpty(query)) {
-			return new ArrayList<SearchResult>();
-		}
-		List<Repository> repos = new ArrayList<Repository>();
-		for (String r : repositories) {
-			repos.add(GitBlit.self().getRepository(r));
-		}
-		List<SearchResult> srs = LuceneUtils.search(query, 100, repos.toArray(new Repository[repos.size()]));
-		for (Repository r : repos) {
-			r.close();
-		}
-		return srs;
-	}
+	}	
 }

--
Gitblit v1.9.1