| @@ -1,15 +1,16 @@ | ||
| 1 | 1 | <?php |
| 2 | + | |
| 3 | +use wpCloud\StatelessMedia\Migrator; | |
| 4 | +use \wpCloud\StatelessMedia\Batch\BatchTaskManager; | |
| 5 | + | |
| 2 | 6 | /** |
| 3 | 7 | * WP CLI SM Commands |
| 4 | - * | |
| 5 | 8 | */ |
| 9 | +if (defined('WP_CLI') && WP_CLI && class_exists('WP_CLI_Command')) { | |
| 6 | 10 | |
| 7 | -if( defined( 'WP_CLI' ) && WP_CLI ) { | |
| 8 | - | |
| 9 | 11 | /** |
| 10 | 12 | * WP-CLI command |
| 11 | - * | |
| 12 | 13 | */ |
| 13 | 14 | class SM_CLI_Command extends WP_CLI_Command { |
| 14 | 15 | |
| 15 | 16 | public $url; |
| @@ -17,35 +18,135 @@ | ||
| 17 | 18 | /** |
| 18 | 19 | * @param $args |
| 19 | 20 | * @param $assoc_args |
| 20 | 21 | */ |
| 21 | - public function __construct( $args = array(), $assoc_args = array() ) { | |
| 22 | + public function __construct($args = array(), $assoc_args = array()) { | |
| 22 | 23 | parent::__construct(); |
| 23 | 24 | |
| 24 | - if ( php_sapi_name() != 'cli' ) { | |
| 25 | + if (php_sapi_name() != 'cli') { | |
| 25 | 26 | die('Must run from command line'); |
| 26 | 27 | } |
| 27 | 28 | |
| 28 | 29 | //** Setup some server settings */ |
| 29 | - set_time_limit( 0 ); | |
| 30 | - ini_set( 'memory_limit', '2G' ); | |
| 30 | + set_time_limit(0); | |
| 31 | + ini_set('memory_limit', '2G'); | |
| 31 | 32 | //** Setup error handling */ |
| 32 | - error_reporting( E_ALL ); | |
| 33 | - ini_set( 'display_errors', 1 ); | |
| 34 | - ini_set( 'log_errors', 0 ); | |
| 35 | - ini_set( 'html_errors', 0 ); | |
| 33 | + ini_set('display_errors', 1); | |
| 34 | + ini_set('log_errors', 0); | |
| 35 | + ini_set('html_errors', 0); | |
| 36 | 36 | |
| 37 | - if( !class_exists( 'SM_CLI_Process' ) ) { | |
| 38 | - require_once( dirname( __FILE__ ) . '/class-sm-cli-process.php' ); | |
| 37 | + if (!class_exists('SM_CLI_Process')) { | |
| 38 | + require_once(dirname(__FILE__) . '/class-sm-cli-process.php'); | |
| 39 | 39 | } |
| 40 | 40 | |
| 41 | - if( !class_exists( 'SM_CLI' ) ) { | |
| 42 | - require_once( dirname( __FILE__ ) . '/class-sm-cli.php' ); | |
| 41 | + if (!class_exists('SM_CLI')) { | |
| 42 | + require_once(dirname(__FILE__) . '/class-sm-cli.php'); | |
| 43 | 43 | } |
| 44 | 44 | |
| 45 | 45 | /** Be sure that we add url parameter to commands if we have MultiSite installation. */ |
| 46 | - $this->url = is_multisite() ? WP_CLI::get_runner()->config[ 'url' ] : false; | |
| 46 | + $this->url = is_multisite() ? WP_CLI::get_runner()->config['url'] : false; | |
| 47 | + } | |
| 47 | 48 | |
| 49 | + /** | |
| 50 | + * Sync Data | |
| 51 | + * | |
| 52 | + * ## OPTIONS | |
| 53 | + * | |
| 54 | + * <type> | |
| 55 | + * : Which data we want to sync. May be images or files. | |
| 56 | + * | |
| 57 | + * --url | |
| 58 | + * : Blog URL if multisite installation. | |
| 59 | + * | |
| 60 | + * --start | |
| 61 | + * : Indent (sql start). It's ignored on batches. | |
| 62 | + * | |
| 63 | + * --limit | |
| 64 | + * : Limit per query (sql limit) | |
| 65 | + * | |
| 66 | + * --end | |
| 67 | + * : Where ( on which row ) we should stop script. It's ignored on batches | |
| 68 | + * | |
| 69 | + * --batch | |
| 70 | + * : Number of Batch. Default is 1. | |
| 71 | + * | |
| 72 | + * --batches | |
| 73 | + * : General amount of batches. | |
| 74 | + * | |
| 75 | + * --b | |
| 76 | + * : Runs command using batches till it's done. Other parameters will be ignored. There are 10 batches by default. Batch is external command process | |
| 77 | + * | |
| 78 | + * --log | |
| 79 | + * : Show more information in command line | |
| 80 | + * | |
| 81 | + * --o | |
| 82 | + * : Process includes database optimization and transient removing. | |
| 83 | + * | |
| 84 | + * --order | |
| 85 | + * : Order. May be ASC or DESC | |
| 86 | + * | |
| 87 | + * ## EXAMPLES | |
| 88 | + * | |
| 89 | + * wp stateless sync images --url=example.com --b | |
| 90 | + * : Run process looping 10 batches. Every batch is external command 'wp stateless sync images --url=example.com --batch=<number> --batches=10' | |
| 91 | + * | |
| 92 | + * wp stateless sync images --url=example.com --b --batches=100 | |
| 93 | + * : Run process looping 100 batches. | |
| 94 | + * | |
| 95 | + * wp stateless sync images --url=example.com --b --batches=10 --batch=2 | |
| 96 | + * : Run second batch from 10 batches manually. | |
| 97 | + * | |
| 98 | + * wp stateless sync images --url=example.com --log | |
| 99 | + * : Run default process showing additional information in command line. | |
| 100 | + * | |
| 101 | + * wp stateless sync images --url=example.com --end=3000 --limit=50 | |
| 102 | + * : Run process from 1 to 3000 row. Splits process by limiting queries to 50 rows. So, the current example does 60 queries ( 3000 / 50 = 60 ) | |
| 103 | + * | |
| 104 | + * wp stateless sync images --url=example.com --start=777 --end=3000 --o | |
| 105 | + * : Run process from 777 to 3000 row. Also does database optimization and removes transient in the end. | |
| 106 | + * | |
| 107 | + * @synopsis <type> [--url=<val>] [--start=<val>] [--limit=<val>] [--end=<val>] [--batch=<val>] [--batches=<val>] [--b] [--log] [--o] [--order=<val>] | |
| 108 | + * @param $args | |
| 109 | + * @param $assoc_args | |
| 110 | + */ | |
| 111 | + public function sync($args, $assoc_args) { | |
| 112 | + | |
| 113 | + $sm_mode = ud_get_stateless_media()->get('sm.mode'); | |
| 114 | + if ($sm_mode === 'stateless') { | |
| 115 | + WP_CLI::error('Sync is not supported in Stateless mode'); | |
| 116 | + } | |
| 117 | + //** DB Optimization process */ | |
| 118 | + if (isset($assoc_args['o'])) { | |
| 119 | + $this->_before_command_run(); | |
| 120 | + } | |
| 121 | + //** Run batches */ | |
| 122 | + if (isset($assoc_args['b'])) { | |
| 123 | + if (empty($args[0])) { | |
| 124 | + WP_CLI::error('Invalid type parameter'); | |
| 125 | + } | |
| 126 | + $this->_run_batches('sync', $args[0], $assoc_args); | |
| 127 | + } | |
| 128 | + //** Or run command as is. */ | |
| 129 | + else { | |
| 130 | + if (!class_exists('SM_CLI_Sync')) { | |
| 131 | + require_once(dirname(__FILE__) . '/class-sm-cli-sync.php'); | |
| 132 | + } | |
| 133 | + if (class_exists('SM_CLI_Sync')) { | |
| 134 | + $object = new SM_CLI_Sync($args, $assoc_args); | |
| 135 | + $controller = !empty($args[0]) ? $args[0] : false; | |
| 136 | + if ($controller && is_callable(array($object, $controller))) { | |
| 137 | + call_user_func(array($object, $controller)); | |
| 138 | + } else { | |
| 139 | + WP_CLI::error('Invalid type parameter'); | |
| 140 | + } | |
| 141 | + } else { | |
| 142 | + WP_CLI::error('Class SM_CLI_Sync is undefined.'); | |
| 143 | + } | |
| 144 | + } | |
| 145 | + //** Get rid of all transients and run DB optimization again */ | |
| 146 | + if (isset($assoc_args['o'])) { | |
| 147 | + $this->_after_command_run(); | |
| 148 | + } | |
| 48 | 149 | } |
| 49 | 150 | |
| 50 | 151 | /** |
| 51 | 152 | * Upgrade Data |
| @@ -52,9 +153,9 @@ | ||
| 52 | 153 | * |
| 53 | 154 | * ## OPTIONS |
| 54 | 155 | * |
| 55 | 156 | * <type> |
| 56 | - * : Which data we want to upgrade | |
| 157 | + * : Which data we want to upgrade. Currently only 'meta' type is supported. | |
| 57 | 158 | * |
| 58 | 159 | * --start |
| 59 | 160 | * : Indent (sql start). It's ignored on batches. |
| 60 | 161 | * |
| @@ -77,104 +178,406 @@ | ||
| 77 | 178 | * : Show more information in command line |
| 78 | 179 | * |
| 79 | 180 | * --o |
| 80 | 181 | * : Process includes database optimization and transient removing. |
| 182 | + * | |
| 183 | + * --url | |
| 184 | + * : Blog URL if multisite installation. | |
| 81 | 185 | * |
| 82 | 186 | * ## EXAMPLES |
| 83 | 187 | * |
| 84 | - * wp sm upgrade meta --url=example.com --b | |
| 85 | - * : Run process looping 10 batches. Every batch is external command 'wp sm upgrade meta --url=example.com --batch=<number> --batches=10' | |
| 188 | + * wp stateless upgrade meta --url=example.com --b | |
| 189 | + * : Run process looping 10 batches. Every batch is external command 'wp stateless upgrade meta --url=example.com --batch=<number> --batches=10' | |
| 86 | 190 | * |
| 87 | - * wp sm upgrade meta --url=example.com --b --batches=100 | |
| 191 | + * wp stateless upgrade meta --url=example.com --b --batches=100 | |
| 88 | 192 | * : Run process looping 100 batches. |
| 89 | 193 | * |
| 90 | - * wp sm upgrade meta --url=example.com --b --batches=10 --batch=2 | |
| 194 | + * wp stateless upgrade meta --url=example.com --b --batches=10 --batch=2 | |
| 91 | 195 | * : Run second batch from 10 batches manually. |
| 92 | 196 | * |
| 93 | - * wp sm upgrade meta --url=example.com --log | |
| 197 | + * wp stateless upgrade meta --url=example.com --log | |
| 94 | 198 | * : Run default process showing additional information in command line. |
| 95 | 199 | * |
| 96 | - * wp sm upgrade meta --url=example.com --end=3000 --limit=50 | |
| 200 | + * wp stateless upgrade meta --url=example.com --end=3000 --limit=50 | |
| 97 | 201 | * : Run process from 1 to 3000 row. Splits process by limiting queries to 50 rows. So, the current example does 60 queries ( 3000 / 50 = 60 ) |
| 98 | 202 | * |
| 99 | - * wp sm upgrade meta --url=example.com --start=777 --end=3000 --o | |
| 203 | + * wp stateless upgrade meta --url=example.com --start=777 --end=3000 --o | |
| 100 | 204 | * : Run process from 777 to 3000 row. Also does database optimization and removes transient in the end. |
| 101 | 205 | * |
| 102 | - * @synopsis [<type>] [--start=<val>] [--limit=<val>] [--end=<val>] [--batch=<val>] [--batches=<val>] [--b] [--log] [--o] | |
| 206 | + * @synopsis <type> [--url=<val>] [--start=<val>] [--limit=<val>] [--end=<val>] [--batch=<val>] [--batches=<val>] [--b] [--log] [--o] | |
| 103 | 207 | * @param $args |
| 104 | 208 | * @param $assoc_args |
| 105 | 209 | */ |
| 106 | - public function upgrade( $args, $assoc_args ) { | |
| 210 | + public function upgrade($args, $assoc_args) { | |
| 107 | 211 | //** DB Optimization process */ |
| 108 | - if( isset( $assoc_args[ 'o' ] ) ) { | |
| 212 | + if (isset($assoc_args['o'])) { | |
| 109 | 213 | $this->_before_command_run(); |
| 110 | 214 | } |
| 111 | 215 | //** Run batches */ |
| 112 | - if( isset( $assoc_args[ 'b' ] ) ) { | |
| 113 | - if( empty( $args[0] ) ) { | |
| 114 | - WP_CLI::error( 'Invalid type parameter' ); | |
| 216 | + if (isset($assoc_args['b'])) { | |
| 217 | + if (empty($args[0])) { | |
| 218 | + WP_CLI::error('Invalid type parameter'); | |
| 115 | 219 | } |
| 116 | - $this->_run_batches( 'upgrade', $args[0], $assoc_args ); | |
| 220 | + $this->_run_batches('upgrade', $args[0], $assoc_args); | |
| 117 | 221 | } |
| 118 | 222 | //** Or run command as is. */ |
| 119 | 223 | else { |
| 120 | - if( !class_exists( 'SM_CLI_Upgrade' ) ) { | |
| 121 | - require_once( dirname( __FILE__ ) . '/class-sm-cli-upgrade.php' ); | |
| 224 | + if (!class_exists('SM_CLI_Upgrade')) { | |
| 225 | + require_once(dirname(__FILE__) . '/class-sm-cli-upgrade.php'); | |
| 122 | 226 | } |
| 123 | - if( class_exists( 'SM_CLI_Upgrade' ) ) { | |
| 124 | - $object = new SM_CLI_Upgrade( $args, $assoc_args ); | |
| 125 | - $controller = !empty( $args[0] ) ? $args[0] : false; | |
| 126 | - if( $controller && is_callable( array( $object, $controller ) ) ) { | |
| 127 | - call_user_func( array( $object, $controller ) ); | |
| 227 | + if (class_exists('SM_CLI_Upgrade')) { | |
| 228 | + $object = new SM_CLI_Upgrade($args, $assoc_args); | |
| 229 | + $controller = !empty($args[0]) ? $args[0] : false; | |
| 230 | + if ($controller && is_callable(array($object, $controller))) { | |
| 231 | + call_user_func(array($object, $controller)); | |
| 128 | 232 | } else { |
| 129 | - WP_CLI::error( 'Invalid type parameter' ); | |
| 233 | + WP_CLI::error('Invalid type parameter'); | |
| 130 | 234 | } |
| 131 | 235 | } else { |
| 132 | - WP_CLI::error( 'Class SM_CLI_Upgrade is undefined.' ); | |
| 236 | + WP_CLI::error('Class SM_CLI_Upgrade is undefined.'); | |
| 133 | 237 | } |
| 134 | 238 | } |
| 135 | 239 | //** Get rid of all transients and run DB optimization again */ |
| 136 | - if( isset( $assoc_args[ 'o' ] ) ) { | |
| 240 | + if (isset($assoc_args['o'])) { | |
| 137 | 241 | $this->_after_command_run(); |
| 138 | 242 | } |
| 139 | 243 | } |
| 140 | 244 | |
| 141 | 245 | /** |
| 246 | + * Run migrations | |
| 247 | + * | |
| 248 | + * ## OPTIONS | |
| 249 | + * | |
| 250 | + * [<id|auto>] | |
| 251 | + * : start migration by its ID, or automatically run all pending migrations (auto). Auto mode does not support '--force' parameter. | |
| 252 | + * | |
| 253 | + * --force | |
| 254 | + * : Force starting migration even if it is not pending | |
| 255 | + * | |
| 256 | + * --progress=<interval> | |
| 257 | + * : Monitor migration progress every <interval> seconds (minimum 1) | |
| 258 | + * | |
| 259 | + * --email=<email> | |
| 260 | + * : Send email notification to specified email when migration is finished. By default it uses email from plugin settings. You can also use a list of emails, comma separated. | |
| 261 | + * | |
| 262 | + * --url | |
| 263 | + * : Blog URL if multisite installation. | |
| 264 | + * | |
| 265 | + * --yes | |
| 266 | + * : Confirm automatically. | |
| 267 | + * | |
| 268 | + * | |
| 269 | + * ## EXAMPLES | |
| 270 | + * | |
| 271 | + * wp stateless migrate | |
| 272 | + * : List migrations information. | |
| 273 | + * | |
| 274 | + * wp stateless migrate --url=example.com | |
| 275 | + * : List migrations information for specific blog in multisite network. | |
| 276 | + * | |
| 277 | + * wp stateless migrate --progress=3 | |
| 278 | + * : Display current migration progress every 3 seconds. | |
| 279 | + * | |
| 280 | + * wp stateless migrate 20240216150177 | |
| 281 | + * : Start migration with ID 20240216150177. | |
| 282 | + * | |
| 283 | + * wp stateless migrate auto --email=mail@example.com --yes | |
| 284 | + * : Automatically run all pending migrations without confirmation and send notifications to mail@example.com. | |
| 285 | + * | |
| 286 | + * wp stateless migrate 20240216150177 --progress=2 --yes | |
| 287 | + * : Start migration with ID 20240216150177 without confirmation and display progress every 2 seconds. | |
| 288 | + * | |
| 289 | + * wp stateless migrate 20240216150177 --force --email=mail@example.com,user@domain.com --url=example.com | |
| 290 | + * : Start migration with ID 20240216150177 for specific blog in multisite network. Start migration even if it was already finished or failed. After finishing send email notification to mail@example.com and user@domain.com. | |
| 291 | + * | |
| 292 | + * @synopsis [<id|auto>] [--force] [--progress=<val>] [--email=<val>] [--yes] [--url=<val>] | |
| 293 | + * @param $args | |
| 294 | + * @param $assoc_args | |
| 295 | + */ | |
| 296 | + public function migrate($args, $assoc_args) { | |
| 297 | + $id = $args[0] ?? ''; | |
| 298 | + | |
| 299 | + // No migration ID provided, list all migrations and exit | |
| 300 | + if ( empty($id) && !isset($assoc_args['progress']) ) { | |
| 301 | + $this->_list_migrations(); | |
| 302 | + | |
| 303 | + return; | |
| 304 | + } else if ( !empty($id) ) { | |
| 305 | + if ( $id === 'auto' ) { | |
| 306 | + if ( isset($assoc_args['force']) ) { | |
| 307 | + WP_CLI::error( 'The parameter --force is not supported for auto mode.' ); | |
| 308 | + | |
| 309 | + return; | |
| 310 | + } | |
| 311 | + | |
| 312 | + $this->_auto_migrate($assoc_args); | |
| 313 | + | |
| 314 | + return; | |
| 315 | + } else { | |
| 316 | + $this->_run_migration($id, $assoc_args); | |
| 317 | + } | |
| 318 | + } | |
| 319 | + | |
| 320 | + if ( $id !== 'auto' && isset($assoc_args['progress']) ) { | |
| 321 | + $this->_check_progress($assoc_args['progress']); | |
| 322 | + } | |
| 323 | + } | |
| 324 | + | |
| 325 | + /** | |
| 326 | + * Sets cacheControl meta for all files on Google Cloud Storage to the default value from WP-Stateless settings. | |
| 327 | + * | |
| 328 | + * ## OPTIONS | |
| 329 | + * | |
| 330 | + * --url | |
| 331 | + * : Blog URL if multisite installation. | |
| 332 | + * | |
| 333 | + * ## EXAMPLES | |
| 334 | + * | |
| 335 | + * wp stateless reset_cache_control | |
| 336 | + * : Sync cache control for all files on Google Cloud Storage. | |
| 337 | + * | |
| 338 | + * @synopsis [--url=<val>] | |
| 339 | + * @param $args | |
| 340 | + * @param $assoc_args | |
| 341 | + */ | |
| 342 | + public function reset_cache_control($args, $assoc_args) { | |
| 343 | + $sm_mode = ud_get_stateless_media()->get('sm.mode'); | |
| 344 | + if ( ud_get_stateless_media()->is_mode('stateless') ) { | |
| 345 | + WP_CLI::error('Sync cache control is not supported in Stateless mode'); | |
| 346 | + } | |
| 347 | + | |
| 348 | + global $wpdb; | |
| 349 | + | |
| 350 | + $gs_client = ud_get_stateless_media()->get_client(); | |
| 351 | + $cache_control = ud_get_stateless_media()->get_default_cache_control(); | |
| 352 | + $cache_control = apply_filters('sm:item:cacheControl', ud_get_stateless_media()->get_default_cache_control() ); | |
| 353 | + | |
| 354 | + $table_name = ud_stateless_db()->files; | |
| 355 | + $names = $wpdb->get_col(" | |
| 356 | + SELECT name | |
| 357 | + FROM {$table_name} | |
| 358 | + WHERE post_id IS NOT NULL | |
| 359 | + "); | |
| 360 | + | |
| 361 | + foreach ($names as $name) { | |
| 362 | + if ( !$gs_client->media_exists($name) ) { | |
| 363 | + continue; | |
| 364 | + } | |
| 365 | + | |
| 366 | + $args = [ | |
| 367 | + 'skipLocalCheck' => true, | |
| 368 | + 'force' => false, | |
| 369 | + 'cacheControl' => $cache_control, | |
| 370 | + 'name' => $name, | |
| 371 | + ]; | |
| 372 | + | |
| 373 | + $gs_client->add_media($args); | |
| 374 | + | |
| 375 | + WP_CLI::line("Processed file: '{$name}'"); | |
| 376 | + } | |
| 377 | + } | |
| 378 | + | |
| 379 | + /** | |
| 380 | + * Run all pending migrations | |
| 381 | + * | |
| 382 | + * @param array $assoc_args | |
| 383 | + */ | |
| 384 | + private function _auto_migrate($assoc_args) { | |
| 385 | + $progress = $assoc_args['progress'] ?? 1; | |
| 386 | + | |
| 387 | + do { | |
| 388 | + // We need to omit the cache and get the data directly from the db | |
| 389 | + $migrations = apply_filters('wp_stateless_get_migrations', []); | |
| 390 | + | |
| 391 | + $keys = array_reverse( array_keys($migrations) ); | |
| 392 | + $id = null; | |
| 393 | + | |
| 394 | + // Do we have next pending migration? | |
| 395 | + foreach ($keys as $key) { | |
| 396 | + if ( $migrations[$key]['status'] === Migrator::STATUS_PENDING ) { | |
| 397 | + $id = $key; | |
| 398 | + break; | |
| 399 | + } | |
| 400 | + } | |
| 401 | + | |
| 402 | + if ( !empty($id) ) { | |
| 403 | + $command = "wp stateless migrate $id --yes --progress=$progress"; | |
| 404 | + | |
| 405 | + WP_CLI::line('...'); | |
| 406 | + WP_CLI::line("Launching external command '{$command}'"); | |
| 407 | + WP_CLI::line('Waiting...'); | |
| 408 | + | |
| 409 | + @ob_flush(); | |
| 410 | + flush(); | |
| 411 | + | |
| 412 | + $r = SM_CLI::launch($command, false, true); | |
| 413 | + | |
| 414 | + if ($r->return_code) { | |
| 415 | + WP_CLI::error("Something went wrong. External command process failed."); | |
| 416 | + } else { | |
| 417 | + echo $r->stdout; | |
| 418 | + } | |
| 419 | + | |
| 420 | + continue; | |
| 421 | + } | |
| 422 | + | |
| 423 | + break; | |
| 424 | + | |
| 425 | + } while(true); | |
| 426 | + | |
| 427 | + WP_CLI::success('No pending migrations left.'); | |
| 428 | + } | |
| 429 | + | |
| 430 | + /** | |
| 431 | + * Run the specific migration | |
| 432 | + * | |
| 433 | + * @param string $id | |
| 434 | + * @param array $assoc_args | |
| 435 | + */ | |
| 436 | + private function _run_migration($id, $assoc_args) { | |
| 437 | + $migrations = $this->_get_migrations(); | |
| 438 | + | |
| 439 | + if ( !isset($migrations[$id]) ) { | |
| 440 | + WP_CLI::error("Invalid migration ID: $id"); | |
| 441 | + } | |
| 442 | + | |
| 443 | + $migration = $migrations[$id]; | |
| 444 | + | |
| 445 | + // Check if we can run migration | |
| 446 | + if ( !$migration['can_start'] && !isset($assoc_args['force']) ) { | |
| 447 | + WP_CLI::error( 'Migration ' . $migration['description'] . ' is not ready for starting. ' . PHP_EOL . | |
| 448 | + 'Migration status: ' . $migration['status_text'] . ', ' . strip_tags($migration['message']) . PHP_EOL . | |
| 449 | + 'Please use --force to run it anyway.' | |
| 450 | + ); | |
| 451 | + } | |
| 452 | + | |
| 453 | + $email = $assoc_args['email'] ?? ud_get_stateless_media()->get_notification_email(); | |
| 454 | + | |
| 455 | + WP_CLI::line( 'Please make a backup copy of your database and try not to upload, change or delete your media while the process continues.' . PHP_EOL . | |
| 456 | + "After the process finishes an email will be sent to: $email" . PHP_EOL | |
| 457 | + ); | |
| 458 | + | |
| 459 | + WP_CLI::confirm( "Are you sure you want to run the migration $id?", $assoc_args ); | |
| 460 | + | |
| 461 | + // Run migration | |
| 462 | + Migrator::instance()->start_migration([], [ | |
| 463 | + 'id' => $id, | |
| 464 | + 'is_migration' => true, | |
| 465 | + 'force' => true, | |
| 466 | + ]); | |
| 467 | + | |
| 468 | + WP_CLI::success( "Started migration $id" ); | |
| 469 | + } | |
| 470 | + | |
| 471 | + /** | |
| 472 | + * Get migrations state | |
| 473 | + * | |
| 474 | + * @return array | |
| 475 | + */ | |
| 476 | + private function _get_migrations() { | |
| 477 | + $migrations = apply_filters('wp_stateless_batch_state', [], ['force_migrations' => true]); | |
| 478 | + return $migrations['migrations'] ?? []; | |
| 479 | + } | |
| 480 | + | |
| 481 | + /** | |
| 482 | + * List migrations | |
| 483 | + */ | |
| 484 | + private function _list_migrations() { | |
| 485 | + $migrations = $this->_get_migrations(); | |
| 486 | + | |
| 487 | + if ( empty($migrations) ) { | |
| 488 | + WP_CLI::success('No migrations found'); | |
| 489 | + } | |
| 490 | + | |
| 491 | + $data = []; | |
| 492 | + | |
| 493 | + foreach ($migrations as $id => $migration) { | |
| 494 | + $data[$id] = [ | |
| 495 | + 'id' => $id, | |
| 496 | + 'description' => $migration['description'], | |
| 497 | + 'status' => $migration['status_text'], | |
| 498 | + 'message' => strip_tags($migration['message']), | |
| 499 | + ]; | |
| 500 | + } | |
| 501 | + | |
| 502 | + WP_CLI\Utils\format_items('table', $data, ['id', 'description', 'status', 'message']); | |
| 503 | + } | |
| 504 | + | |
| 505 | + /** | |
| 506 | + * Check progress | |
| 507 | + */ | |
| 508 | + private function _check_progress($progress) { | |
| 509 | + global $wpdb; | |
| 510 | + | |
| 511 | + $sleep = max($progress, 1); | |
| 512 | + $key = BatchTaskManager::instance()->get_state_key(); | |
| 513 | + | |
| 514 | + $sql = "SELECT option_value FROM $wpdb->options WHERE option_name = '%s' LIMIT 1"; | |
| 515 | + $sql = $wpdb->prepare($sql, $key); | |
| 516 | + | |
| 517 | + $description = ''; | |
| 518 | + | |
| 519 | + do { | |
| 520 | + // We need to omit the cache and get the data directly from the db | |
| 521 | + $state = $wpdb->get_var($sql); | |
| 522 | + $state = maybe_unserialize($state); | |
| 523 | + | |
| 524 | + if ( empty($state) || !isset($state['is_migration']) || !$state['is_migration'] ) { | |
| 525 | + $message = empty($description) ? 'Migration finished' : "Migration '$description' finished"; | |
| 526 | + WP_CLI::success($message); | |
| 527 | + | |
| 528 | + return; | |
| 529 | + } | |
| 530 | + | |
| 531 | + $description = $state['description'] ?? ''; | |
| 532 | + $completed = $state['completed'] ?? 0; | |
| 533 | + $total = $state['total'] ?? 0; | |
| 534 | + | |
| 535 | + $percent = $total > 0 ? round($completed / $total * 100, 2) : 0; | |
| 536 | + | |
| 537 | + $message = sprintf("Migration '%s' compeleted %.2f%%: %d of %d items processed", $description, $percent, $completed, $total); | |
| 538 | + WP_CLI::line($message); | |
| 539 | + | |
| 540 | + sleep($sleep); | |
| 541 | + } while (true); | |
| 542 | + } | |
| 543 | + | |
| 544 | + /** | |
| 142 | 545 | * Runs batches |
| 143 | 546 | */ |
| 144 | - private function _run_batches( $method, $type, $assoc_args ) { | |
| 145 | - $batches = isset( $assoc_args[ 'batches' ] ) ? $assoc_args[ 'batches' ] : 10; | |
| 146 | - if( !is_numeric( $batches ) || $batches <= 0 ) { | |
| 147 | - WP_CLI::error( 'Parameter --batches must have numeric value.' ); | |
| 547 | + private function _run_batches($method, $type, $assoc_args) { | |
| 548 | + $batches = isset($assoc_args['batches']) ? $assoc_args['batches'] : 10; | |
| 549 | + if (!is_numeric($batches) || $batches <= 0) { | |
| 550 | + WP_CLI::error('Parameter --batches must have numeric value.'); | |
| 148 | 551 | } |
| 149 | - $limit = isset( $assoc_args[ 'limit' ] ) ? $assoc_args[ 'limit' ] : 100; | |
| 150 | - if( !is_numeric( $limit ) || $limit <= 0 ) { | |
| 151 | - WP_CLI::error( 'Parameter --limit must have numeric value.' ); | |
| 552 | + $limit = isset($assoc_args['limit']) ? $assoc_args['limit'] : 100; | |
| 553 | + if (!is_numeric($limit) || $limit <= 0) { | |
| 554 | + WP_CLI::error('Parameter --limit must have numeric value.'); | |
| 152 | 555 | } |
| 556 | + $force = isset($assoc_args['force']) ? '--force' : ''; | |
| 153 | 557 | |
| 154 | - for( $i=1; $i<=$batches; $i++ ) { | |
| 558 | + for ($i = 1; $i <= $batches; $i++) { | |
| 155 | 559 | |
| 156 | - if( !empty( $this->url ) ) { | |
| 157 | - $command = "wp sm {$method} {$type} --batch={$i} --batches={$batches} --limit={$limit} --url={$this->url}"; | |
| 560 | + if (!empty($this->url)) { | |
| 561 | + $command = "wp stateless {$method} {$type} {$force} --batch={$i} --batches={$batches} --limit={$limit} --url={$this->url}"; | |
| 158 | 562 | } else { |
| 159 | - $command = "wp sm {$method} {$type} --batch={$i} --batches={$batches} --limit={$limit}"; | |
| 563 | + $command = "wp stateless {$method} {$type} {$force} --batch={$i} --batches={$batches} --limit={$limit}"; | |
| 160 | 564 | } |
| 161 | 565 | |
| 162 | - WP_CLI::line( '...' ); | |
| 163 | - WP_CLI::line( "Launching external command '{$command}'" ); | |
| 164 | - WP_CLI::line( 'Waiting...' ); | |
| 566 | + WP_CLI::line('...'); | |
| 567 | + WP_CLI::line("Launching external command '{$command}'"); | |
| 568 | + WP_CLI::line('Waiting...'); | |
| 165 | 569 | |
| 166 | 570 | @ob_flush(); |
| 167 | 571 | flush(); |
| 168 | 572 | |
| 169 | - $r = SM_CLI::launch( $command, false, true ); | |
| 573 | + $r = SM_CLI::launch($command, false, true); | |
| 170 | 574 | |
| 171 | - if( $r->return_code ) { | |
| 172 | - WP_CLI::error( "Something went wrong. External command process failed." ); | |
| 575 | + if ($r->return_code) { | |
| 576 | + WP_CLI::error("Something went wrong. External command process failed."); | |
| 173 | 577 | } else { |
| 174 | 578 | echo $r->stdout; |
| 175 | 579 | } |
| 176 | - | |
| 177 | 580 | } |
| 178 | 581 | } |
| 179 | 582 | |
| 180 | 583 | /** |
| @@ -180,18 +583,18 @@ | ||
| 180 | 583 | /** |
| 181 | 584 | * Optimization process |
| 182 | 585 | * Runs before command's process |
| 183 | 586 | */ |
| 184 | - private function _before_command_run(){ | |
| 185 | - WP_CLI::line( "Starting Database optimization process. Waiting..." ); | |
| 587 | + private function _before_command_run() { | |
| 588 | + WP_CLI::line("Starting Database optimization process. Waiting..."); | |
| 186 | 589 | @ob_flush(); |
| 187 | 590 | flush(); |
| 188 | - $command = !empty( $this->url ) ? "wp db optimize --url={$this->url}" : "wp db optimize"; | |
| 189 | - $r = SM_CLI::launch( $command, false, true ); | |
| 190 | - if( $r->return_code ) { | |
| 191 | - WP_CLI::error( "Something went wrong. Database optimization process failed." ); | |
| 591 | + $command = !empty($this->url) ? "wp db optimize --url={$this->url}" : "wp db optimize"; | |
| 592 | + $r = SM_CLI::launch($command, false, true); | |
| 593 | + if ($r->return_code) { | |
| 594 | + WP_CLI::error("Something went wrong. Database optimization process failed."); | |
| 192 | 595 | } else { |
| 193 | - WP_CLI::success( "Database is optimized" ); | |
| 596 | + WP_CLI::success("Database is optimized"); | |
| 194 | 597 | } |
| 195 | 598 | } |
| 196 | 599 | |
| 197 | 600 | /** |
| @@ -197,35 +600,33 @@ | ||
| 197 | 600 | /** |
| 198 | 601 | * Optimization process |
| 199 | 602 | * Runs after command's process |
| 200 | 603 | */ |
| 201 | - private function _after_command_run(){ | |
| 604 | + private function _after_command_run() { | |
| 202 | 605 | //** Run transient flushing */ |
| 203 | - WP_CLI::line( "Starting remove transient. Waiting..." ); | |
| 606 | + WP_CLI::line("Starting remove transient. Waiting..."); | |
| 204 | 607 | @ob_flush(); |
| 205 | 608 | flush(); |
| 206 | - $command = !empty( $this->url ) ? "wp transient delete-all --url={$this->url}" : "wp transient delete-all"; | |
| 207 | - $r = SM_CLI::launch( $command, false, true ); | |
| 208 | - if( $r->return_code ) { | |
| 209 | - WP_CLI::error( "Something went wrong. Transient process failed." ); | |
| 609 | + $command = !empty($this->url) ? "wp transient delete-all --url={$this->url}" : "wp transient delete-all"; | |
| 610 | + $r = SM_CLI::launch($command, false, true); | |
| 611 | + if ($r->return_code) { | |
| 612 | + WP_CLI::error("Something went wrong. Transient process failed."); | |
| 210 | 613 | } else { |
| 211 | - WP_CLI::success( "Transient is removed" ); | |
| 614 | + WP_CLI::success("Transient is removed"); | |
| 212 | 615 | } |
| 213 | 616 | //** Run MySQL optimization */ |
| 214 | - WP_CLI::line( "Starting Database optimization process. Waiting..." ); | |
| 617 | + WP_CLI::line("Starting Database optimization process. Waiting..."); | |
| 215 | 618 | @ob_flush(); |
| 216 | 619 | flush(); |
| 217 | - $command = !empty( $this->url ) ? "wp db optimize --url={$this->url}" : "wp db optimize"; | |
| 218 | - $r = SM_CLI::launch( $command, false, true ); | |
| 219 | - if( $r->return_code ) { | |
| 220 | - WP_CLI::error( "Something went wrong. Database optimization process failed." ); | |
| 620 | + $command = !empty($this->url) ? "wp db optimize --url={$this->url}" : "wp db optimize"; | |
| 621 | + $r = SM_CLI::launch($command, false, true); | |
| 622 | + if ($r->return_code) { | |
| 623 | + WP_CLI::error("Something went wrong. Database optimization process failed."); | |
| 221 | 624 | } else { |
| 222 | - WP_CLI::success( "Database is optimized" ); | |
| 625 | + WP_CLI::success("Database is optimized"); | |
| 223 | 626 | } |
| 224 | 627 | } |
| 225 | - | |
| 226 | 628 | } |
| 227 | 629 | |
| 228 | 630 | /** Add the commands from above */ |
| 229 | - WP_CLI::add_command( 'sm', 'SM_CLI_Command' ); | |
| 230 | - | |
| 231 | -} | |
| 631 | + WP_CLI::add_command('stateless', 'SM_CLI_Command'); | |
| 632 | +} | |
