Show root commands in welcome shell and improve command registration

James Moger

2014-03-17 01b529edac600feb3c0750d27d0759b5f49fde72

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

@@ -34,6 +34,7 @@
import com.gitblit.IStoredSettings;
import com.gitblit.Keys;
import com.gitblit.models.UserModel;
import com.gitblit.transport.ssh.commands.DispatchCommand;
import com.gitblit.utils.StringUtils;

/**
@@ -56,7 +57,7 @@
   private static class SendMessage implements Command, SessionAware {

      private final IStoredSettings settings;
      private SshDaemonClient client;
      private ServerSession session;

      private InputStream in;
      private OutputStream out;
@@ -89,7 +90,7 @@

      @Override
      public void setSession(final ServerSession session) {
         this.client = session.getAttribute(SshDaemonClient.KEY);
         this.session = session;
      }

      @Override
@@ -105,16 +106,21 @@

      @Override
      public void destroy() {
         this.session = null;
      }

      String getMessage() {
         SshDaemonClient client = session.getAttribute(SshDaemonClient.KEY);
         UserModel user = client.getUser();

         StringBuilder msg = new StringBuilder();
         msg.append("\r\n");
         msg.append("  Hi ");
         msg.append(user.getDisplayName());
         msg.append(", you have successfully connected to Gitblit over SSH.");
         msg.append(", you have successfully connected to Gitblit over SSH");
         msg.append("\r\n");
         msg.append("with client: ");
         msg.append(session.getClientVersion());
         msg.append("\r\n");
         msg.append("\r\n");

@@ -126,6 +132,12 @@
         msg.append("\r\n");
         msg.append("\r\n");

         // display the core commands
         SshCommandFactory cmdFactory = (SshCommandFactory) session.getFactoryManager().getCommandFactory();
         DispatchCommand root = cmdFactory.createRootDispatcher(client, "");
         String usage = root.usage().replace("\n", "\r\n");
         msg.append(usage);

         return msg.toString();
      }