]> scripts.mit.edu Git - autoinstalls/wordpress.git/blob - wp-admin/includes/class-wp-importer.php
WordPress 3.4.2-scripts
[autoinstalls/wordpress.git] / wp-admin / includes / class-wp-importer.php
1 <?php
2 /**
3  * WP_Importer base class
4  */
5 class WP_Importer {
6         /**
7          * Class Constructor
8          *
9          * @return void
10          */
11         function __construct() {}
12
13         /**
14          * Returns array with imported permalinks from WordPress database
15          *
16          * @param string $bid
17          * @return array
18          */
19         function get_imported_posts( $importer_name, $bid ) {
20                 global $wpdb;
21
22                 $hashtable = array();
23
24                 $limit = 100;
25                 $offset = 0;
26
27                 // Grab all posts in chunks
28                 do {
29                         $meta_key = $importer_name . '_' . $bid . '_permalink';
30                         $sql = $wpdb->prepare( "SELECT post_id, meta_value FROM $wpdb->postmeta WHERE meta_key = '%s' LIMIT %d,%d", $meta_key, $offset, $limit );
31                         $results = $wpdb->get_results( $sql );
32
33                         // Increment offset
34                         $offset = ( $limit + $offset );
35
36                         if ( !empty( $results ) ) {
37                                 foreach ( $results as $r ) {
38                                         // Set permalinks into array
39                                         $hashtable[$r->meta_value] = intval( $r->post_id );
40                                 }
41                         }
42                 } while ( count( $results ) == $limit );
43
44                 // unset to save memory
45                 unset( $results, $r );
46
47                 return $hashtable;
48         }
49
50         /**
51          * Return count of imported permalinks from WordPress database
52          *
53          * @param string $bid
54          * @return int
55          */
56         function count_imported_posts( $importer_name, $bid ) {
57                 global $wpdb;
58
59                 $count = 0;
60
61                 // Get count of permalinks
62                 $meta_key = $importer_name . '_' . $bid . '_permalink';
63                 $sql = $wpdb->prepare( "SELECT COUNT( post_id ) AS cnt FROM $wpdb->postmeta WHERE meta_key = '%s'", $meta_key );
64
65                 $result = $wpdb->get_results( $sql );
66
67                 if ( !empty( $result ) )
68                         $count = intval( $result[0]->cnt );
69
70                 // unset to save memory
71                 unset( $results );
72
73                 return $count;
74         }
75
76         /**
77          * Set array with imported comments from WordPress database
78          *
79          * @param string $bid
80          * @return array
81          */
82         function get_imported_comments( $bid ) {
83                 global $wpdb;
84
85                 $hashtable = array();
86
87                 $limit = 100;
88                 $offset = 0;
89
90                 // Grab all comments in chunks
91                 do {
92                         $sql = $wpdb->prepare( "SELECT comment_ID, comment_agent FROM $wpdb->comments LIMIT %d,%d", $offset, $limit );
93                         $results = $wpdb->get_results( $sql );
94
95                         // Increment offset
96                         $offset = ( $limit + $offset );
97
98                         if ( !empty( $results ) ) {
99                                 foreach ( $results as $r ) {
100                                         // Explode comment_agent key
101                                         list ( $ca_bid, $source_comment_id ) = explode( '-', $r->comment_agent );
102                                         $source_comment_id = intval( $source_comment_id );
103
104                                         // Check if this comment came from this blog
105                                         if ( $bid == $ca_bid ) {
106                                                 $hashtable[$source_comment_id] = intval( $r->comment_ID );
107                                         }
108                                 }
109                         }
110                 } while ( count( $results ) == $limit );
111
112                 // unset to save memory
113                 unset( $results, $r );
114
115                 return $hashtable;
116         }
117
118         function set_blog( $blog_id ) {
119                 if ( is_numeric( $blog_id ) ) {
120                         $blog_id = (int) $blog_id;
121                 } else {
122                         $blog = 'http://' . preg_replace( '#^https?://#', '', $blog_id );
123                         if ( ( !$parsed = parse_url( $blog ) ) || empty( $parsed['host'] ) ) {
124                                 fwrite( STDERR, "Error: can not determine blog_id from $blog_id\n" );
125                                 exit();
126                         }
127                         if ( empty( $parsed['path'] ) )
128                                 $parsed['path'] = '/';
129                         $blog = get_blog_details( array( 'domain' => $parsed['host'], 'path' => $parsed['path'] ) );
130                         if ( !$blog ) {
131                                 fwrite( STDERR, "Error: Could not find blog\n" );
132                                 exit();
133                         }
134                         $blog_id = (int) $blog->blog_id;
135                         // Restore global $current_blog
136                         global $current_blog;
137                         $current_blog = $blog;
138                 }
139
140                 if ( function_exists( 'is_multisite' ) ) {
141                         if ( is_multisite() )
142                                 switch_to_blog( $blog_id );
143                 }
144
145                 return $blog_id;
146         }
147
148         function set_user( $user_id ) {
149                 if ( is_numeric( $user_id ) ) {
150                         $user_id = (int) $user_id;
151                 } else {
152                         $user_id = (int) username_exists( $user_id );
153                 }
154
155                 if ( !$user_id || !wp_set_current_user( $user_id ) ) {
156                         fwrite( STDERR, "Error: can not find user\n" );
157                         exit();
158                 }
159
160                 return $user_id;
161         }
162
163         /**
164          * Sort by strlen, longest string first
165          *
166          * @param string $a
167          * @param string $b
168          * @return int
169          */
170         function cmpr_strlen( $a, $b ) {
171                 return strlen( $b ) - strlen( $a );
172         }
173
174         /**
175          * GET URL
176          *
177          * @param string $url
178          * @param string $username
179          * @param string $password
180          * @param bool $head
181          * @return array
182          */
183         function get_page( $url, $username = '', $password = '', $head = false ) {
184                 // Increase the timeout
185                 add_filter( 'http_request_timeout', array( &$this, 'bump_request_timeout' ) );
186
187                 $headers = array();
188                 $args = array();
189                 if ( true === $head )
190                         $args['method'] = 'HEAD';
191                 if ( !empty( $username ) && !empty( $password ) )
192                         $headers['Authorization'] = 'Basic ' . base64_encode( "$username:$password" );
193
194                 $args['headers'] = $headers;
195
196                 return wp_remote_request( $url, $args );
197         }
198
199         /**
200          * Bump up the request timeout for http requests
201          *
202          * @param int $val
203          * @return int
204          */
205         function bump_request_timeout( $val ) {
206                 return 60;
207         }
208
209         /**
210          * Check if user has exceeded disk quota
211          *
212          * @return bool
213          */
214         function is_user_over_quota() {
215                 global $current_blog;
216
217                 if ( function_exists( 'upload_is_user_over_quota' ) ) {
218                         if ( upload_is_user_over_quota( 1 ) ) {
219                                 echo "Sorry, you have used your upload quota.\n";
220                                 return true;
221                         }
222                 }
223
224                 return false;
225         }
226
227         /**
228          * Replace newlines, tabs, and multiple spaces with a single space
229          *
230          * @param string $string
231          * @return string
232          */
233         function min_whitespace( $string ) {
234                 return preg_replace( '|[\r\n\t ]+|', ' ', $string );
235         }
236
237         /**
238          * Reset global variables that grow out of control during imports
239          *
240          * @return void
241          */
242         function stop_the_insanity() {
243                 global $wpdb, $wp_actions;
244                 // Or define( 'WP_IMPORTING', true );
245                 $wpdb->queries = array();
246                 // Reset $wp_actions to keep it from growing out of control
247                 $wp_actions = array();
248         }
249 }
250
251 /**
252  * Returns value of command line params.
253  * Exits when a required param is not set.
254  *
255  * @param string $param
256  * @param bool $required
257  * @return mixed
258  */
259 function get_cli_args( $param, $required = false ) {
260         $args = $_SERVER['argv'];
261
262         $out = array();
263
264         $last_arg = null;
265         $return = null;
266
267         $il = sizeof( $args );
268
269         for ( $i = 1, $il; $i < $il; $i++ ) {
270                 if ( (bool) preg_match( "/^--(.+)/", $args[$i], $match ) ) {
271                         $parts = explode( "=", $match[1] );
272                         $key = preg_replace( "/[^a-z0-9]+/", "", $parts[0] );
273
274                         if ( isset( $parts[1] ) ) {
275                                 $out[$key] = $parts[1];
276                         } else {
277                                 $out[$key] = true;
278                         }
279
280                         $last_arg = $key;
281                 } else if ( (bool) preg_match( "/^-([a-zA-Z0-9]+)/", $args[$i], $match ) ) {
282                         for ( $j = 0, $jl = strlen( $match[1] ); $j < $jl; $j++ ) {
283                                 $key = $match[1]{$j};
284                                 $out[$key] = true;
285                         }
286
287                         $last_arg = $key;
288                 } else if ( $last_arg !== null ) {
289                         $out[$last_arg] = $args[$i];
290                 }
291         }
292
293         // Check array for specified param
294         if ( isset( $out[$param] ) ) {
295                 // Set return value
296                 $return = $out[$param];
297         }
298
299         // Check for missing required param
300         if ( !isset( $out[$param] ) && $required ) {
301                 // Display message and exit
302                 echo "\"$param\" parameter is required but was not specified\n";
303                 exit();
304         }
305
306         return $return;
307 }