Changed the repository filter options to those related to the tickets owned...

Eddy Young

2015-05-22 3004fe27640f6851bc0d230a954baa3b157746ee

 src/main/java/com/gitblit/wicket/pages/MyTicketsPage.java

@@ -17,6 +17,10 @@

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;

import org.apache.wicket.PageParameters;
@@ -69,14 +73,17 @@
      }

      final String username = currentUser.getName();
        final String defaultQueryParam = "watchedby:"+username;

      final String[] statiiParam = (params == null) ? TicketsUI.openStatii : params.getStringArray(Lucene.status.name());
      final String assignedToParam = (params == null) ? "" : params.getString(Lucene.responsible.name(), null);
      final String milestoneParam = (params == null) ? "" : params.getString(Lucene.milestone.name(), null);
      final String queryParam = (params == null || StringUtils.isEmpty(params.getString("q", null))) ? "watchedby:" + username : params.getString("q", null);
        final String queryParam = (params == null || StringUtils.isEmpty(params.getString("q", null))) ? defaultQueryParam : params.getString("q", null);
      final String searchParam = (params == null) ? "" : params.getString("s", null);
      final String sortBy = (params == null) ? "" : Lucene.fromString(params.getString("sort", Lucene.created.name())).name();
      final String repositoryId = (params == null) ? "" : params.getString(Lucene.rid.name(), null);
      final boolean desc = (params == null) ? true : !"asc".equals(params.getString("direction", "desc"));


      // add the user title panel
      add(new UserTitlePanel("userTitlePanel", currentUser, getString("gb.myTickets")));
@@ -272,10 +279,22 @@
      add(sortMenu);

        // by repository
      List<RepositoryModel> repositoryChoices = getRepositoryModels();
        RepositoryModel noneChoice = new RepositoryModel();
        final List<QueryResult> tickets =
            query(initializeQueryBuilder(defaultQueryParam, username), 1, Integer.MAX_VALUE, sortBy, desc);
        final List<RepositoryModel> repositoryChoices = correspondingRepositories(tickets);
        Collections.sort(repositoryChoices, new Comparator<RepositoryModel>() {
            private static final long serialVersionUID = 1L;

            @Override
            public int compare(RepositoryModel repository1, RepositoryModel repository2) {
                return repository1.toString().compareTo(repository2.toString());
            }
        });

        final RepositoryModel noneChoice = new RepositoryModel();
        noneChoice.name = getString("gb.all");
        repositoryChoices.add(0, noneChoice);

      RepositoryModel currentRepository = repositoryChoices.size() > 0 ? repositoryChoices.get(0) : null;
      for (RepositoryModel r : repositoryChoices) {
         if (r.getRID().equals(repositoryId)) {
@@ -298,8 +317,9 @@
      };
      add(repositoryMenu);

      // Build Query here
      QueryBuilder qb = new QueryBuilder(queryParam);
        // Update query with filter criteria
        final QueryBuilder qb = initializeQueryBuilder(queryParam, username);

      if (!qb.containsField(Lucene.status.name()) && !ArrayUtils.isEmpty(statiiParam)) {
         // specify the states
         boolean not = false;
@@ -312,6 +332,7 @@
               q.or(Lucene.status.matches(state));
            }
         }

         if (not) {
            qb.and(q.toString());
         } else {
@@ -325,32 +346,12 @@
            qb.and(q1.toSubquery().toString());
        }

      final String luceneQuery;
      if (qb.containsField(Lucene.createdby.name())
            || qb.containsField(Lucene.responsible.name())
            || qb.containsField(Lucene.watchedby.name())) {
         // focused "my tickets" query
         luceneQuery = qb.build();
      } else {
         // general "my tickets" query
         QueryBuilder myQuery = new QueryBuilder();
         myQuery.or(Lucene.createdby.matches(username));
         myQuery.or(Lucene.responsible.matches(username));
         myQuery.or(Lucene.watchedby.matches(username));
         myQuery.and(qb.toSubquery().toString());
         luceneQuery = myQuery.build();
      }

      // paging links
      int page = (params != null) ? Math.max(1, WicketUtils.getPage(params)) : 1;
      int pageSize = app().settings().getInteger(Keys.tickets.perPage, 25);

      List<QueryResult> results;
      if(StringUtils.isEmpty(searchParam)) {
         results = app().tickets().queryFor(luceneQuery, page, pageSize, sortBy, desc);
      } else {
         results = app().tickets().searchFor(null, searchParam, page, pageSize);
      }
        final List<QueryResult> results = 
            StringUtils.isEmpty(searchParam) ? query(qb, page, pageSize, sortBy, desc) : search(searchParam, page, pageSize);

      int totalResults = results.size() == 0 ? 0 : results.get(0).totalResults;
      buildPager(queryParam, milestoneParam, statiiParam, assignedToParam, sortBy, desc, repositoryId, page, pageSize, results.size(), totalResults);
@@ -455,4 +456,43 @@
      };
      add(pagesView);
   }

    private QueryBuilder initializeQueryBuilder(String queryparam, String username) {
        final QueryBuilder qb = new QueryBuilder(queryparam);

        // focused "my tickets"
        if (qb.containsField(Lucene.createdby.name())
                || qb.containsField(Lucene.responsible.name())
                || qb.containsField(Lucene.watchedby.name())) {

            return qb;
        }

        // general "my tickets"
        return qb.andSubquery()
                 .or(Lucene.createdby.matches(username))
                 .or(Lucene.responsible.matches(username))
                 .or(Lucene.watchedby.matches(username))
                 .endSubquery();
    }

    private List<QueryResult> query(QueryBuilder qb, int page, int pageSize, String sortBy, boolean descending) {
        return app().tickets().queryFor(qb.build(), page, pageSize, sortBy, descending);
    }

    private List<QueryResult> search(String searchParam, int page, int pageSize) {
        return app().tickets().searchFor(null, searchParam, page, pageSize);
    }

    private List<RepositoryModel> correspondingRepositories(Collection<QueryResult> tickets) { 
        final HashMap<String, RepositoryModel> result = new HashMap<>();
        for (QueryResult ticket : tickets) {
            RepositoryModel repository = app().repositories().getRepositoryModel(ticket.repository);
            if (!result.containsKey(repository.getRID())) {
                result.put(repository.getRID(), repository);
            }
        }

        return new ArrayList<>(result.values());
    }
}