| 1 | diff --git a/src/wp-admin/includes/export.php b/src/wp-admin/includes/export.php |
|---|
| 2 | index 6d01a2c..b200c0d 100644 |
|---|
| 3 | --- a/src/wp-admin/includes/export.php |
|---|
| 4 | +++ b/src/wp-admin/includes/export.php |
|---|
| 5 | @@ -42,6 +42,7 @@ function export_wp( $args = array() ) { |
|---|
| 6 | */ |
|---|
| 7 | do_action( 'export_wp', $args ); |
|---|
| 8 | |
|---|
| 9 | + // Set export file name |
|---|
| 10 | $sitename = sanitize_key( get_bloginfo( 'name' ) ); |
|---|
| 11 | if ( ! empty( $sitename ) ) { |
|---|
| 12 | $sitename .= '.'; |
|---|
| 13 | @@ -59,48 +60,98 @@ function export_wp( $args = array() ) { |
|---|
| 14 | */ |
|---|
| 15 | $filename = apply_filters( 'export_wp_filename', $wp_filename, $sitename, $date ); |
|---|
| 16 | |
|---|
| 17 | + // Set content headers |
|---|
| 18 | header( 'Content-Description: File Transfer' ); |
|---|
| 19 | header( 'Content-Disposition: attachment; filename=' . $filename ); |
|---|
| 20 | header( 'Content-Type: text/xml; charset=' . get_option( 'blog_charset' ), true ); |
|---|
| 21 | |
|---|
| 22 | - if ( 'all' != $args['content'] && post_type_exists( $args['content'] ) ) { |
|---|
| 23 | - $ptype = get_post_type_object( $args['content'] ); |
|---|
| 24 | - if ( ! $ptype->can_export ) |
|---|
| 25 | - $args['content'] = 'post'; |
|---|
| 26 | + $query_args = array( |
|---|
| 27 | + 'post_status' => 'any', |
|---|
| 28 | + 'posts_per_page' => -1 |
|---|
| 29 | + ); |
|---|
| 30 | |
|---|
| 31 | - $where = $wpdb->prepare( "{$wpdb->posts}.post_type = %s", $args['content'] ); |
|---|
| 32 | + if ( 'all' != $args['content'] ) { |
|---|
| 33 | + $post_types = (array) $args['content']; |
|---|
| 34 | + |
|---|
| 35 | + $query_args['post_type'] = array(); |
|---|
| 36 | + |
|---|
| 37 | + foreach( $post_types as $post_type ) { |
|---|
| 38 | + $post_type_object = get_post_type_object( $post_type ); |
|---|
| 39 | + |
|---|
| 40 | + if ( $post_type_object && $post_type_object->can_export ) { |
|---|
| 41 | + $query_args['post_type'][] = $post_type; |
|---|
| 42 | + } |
|---|
| 43 | + } |
|---|
| 44 | + |
|---|
| 45 | + if ( empty( $query_args['post_type'] ) ) { |
|---|
| 46 | + $query_args['post_type'] = 'post'; |
|---|
| 47 | + } |
|---|
| 48 | } else { |
|---|
| 49 | - $post_types = get_post_types( array( 'can_export' => true ) ); |
|---|
| 50 | - $esses = array_fill( 0, count($post_types), '%s' ); |
|---|
| 51 | - $where = $wpdb->prepare( "{$wpdb->posts}.post_type IN (" . implode( ',', $esses ) . ')', $post_types ); |
|---|
| 52 | + $query_args['post_type'] = array_values( get_post_types( array( 'can_export' => true ) ) ); |
|---|
| 53 | } |
|---|
| 54 | |
|---|
| 55 | - if ( $args['status'] && ( 'post' == $args['content'] || 'page' == $args['content'] ) ) |
|---|
| 56 | - $where .= $wpdb->prepare( " AND {$wpdb->posts}.post_status = %s", $args['status'] ); |
|---|
| 57 | - else |
|---|
| 58 | - $where .= " AND {$wpdb->posts}.post_status != 'auto-draft'"; |
|---|
| 59 | + if ( $args['status'] && ( 'post' == $args['content'] || 'page' == $args['content'] ) ) { |
|---|
| 60 | + $query_args['post_status'] = $args['status']; |
|---|
| 61 | + } |
|---|
| 62 | |
|---|
| 63 | - $join = ''; |
|---|
| 64 | if ( $args['category'] && 'post' == $args['content'] ) { |
|---|
| 65 | if ( $term = term_exists( $args['category'], 'category' ) ) { |
|---|
| 66 | - $join = "INNER JOIN {$wpdb->term_relationships} ON ({$wpdb->posts}.ID = {$wpdb->term_relationships}.object_id)"; |
|---|
| 67 | - $where .= $wpdb->prepare( " AND {$wpdb->term_relationships}.term_taxonomy_id = %d", $term['term_taxonomy_id'] ); |
|---|
| 68 | + $query_args['tax_query'] = array( |
|---|
| 69 | + array( |
|---|
| 70 | + 'taxonomy' => 'category', |
|---|
| 71 | + 'terms' => $term->term_id, |
|---|
| 72 | + 'include_children' => false |
|---|
| 73 | + ) |
|---|
| 74 | + ); |
|---|
| 75 | } |
|---|
| 76 | } |
|---|
| 77 | |
|---|
| 78 | - if ( 'post' == $args['content'] || 'page' == $args['content'] || 'attachment' == $args['content'] ) { |
|---|
| 79 | + if ( 'post' == $args['content'] || 'page' == $args['content'] ) { |
|---|
| 80 | if ( $args['author'] ) |
|---|
| 81 | - $where .= $wpdb->prepare( " AND {$wpdb->posts}.post_author = %d", $args['author'] ); |
|---|
| 82 | + $query_args['author'] = (int) $args['author']; |
|---|
| 83 | + |
|---|
| 84 | + $date_query = array(); |
|---|
| 85 | + |
|---|
| 86 | |
|---|
| 87 | if ( $args['start_date'] ) |
|---|
| 88 | - $where .= $wpdb->prepare( " AND {$wpdb->posts}.post_date >= %s", date( 'Y-m-d', strtotime($args['start_date']) ) ); |
|---|
| 89 | + $date_query[]['after'] = $args['start_date']; |
|---|
| 90 | |
|---|
| 91 | - if ( $args['end_date'] ) |
|---|
| 92 | - $where .= $wpdb->prepare( " AND {$wpdb->posts}.post_date < %s", date( 'Y-m-d', strtotime('+1 month', strtotime($args['end_date'])) ) ); |
|---|
| 93 | + if ( $args['end_date'] ) { |
|---|
| 94 | + $date_query[]['before'] = '@' . strtotime( '+1 month', strtotime( $args['end_date'] ) ); |
|---|
| 95 | + } |
|---|
| 96 | + |
|---|
| 97 | + |
|---|
| 98 | + if ( ! empty( $date_query ) ) { |
|---|
| 99 | + $query_args['date_query'] = $date_query; |
|---|
| 100 | + } |
|---|
| 101 | } |
|---|
| 102 | |
|---|
| 103 | - // Grab a snapshot of post IDs, just in case it changes during the export. |
|---|
| 104 | - $post_ids = $wpdb->get_col( "SELECT ID FROM {$wpdb->posts} $join WHERE $where" ); |
|---|
| 105 | + /** |
|---|
| 106 | + * Filter WP_Query arguments for which posts to export. |
|---|
| 107 | + * |
|---|
| 108 | + * @todo Update since |
|---|
| 109 | + * @since 4.X |
|---|
| 110 | + * |
|---|
| 111 | + * @param array $query_args An array of WP_Query arguments. |
|---|
| 112 | + * @param array $args An array of export arguments. |
|---|
| 113 | + */ |
|---|
| 114 | + $query_args = apply_filters( 'export_wp_query_args', $query_args, $args ); |
|---|
| 115 | + |
|---|
| 116 | + $query_args['fields'] = 'ids'; |
|---|
| 117 | + |
|---|
| 118 | + $post_ids = get_posts( $query_args ); |
|---|
| 119 | + |
|---|
| 120 | + /** |
|---|
| 121 | + * Filter Post IDs to be exported. |
|---|
| 122 | + * |
|---|
| 123 | + * @todo Update since |
|---|
| 124 | + * @since 4.X |
|---|
| 125 | + * |
|---|
| 126 | + * @param array $post_ids An array of Post IDs to export. |
|---|
| 127 | + * @param array $query_args An array of WP_Query arguments. |
|---|
| 128 | + * @param array $args An array of export arguments. |
|---|
| 129 | + */ |
|---|
| 130 | + $post_ids = apply_filters( 'export_wp_post_ids', $post_ids, $query_args, $args ); |
|---|
| 131 | |
|---|
| 132 | /* |
|---|
| 133 | * Get the requested terms ready, empty unless posts filtered by category |
|---|
| 134 | @@ -111,29 +162,36 @@ function export_wp( $args = array() ) { |
|---|
| 135 | $cat = get_term( $term['term_id'], 'category' ); |
|---|
| 136 | $cats = array( $cat->term_id => $cat ); |
|---|
| 137 | unset( $term, $cat ); |
|---|
| 138 | - } elseif ( 'all' == $args['content'] ) { |
|---|
| 139 | + } else if ( 'all' == $args['content'] ) { |
|---|
| 140 | $categories = (array) get_categories( array( 'get' => 'all' ) ); |
|---|
| 141 | $tags = (array) get_tags( array( 'get' => 'all' ) ); |
|---|
| 142 | |
|---|
| 143 | $custom_taxonomies = get_taxonomies( array( '_builtin' => false ) ); |
|---|
| 144 | $custom_terms = (array) get_terms( $custom_taxonomies, array( 'get' => 'all' ) ); |
|---|
| 145 | |
|---|
| 146 | - // Put categories in order with no child going before its parent. |
|---|
| 147 | - while ( $cat = array_shift( $categories ) ) { |
|---|
| 148 | - if ( $cat->parent == 0 || isset( $cats[$cat->parent] ) ) |
|---|
| 149 | - $cats[$cat->term_id] = $cat; |
|---|
| 150 | - else |
|---|
| 151 | - $categories[] = $cat; |
|---|
| 152 | + // Put categories in order with no child going before its parent |
|---|
| 153 | + if( $categories ) { |
|---|
| 154 | + while ( $cat = array_shift( $categories ) ) { |
|---|
| 155 | + if ( $cat->parent == 0 || isset( $cats[$cat->parent] ) ) { |
|---|
| 156 | + $cats[$cat->term_id] = $cat; |
|---|
| 157 | + } else { |
|---|
| 158 | + $categories[] = $cat; |
|---|
| 159 | + } |
|---|
| 160 | + } |
|---|
| 161 | } |
|---|
| 162 | |
|---|
| 163 | - // Put terms in order with no child going before its parent. |
|---|
| 164 | - while ( $t = array_shift( $custom_terms ) ) { |
|---|
| 165 | - if ( $t->parent == 0 || isset( $terms[$t->parent] ) ) |
|---|
| 166 | - $terms[$t->term_id] = $t; |
|---|
| 167 | - else |
|---|
| 168 | - $custom_terms[] = $t; |
|---|
| 169 | + // Put terms in order with no child going before its parent |
|---|
| 170 | + if( $custom_terms ) { |
|---|
| 171 | + while ( $t = array_shift( $custom_terms ) ) { |
|---|
| 172 | + if ( $t->parent == 0 || isset( $terms[$t->parent] ) ) { |
|---|
| 173 | + $terms[$t->term_id] = $t; |
|---|
| 174 | + } else { |
|---|
| 175 | + $custom_terms[] = $t; |
|---|
| 176 | + } |
|---|
| 177 | + } |
|---|
| 178 | } |
|---|
| 179 | |
|---|
| 180 | + // Clean up |
|---|
| 181 | unset( $categories, $custom_taxonomies, $custom_terms ); |
|---|
| 182 | } |
|---|
| 183 | |
|---|
| 184 | @@ -401,24 +459,27 @@ function export_wp( $args = array() ) { |
|---|
| 185 | do_action( 'rss2_head' ); |
|---|
| 186 | ?> |
|---|
| 187 | |
|---|
| 188 | -<?php if ( $post_ids ) { |
|---|
| 189 | - /** |
|---|
| 190 | - * @global WP_Query $wp_query |
|---|
| 191 | - */ |
|---|
| 192 | - global $wp_query; |
|---|
| 193 | - |
|---|
| 194 | - // Fake being in the loop. |
|---|
| 195 | - $wp_query->in_the_loop = true; |
|---|
| 196 | - |
|---|
| 197 | - // Fetch 20 posts at a time rather than loading the entire table into memory. |
|---|
| 198 | - while ( $next_posts = array_splice( $post_ids, 0, 20 ) ) { |
|---|
| 199 | - $where = 'WHERE ID IN (' . join( ',', $next_posts ) . ')'; |
|---|
| 200 | - $posts = $wpdb->get_results( "SELECT * FROM {$wpdb->posts} $where" ); |
|---|
| 201 | - |
|---|
| 202 | - // Begin Loop. |
|---|
| 203 | - foreach ( $posts as $post ) { |
|---|
| 204 | - setup_postdata( $post ); |
|---|
| 205 | - $is_sticky = is_sticky( $post->ID ) ? 1 : 0; |
|---|
| 206 | + <?php |
|---|
| 207 | + if ( $post_ids ) { |
|---|
| 208 | + $posts_per_page = 20; |
|---|
| 209 | + |
|---|
| 210 | + $wp_query_args = array( |
|---|
| 211 | + 'post_type' => 'any', |
|---|
| 212 | + 'post_status' => 'any', |
|---|
| 213 | + 'posts_per_page' => $posts_per_page, |
|---|
| 214 | + 'post__in' => array_slice( $post_ids, 0, $posts_per_page ) |
|---|
| 215 | + ); |
|---|
| 216 | + |
|---|
| 217 | + $query = new WP_Query( $wp_query_args ); |
|---|
| 218 | + |
|---|
| 219 | + $page = 0; |
|---|
| 220 | + |
|---|
| 221 | + // Paginate posts 20 at a time |
|---|
| 222 | + while ( $query->have_posts() ) { |
|---|
| 223 | + // Begin Loop. |
|---|
| 224 | + while ( $query->have_posts() ) { |
|---|
| 225 | + $query->the_post(); |
|---|
| 226 | + $is_sticky = is_sticky( $post->ID ) ? 1 : 0; |
|---|
| 227 | ?> |
|---|
| 228 | <item> |
|---|
| 229 | <title><?php |
|---|
| 230 | @@ -532,9 +593,17 @@ function export_wp( $args = array() ) { |
|---|
| 231 | <?php endforeach; ?> |
|---|
| 232 | </item> |
|---|
| 233 | <?php |
|---|
| 234 | + } |
|---|
| 235 | + |
|---|
| 236 | + // Fetch next set of posts |
|---|
| 237 | + $wp_query_args['post__in'] = array_slice( $post_ids, ( $page * $posts_per_page ), $posts_per_page ); |
|---|
| 238 | + |
|---|
| 239 | + $query->query( $wp_query_args ); |
|---|
| 240 | + |
|---|
| 241 | + $page++; |
|---|
| 242 | + } |
|---|
| 243 | } |
|---|
| 244 | - } |
|---|
| 245 | -} ?> |
|---|
| 246 | +?> |
|---|
| 247 | </channel> |
|---|
| 248 | </rss> |
|---|
| 249 | <?php |
|---|