James Moger
2015-11-22 ed552ba47c02779c270ffd62841d6d1048dade70
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;
@@ -27,6 +31,7 @@
import org.apache.wicket.markup.repeater.data.ListDataProvider;
import com.gitblit.Keys;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.TicketModel;
import com.gitblit.models.TicketModel.Status;
import com.gitblit.models.UserModel;
@@ -68,13 +73,16 @@
      }
      final String username = currentUser.getName();
      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) ? null : 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")));
@@ -91,6 +99,7 @@
                  assignedToParam,
                  sortBy,
                  desc,
                        repositoryId,
                  1)));
      add(new BookmarkablePageLink<Void>("bugsQuery", MyTicketsPage.class,
@@ -101,6 +110,7 @@
                  assignedToParam,
                  sortBy,
                  desc,
                        repositoryId,
                  1)));
      add(new BookmarkablePageLink<Void>("enhancementsQuery", MyTicketsPage.class,
@@ -111,6 +121,7 @@
                  assignedToParam,
                  sortBy,
                  desc,
                        repositoryId,
                  1)));
      add(new BookmarkablePageLink<Void>("tasksQuery", MyTicketsPage.class,
@@ -121,6 +132,7 @@
                  assignedToParam,
                  sortBy,
                  desc,
                        repositoryId,
                  1)));
      add(new BookmarkablePageLink<Void>("questionsQuery", MyTicketsPage.class,
@@ -131,6 +143,18 @@
                  assignedToParam,
                  sortBy,
                  desc,
                        repositoryId,
                        1)));
        add(new BookmarkablePageLink<Void>("maintenanceQuery", MyTicketsPage.class,
                queryParameters(
                        Lucene.type.matches(TicketModel.Type.Maintenance.name()),
                        milestoneParam,
                        statiiParam,
                        assignedToParam,
                        sortBy,
                        desc,
                        repositoryId,
                  1)));
      add(new BookmarkablePageLink<Void>("resetQuery", MyTicketsPage.class,
@@ -141,6 +165,7 @@
                  null,
                  null,
                  true,
                        null,
                  1)));
      add(new Label("userDivider"));
@@ -152,6 +177,7 @@
                  assignedToParam,
                  sortBy,
                  desc,
                        repositoryId,
                  1)));
      add(new BookmarkablePageLink<Void>("watchedQuery", MyTicketsPage.class,
@@ -162,6 +188,7 @@
                  assignedToParam,
                  sortBy,
                  desc,
                        repositoryId,
                  1)));
      add(new BookmarkablePageLink<Void>("mentionsQuery", MyTicketsPage.class,
            queryParameters(
@@ -171,6 +198,7 @@
                  assignedToParam,
                  sortBy,
                  desc,
                        repositoryId,
                  1)));
      add(new BookmarkablePageLink<Void>("responsibleQuery", MyTicketsPage.class,
            queryParameters(
@@ -180,6 +208,7 @@
                  assignedToParam,
                  sortBy,
                  desc,
                        repositoryId,
                  1)));
      // states
@@ -188,9 +217,9 @@
      } else {
         add(new Label("selectedStatii", StringUtils.flattenStrings(Arrays.asList(statiiParam), ",")));
      }
      add(new BookmarkablePageLink<Void>("openTickets", MyTicketsPage.class, queryParameters(queryParam, milestoneParam, TicketsUI.openStatii, assignedToParam, sortBy, desc, 1)));
      add(new BookmarkablePageLink<Void>("closedTickets", MyTicketsPage.class, queryParameters(queryParam, milestoneParam, TicketsUI.closedStatii, assignedToParam, sortBy, desc, 1)));
      add(new BookmarkablePageLink<Void>("allTickets", MyTicketsPage.class, queryParameters(queryParam, milestoneParam, null, assignedToParam, sortBy, desc, 1)));
        add(new BookmarkablePageLink<Void>("openTickets", MyTicketsPage.class, queryParameters(queryParam, milestoneParam, TicketsUI.openStatii, assignedToParam, sortBy, desc, repositoryId, 1)));
        add(new BookmarkablePageLink<Void>("closedTickets", MyTicketsPage.class, queryParameters(queryParam, milestoneParam, TicketsUI.closedStatii, assignedToParam, sortBy, desc, repositoryId, 1)));
        add(new BookmarkablePageLink<Void>("allTickets", MyTicketsPage.class, queryParameters(queryParam, milestoneParam, null, assignedToParam, sortBy, desc, repositoryId, 1)));
      // by status
      List<Status> statii = new ArrayList<Status>(Arrays.asList(Status.values()));
