Add a ./bin/storage shell command

Summary: Fixes T7078. Adds a `./bin/storage shell` command which passes through to a MySQL shell. This is slightly more convenient than running `mysql` manually.

Test Plan: Ran `./bin/storage shell` and got a MySQL shell.

Reviewers: epriestley, #blessed_reviewers

Reviewed By: epriestley, #blessed_reviewers

Subscribers: Korvin, epriestley

Maniphest Tasks: T7078

Differential Revision: https://secure.phabricator.com/D11548

src/__phutil_library_map__.php

@@ -2462,6 +2462,7 @@ phutil_register_library_map(array(
     'PhabricatorStorageManagementDumpWorkflow' => 'infrastructure/storage/management/workflow/PhabricatorStorageManagementDumpWorkflow.php',
     'PhabricatorStorageManagementProbeWorkflow' => 'infrastructure/storage/management/workflow/PhabricatorStorageManagementProbeWorkflow.php',
     'PhabricatorStorageManagementQuickstartWorkflow' => 'infrastructure/storage/management/workflow/PhabricatorStorageManagementQuickstartWorkflow.php',
+    'PhabricatorStorageManagementShellWorkflow' => 'infrastructure/storage/management/workflow/PhabricatorStorageManagementShellWorkflow.php',
     'PhabricatorStorageManagementStatusWorkflow' => 'infrastructure/storage/management/workflow/PhabricatorStorageManagementStatusWorkflow.php',
     'PhabricatorStorageManagementUpgradeWorkflow' => 'infrastructure/storage/management/workflow/PhabricatorStorageManagementUpgradeWorkflow.php',
     'PhabricatorStorageManagementWorkflow' => 'infrastructure/storage/management/workflow/PhabricatorStorageManagementWorkflow.php',
@@ -5753,6 +5754,7 @@ phutil_register_library_map(array(
     'PhabricatorStorageManagementDumpWorkflow' => 'PhabricatorStorageManagementWorkflow',
     'PhabricatorStorageManagementProbeWorkflow' => 'PhabricatorStorageManagementWorkflow',
     'PhabricatorStorageManagementQuickstartWorkflow' => 'PhabricatorStorageManagementWorkflow',
+    'PhabricatorStorageManagementShellWorkflow' => 'PhabricatorStorageManagementWorkflow',
     'PhabricatorStorageManagementStatusWorkflow' => 'PhabricatorStorageManagementWorkflow',
     'PhabricatorStorageManagementUpgradeWorkflow' => 'PhabricatorStorageManagementWorkflow',
     'PhabricatorStorageManagementWorkflow' => 'PhabricatorManagementWorkflow',

src/infrastructure/storage/management/workflow/PhabricatorStorageManagementDumpWorkflow.php

@@ -32,7 +32,6 @@ final class PhabricatorStorageManagementDumpWorkflow
     list($host, $port) = $this->getBareHostAndPort($api->getHost());
 
     $flag_password = '';
-
     $password = $api->getPassword();
     if ($password) {
       if (strlen($password->openEnvelope())) {
@@ -54,19 +53,4 @@ final class PhabricatorStorageManagementDumpWorkflow
       $databases);
   }
 
-  private function getBareHostAndPort($host) {
-    // Split out port information, since the command-line client requires a
-    // separate flag for the port.
-    $uri = new PhutilURI('mysql://'.$host);
-    if ($uri->getPort()) {
-      $port = $uri->getPort();
-      $bare_hostname = $uri->getDomain();
-    } else {
-      $port = null;
-      $bare_hostname = $host;
-    }
-
-    return array($bare_hostname, $port);
-  }
-
 }

src/infrastructure/storage/management/workflow/PhabricatorStorageManagementShellWorkflow.php

@@ -0,0 +1,38 @@
+<?php
+
+final class PhabricatorStorageManagementShellWorkflow
+  extends PhabricatorStorageManagementWorkflow {
+
+  protected function didConstruct() {
+    $this
+      ->setName('shell')
+      ->setExamples('**shell** [__options__]')
+      ->setSynopsis('Launch an interactive shell.');
+  }
+
+  public function execute(PhutilArgumentParser $args) {
+    $api = $this->getAPI();
+    list($host, $port) = $this->getBareHostAndPort($api->getHost());
+
+    $flag_port = $port
+      ? csprintf('--port %d', $port)
+      : '';
+
+    $flag_password = '';
+    $password = $api->getPassword();
+    if ($password) {
+      if (strlen($password->openEnvelope())) {
+        $flag_password = csprintf('--password=%P', $password);
+      }
+    }
+
+    return phutil_passthru(
+      'mysql --default-character-set=utf8 '.
+      '-u %s %C -h %s %C',
+      $api->getUser(),
+      $flag_password,
+      $host,
+      $flag_port);
+  }
+
+}

src/infrastructure/storage/management/workflow/PhabricatorStorageManagementWorkflow.php

@@ -663,4 +663,19 @@ abstract class PhabricatorStorageManagementWorkflow
     return 2;
   }
 
+  protected final function getBareHostAndPort($host) {
+    // Split out port information, since the command-line client requires a
+    // separate flag for the port.
+    $uri = new PhutilURI('mysql://'.$host);
+    if ($uri->getPort()) {
+      $port = $uri->getPort();
+      $bare_hostname = $uri->getDomain();
+    } else {
+      $port = null;
+      $bare_hostname = $host;
+    }
+
+    return array($bare_hostname, $port);
+  }
+
 }