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 |  172 +++++++++++++++++++++++++++++++-------------------------
 1 files changed, 95 insertions(+), 77 deletions(-)

diff --git a/src/com/gitblit/wicket/pages/LucenePage.java b/src/com/gitblit/wicket/pages/LucenePage.java
index f72aeda..099471a 100644
--- a/src/com/gitblit/wicket/pages/LucenePage.java
+++ b/src/com/gitblit/wicket/pages/LucenePage.java
@@ -28,13 +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.utils.LuceneUtils;
+import com.gitblit.models.UserModel;
+import com.gitblit.utils.ArrayUtils;
 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;
 
@@ -53,18 +56,98 @@
 	private void setup(PageParameters params) {
 		setupPage("", "");
 		
-		String repository = null;		
-		String value = null;
-		com.gitblit.Constants.SearchType searchType = null;
+		// default values
+		ArrayList<String> repositories = new ArrayList<String>();				
+		String query = "";
 
 		if (params != null) {
-			repository = WicketUtils.getRepositoryName(params);
-			value = WicketUtils.getSearchString(params);
-			String type = WicketUtils.getSearchType(params);
-			searchType = com.gitblit.Constants.SearchType.forName(type);
+			String repository = WicketUtils.getRepositoryName(params);
+			if (!StringUtils.isEmpty(repository)) {
+				repositories.add(repository);
+			}
+			
+			if (params.containsKey("repositories")) {
+				String value = params.getString("repositories", "");
+				List<String> list = StringUtils.getStringsFromValue(value);			
+				repositories.addAll(list);
+			}
+
+			if (params.containsKey("query")) {
+				query = params.getString("query", "");	
+			} else {
+				String value = WicketUtils.getSearchString(params);
+				String type = WicketUtils.getSearchType(params);
+				com.gitblit.Constants.SearchType searchType = com.gitblit.Constants.SearchType.forName(type);
+				if (!StringUtils.isEmpty(value)) {
+					if (searchType == SearchType.COMMIT) {
+						query = "type:" + searchType.name().toLowerCase() + " AND \"" + value + "\"";	
+					} else {
+						query = searchType.name().toLowerCase() + ":\"" + value + "\"";
+					}
+				}
+			}
 		}
 		
+		// 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>>(searchRepositories);
+		StatelessForm<Void> form = new StatelessForm<Void>("searchForm") {
+			
+			private static final long serialVersionUID = 1L;
+
+			@Override
+			public void onSubmit() {
+				String q = queryModel.getObject();
+				if (StringUtils.isEmpty(q)) {
+					error(getString("gb.undefinedQueryWarning"));
+					return;
+				}				
+				if (repositoriesModel.getObject().size() == 0) {
+					error(getString("gb.noSelectedRepositoriesWarning"));
+					return;
+				}
+				PageParameters params = new PageParameters();
+				params.put("repositories", StringUtils.flattenStrings(repositoriesModel.getObject()));
+				params.put("query", queryModel.getObject());
+				setResponsePage(LucenePage.class, params);
+			}
+		};
+				
+		ListMultipleChoice<String> selections = new ListMultipleChoice<String>("repositories", 
+				repositoriesModel, availableRepositories, new StringChoiceRenderer());
+		selections.setMaxRows(10);
+		form.add(selections);
+		form.add(new TextField<String>("query", queryModel));
+		add(form);
+				
+		// execute search
 		final List<SearchResult> results = new ArrayList<SearchResult>();
+		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);
 		final DataView<SearchResult> resultsView = new DataView<SearchResult>("searchResults", resultsDp) {
 			private static final long serialVersionUID = 1L;
@@ -97,74 +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()));
-			}
-		};		
-		
-		// initial query
-		final Model<String> fragment = new Model<String>();
-		if (!StringUtils.isEmpty(value)) {
-			if (searchType == SearchType.COMMIT) {
-				fragment.setObject("type:" + searchType.name().toLowerCase() + " AND \"" + value + "\"");	
-			} else {
-				fragment.setObject(searchType.name().toLowerCase() + ":\"" + value + "\"");
-			}
-		}
-		
-		// selected repositories
-		final Model<ArrayList<String>> repositories = new Model<ArrayList<String>>();
-		if (!StringUtils.isEmpty(repository)) {			
-			ArrayList<String> list = new ArrayList<String>();
-			list.add(repository);
-			repositories.setObject(list);
-		}
-		
-		// search form
-		StatelessForm<Void> form = new StatelessForm<Void>("searchForm") {
-			
-			private static final long serialVersionUID = 1L;
-
-			@Override
-			public void onSubmit() {
-				String f = fragment.getObject();
-				if (StringUtils.isEmpty(f)) {
-					error("Query is empty!");
-					return;
-				}				
-				if (repositories.getObject().size() == 0) {
-					error("Please select one or more repositories!");
-					return;
-				}
-				results.clear();
-				results.addAll(search(repositories, fragment));
-				resultsView.setVisible(true);
+				item.add(WicketUtils.createDatestampLabel("date", sr.date, getTimeZone()));
 			}
 		};
-		ListMultipleChoice<String> selections = new ListMultipleChoice<String>("repositories", repositories, GitBlit.self().getRepositoryList());
-		selections.setMaxRows(10);
-		form.add(selections);
-		form.add(new TextField<String>("fragment", fragment));
-		add(form);
-		if (!StringUtils.isEmpty(repository) && !StringUtils.isEmpty(fragment.getObject())) {
-			// search is defined by url parameters
-			results.clear();
-			results.addAll(search(repositories, fragment));
-			add(resultsView.setVisible(true));
-		} else {
-			// no pre-defined search
-			add(resultsView.setVisible(false));
-		}
-	}
-	
-	private List<SearchResult> search(Model<ArrayList<String>> repositories, Model<String> fragment) {
-		List<Repository> repos = new ArrayList<Repository>();
-		for (String r : repositories.getObject()) {
-			repos.add(GitBlit.self().getRepository(r));
-		}
-		List<SearchResult> srs = LuceneUtils.search(fragment.getObject(), 100, repos.toArray(new Repository[repos.size()]));
-		for (Repository r : repos) {
-			r.close();
-		}
-		return srs;
-	}
+		add(resultsView.setVisible(results.size() > 0));
+	}	
 }

--
Gitblit v1.9.1