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
76
77
78
79
80
81
82
83
84
85
86
87
88
89

<?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;
use Automattic\WooCommerce\Internal\Integrations\WPPostsImporter;
use Automattic\WooCommerce\Utilities\FeaturesUtil;
use Automattic\WooCommerce\Internal\CLI\Migrator\Runner;

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();

        /**
         * Adds the blueprint CLI initialization to the 'init' hook to prevent premature translation loading.
         *
         * The hook is required because FeaturesUtil::feature_is_enabled() loads translations during the
         * blueprint CLI check. This hook can be removed once FeaturesUtil::feature_is_enabled() is
         * refactored to not load translations.
         *
         * @see https://github.com/woocommerce/woocommerce/issues/56305
         */
        add_action( 'init', array( $this, 'add_blueprint_cli_hook' ) );

        /**
         * Register the WP Posts importer.
         */
        $wp_posts_importer = wc_get_container()->get( WPPostsImporter::class );
        $wp_posts_importer->register();
    }

    /**
     * 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';
    }

    /**
     * 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' );
        if ( defined( 'WOOCOMMERCE_MIGRATOR_ENABLED' ) && WOOCOMMERCE_MIGRATOR_ENABLED ) {
            WP_CLI::add_hook( 'after_wp_load', 'Automattic\WooCommerce\Internal\CLI\Migrator\Runner::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 ) );
    }

    /**
     * Include Blueprint CLI if it's available.
     */
    public function add_blueprint_cli_hook() {
        if ( FeaturesUtil::feature_is_enabled( 'blueprint' ) && class_exists( \Automattic\WooCommerce\Blueprint\Cli::class ) ) {
            require_once dirname( WC_PLUGIN_FILE ) . '/packages/blueprint/src/Cli.php';
            WP_CLI::add_hook( 'after_wp_load', 'Automattic\WooCommerce\Blueprint\Cli::register_commands' );
        }
    }
}

new WC_CLI();