| 1 | <?php
|
|---|
| 2 |
|
|---|
| 3 | class 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 |
|
|---|
| 82 | new CWS_Non_Verbose_Page_Rules;
|
|---|