initial commit

includes/cli/class-wc-cli-tool-command.php

@@ -0,0 +1,107 @@
+<?php
+/**
+ * WC_CLI_Tool_Command class file.
+ *
+ * @package WooCommerce\CLI
+ */
+
+if ( ! defined( 'ABSPATH' ) ) {
+	exit;
+}
+
+/**
+ * Hooks up our system status tools to the CLI.
+ *
+ * Forked from wp-cli/restful (by Daniel Bachhuber, released under the MIT license https://opensource.org/licenses/MIT).
+ * https://github.com/wp-cli/restful
+ *
+ * @version 3.0.0
+ * @package WooCommerce
+ */
+class WC_CLI_Tool_Command {
+
+	/**
+	 * Registers just a 'list' and 'run' command to the WC CLI
+	 * since we only want to enable certain actions on the system status
+	 * tools endpoints.
+	 */
+	public static function register_commands() {
+		global $wp_rest_server;
+
+		$request       = new WP_REST_Request( 'OPTIONS', '/wc/v2/system_status/tools' );
+		$response      = $wp_rest_server->dispatch( $request );
+		$response_data = $response->get_data();
+		if ( empty( $response_data ) ) {
+			return;
+		}
+
+		$parent             = 'wc tool';
+		$supported_commands = array( 'list', 'run' );
+		foreach ( $supported_commands as $command ) {
+			$synopsis = array();
+			if ( 'run' === $command ) {
+				$synopsis[] = array(
+					'name'        => 'id',
+					'type'        => 'positional',
+					'description' => __( 'The id for the resource.', 'woocommerce' ),
+					'optional'    => false,
+				);
+				$method     = 'update_item';
+				$route      = '/wc/v2/system_status/tools/(?P<id>[\w-]+)';
+			} elseif ( 'list' === $command ) {
+				$synopsis[] = array(
+					'name'        => 'fields',
+					'type'        => 'assoc',
+					'description' => __( 'Limit response to specific fields. Defaults to all fields.', 'woocommerce' ),
+					'optional'    => true,
+				);
+				$synopsis[] = array(
+					'name'        => 'field',
+					'type'        => 'assoc',
+					'description' => __( 'Get the value of an individual field.', 'woocommerce' ),
+					'optional'    => true,
+				);
+				$synopsis[] = array(
+					'name'        => 'format',
+					'type'        => 'assoc',
+					'description' => __( 'Render response in a particular format.', 'woocommerce' ),
+					'optional'    => true,
+					'default'     => 'table',
+					'options'     => array(
+						'table',
+						'json',
+						'csv',
+						'ids',
+						'yaml',
+						'count',
+						'headers',
+						'body',
+						'envelope',
+					),
+				);
+				$method     = 'list_items';
+				$route      = '/wc/v2/system_status/tools';
+			}
+
+			$before_invoke = null;
+			if ( empty( $command_args['when'] ) && WP_CLI::get_config( 'debug' ) ) {
+				$before_invoke = function() {
+					wc_maybe_define_constant( 'SAVEQUERIES', true );
+				};
+			}
+
+			$rest_command = new WC_CLI_REST_Command( 'system_status_tool', $route, $response_data['schema'] );
+
+			WP_CLI::add_command(
+				"{$parent} {$command}",
+				array( $rest_command, $method ),
+				array(
+					'synopsis'      => $synopsis,
+					'when'          => ! empty( $command_args['when'] ) ? $command_args['when'] : '',
+					'before_invoke' => $before_invoke,
+				)
+			);
+		}
+	}
+
+}