Support regex filtering the projects, repositories, and users list

James Moger

2014-03-22 5cc48016caf4de3ecce6fbf2f2a34d3bde99edb6

 src/main/java/com/gitblit/transport/ssh/gitblit/ProjectsDispatcher.java

@@ -15,19 +15,23 @@
 */
package com.gitblit.transport.ssh.gitblit;

import java.text.MessageFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.List;

import org.kohsuke.args4j.Argument;
import org.kohsuke.args4j.Option;
import org.parboiled.common.StringUtils;

import com.gitblit.manager.IGitblit;
import com.gitblit.models.ProjectModel;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.UserModel;
import com.gitblit.transport.ssh.commands.CommandMetaData;
import com.gitblit.transport.ssh.commands.DispatchCommand;
import com.gitblit.transport.ssh.commands.SshCommand;
import com.gitblit.utils.FlipTable;
import com.gitblit.utils.FlipTable.Borders;

@CommandMetaData(name = "projects", description = "Project management commands")
public class ProjectsDispatcher extends DispatchCommand {
@@ -44,36 +48,73 @@
      @Option(name = "--verbose", aliases = { "-v" }, usage = "verbose")
      private boolean verbose;

      @Option(name = "--tabbed", aliases = { "-t" }, usage = "as tabbed output")
      private boolean tabbed;

      @Argument(index = 0, metaVar = "REGEX", usage = "regex filter expression")
      protected String regexFilter;

      @Override
      public void run() {
         IGitblit gitblit = getContext().getGitblit();
         UserModel user = getContext().getClient().getUser();
         SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd");

         List<ProjectModel> projects = gitblit.getProjectModels(user, false);
         int nameLen = 0;
         int descLen = 0;
         for (ProjectModel project : projects) {
            int len = project.name.length();
            if (len > nameLen) {
               nameLen = len;
            }
            if (!StringUtils.isEmpty(project.description)) {
               len = project.description.length();
               if (len > descLen) {
                  descLen = len;
         List<ProjectModel> filtered;
         if (StringUtils.isEmpty(regexFilter)) {
            // no regex filter 
            filtered = projects;
         } else {
            // regex filter the list
            filtered = new ArrayList<ProjectModel>();
            for (ProjectModel p : projects) {
               if (p.name.matches(regexFilter)) {
                  filtered.add(p);
               }
            }
         }

         if (tabbed) {
            asTabbed(filtered);
         } else {
            asTable(filtered);
         }
      }

      protected void asTable(List<ProjectModel> list) {
         String[] headers;
         if (verbose) {
            String[] h = { "Name", "Description", "Last Modified", "# Repos" };
            headers = h;
         } else {
            String[] h = { "Name", "Last Modified", "# Repos" };
            headers = h;
         }

         SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd");
         String[][] data = new String[list.size()][];
         for (int i = 0; i < list.size(); i++) {
            ProjectModel p = list.get(i);

            if (verbose) {
               data[i] = new String[] { p.name, p.description, df.format(p.lastChange), "" + p.repositories.size() };
            } else {
               data[i] = new String[] { p.name, df.format(p.lastChange), "" + p.repositories.size() };
            }
         }
         stdout.println(FlipTable.of(headers, data, Borders.BODY_HCOLS));
      }

      protected void asTabbed(List<ProjectModel> list) {
         String pattern;
         if (verbose) {
            pattern = MessageFormat.format("%-{0,number,0}s\t%-{1,number,0}s\t%s", nameLen, descLen);
            pattern = "%s\t%s\t%s";
         } else {
            pattern = "%s";
         }

         for (ProjectModel project : projects) {
         SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd");
         for (ProjectModel project : list) {
            stdout.println(String.format(pattern,
                  project.name,
                  project.description == null ? "" : project.description,