Пример #1
0
 /**
  * Perform CLI migration.
  *
  * @param mixed $profile Profile key or array
  *
  * @return void
  */
 protected function _perform_cli_migration($profile)
 {
     $wpmdb_cli = null;
     //load correct cli class
     if (function_exists('wpmdb_pro_cli')) {
         $wpmdb_cli = wpmdb_pro_cli();
     } else {
         $wpmdb_cli = wpmdb_cli();
     }
     if (empty($wpmdb_cli)) {
         WP_CLI::error(__('WP Migrate DB CLI class not available.', 'wp-migrate-db-cli'));
         return;
     }
     $result = $wpmdb_cli->cli_migration($profile);
     if (!is_wp_error($result)) {
         $success_msg = sprintf(__('Export saved to: %s', 'wp-migrate-db-cli'), $result);
         if ('find_replace' === $profile['action']) {
             $success_msg = __('Find & Replace complete', 'wp-migrate-db-cli');
         }
         WP_CLI::success($success_msg);
     } elseif (is_wp_error($result)) {
         WP_CLI::error(WPMDB_CLI::cleanup_message($result->get_error_message()));
     }
 }
Пример #2
0
 /**
  * Get profile data from CLI args.
  *
  * @param array $args
  * @param array $assoc_args
  *
  * @return array|WP_Error
  */
 public function get_profile_data_from_args($args, $assoc_args)
 {
     //load correct cli class
     if (function_exists('wp_migrate_db_pro_cli_addon')) {
         $wpmdb_cli = wp_migrate_db_pro_cli_addon();
     } elseif (function_exists('wpmdb_pro_cli')) {
         $wpmdb_cli = wpmdb_pro_cli();
     } else {
         $wpmdb_cli = wpmdb_cli();
     }
     $unknown_args = $this->get_unknown_args($assoc_args);
     if (!empty($unknown_args)) {
         $message = __('Parameter errors: ', 'wp-migrate-db-cli');
         foreach ($unknown_args as $unknown_arg) {
             $message .= "\n " . sprintf(__('unknown %s parameter', 'wp-migrate-db-cli'), '--' . $unknown_arg);
         }
         if (is_a($wpmdb_cli, 'WPMDBPro_CLI')) {
             $message .= "\n" . __('Please make sure that you have activated the appropriate addons for WP Migrate DB Pro.', 'wp-migrate-db-cli');
         }
         return $wpmdb_cli->cli_error($message);
     }
     if (empty($assoc_args['action'])) {
         return $wpmdb_cli->cli_error(__('Missing action parameter', 'wp-migrate-db-cli'));
     }
     if ('savefile' === $assoc_args['action'] && !empty($assoc_args['export_dest'])) {
         $export_dest = $assoc_args['export_dest'];
     }
     $action = $assoc_args['action'];
     // --find=<old> and --replace=<new>
     $replace_old = array();
     $replace_new = array();
     if (!empty($assoc_args['find'])) {
         $replace_old = str_getcsv($assoc_args['find']);
     } else {
         if ('find_replace' === $assoc_args['action']) {
             return $wpmdb_cli->cli_error(__('Missing find and replace values.', 'wp-migrate-db-cli'));
         }
     }
     if (!empty($assoc_args['replace'])) {
         $replace_new = str_getcsv($assoc_args['replace']);
     }
     if (count($replace_old) !== count($replace_new)) {
         return $wpmdb_cli->cli_error(sprintf(__('%1$s and %2$s must contain the same number of values', 'wp-migrate-db-cli'), '--find', '--replace'));
     }
     array_unshift($replace_old, '');
     array_unshift($replace_new, '');
     // --exclude-spam
     $exclude_spam = intval(isset($assoc_args['exclude-spam']));
     // --gzip-file
     $gzip_file = intval(isset($assoc_args['gzip-file']));
     $select_post_types = array();
     // --exclude-post-revisions
     if (!empty($assoc_args['exclude-post-revisions'])) {
         $select_post_types[] = 'revision';
     }
     $exclude_post_types = count($select_post_types) > 0 ? 1 : 0;
     // --skip-replace-guids
     $replace_guids = 1;
     if (isset($assoc_args['skip-replace-guids'])) {
         $replace_guids = 0;
     }
     $select_tables = array();
     $table_migrate_option = 'migrate_only_with_prefix';
     // --include-transients.
     $exclude_transients = intval(!isset($assoc_args['include-transients']));
     //cleanup filename for exports
     if (!empty($export_dest)) {
         if ($gzip_file) {
             if ('gz' !== pathinfo($export_dest, PATHINFO_EXTENSION)) {
                 if ('sql' === pathinfo($export_dest, PATHINFO_EXTENSION)) {
                     $export_dest .= '.gz';
                 } else {
                     $export_dest .= '.sql.gz';
                 }
             }
         } elseif ('sql' !== pathinfo($export_dest, PATHINFO_EXTENSION)) {
             $export_dest = preg_replace('/(\\.sql)?(\\.gz)?$/i', '', $export_dest) . '.sql';
         }
         // ensure export destination is writable
         if (!@touch($export_dest)) {
             return $wpmdb_cli->cli_error(sprintf(__('Cannot write to file "%1$s". Please ensure that the specified directory exists and is writable.', 'wp-migrate-db-cli'), $export_dest));
         }
     }
     $profile = compact('action', 'replace_old', 'table_migrate_option', 'replace_new', 'select_tables', 'exclude_post_types', 'select_post_types', 'replace_guids', 'exclude_spam', 'gzip_file', 'exclude_transients', 'export_dest');
     $profile = apply_filters('wpmdb_cli_filter_get_profile_data_from_args', $profile, $args, $assoc_args);
     return $profile;
 }