WordPress.org

Make WordPress Core

Ticket #16687: non-verbose-page-rules.002.php

File non-verbose-page-rules.002.php, 3.0 KB (added by duck_, 4 years ago)
Line 
1<?php
2
3class CWS_Non_Verbose_Page_Rules {
4        public function __construct() {
5                add_filter( 'post_rewrite_rules', array( $this, 'post_rewrite_rules' ) );
6                add_filter( 'query_vars', array( $this, 'query_vars' ) );
7                add_filter( 'request', array( $this, 'request' ) );
8        }
9
10        public function page_rewrite_rules( $rules ) {
11                foreach ( $rules as $k => $v ) {
12                        $rules[$k] = str_replace( 'pagename=', 'post_or_page=', $v );
13                }
14                return $rules;
15        }
16
17        public function post_rewrite_rules( $rules ) {
18                global $wp_rewrite;
19                if ( $wp_rewrite->use_verbose_page_rules && preg_match( '#^/%postname%/?$#', $wp_rewrite->permalink_structure ) ) {
20                        $wp_rewrite->use_verbose_page_rules = false;
21                        add_filter( 'page_rewrite_rules', array( $this, 'page_rewrite_rules' ) );
22                        foreach ( $rules as $k => $v ) {
23                                // post attachment rule blocks 2nd level pages (e.g. level-1/level-2/), deal with it in page catchall
24                                if ( substr( $k, 0, 14 ) == '[^/]+/([^/]+)/' )
25                                        unset( $rules[$k] );
26                                else
27                                        $rules[$k] = str_replace( 'name=', 'post_or_page=', $v );
28                        }
29                }
30                return $rules;
31        }
32
33        public function query_vars( $vars ) {
34                $vars[] = 'post_or_page';
35                return $vars;
36        }
37
38        public function request( $request ) {
39                if ( ! isset( $request['post_or_page'] ) )
40                        return $request;
41
42                global $wpdb;
43                if ( strpos( $request['post_or_page'], '/' ) !== false ) {
44                        $pieces = explode( '/', $request['post_or_page'] );
45                        $count = count( $pieces );
46                        // check for attachment matches first
47                        if ( ( $count == 2 ||
48                                $count == 3 && $pieces[2] == 'trackback' ||
49                                $count == 4 && $pieces[2] == 'feed' && in_array( $pieces[3], array('feed','rdf','rss','rss2','atom') ) ||
50                                $count == 3 && in_array( $pieces[2], array('feed','rdf','rss','rss2','atom') ) ||
51                                $count == 3 && preg_match( '/^comment-page-([0-9]{1,})$/', $pieces[2] ) ) &&
52                                        $id = $wpdb->get_var( $wpdb->prepare( "SELECT ID FROM $wpdb->posts WHERE post_name = %s AND post_type = 'attachment' LIMIT 1", $pieces[1] ) ) )
53                                $request['attachment_id'] = $id;
54                        // otherwise this is definitely a page
55                        else
56                                $request['pagename'] = $request['post_or_page'];
57                } else {
58                        //*
59                        $post_or_page = $wpdb->get_var( $wpdb->prepare( "SELECT post_type FROM $wpdb->posts WHERE post_name = %s AND post_type IN ('post','page') LIMIT 1", $request['post_or_page'] ) );
60                        if ( 'page' == $post_or_page )
61                                $request['pagename'] = $request['post_or_page'];
62                        else
63                                $request['name'] = $request['post_or_page'];
64                        //*/
65
66                        /* could try to be a bit smarter and get an ID now
67                        $results = $wpdb->get_results( $wpdb->prepare( "SELECT ID, post_type FROM $wpdb->posts WHERE post_name = %s AND post_type IN ('post','page') LIMIT 1", $request['post_or_page'] ) );
68                        if ( $results && $results[0]->post_type == 'page' )
69                                $request['page_id'] = $results[0]->ID;
70                        elseif ( $results )
71                                $request['p'] = $results[0]->ID;
72                        else
73                                $request['name'] = $request['post_or_page']; // 404
74                        //*/
75                }
76
77                unset( $request['post_or_page'] );
78                return $request;
79        }
80}
81
82new CWS_Non_Verbose_Page_Rules;