1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

<?php
/**
 * Enables WooCommerce, via the command line.
 *
 * @package WooCommerce\CLI
 * @version 3.0.0
 */

use Automattic\WooCommerce\Database\Migrations\CustomOrderTable\CLIRunner as CustomOrdersTableCLIRunner;
use Automattic\WooCommerce\Internal\ProductAttributesLookup\CLIRunner as ProductAttributesLookupCLIRunner;

defined( 'ABSPATH' ) || exit;

/**
 * CLI class.
 */
class WC_CLI {
    /**
     * Load required files and hooks to make the CLI work.
     */
    public function __construct() {
        $this->includes();
        $this->hooks();
    }

    /**
     * Load command files.
     */
    private function includes() {
        require_once __DIR__ . '/cli/class-wc-cli-runner.php';
        require_once __DIR__ . '/cli/class-wc-cli-rest-command.php';
        require_once __DIR__ . '/cli/class-wc-cli-tool-command.php';
        require_once __DIR__ . '/cli/class-wc-cli-update-command.php';
        require_once __DIR__ . '/cli/class-wc-cli-tracker-command.php';
        require_once __DIR__ . '/cli/class-wc-cli-com-command.php';
        require_once __DIR__ . '/cli/class-wc-cli-com-extension-command.php';
        $this->maybe_include_blueprint_cli();
    }

    /**
     * Sets up and hooks WP CLI to our CLI code.
     */
    private function hooks() {
        WP_CLI::add_hook( 'after_wp_load', 'WC_CLI_Runner::after_wp_load' );
        WP_CLI::add_hook( 'after_wp_load', 'WC_CLI_Tool_Command::register_commands' );
        WP_CLI::add_hook( 'after_wp_load', 'WC_CLI_Update_Command::register_commands' );
        WP_CLI::add_hook( 'after_wp_load', 'WC_CLI_Tracker_Command::register_commands' );
        WP_CLI::add_hook( 'after_wp_load', 'WC_CLI_COM_Command::register_commands' );
        WP_CLI::add_hook( 'after_wp_load', 'WC_CLI_COM_Extension_Command::register_commands' );
        $cli_runner = wc_get_container()->get( CustomOrdersTableCLIRunner::class );
        WP_CLI::add_hook( 'after_wp_load', array( $cli_runner, 'register_commands' ) );
        $cli_runner = wc_get_container()->get( ProductAttributesLookupCLIRunner::class );
        WP_CLI::add_hook( 'after_wp_load', fn() => \WP_CLI::add_command( 'wc palt', $cli_runner ) );

        if ( class_exists( \Automattic\WooCommerce\Blueprint\Cli::class ) ) {
            WP_CLI::add_hook( 'after_wp_load', 'Automattic\WooCommerce\Blueprint\Cli::register_commands' );
        }
    }

    /**
     * Include Blueprint CLI if it's available.
     */
    private function maybe_include_blueprint_cli() {
        if ( ! function_exists( 'wc_admin_get_feature_config' ) ) {
            require_once WC_ABSPATH . 'includes/react-admin/feature-config.php';
        }

        $features = wc_admin_get_feature_config();
        if ( isset( $features['blueprint'] ) ) {
            require_once dirname( WC_PLUGIN_FILE ) . '/vendor/woocommerce/blueprint/src/Cli.php';
        }
    }
}

new WC_CLI();