Ticket #12726: 12726-2.patch

wp-includes/post-functions.php

@@ -4059 +4059 @@
  * @return WP_Post|array|void WP_Post on success.
  */
 function get_page_by_path( $page_path, $output = OBJECT, $post_type = 'page' ) {
-        global $wpdb;
 
-        $page_path = rawurlencode(urldecode($page_path));
-        $page_path = str_replace('%2F', '/', $page_path);
-        $page_path = str_replace('%20', ' ', $page_path);
-        $parts = explode( '/', trim( $page_path, '/' ) );
-        $parts = esc_sql( $parts );
-        $parts = array_map( 'sanitize_title_for_query', $parts );
+    return get_post_by( 'path', $page_path, array(
+        'post_type' => $post_type,
+                'output'    => $output,
+    ));
 
-        $in_string = "'" . implode( "','", $parts ) . "'";
-
-        if ( is_array( $post_type ) ) {
-                $post_types = $post_type;
-        } else {
-                $post_types = array( $post_type, 'attachment' );
-        }
+}
 
-        $post_types = esc_sql( $post_types );
-        $post_type_in_string = "'" . implode( "','", $post_types ) . "'";
-        $sql = "
-                SELECT ID, post_name, post_parent, post_type
-                FROM $wpdb->posts
-                WHERE post_name IN ($in_string)
-                AND post_type IN ($post_type_in_string)
-        ";
-
-        $pages = $wpdb->get_results( $sql, OBJECT_K );
-
-        $revparts = array_reverse( $parts );
-
-        $foundid = 0;
-        foreach ( (array) $pages as $page ) {
-                if ( $page->post_name == $revparts[0] ) {
-                        $count = 0;
-                        $p = $page;
-                        while ( $p->post_parent != 0 && isset( $pages[ $p->post_parent ] ) ) {
-                                $count++;
-                                $parent = $pages[ $p->post_parent ];
-                                if ( ! isset( $revparts[ $count ] ) || $parent->post_name != $revparts[ $count ] )
-                                        break;
-                                $p = $parent;
-                        }
-
-                        if ( $p->post_parent == 0 && $count+1 == count( $revparts ) && $p->post_name == $revparts[ $count ] ) {
-                                $foundid = $page->ID;
-                                if ( $page->post_type == $post_type )
-                                        break;
-                        }
-                }
-        }
-
-        if ( $foundid ) {
-                return get_post( $foundid, $output );
-        }
-}
-
 /**
  * Retrieve a page given its title.
  *
@@ -4129 +4081 @@
  * @return WP_Post|array|void WP_Post on success or null on failure
  */
 function get_page_by_title( $page_title, $output = OBJECT, $post_type = 'page' ) {
-        global $wpdb;
 
-        if ( is_array( $post_type ) ) {
-                $post_type = esc_sql( $post_type );
-                $post_type_in_string = "'" . implode( "','", $post_type ) . "'";
-                $sql = $wpdb->prepare( "
-                        SELECT ID
-                        FROM $wpdb->posts
-                        WHERE post_title = %s
-                        AND post_type IN ($post_type_in_string)
-                ", $page_title );
-        } else {
-                $sql = $wpdb->prepare( "
-                        SELECT ID
-                        FROM $wpdb->posts
-                        WHERE post_title = %s
-                        AND post_type = %s
-                ", $page_title, $post_type );
-        }
+        return get_post_by( '$title', $page_title, array(
+                'post_type' => $post_type,
+                'output'    => $output,
+        ));
 
-        $page = $wpdb->get_var( $sql );
-
-        if ( $page ) {
-                return get_post( $page, $output );
-        }
-}
+}
 
 /**
  * Identify descendants of a given page ID in a list of page objects.
@@ -5857 +5790 @@
 
                 update_post_caches( $fresh_posts, 'any', $update_term_cache, $update_meta_cache );
         }
+}
+
+/**
+ * Retrieves post data by a given field.
+ *
+ * See {@link &get_post()} for optional filter and output arguments values.
+ *
+ * @since 4.4.0
+ *
+ * @param string $field The field to retrieve the post with.  id | title | path
+ * @param int|string $value A value for $field.  A post ID, title, or path.
+ * @param string|array $args {
+ *     Arguments to modify behavior of get_post_by().
+ *
+ *     @type string $post_type The post type to constrain the result by. Defaults to 'post'
+ *     @type string $output Optional. Either OBJECT, ARRAY_N, or ARRAY_A constant
+ *     @type string $filter Optional. Type of filter to apply. Accepts 'raw', 'edit', 'db', or 'display'. Default 'raw'.
+ *     @type bool   $use_like Optional. If true then uses a LIKE query for post_title and post_name. Defaults to false.
+ * }
+ * @return WP_Post|null
+ */
+function get_post_by( $field, $value, $args = array() ) {
+
+    global $wpdb;
+
+        $args = wp_parse_args( $args, array(
+        'post_type' => 'post',
+        'output'    => 'OBJECT',
+        'filter'    => 'raw',
+                'use_like'  => false,
+    ));
+
+        $post_type = $args['post_type'];
+
+    switch ( $field ) {
+            case 'id':
+                    $post = get_post( $value, $args['output'], $args['filter'] );
+                    break;
+
+            case 'title':
+                $sql = $args['use_like'] ? 'post_title LIKE %s' : '%s=post_title';
+                    $post = $wpdb->get_var( $wpdb->prepare( "SELECT * FROM {$wpdb->posts} WHERE {$sql} AND %s=post_type", $value, $post_type ) );
+                    break;
+
+            case 'path':
+                    $page_path  = rawurlencode( urldecode( $value ) );
+                    $page_path  = str_replace( '%2F', '/', $page_path );
+                    $page_path  = str_replace( '%20', ' ', $page_path );
+                    $page_paths = '/' . trim( $page_path, '/' );
+                    $leaf_path  = sanitize_title( basename( $page_paths ) );
+                    $page_paths = explode( '/', $page_paths );
+                    $full_path  = '';
+                    foreach ( (array) $page_paths as $pathdir ) {
+                            $full_path .= ( $pathdir != '' ? '/' : '' ) . sanitize_title( $pathdir );
+                    }
+
+                    $sql = $args['use_like'] ? 'post_name LIKE %s' : '%s=post_name';
+                    $pages = $wpdb->get_results( $wpdb->prepare( "SELECT ID,post_name,post_parent FROM {$wpdb->posts} WHERE {$sql} AND (%s=post_type OR 'attachment'=post_type)",
+                                    $leaf_path, $post_type ) );
+
+                    if ( empty( $pages ) ) {
+                            $post = null;
+                            break;
+                    }
+
+                    foreach ( $pages as $page ) {
+                    $path = "/{$leaf_path}";
+                    $curpage = $page;
+                    while ( 0 !== intval( $curpage->post_parent ) ) {
+                    $curpage = $wpdb->get_row( $wpdb->prepare( "SELECT ID,post_name,post_parent FROM {$wpdb->posts} WHERE ID=%d and %s=post_type", $curpage->post_parent, $post_type ) );
+                    $path = "/{$curpage->post_name}{$path}";
+                    }
+
+                    if ( $path === $full_path ) {
+                            $post = get_post( $page->ID, $args['output'], $args['filter'] );
+                            break;
+                    }
+                    }
+                    break;
+
+                default:
+                        $post = null;
+        }
+
+        /*
+         * Check to make sure that the post is what we expected.  Set to null if not.
+         */
+        if ( ! is_object( $post ) || ! property_exists( $post, 'post_type' ) || $post_type !== $post->post_type ) {
+          $post = null;
+    }
+
+        return $post;
+
 }