Support querying in tickets ls and accept Object[][] for table data

James Moger

2014-03-23 f81ed74578eb1d0fed26c56ddcfbb39f08a4e47c

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

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

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.RegistrantAccessPermission;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.TeamModel;
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.ListRegexCommand;
import com.gitblit.transport.ssh.commands.SshCommand;
import com.gitblit.utils.FlipTable;
import com.gitblit.utils.FlipTable.Borders;

@CommandMetaData(name = "users", description = "User management commands", admin = true)
public class UsersDispatcher extends DispatchCommand {

   private static final String banner1 = "===========================================================";

   private static final String banner2 = "-----------------------------------------------------------";

   @Override
   protected void setup(UserModel user) {
@@ -61,34 +54,34 @@

         // fields
         String [] fheaders = new String [] { "Field", "Value" };
         String [][] fdata = new String[5][];
         fdata[0] = new String [] { "Email", u.emailAddress };
         fdata[1] = new String [] { "Type", u.accountType.toString() };
         fdata[2] = new String [] { "Can Admin", u.canAdmin() ? "Y":"N" };
         fdata[3] = new String [] { "Can Fork", u.canFork() ? "Y":"N" };
         fdata[4] = new String [] { "Can Create", u.canCreate() ? "Y":"N" };
         Object [][] fdata = new Object[5][];
         fdata[0] = new Object [] { "Email", u.emailAddress };
         fdata[1] = new Object [] { "Type", u.accountType };
         fdata[2] = new Object [] { "Can Admin", u.canAdmin() ? "Y":"N" };
         fdata[3] = new Object [] { "Can Fork", u.canFork() ? "Y":"N" };
         fdata[4] = new Object [] { "Can Create", u.canCreate() ? "Y":"N" };
         String fields = FlipTable.of(fheaders, fdata, Borders.COLS);
			

         // teams
         String [] theaders = new String [] { "Team", "Type" };
         String [][] tdata = new String[u.teams.size()][];
         Object [][] tdata = new Object[u.teams.size()][];
         int i = 0;
         for (TeamModel t : u.teams) {
            tdata[i] = new String [] { t.name, t.accountType.toString() };
            tdata[i] = new Object [] { t.name, t.accountType };
            i++;
         }
         String teams = FlipTable.of(theaders, tdata, Borders.COLS);
			

         // permissions
         List<RegistrantAccessPermission> perms = u.getRepositoryPermissions();
         String[] pheaders = { "Repository", "Permission", "Type", "Source", "Mutable" };
         String [][] pdata = new String[perms.size()][];
         Object [][] pdata = new Object[perms.size()][];
         for (i = 0; i < perms.size(); i++) {
            RegistrantAccessPermission ap = perms.get(i);
            pdata[i] = new String[] { ap.registrant, ap.permission.toString(), ap.permissionType.toString(), ap.source, ap.mutable ? "Y":"N" };
            pdata[i] = new Object[] { ap.registrant, ap.permission, ap.permissionType, ap.source, ap.mutable ? "Y":"N" };
         }
         String permissions = FlipTable.of(pheaders, pdata, Borders.COLS);
			

         // assemble user table
         String [] headers = new String[] { u.getDisplayName() + (u.username.equals(u.getDisplayName()) ? "" : (" (" + u.username + ")")) };
         String[][] data = new String[6][];
@@ -103,43 +96,21 @@
   }

   @CommandMetaData(name = "list", aliases= { "ls" }, description = "List users")
   public static class ListUsers extends SshCommand {

      @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;
   public static class ListUsers extends ListRegexCommand<UserModel> {

      @Override
      public void run() {
      protected List<UserModel> getItems() {
         IGitblit gitblit = getContext().getGitblit();
         List<UserModel> users = gitblit.getAllUsers();

         List<UserModel> filtered;
         if (StringUtils.isEmpty(regexFilter)) {
            // no regex filter 
            filtered = users;
         } else {
            // regex filter the list
            filtered = new ArrayList<UserModel>();
            for (UserModel u : users) {
               if (u.username.matches(regexFilter)) {
                  filtered.add(u);
               }
            }
         }

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

      @Override
      protected boolean matches(UserModel u) {
         return u.username.matches(regexFilter);
      }

      @Override
      protected void asTable(List<UserModel> list) {
         String[] headers;
         if (verbose) {
@@ -150,16 +121,16 @@
            headers = h;
         }

         String[][] data = new String[list.size()][];
         Object[][] data = new Object[list.size()][];
         for (int i = 0; i < list.size(); i++) {
            UserModel u = list.get(i);

            String name = u.disabled ? "-" : ((u.canAdmin() ? "*" : " ")) + u.username;
            if (verbose) {
               data[i] = new String[] { name, u.displayName, u.accountType.name(),
               data[i] = new Object[] { name, u.displayName, u.accountType,
                     u.emailAddress,   u.canCreate() ? "Y":"", u.canFork() ? "Y" : ""};
            } else {
               data[i] = new String[] { name, u.displayName, u.accountType.name(),
               data[i] = new Object[] { name, u.displayName, u.accountType,
                     u.emailAddress };
            }
         }
@@ -168,22 +139,21 @@
         stdout.println();
      }

      @Override
      protected void asTabbed(List<UserModel> users) {
         String pattern;
         if (verbose) {
            pattern = "%s\ts\t%s\t%s\t%s\t%s";
            for (UserModel u : users) {
               outTabbed(u.disabled ? "-" : ((u.canAdmin() ? "*" : " ")) + u.username,
                     u.getDisplayName(),
                     u.accountType,
                     u.emailAddress == null ? "" : u.emailAddress,
                     u.canCreate() ? "Y":"",
                     u.canFork() ? "Y" : "");
            }
         } else {
            pattern = "%s";
         }

         for (UserModel u : users) {
            stdout.println(String.format(pattern,
                  u.disabled ? "-" : ((u.canAdmin() ? "*" : " ")) + u.username,
                  u.getDisplayName(),
                  u.accountType,
                  u.emailAddress == null ? "" : u.emailAddress,
                  u.canCreate() ? "Y":"",
                  u.canFork() ? "Y" : ""));
            for (UserModel u : users) {
               outTabbed(u.disabled ? "-" : ((u.canAdmin() ? "*" : " ")) + u.username);
            }
         }
      }
   }