@@ -202,13 +231,14 @@
         @Override
         public void populateItem(final Item<Status> item) {
            final Status status = item.getModelObject();
            PageParameters p = queryParameters(queryParam, milestoneParam, new String [] { status.name().toLowerCase() }, assignedToParam, sortBy, desc, 1);
                PageParameters p = queryParameters(queryParam, milestoneParam, new String [] { status.name().toLowerCase() }, assignedToParam, sortBy, desc, repositoryId, 1);
            String css = TicketsUI.getStatusClass(status);
            item.add(new LinkPanel("statusLink", css, status.toString(), MyTicketsPage.class, p).setRenderBodyOnly(true));
         }
      };
      add(statiiLinks);
        // by sort
      List<TicketSort> sortChoices = new ArrayList<TicketSort>();
      sortChoices.add(new TicketSort(getString("gb.sortNewest"), Lucene.created.name(), true));
      sortChoices.add(new TicketSort(getString("gb.sortOldest"), Lucene.created.name(), false));
@@ -220,6 +250,10 @@
      sortChoices.add(new TicketSort(getString("gb.sortLeastPatchsetRevisions"), Lucene.patchsets.name(), false));
      sortChoices.add(new TicketSort(getString("gb.sortMostVotes"), Lucene.votes.name(), true));
      sortChoices.add(new TicketSort(getString("gb.sortLeastVotes"), Lucene.votes.name(), false));
        sortChoices.add(new TicketSort(getString("gb.sortHighestPriority"), Lucene.priority.name(), true));
        sortChoices.add(new TicketSort(getString("gb.sortLowestPriority"), Lucene.priority.name(), false));
        sortChoices.add(new TicketSort(getString("gb.sortHighestSeverity"), Lucene.severity.name(), true));
        sortChoices.add(new TicketSort(getString("gb.sortLowestSeverity"), Lucene.severity.name(), false));
      TicketSort currentSort = sortChoices.get(0);
      for (TicketSort ts : sortChoices) {
@@ -237,14 +271,48 @@
         @Override
         public void populateItem(final Item<TicketSort> item) {
            final TicketSort ts = item.getModelObject();
            PageParameters params = queryParameters(queryParam, milestoneParam, statiiParam, assignedToParam, ts.sortBy, ts.desc, 1);
                PageParameters params = queryParameters(queryParam, milestoneParam, statiiParam, assignedToParam, ts.sortBy, ts.desc, repositoryId, 1);
            item.add(new LinkPanel("sortLink", null, ts.name, MyTicketsPage.class, params).setRenderBodyOnly(true));
         }
      };
      add(sortMenu);
      // Build Query here
      QueryBuilder qb = new QueryBuilder(queryParam);
        // by repository
        final List<QueryResult> tickets =
            query(initializeQueryBuilder(null, username), 1, Integer.MAX_VALUE, sortBy, desc);
        final List<RepositoryModel> repositoryChoices = correspondingRepositories(tickets);
        Collections.sort(repositoryChoices);
        final RepositoryModel noneChoice = new RepositoryModel();
        noneChoice.name = getString("gb.all");
        repositoryChoices.add(0, noneChoice);
        RepositoryModel currentRepository = repositoryChoices.get(0);
        for (RepositoryModel r : repositoryChoices) {
            if (r.getRID().equals(repositoryId)) {
                currentRepository = r;
                break;
            }
        }
        add(new Label("currentRepository", currentRepository.toString()));
        ListDataProvider<RepositoryModel> repositoryChoicesDp = new ListDataProvider<RepositoryModel>(repositoryChoices);
        DataView<RepositoryModel> repositoryMenu = new DataView<RepositoryModel>("repository", repositoryChoicesDp) {
            private static final long serialVersionUID = 1L;
            @Override
            public void populateItem(final Item<RepositoryModel> item) {
                final RepositoryModel r = item.getModelObject();
                String rid = r == noneChoice ? null : r.getRID();
                PageParameters params = queryParameters(queryParam, milestoneParam, statiiParam, assignedToParam, sortBy, desc, rid, 1);
                item.add(new LinkPanel("repositoryLink", null, r.toString(), MyTicketsPage.class, params).setRenderBodyOnly(true));
            }
        };
        add(repositoryMenu);
        // 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;
