Merge pull request #60 from ernilambar/59-ids-format-super-admin-list

swissspidy · web-flow · commit 47a2eb3f8fea · 2024-02-26T13:02:52.000+01:00
diff --git a/features/super-admin.feature b/features/super-admin.feature
@@ -180,6 +180,22 @@ Feature: Manage super admins associated with a multisite instance
     And STDOUT should be empty
     And the return code should be 1
 
+  Scenario: List super admins in ids format.
+    Given a WP multisite installation
+
+    When I run `wp user get admin --field=ID`
+    And save STDOUT as {USER_1}
+
+    When I run `wp user create admin2 admin2@example.com --porcelain`
+    And save STDOUT as {USER_2}
+
+    When I run `wp super-admin add admin2`
+    And I run `wp super-admin list --format=ids`
+    Then STDOUT should be:
+      """
+      {USER_1} {USER_2}
+      """
+
   Scenario: Manage a super admin user_login 'admin'
     Given a WP multisite installation
 
diff --git a/src/Super_Admin_Command.php b/src/Super_Admin_Command.php
@@ -49,6 +49,7 @@ public function __construct() {
 	 *   - csv
 	 *   - json
 	 *   - count
+	 *   - ids
 	 *   - yaml
 	 * ---
 	 *
@@ -77,8 +78,20 @@ public function list_subcommand( $_, $assoc_args ) {
 
 				$output_users[] = $output_user;
 			}
-			$formatter = new \WP_CLI\Formatter( $assoc_args, $this->fields );
-			$formatter->display_items( $output_users );
+
+			if ( ! empty( $assoc_args['format'] ) && 'ids' === $assoc_args['format'] ) {
+				$formatter = new \WP_CLI\Formatter( $assoc_args );
+
+				$user_ids = [];
+				foreach ( $super_admins as $user_login ) {
+					$user_obj   = get_user_by( 'login', $user_login );
+					$user_ids[] = $user_obj->ID;
+				}
+				$formatter->display_items( $user_ids );
+			} else {
+				$formatter = new \WP_CLI\Formatter( $assoc_args, $this->fields );
+				$formatter->display_items( $output_users );
+			}
 		}
 	}
 
