Added repository counts to filters menu

James Moger

2011-12-26 a22eb4f5fdf2adeef9d7f9bfa0a72736ef4b0b5c

 src/com/gitblit/wicket/pages/RootPage.java

@@ -17,9 +17,13 @@


import java.text.MessageFormat;

import java.util.ArrayList;

import java.util.Collections;

import java.util.HashMap;

import java.util.LinkedHashSet;

import java.util.List;

import java.util.Map;

import java.util.Set;

import java.util.concurrent.atomic.AtomicInteger;

import java.util.regex.Pattern;



import org.apache.wicket.PageParameters;

@@ -70,8 +74,8 @@
      boolean authenticateView = GitBlit.getBoolean(Keys.web.authenticateViewPages, false);

      boolean authenticateAdmin = GitBlit.getBoolean(Keys.web.authenticateAdminPages, true);

      boolean allowAdmin = GitBlit.getBoolean(Keys.web.allowAdministration, true);

		
      if (authenticateAdmin) {			


      if (authenticateAdmin) {

         showAdmin = allowAdmin && GitBlitWebSession.get().canAdmin();

         // authentication requires state and session

         setStatelessHint(false);

@@ -136,7 +140,7 @@
      WicketUtils.setInputPlaceholder(pwField, getString("gb.password"));

      loginForm.add(pwField);

      add(loginForm);

		


      if (authenticateView || authenticateAdmin) {

         loginForm.setVisible(!GitBlitWebSession.get().isLoggedIn());

      } else {

@@ -185,22 +189,38 @@
   protected List<DropDownMenuItem> getFilterMenuItems() {

      final UserModel user = GitBlitWebSession.get().getUser();

      Set<DropDownMenuItem> filters = new LinkedHashSet<DropDownMenuItem>();

      List<RepositoryModel> repositories = GitBlit.self().getRepositoryModels(user);



      // accessible repositories by federation set

      for (RepositoryModel repository : GitBlit.self().getRepositoryModels(user)) {

      Map<String, AtomicInteger> setMap = new HashMap<String, AtomicInteger>();

      for (RepositoryModel repository : repositories) {

         for (String set : repository.federationSets) {

            filters.add(new DropDownMenuItem(set, "set", set));

            String key = set.toLowerCase();

            if (setMap.containsKey(key)) {

               setMap.get(key).incrementAndGet();

            } else {

               setMap.put(key, new AtomicInteger(1));

            }

         }

      }

      if (filters.size() > 0) {

      if (setMap.size() > 0) {

         List<String> sets = new ArrayList<String>(setMap.keySet());

         Collections.sort(sets);

         for (String set : sets) {

            filters.add(new DropDownMenuItem(MessageFormat.format("{0} ({1})", set,

                  setMap.get(set).get()), "set", set));

         }

         // divider

         filters.add(new DropDownMenuItem());

      }



      // user's team memberships

      if (user != null && user.teams.size() > 0) {

         for (TeamModel team : user.teams) {

            filters.add(new DropDownMenuItem(team.name, "team", team.name));

         List<TeamModel> teams = new ArrayList<TeamModel>(user.teams);

         Collections.sort(teams);

         for (TeamModel team : teams) {

            filters.add(new DropDownMenuItem(MessageFormat.format("{0} ({1})", team.name,

                  team.repositories.size()), "team", team.name));

         }

         // divider

         filters.add(new DropDownMenuItem());

@@ -225,7 +245,8 @@


      if (filters.size() > 0) {

         // add All Repositories

         filters.add(new DropDownMenuItem("All Repositories", null, null));

         filters.add(new DropDownMenuItem(MessageFormat.format("All Repositories ({0})",

               repositories.size()), null, null));

      }



      return new ArrayList<DropDownMenuItem>(filters);