/*
|
* Copyright 2012 gitblit.com.
|
*
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
*/
|
package com.gitblit.wicket.pages;
|
|
import java.text.MessageFormat;
|
import java.util.ArrayList;
|
import java.util.List;
|
|
import org.apache.wicket.Component;
|
import org.apache.wicket.PageParameters;
|
import org.apache.wicket.markup.html.basic.Label;
|
import org.apache.wicket.markup.html.form.ListMultipleChoice;
|
import org.apache.wicket.markup.html.form.StatelessForm;
|
import org.apache.wicket.markup.html.form.TextField;
|
import org.apache.wicket.markup.html.panel.Fragment;
|
import org.apache.wicket.markup.repeater.Item;
|
import org.apache.wicket.markup.repeater.data.DataView;
|
import org.apache.wicket.markup.repeater.data.ListDataProvider;
|
import org.apache.wicket.model.Model;
|
import org.eclipse.jgit.lib.Constants;
|
|
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.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 LuceneSearchPage extends RootPage {
|
|
public LuceneSearchPage() {
|
super();
|
setup(null);
|
}
|
|
public LuceneSearchPage(PageParameters params) {
|
super(params);
|
setup(params);
|
}
|
|
private void setup(PageParameters params) {
|
setupPage("", "");
|
|
// default values
|
ArrayList<String> repositories = new ArrayList<String>();
|
String query = "";
|
int page = 1;
|
int pageSize = GitBlit.getInteger(Keys.web.itemsPerPage, 50);
|
|
if (params != null) {
|
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(LuceneSearchPage.class, params);
|
}
|
};
|
|
ListMultipleChoice<String> selections = new ListMultipleChoice<String>("repositories",
|
repositoriesModel, availableRepositories, new StringChoiceRenderer());
|
selections.setMaxRows(8);
|
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) {
|
if (!ArrayUtils.isEmpty(searchRepositories) && !StringUtils.isEmpty(query)) {
|
add(new Label("resultsHeader", query).setRenderBodyOnly(true));
|
add(new Label("resultsCount", "0 hits").setRenderBodyOnly(true));
|
} else {
|
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;
|
public void populateItem(final Item<SearchResult> item) {
|
final SearchResult sr = item.getModelObject();
|
switch(sr.type) {
|
case commit: {
|
Label icon = WicketUtils.newIcon("type", "icon-refresh");
|
WicketUtils.setHtmlTooltip(icon, "commit");
|
item.add(icon);
|
item.add(new LinkPanel("summary", null, sr.summary, CommitPage.class, WicketUtils.newObjectParameter(sr.repository, sr.commitId)));
|
// show tags
|
Fragment fragment = new Fragment("tags", "tagsPanel", LuceneSearchPage.this);
|
List<String> tags = sr.tags;
|
if (tags == null) {
|
tags = new ArrayList<String>();
|
}
|
ListDataProvider<String> tagsDp = new ListDataProvider<String>(tags);
|
final DataView<String> tagsView = new DataView<String>("tag", tagsDp) {
|
private static final long serialVersionUID = 1L;
|
public void populateItem(final Item<String> item) {
|
String tag = item.getModelObject();
|
Component c = new LinkPanel("tagLink", null, tag, TagPage.class,
|
WicketUtils.newObjectParameter(sr.repository, Constants.R_TAGS + tag));
|
WicketUtils.setCssClass(c, "tagRef");
|
item.add(c);
|
}
|
};
|
fragment.add(tagsView);
|
item.add(fragment);
|
break;
|
}
|
case blob: {
|
Label icon = WicketUtils.newIcon("type", "icon-file");
|
WicketUtils.setHtmlTooltip(icon, "blob");
|
item.add(icon);
|
item.add(new LinkPanel("summary", null, sr.path, BlobPage.class, WicketUtils.newPathParameter(sr.repository, sr.branch, sr.path)));
|
item.add(new Label("tags").setVisible(false));
|
break;
|
}
|
case issue: {
|
Label icon = WicketUtils.newIcon("type", "icon-file");
|
WicketUtils.setHtmlTooltip(icon, "issue");
|
item.add(icon);
|
item.add(new Label("summary", "issue: " + sr.issueId));
|
item.add(new Label("tags").setVisible(false));
|
break;
|
}
|
}
|
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)));
|
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)));
|
}
|
item.add(new Label("author", sr.author));
|
item.add(WicketUtils.createDatestampLabel("date", sr.date, getTimeZone()));
|
}
|
};
|
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, LuceneSearchPage.class, pagerParams));
|
add(new PagerPanel("bottomPager", page, totalPages, LuceneSearchPage.class, pagerParams));
|
}
|
}
|
