From e9de3f7fd55dd4d4535cd7b2fc2bee8094152e39 Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Sat, 17 Mar 2012 12:09:01 -0400
Subject: [PATCH] Implemented RPC administration of indexed branches
---
src/com/gitblit/wicket/pages/LucenePage.java | 249 ++++++++++++++++++++++++++++++++++---------------
1 files changed, 174 insertions(+), 75 deletions(-)
diff --git a/src/com/gitblit/wicket/pages/LucenePage.java b/src/com/gitblit/wicket/pages/LucenePage.java
index f72aeda..10de0bf 100644
--- a/src/com/gitblit/wicket/pages/LucenePage.java
+++ b/src/com/gitblit/wicket/pages/LucenePage.java
@@ -15,6 +15,7 @@
*/
package com.gitblit.wicket.pages;
+import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.List;
@@ -28,15 +29,20 @@
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.Keys;
+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;
+import com.gitblit.wicket.panels.PagerPanel;
public class LucenePage extends RootPage {
@@ -53,18 +59,113 @@
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 = "";
+ int page = 1;
+ int pageSize = GitBlit.getInteger(Keys.web.itemsPerPage, 50);
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);
+ }
+
+ page = WicketUtils.getPage(params);
+
+ 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, page, pageSize, searchRepositories));
+ }
+
+ // results header
+ if (results.size() == 0) {
+ add(new Label("resultsHeader").setVisible(false));
+ add(new Label("resultsCount").setVisible(false));
+ } else {
+ add(new Label("resultsHeader", query).setRenderBodyOnly(true));
+ add(new Label("resultsCount", MessageFormat.format("results {0} - {1} ({2} hits)",
+ results.get(0).hitId, results.get(results.size() - 1).hitId, results.get(0).totalHits)).
+ setRenderBodyOnly(true));
+ }
+
+ // 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;
@@ -95,76 +196,74 @@
}
item.add(new Label("fragment", sr.fragment).setEscapeModelStrings(false).setVisible(!StringUtils.isEmpty(sr.fragment)));
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;
+ if (StringUtils.isEmpty(sr.branch)) {
+ item.add(new Label("branch", "null"));
+ } else {
+ item.add(new LinkPanel("branch", "branch", StringUtils.getRelativePath(Constants.R_HEADS, sr.branch), LogPage.class, WicketUtils.newObjectParameter(sr.repository, sr.branch)));
}
- results.clear();
- results.addAll(search(repositories, fragment));
- resultsView.setVisible(true);
+ item.add(new Label("author", sr.author));
+ 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));
+ add(resultsView.setVisible(results.size() > 0));
+
+ PageParameters pagerParams = new PageParameters();
+ pagerParams.put("repositories", StringUtils.flattenStrings(repositoriesModel.getObject()));
+ pagerParams.put("query", queryModel.getObject());
+
+ int totalPages = 0;
+ if (results.size() > 0) {
+ totalPages = (results.get(0).totalHits / pageSize) + (results.get(0).totalHits % pageSize > 0 ? 1 : 0);
}
+
+ add(new PagerPanel("topPager", page, totalPages, LucenePage.class, pagerParams));
+ add(new PagerPanel("bottomPager", page, totalPages, LucenePage.class, pagerParams));
}
- 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;
- }
+// private String buildPager(int currentPage, int count, int total) {
+// int pages = (total / count) + (total % count == 0 ? 0 : 1);
+//
+// // pages are 1-indexed
+// // previous page link
+// if (currentPage <= 1) {
+// sb.append(MessageFormat.format(li, "disabled", "#", "←"));
+// } else {
+// List<String> parameters = new ArrayList<String>();
+// if (!StringUtils.isEmpty(penString)) {
+// parameters.add(penString);
+// }
+// parameters.add(MessageFormat.format(pg, currentPage - 1));
+// sb.append(MessageFormat.format(li, "", StringUtils.flattenStrings(parameters, "&"), "←"));
+// }
+//
+// // page links in middle
+// int minpage = Math.max(1, currentPage - Math.min(2, 2));
+// int maxpage = Math.min(pages, minpage + 4);
+// for (int i = minpage; i <= maxpage; i++) {
+// String cssClass = "";
+// if (i == currentPage) {
+// cssClass = "active";
+// }
+// List<String> parameters = new ArrayList<String>();
+// if (!StringUtils.isEmpty(penString)) {
+// parameters.add(penString);
+// }
+// parameters.add(MessageFormat.format(pg, i));
+// sb.append(MessageFormat.format(li, cssClass, StringUtils.flattenStrings(parameters, "&"), i));
+// }
+//
+// // next page link
+// if (currentPage == pages) {
+// sb.append(MessageFormat.format(li, "disabled", "#", "→"));
+// } else {
+// List<String> parameters = new ArrayList<String>();
+// if (!StringUtils.isEmpty(penString)) {
+// parameters.add(penString);
+// }
+// parameters.add(MessageFormat.format(pg, currentPage + 1));
+// sb.append(MessageFormat.format(li, "", StringUtils.flattenStrings(parameters, "&"), "→"));
+// }
+// return sb.toString();
+// }
+
}
--
Gitblit v1.9.1