Update `wp user-meta` to support `user_login` as a first positional argument

diff --git a/php/commands/user-meta.php b/php/commands/user-meta.php
index 1e2dda0..045bf4d 100644 (file)
--- a/php/commands/user-meta.php
+++ b/php/commands/user-meta.php
@@ -14,6 +14,64 @@
  */
 class User_Meta_Command extends \WP_CLI\CommandWithMeta {
        protected $meta_type = 'user';
+
+       public function __construct() {
+               $this->fetcher = new \WP_CLI\Fetchers\User;
+       }
+
+       /**
+        * Get meta field value.
+        *
+        * @synopsis <user> <key> [--format=<format>]
+        */
+       public function get( $args, $assoc_args ) {
+               $args = $this->replace_login_with_user_id( $args );
+               parent::get( $args, $assoc_args );
+       }
+
+       /**
+        * Delete a meta field.
+        *
+        * @synopsis <user> <key>
+        */
+       public function delete( $args, $assoc_args ) {
+               $args = $this->replace_login_with_user_id( $args );
+               parent::delete( $args, $assoc_args );
+       }
+
+       /**
+        * Add a meta field.
+        *
+        * @synopsis <user> <key> <value> [--format=<format>]
+        */
+       public function add( $args, $assoc_args ) {
+               $args = $this->replace_login_with_user_id( $args );
+               parent::add( $args, $assoc_args );
+       }
+
+       /**
+        * Update a meta field.
+        *
+        * @alias set
+        * @synopsis <user> <key> <value> [--format=<format>]
+        */
+       public function update( $args, $assoc_args ) {
+               $args = $this->replace_login_with_user_id( $args );
+               parent::update( $args, $assoc_args );
+       }
+
+       /**
+        * Replace user_login value with user ID
+        * user-meta is a special case that also supports user_login
+        * 
+        * @param array
+        * @return array
+        */
+       private function replace_login_with_user_id( $args ) {
+               $user = $this->fetcher->get_check( $args[0] );
+               $args[0] = $user->ID;
+               return $args;
+       }
 }
 
 WP_CLI::add_command( 'user-meta', 'User_Meta_Command' );