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

<?php

class CWS_Non_Verbose_Page_Rules {
        public function __construct() {
                add_action( 'init', array( $this, 'init' ) );
        }

        public function init() {
                add_filter( 'post_rewrite_rules', array( $this, 'post_rewrite_rules' ) );
                add_filter( 'query_vars', array( $this, 'query_vars' ) );
                add_filter( 'request', array( $this, 'request' ) );
        }

        public function page_rewrite_rules( $rules ) {
                foreach ( $rules as $k => $v ) {
                        $rules[$k] = str_replace( 'pagename=', 'post_or_page=', $v );
                }
                return $rules;
        }

        public function post_rewrite_rules( $rules ) {
                global $wp_rewrite;
                if ( $wp_rewrite->use_verbose_page_rules ) {
                        $wp_rewrite->use_verbose_page_rules = false;
                        add_filter( 'page_rewrite_rules', array( $this, 'page_rewrite_rules' ) );
                        foreach ( $rules as $k => $v ) {
                                $rules[$k] = str_replace( 'name=', 'post_or_page=', $v );
                        }
                }
                return $rules;
        }

        public function query_vars( $vars ) {
                $vars[] = 'post_or_page';
                return $vars;
        }

        public function request( $request ) {
                if ( !isset( $request['post_or_page'] ) )
                        return $request;
                global $wpdb;
                $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'] ) );
                if ( 'page' == $post_or_page )
                        $request['pagename'] = $request['post_or_page'];
                else
                        $request['name'] = $request['post_or_page'];
                unset( $request['post_or_page'] );
                return $request;
        }

}

new CWS_Non_Verbose_Page_Rules;