@@ -257,6 +325,7 @@
               q.or(Lucene.status.matches(state));
            }
         }
         if (not) {
            qb.and(q.toString());
         } else {
@@ -264,35 +333,21 @@
         }
      }
      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();
        if (noneChoice != currentRepository && !qb.containsField(Lucene.rid.name())) {
            QueryBuilder q1 = new QueryBuilder();
            q1.and(Lucene.rid.matches(repositoryId));
            qb.and(q1.toSubquery().toString());
      }
      // 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, page, pageSize, results.size(), totalResults);
        buildPager(queryParam, milestoneParam, statiiParam, assignedToParam, sortBy, desc, repositoryId, page, pageSize, results.size(), totalResults);
      final boolean showSwatch = app().settings().getBoolean(Keys.web.repositoryListSwatches, true);
      add(new TicketListPanel("ticketList", results, showSwatch, true));
@@ -305,6 +360,7 @@
         String assignedTo,
         String sort,
         boolean descending,
            String repositoryId,
         int page) {
      PageParameters params = WicketUtils.newRepositoryParameter("");
@@ -328,6 +384,9 @@
      if (!descending) {
         params.add("direction", "asc");
      }
        if (!StringUtils.isEmpty(repositoryId)) {
            params.add(Lucene.rid.name(), repositoryId);
        }
      if (page > 1) {
         params.add("pg", "" + page);
      }
@@ -341,6 +400,7 @@
         final String assignedTo,
         final String sort,
         final boolean desc,
            final String repositoryId,
         final int page,
         int pageSize,
         int count,
@@ -349,8 +409,8 @@
      boolean showNav = total > (2 * pageSize);
      boolean allowPrev = page > 1;
      boolean allowNext = (pageSize * (page - 1) + count) < total;
      add(new BookmarkablePageLink<Void>("prevLink", MyTicketsPage.class, queryParameters(query, milestone, states, assignedTo, sort, desc, page - 1)).setEnabled(allowPrev).setVisible(showNav));
      add(new BookmarkablePageLink<Void>("nextLink", MyTicketsPage.class, queryParameters(query, milestone, states, assignedTo, sort, desc, page + 1)).setEnabled(allowNext).setVisible(showNav));
        add(new BookmarkablePageLink<Void>("prevLink", MyTicketsPage.class, queryParameters(query, milestone, states, assignedTo, sort, desc, repositoryId, page - 1)).setEnabled(allowPrev).setVisible(showNav));
        add(new BookmarkablePageLink<Void>("nextLink", MyTicketsPage.class, queryParameters(query, milestone, states, assignedTo, sort, desc, repositoryId, page + 1)).setEnabled(allowNext).setVisible(showNav));
      if (total <= pageSize) {
         add(new Label("pageLink").setVisible(false));
@@ -379,7 +439,7 @@
         @Override
         public void populateItem(final Item<Integer> item) {
            final Integer i = item.getModelObject();
            LinkPanel link = new LinkPanel("page", null, "" + i, MyTicketsPage.class, queryParameters(query, milestone, states, assignedTo, sort, desc, i));
                LinkPanel link = new LinkPanel("page", null, "" + i, MyTicketsPage.class, queryParameters(query, milestone, states, assignedTo, sort, desc, repositoryId, i));
            link.setRenderBodyOnly(true);
            if (i == page) {
               WicketUtils.setCssClass(item, "active");
@@ -389,4 +449,45 @@
      };
      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())
                || qb.containsField(Lucene.mentions.name())) {
            return qb;
        }
        // general "my tickets"
        return qb.andSubquery()
                 .or(Lucene.createdby.matches(username))
                 .or(Lucene.responsible.matches(username))
                 .or(Lucene.watchedby.matches(username))
                 .or(Lucene.mentions.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());
    }
}