feat: add ability to select which databases to give a user access to

carlalexander · carlalexander · commit 0351114db3d2 · 2020-11-17T01:22:18.000-05:00
diff --git a/src/ApiClient.php b/src/ApiClient.php
@@ -133,9 +133,10 @@ public function createDatabaseServer(string $name, int $networkId, string $type,
     /**
      * Create a new user on the given database server.
      */
-    public function createDatabaseUser(int $databaseId, string $username): Collection
+    public function createDatabaseUser(int $databaseId, string $username, array $databases = []): Collection
     {
         return $this->request('post', "/database-servers/{$databaseId}/users", [
+            'databases' => $databases,
             'username' => $username,
         ]);
     }
diff --git a/src/Command/Database/CreateDatabaseUserCommand.php b/src/Command/Database/CreateDatabaseUserCommand.php
@@ -44,13 +44,18 @@ protected function configure()
     protected function perform(InputInterface $input, ConsoleOutput $output)
     {
         $databaseId = $this->determineDatabaseServer('On which database server would you like to create the new database user?', $input, $output);
+        $databases = [];
         $username = $this->getStringArgument($input, 'username');
 
         if (empty($username) && $input->isInteractive()) {
             $username = $output->ask('What is the username of the new database user');
         }
 
-        $user = $this->apiClient->createDatabaseUser($databaseId, $username);
+        if (!$output->confirm('Do you want to the new user to have access to all databases?', false)) {
+            $databases = $output->multichoice('Please enter the comma-separated list of databases that you want the user to have access to', $this->apiClient->getDatabases($databaseId)->all());
+        }
+
+        $user = $this->apiClient->createDatabaseUser($databaseId, $username, $databases);
 
         $output->horizontalTable(
             ['Username', 'Password'],
diff --git a/src/Console/ConsoleOutput.php b/src/Console/ConsoleOutput.php
@@ -116,6 +116,22 @@ public function infoWithWarning(string $message, string $warning, bool $newline
         $this->write(sprintf('<info>%s</info> (<comment>%s</comment>)', $message, $warning), $newline);
     }
 
+    /**
+     * Ask a multiselect choice question.
+     */
+    public function multichoice($question, array $choices, $default = null): array
+    {
+        if (null !== $default) {
+            $values = array_flip($choices);
+            $default = $values[$default];
+        }
+
+        $question = new ChoiceQuestion($question, $choices, $default);
+        $question->setMultiselect(true);
+
+        return (array) $this->askQuestion($question);
+    }
+
     /**
      * Write out a warning message.
      */

Original file line number	Diff line number	Diff line change
`@@ -133,9 +133,10 @@ public function createDatabaseServer(string $name, int $networkId, string $type,`
`133`	`133`	`/**`
`134`	`134`	`* Create a new user on the given database server.`
`135`	`135`	`*/`
`136`		`- public function createDatabaseUser(int $databaseId, string $username): Collection`
	`136`	`+ public function createDatabaseUser(int $databaseId, string $username, array $databases = []): Collection`
`137`	`137`	`{`
`138`	`138`	`return $this->request('post', "/database-servers/{$databaseId}/users", [`
	`139`	`+ 'databases' => $databases,`
`139`	`140`	`'username' => $username,`
`140`	`141`	`]);`
`141`	`142`	`}`