/* * 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.util.ArrayList; import java.util.List; 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.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 org.eclipse.jgit.lib.Repository; import com.gitblit.Constants.SearchType; import com.gitblit.GitBlit; import com.gitblit.models.SearchResult; import com.gitblit.utils.LuceneUtils; import com.gitblit.utils.StringUtils; import com.gitblit.wicket.WicketUtils; import com.gitblit.wicket.panels.LinkPanel; public class LucenePage extends RootPage { public LucenePage() { super(); setup(null); } public LucenePage(PageParameters params) { super(params); setup(params); } private void setup(PageParameters params) { setupPage("", ""); String repository = null; String value = null; com.gitblit.Constants.SearchType searchType = null; if (params != null) { repository = WicketUtils.getRepositoryName(params); value = WicketUtils.getSearchString(params); String type = WicketUtils.getSearchType(params); searchType = com.gitblit.Constants.SearchType.forName(type); } final List results = new ArrayList(); ListDataProvider resultsDp = new ListDataProvider(results); final DataView resultsView = new DataView("searchResults", resultsDp) { private static final long serialVersionUID = 1L; public void populateItem(final Item item) { 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.id))); break; } case blob: { Label icon = WicketUtils.newIcon("type", "icon-file"); WicketUtils.setHtmlTooltip(icon, "blob"); item.add(icon); item.add(new LinkPanel("summary", null, sr.id, BlobPage.class, WicketUtils.newPathParameter(sr.repository, sr.branch, sr.id))); break; } case issue: { Label icon = WicketUtils.newIcon("type", "icon-file"); WicketUtils.setHtmlTooltip(icon, "issue"); item.add(icon); item.add(new Label("summary", "issue: " + sr.id)); break; } } 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 fragment = new Model(); 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> repositories = new Model>(); if (!StringUtils.isEmpty(repository)) { ArrayList list = new ArrayList(); list.add(repository); repositories.setObject(list); } // search form StatelessForm form = new StatelessForm("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); } }; ListMultipleChoice selections = new ListMultipleChoice("repositories", repositories, GitBlit.self().getRepositoryList()); selections.setMaxRows(11); form.add(selections); form.add(new TextField("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 search(Model> repositories, Model fragment) { List repos = new ArrayList(); for (String r : repositories.getObject()) { repos.add(GitBlit.self().getRepository(r)); } List srs = LuceneUtils.search(fragment.getObject(), 100, repos.toArray(new Repository[repos.size()])); for (Repository r : repos) { r.close(); } return srs; } }