| commit | author | age | ||
| 4c13bf | 1 | /* |
| JM | 2 | * Copyright 2012 gitblit.com. |
| 3 | * | |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); | |
| 5 | * you may not use this file except in compliance with the License. | |
| 6 | * You may obtain a copy of the License at | |
| 7 | * | |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 | |
| 9 | * | |
| 10 | * Unless required by applicable law or agreed to in writing, software | |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, | |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| 13 | * See the License for the specific language governing permissions and | |
| 14 | * limitations under the License. | |
| 15 | */ | |
| 16 | package com.gitblit.wicket.pages; | |
| 17 | ||
| d04009 | 18 | import java.text.MessageFormat; |
| 4c13bf | 19 | import java.util.ArrayList; |
| 548b29 | 20 | import java.util.LinkedHashSet; |
| 4c13bf | 21 | import java.util.List; |
| 548b29 | 22 | import java.util.Set; |
| 4c13bf | 23 | |
| 31a897 | 24 | import org.apache.wicket.Component; |
| 4c13bf | 25 | import org.apache.wicket.PageParameters; |
| JM | 26 | import org.apache.wicket.markup.html.basic.Label; |
| 548b29 | 27 | import org.apache.wicket.markup.html.form.CheckBox; |
| 3d0494 | 28 | import org.apache.wicket.markup.html.form.ListMultipleChoice; |
| 4c13bf | 29 | import org.apache.wicket.markup.html.form.TextField; |
| 31a897 | 30 | import org.apache.wicket.markup.html.panel.Fragment; |
| 4c13bf | 31 | import org.apache.wicket.markup.repeater.Item; |
| JM | 32 | import org.apache.wicket.markup.repeater.data.DataView; |
| 33 | import org.apache.wicket.markup.repeater.data.ListDataProvider; | |
| 34 | import org.apache.wicket.model.Model; | |
| 3f0677 | 35 | import org.eclipse.jgit.lib.Constants; |
| 4c13bf | 36 | |
| 3d0494 | 37 | import com.gitblit.Constants.SearchType; |
| d04009 | 38 | import com.gitblit.Keys; |
| 672296 | 39 | import com.gitblit.models.RepositoryModel; |
| 4c13bf | 40 | import com.gitblit.models.SearchResult; |
| 672296 | 41 | import com.gitblit.models.UserModel; |
| 7d97c6 | 42 | import com.gitblit.utils.ArrayUtils; |
| 3f0677 | 43 | import com.gitblit.utils.StringUtils; |
| 672296 | 44 | import com.gitblit.wicket.GitBlitWebSession; |
| ed295f | 45 | import com.gitblit.wicket.SessionlessForm; |
| 672296 | 46 | import com.gitblit.wicket.StringChoiceRenderer; |
| 4c13bf | 47 | import com.gitblit.wicket.WicketUtils; |
| JM | 48 | import com.gitblit.wicket.panels.LinkPanel; |
| d04009 | 49 | import com.gitblit.wicket.panels.PagerPanel; |
| 4c13bf | 50 | |
| 4e1930 | 51 | public class LuceneSearchPage extends RootPage { |
| 4c13bf | 52 | |
| 4e1930 | 53 | public LuceneSearchPage() { |
| 4c13bf | 54 | super(); |
| JM | 55 | setup(null); |
| 56 | } | |
| 57 | ||
| 4e1930 | 58 | public LuceneSearchPage(PageParameters params) { |
| 4c13bf | 59 | super(params); |
| JM | 60 | setup(params); |
| 61 | } | |
| 62 | ||
| 63 | private void setup(PageParameters params) { | |
| 64 | setupPage("", ""); | |
| 699e71 | 65 | |
| 7d97c6 | 66 | // default values |
| 699e71 | 67 | ArrayList<String> repositories = new ArrayList<String>(); |
| 7d97c6 | 68 | String query = ""; |
| b674fe | 69 | boolean allRepos = false; |
| PM | 70 | |
| d04009 | 71 | int page = 1; |
| 99d0d4 | 72 | int pageSize = app().settings().getInteger(Keys.web.itemsPerPage, 50); |
| 3d0494 | 73 | |
| b4d958 | 74 | // display user-accessible selections |
| VF | 75 | UserModel user = GitBlitWebSession.get().getUser(); |
| 76 | List<String> availableRepositories = new ArrayList<String>(); | |
| 77 | for (RepositoryModel model : app().repositories().getRepositoryModels(user)) { | |
| 78 | if (model.hasCommits && !ArrayUtils.isEmpty(model.indexedBranches)) { | |
| 79 | availableRepositories.add(model.name); | |
| 80 | } | |
| 81 | } | |
| 82 | ||
| 3d0494 | 83 | if (params != null) { |
| 7d97c6 | 84 | String repository = WicketUtils.getRepositoryName(params); |
| JM | 85 | if (!StringUtils.isEmpty(repository)) { |
| 86 | repositories.add(repository); | |
| 87 | } | |
| d04009 | 88 | |
| 31a897 | 89 | page = WicketUtils.getPage(params); |
| 699e71 | 90 | |
| 7d97c6 | 91 | if (params.containsKey("repositories")) { |
| JM | 92 | String value = params.getString("repositories", ""); |
| 699e71 | 93 | List<String> list = StringUtils.getStringsFromValue(value); |
| 7d97c6 | 94 | repositories.addAll(list); |
| b4d958 | 95 | } |
| b674fe | 96 | |
| PM | 97 | allRepos = params.getAsBoolean("allrepos", false); |
| 98 | if (allRepos) { | |
| b4d958 | 99 | repositories.addAll(availableRepositories); |
| 7d97c6 | 100 | } |
| JM | 101 | |
| 102 | if (params.containsKey("query")) { | |
| 699e71 | 103 | query = params.getString("query", ""); |
| 7d97c6 | 104 | } else { |
| JM | 105 | String value = WicketUtils.getSearchString(params); |
| 106 | String type = WicketUtils.getSearchType(params); | |
| 107 | com.gitblit.Constants.SearchType searchType = com.gitblit.Constants.SearchType.forName(type); | |
| 108 | if (!StringUtils.isEmpty(value)) { | |
| 109 | if (searchType == SearchType.COMMIT) { | |
| 699e71 | 110 | query = "type:" + searchType.name().toLowerCase() + " AND \"" + value + "\""; |
| 7d97c6 | 111 | } else { |
| JM | 112 | query = searchType.name().toLowerCase() + ":\"" + value + "\""; |
| 113 | } | |
| 114 | } | |
| 115 | } | |
| 3d0494 | 116 | } |
| 699e71 | 117 | |
| 99d0d4 | 118 | boolean luceneEnabled = app().settings().getBoolean(Keys.web.allowLuceneIndexing, true); |
| 7db092 | 119 | if (luceneEnabled) { |
| JM | 120 | if (availableRepositories.size() == 0) { |
| 121 | info(getString("gb.noIndexedRepositoriesWarning")); | |
| 122 | } | |
| 123 | } else { | |
| 124 | error(getString("gb.luceneDisabled")); | |
| 40ca5c | 125 | } |
| JM | 126 | |
| 127 | // enforce user-accessible repository selections | |
| 548b29 | 128 | Set<String> uniqueRepositories = new LinkedHashSet<String>(); |
| 40ca5c | 129 | for (String selectedRepository : repositories) { |
| JM | 130 | if (availableRepositories.contains(selectedRepository)) { |
| 548b29 | 131 | uniqueRepositories.add(selectedRepository); |
| 40ca5c | 132 | } |
| JM | 133 | } |
| 548b29 | 134 | ArrayList<String> searchRepositories = new ArrayList<String>(uniqueRepositories); |
| 699e71 | 135 | |
| 7d97c6 | 136 | // search form |
| JM | 137 | final Model<String> queryModel = new Model<String>(query); |
| 40ca5c | 138 | final Model<ArrayList<String>> repositoriesModel = new Model<ArrayList<String>>(searchRepositories); |
| b674fe | 139 | final Model<Boolean> allreposModel = new Model<Boolean>(allRepos); |
| ed295f | 140 | SessionlessForm<Void> form = new SessionlessForm<Void>("searchForm", getClass()) { |
| 699e71 | 141 | |
| 7d97c6 | 142 | private static final long serialVersionUID = 1L; |
| JM | 143 | |
| 144 | @Override | |
| 145 | public void onSubmit() { | |
| 146 | String q = queryModel.getObject(); | |
| 147 | if (StringUtils.isEmpty(q)) { | |
| 40ca5c | 148 | error(getString("gb.undefinedQueryWarning")); |
| 7d97c6 | 149 | return; |
| 699e71 | 150 | } |
| b4d958 | 151 | if (repositoriesModel.getObject().size() == 0 && !allreposModel.getObject()) { |
| 40ca5c | 152 | error(getString("gb.noSelectedRepositoriesWarning")); |
| 7d97c6 | 153 | return; |
| JM | 154 | } |
| 155 | PageParameters params = new PageParameters(); | |
| 156 | params.put("repositories", StringUtils.flattenStrings(repositoriesModel.getObject())); | |
| 157 | params.put("query", queryModel.getObject()); | |
| b4d958 | 158 | params.put("allrepos", allreposModel.getObject()); |
| 1b84d3 | 159 | LuceneSearchPage page = new LuceneSearchPage(params); |
| JM | 160 | setResponsePage(page); |
| 7d97c6 | 161 | } |
| JM | 162 | }; |
| 699e71 | 163 | |
| JM | 164 | ListMultipleChoice<String> selections = new ListMultipleChoice<String>("repositories", |
| 672296 | 165 | repositoriesModel, availableRepositories, new StringChoiceRenderer()); |
| 4e1930 | 166 | selections.setMaxRows(8); |
| 7db092 | 167 | form.add(selections.setEnabled(luceneEnabled)); |
| JM | 168 | form.add(new TextField<String>("query", queryModel).setEnabled(luceneEnabled)); |
| b4d958 | 169 | form.add(new CheckBox("allrepos", allreposModel)); |
| 7db092 | 170 | add(form.setEnabled(luceneEnabled)); |
| 699e71 | 171 | |
| 7d97c6 | 172 | // execute search |
| 4c13bf | 173 | final List<SearchResult> results = new ArrayList<SearchResult>(); |
| 40ca5c | 174 | if (!ArrayUtils.isEmpty(searchRepositories) && !StringUtils.isEmpty(query)) { |
| 99d0d4 | 175 | results.addAll(app().repositories().search(query, page, pageSize, searchRepositories)); |
| d04009 | 176 | } |
| 699e71 | 177 | |
| d04009 | 178 | // results header |
| JM | 179 | if (results.size() == 0) { |
| 22a8a6 | 180 | if (!ArrayUtils.isEmpty(searchRepositories) && !StringUtils.isEmpty(query)) { |
| JM | 181 | add(new Label("resultsHeader", query).setRenderBodyOnly(true)); |
| 6caa93 | 182 | add(new Label("resultsCount", getString("gb.noHits")).setRenderBodyOnly(true)); |
| 22a8a6 | 183 | } else { |
| JM | 184 | add(new Label("resultsHeader").setVisible(false)); |
| 185 | add(new Label("resultsCount").setVisible(false)); | |
| 186 | } | |
| d04009 | 187 | } else { |
| JM | 188 | add(new Label("resultsHeader", query).setRenderBodyOnly(true)); |
| 6caa93 | 189 | add(new Label("resultsCount", MessageFormat.format(getString("gb.queryResults"), |
| d04009 | 190 | results.get(0).hitId, results.get(results.size() - 1).hitId, results.get(0).totalHits)). |
| JM | 191 | setRenderBodyOnly(true)); |
| d896e6 | 192 | } |
| 699e71 | 193 | |
| 7d97c6 | 194 | // search results view |
| 4c13bf | 195 | ListDataProvider<SearchResult> resultsDp = new ListDataProvider<SearchResult>(results); |
| JM | 196 | final DataView<SearchResult> resultsView = new DataView<SearchResult>("searchResults", resultsDp) { |
| 197 | private static final long serialVersionUID = 1L; | |
| 699e71 | 198 | @Override |
| 4c13bf | 199 | public void populateItem(final Item<SearchResult> item) { |
| 31a897 | 200 | final SearchResult sr = item.getModelObject(); |
| 4c13bf | 201 | switch(sr.type) { |
| JM | 202 | case commit: { |
| 203 | Label icon = WicketUtils.newIcon("type", "icon-refresh"); | |
| 204 | WicketUtils.setHtmlTooltip(icon, "commit"); | |
| 205 | item.add(icon); | |
| 98b4ed | 206 | item.add(new LinkPanel("summary", null, sr.summary, CommitPage.class, WicketUtils.newObjectParameter(sr.repository, sr.commitId))); |
| 31a897 | 207 | // show tags |
| JM | 208 | Fragment fragment = new Fragment("tags", "tagsPanel", LuceneSearchPage.this); |
| 209 | List<String> tags = sr.tags; | |
| 210 | if (tags == null) { | |
| 211 | tags = new ArrayList<String>(); | |
| 212 | } | |
| 213 | ListDataProvider<String> tagsDp = new ListDataProvider<String>(tags); | |
| 214 | final DataView<String> tagsView = new DataView<String>("tag", tagsDp) { | |
| 215 | private static final long serialVersionUID = 1L; | |
| 699e71 | 216 | @Override |
| 31a897 | 217 | public void populateItem(final Item<String> item) { |
| JM | 218 | String tag = item.getModelObject(); |
| 219 | Component c = new LinkPanel("tagLink", null, tag, TagPage.class, | |
| 220 | WicketUtils.newObjectParameter(sr.repository, Constants.R_TAGS + tag)); | |
| 699e71 | 221 | WicketUtils.setCssClass(c, "tagRef"); |
| 31a897 | 222 | item.add(c); |
| JM | 223 | } |
| 224 | }; | |
| 225 | fragment.add(tagsView); | |
| 226 | item.add(fragment); | |
| 4c13bf | 227 | break; |
| JM | 228 | } |
| 229 | case blob: { | |
| 230 | Label icon = WicketUtils.newIcon("type", "icon-file"); | |
| 231 | WicketUtils.setHtmlTooltip(icon, "blob"); | |
| 232 | item.add(icon); | |
| 98b4ed | 233 | item.add(new LinkPanel("summary", null, sr.path, BlobPage.class, WicketUtils.newPathParameter(sr.repository, sr.branch, sr.path))); |
| 31a897 | 234 | item.add(new Label("tags").setVisible(false)); |
| 4c13bf | 235 | break; |
| JM | 236 | } |
| 237 | } | |
| 98b4ed | 238 | item.add(new Label("fragment", sr.fragment).setEscapeModelStrings(false).setVisible(!StringUtils.isEmpty(sr.fragment))); |
| 3f0677 | 239 | item.add(new LinkPanel("repository", null, sr.repository, SummaryPage.class, WicketUtils.newRepositoryParameter(sr.repository))); |
| d04009 | 240 | if (StringUtils.isEmpty(sr.branch)) { |
| JM | 241 | item.add(new Label("branch", "null")); |
| 242 | } else { | |
| 243 | item.add(new LinkPanel("branch", "branch", StringUtils.getRelativePath(Constants.R_HEADS, sr.branch), LogPage.class, WicketUtils.newObjectParameter(sr.repository, sr.branch))); | |
| 244 | } | |
| 4c13bf | 245 | item.add(new Label("author", sr.author)); |
| 9adf62 | 246 | item.add(WicketUtils.createDatestampLabel("date", sr.date, getTimeZone(), getTimeUtils())); |
| 4c13bf | 247 | } |
| JM | 248 | }; |
| 7d97c6 | 249 | add(resultsView.setVisible(results.size() > 0)); |
| 699e71 | 250 | |
| d04009 | 251 | PageParameters pagerParams = new PageParameters(); |
| JM | 252 | pagerParams.put("repositories", StringUtils.flattenStrings(repositoriesModel.getObject())); |
| 253 | pagerParams.put("query", queryModel.getObject()); | |
| 699e71 | 254 | |
| 697def | 255 | boolean showPager = false; |
| d04009 | 256 | int totalPages = 0; |
| JM | 257 | if (results.size() > 0) { |
| 258 | totalPages = (results.get(0).totalHits / pageSize) + (results.get(0).totalHits % pageSize > 0 ? 1 : 0); | |
| 697def | 259 | showPager = results.get(0).totalHits > pageSize; |
| d04009 | 260 | } |
| 699e71 | 261 | |
| 697def | 262 | add(new PagerPanel("topPager", page, totalPages, LuceneSearchPage.class, pagerParams).setVisible(showPager)); |
| JM | 263 | add(new PagerPanel("bottomPager", page, totalPages, LuceneSearchPage.class, pagerParams).setVisible(showPager)); |
| d04009 | 264 | } |
| 4c13bf | 265 | } |
