From a5fe68e002632c190ffbd85167671ed4d4961135 Mon Sep 17 00:00:00 2001 From: donncha Date: Thu, 21 Jun 2007 16:41:02 +0000 Subject: WP Merge to WP 2.2.1 git-svn-id: http://svn.automattic.com/wordpress-mu/trunk@1005 7be80a69-a1ef-0310-a953-fb0f7c49ff36 --- wp-includes/rewrite.php | 1809 +++++++++++++++++++++++------------------------ 1 file changed, 904 insertions(+), 905 deletions(-) (limited to 'wp-includes/rewrite.php') diff --git a/wp-includes/rewrite.php b/wp-includes/rewrite.php index eb10bf7..2f440ab 100644 --- a/wp-includes/rewrite.php +++ b/wp-includes/rewrite.php @@ -1,905 +1,904 @@ -add_rule($regex, $redirect); -} - -//Add a new tag (like %postname%) -//warning: you must call this on init or earlier, otherwise the query var addition stuff won't work -function add_rewrite_tag($tagname, $regex) { - //validation - if (strlen($tagname) < 3 || $tagname{0} != '%' || $tagname{strlen($tagname)-1} != '%') { - return; - } - - $qv = trim($tagname, '%'); - - global $wp_rewrite, $wp; - $wp->add_query_var($qv); - $wp_rewrite->add_rewrite_tag($tagname, $regex, $qv . '='); -} - -//Add a new feed type like /atom1/ -function add_feed($feedname, $function) { - global $wp_rewrite; - if (!in_array($feedname, $wp_rewrite->feeds)) { //override the file if it is - $wp_rewrite->feeds[] = $feedname; - } - $hook = 'do_feed_' . $feedname; - remove_action($hook, $function, 10, 1); - add_action($hook, $function, 10, 1); - return $hook; -} - -define('EP_PERMALINK', 1 ); -define('EP_ATTACHMENT', 2 ); -define('EP_DATE', 4 ); -define('EP_YEAR', 8 ); -define('EP_MONTH', 16 ); -define('EP_DAY', 32 ); -define('EP_ROOT', 64 ); -define('EP_COMMENTS', 128 ); -define('EP_SEARCH', 256 ); -define('EP_CATEGORIES', 512 ); -define('EP_AUTHORS', 2048); -define('EP_PAGES', 4096); -//pseudo-places -define('EP_NONE', 0 ); -define('EP_ALL', 8191); - -//and an endpoint, like /trackback/ -function add_rewrite_endpoint($name, $places) { - global $wp_rewrite; - $wp_rewrite->add_endpoint($name, $places); -} - -// examine a url (supposedly from this blog) and try to -// determine the post ID it represents. -function url_to_postid($url) { - global $wp_rewrite; - - $url = apply_filters('url_to_postid', $url); - - // First, check to see if there is a 'p=N' or 'page_id=N' to match against - preg_match('#[?&](p|page_id)=(\d+)#', $url, $values); - $id = intval($values[2]); - if ( $id ) return $id; - - // Check to see if we are using rewrite rules - $rewrite = $wp_rewrite->wp_rewrite_rules(); - - // Not using rewrite rules, and 'p=N' and 'page_id=N' methods failed, so we're out of options - if ( empty($rewrite) ) - return 0; - - // $url cleanup by Mark Jaquith - // This fixes things like #anchors, ?query=strings, missing 'www.', - // added 'www.', or added 'index.php/' that will mess up our WP_Query - // and return a false negative - - // Get rid of the #anchor - $url_split = explode('#', $url); - $url = $url_split[0]; - - // Get rid of URL ?query=string - $url_split = explode('?', $url); - $url = $url_split[0]; - - // Add 'www.' if it is absent and should be there - if ( false !== strpos(get_option('home'), '://www.') && false === strpos($url, '://www.') ) - $url = str_replace('://', '://www.', $url); - - // Strip 'www.' if it is present and shouldn't be - if ( false === strpos(get_option('home'), '://www.') ) - $url = str_replace('://www.', '://', $url); - - // Strip 'index.php/' if we're not using path info permalinks - if ( !$wp_rewrite->using_index_permalinks() ) - $url = str_replace('index.php/', '', $url); - - if ( false !== strpos($url, get_option('home')) ) { - // Chop off http://domain.com - $url = str_replace(get_option('home'), '', $url); - } else { - // Chop off /path/to/blog - $home_path = parse_url(get_option('home')); - $home_path = $home_path['path']; - $url = str_replace($home_path, '', $url); - } - - // Trim leading and lagging slashes - $url = trim($url, '/'); - - $request = $url; - - // Done with cleanup - - // Look for matches. - $request_match = $request; - foreach ($rewrite as $match => $query) { - // If the requesting file is the anchor of the match, prepend it - // to the path info. - if ( (! empty($url)) && (strpos($match, $url) === 0) ) { - $request_match = $url . '/' . $request; - } - - if ( preg_match("!^$match!", $request_match, $matches) ) { - // Got a match. - // Trim the query of everything up to the '?'. - $query = preg_replace("!^.+\?!", '', $query); - - // Substitute the substring matches into the query. - eval("\$query = \"$query\";"); - $query = new WP_Query($query); - if ( $query->is_single || $query->is_page ) - return $query->post->ID; - else - return 0; - } - } - return 0; -} - -/* WP_Rewrite class -*******************************************************************************/ - -class WP_Rewrite { - var $permalink_structure; - var $use_trailing_slashes; - var $category_base; - var $category_structure; - var $author_base = 'author'; - var $author_structure; - var $date_structure; - var $page_structure; - var $search_base = 'search'; - var $search_structure; - var $comments_base = 'comments'; - var $feed_base = 'feed'; - var $comments_feed_structure; - var $feed_structure; - var $front; - var $root = ''; - var $index = 'index.php'; - var $matches = ''; - var $rules; - var $extra_rules; //those not generated by the class, see add_rewrite_rule() - var $non_wp_rules; //rules that don't redirect to WP's index.php - var $endpoints; - var $use_verbose_rules = false; - var $rewritecode = - array( - '%year%', - '%monthnum%', - '%day%', - '%hour%', - '%minute%', - '%second%', - '%postname%', - '%post_id%', - '%category%', - '%author%', - '%pagename%', - '%search%' - ); - - var $rewritereplace = - array( - '([0-9]{4})', - '([0-9]{1,2})', - '([0-9]{1,2})', - '([0-9]{1,2})', - '([0-9]{1,2})', - '([0-9]{1,2})', - '([^/]+)', - '([0-9]+)', - '(.+?)', - '(.+?)', - '([^/]+)', - '([^/]+)', - '(.+)' - ); - - var $queryreplace = - array ( - 'year=', - 'monthnum=', - 'day=', - 'hour=', - 'minute=', - 'second=', - 'name=', - 'p=', - 'category_name=', - 'author_name=', - 'pagename=', - 's=' - ); - - var $feeds = array ( 'feed', 'rdf', 'rss', 'rss2', 'atom' ); - - function using_permalinks() { - if (empty($this->permalink_structure)) - return false; - else - return true; - } - - function using_index_permalinks() { - if (empty($this->permalink_structure)) { - return false; - } - - // If the index is not in the permalink, we're using mod_rewrite. - if (preg_match('#^/*' . $this->index . '#', $this->permalink_structure)) { - return true; - } - - return false; - } - - function using_mod_rewrite_permalinks() { - if ( $this->using_permalinks() && ! $this->using_index_permalinks()) - return true; - else - return false; - } - - function preg_index($number) { - $match_prefix = '$'; - $match_suffix = ''; - - if (! empty($this->matches)) { - $match_prefix = '$' . $this->matches . '['; - $match_suffix = ']'; - } - - return "$match_prefix$number$match_suffix"; - } - - function page_rewrite_rules() { - $uris = get_option('page_uris'); - $attachment_uris = get_option('page_attachment_uris'); - - $rewrite_rules = array(); - $page_structure = $this->get_page_permastruct(); - if( is_array( $attachment_uris ) ) { - foreach ($attachment_uris as $uri => $pagename) { - $this->add_rewrite_tag('%pagename%', "($uri)", 'attachment='); - $rewrite_rules = array_merge($rewrite_rules, $this->generate_rewrite_rules($page_structure, EP_PAGES)); - } - } - if( is_array( $uris ) ) { - foreach ($uris as $uri => $pagename) { - $this->add_rewrite_tag('%pagename%', "($uri)", 'pagename='); - $rewrite_rules = array_merge($rewrite_rules, $this->generate_rewrite_rules($page_structure, EP_PAGES)); - } - } - - return $rewrite_rules; - } - - function get_date_permastruct() { - if (isset($this->date_structure)) { - return $this->date_structure; - } - - if (empty($this->permalink_structure)) { - $this->date_structure = ''; - return false; - } - - // The date permalink must have year, month, and day separated by slashes. - $endians = array('%year%/%monthnum%/%day%', '%day%/%monthnum%/%year%', '%monthnum%/%day%/%year%'); - - $this->date_structure = ''; - $date_endian = ''; - - foreach ($endians as $endian) { - if (false !== strpos($this->permalink_structure, $endian)) { - $date_endian= $endian; - break; - } - } - - if ( empty($date_endian) ) - $date_endian = '%year%/%monthnum%/%day%'; - - // Do not allow the date tags and %post_id% to overlap in the permalink - // structure. If they do, move the date tags to $front/date/. - $front = $this->front; - preg_match_all('/%.+?%/', $this->permalink_structure, $tokens); - $tok_index = 1; - foreach ($tokens[0] as $token) { - if ( ($token == '%post_id%') && ($tok_index <= 3) ) { - $front = $front . 'date/'; - break; - } - $tok_index++; - } - - $this->date_structure = $front . $date_endian; - - return $this->date_structure; - } - - function get_year_permastruct() { - $structure = $this->get_date_permastruct($this->permalink_structure); - - if (empty($structure)) { - return false; - } - - $structure = str_replace('%monthnum%', '', $structure); - $structure = str_replace('%day%', '', $structure); - - $structure = preg_replace('#/+#', '/', $structure); - - return $structure; - } - - function get_month_permastruct() { - $structure = $this->get_date_permastruct($this->permalink_structure); - - if (empty($structure)) { - return false; - } - - $structure = str_replace('%day%', '', $structure); - - $structure = preg_replace('#/+#', '/', $structure); - - return $structure; - } - - function get_day_permastruct() { - return $this->get_date_permastruct($this->permalink_structure); - } - - function get_category_permastruct() { - if (isset($this->category_structure)) { - return $this->category_structure; - } - - if (empty($this->permalink_structure)) { - $this->category_structure = ''; - return false; - } - - if (empty($this->category_base)) - $this->category_structure = $this->front . 'category/'; - else - $this->category_structure = $this->category_base . '/'; - - $this->category_structure .= '%category%'; - - return $this->category_structure; - } - - function get_author_permastruct() { - if (isset($this->author_structure)) { - return $this->author_structure; - } - - if (empty($this->permalink_structure)) { - $this->author_structure = ''; - return false; - } - - $this->author_structure = $this->front . $this->author_base . '/%author%'; - - return $this->author_structure; - } - - function get_search_permastruct() { - if (isset($this->search_structure)) { - return $this->search_structure; - } - - if (empty($this->permalink_structure)) { - $this->search_structure = ''; - return false; - } - - $this->search_structure = $this->root . $this->search_base . '/%search%'; - - return $this->search_structure; - } - - function get_page_permastruct() { - if (isset($this->page_structure)) { - return $this->page_structure; - } - - if (empty($this->permalink_structure)) { - $this->page_structure = ''; - return false; - } - - $this->page_structure = $this->root . '%pagename%'; - - return $this->page_structure; - } - - function get_feed_permastruct() { - if (isset($this->feed_structure)) { - return $this->feed_structure; - } - - if (empty($this->permalink_structure)) { - $this->feed_structure = ''; - return false; - } - - $this->feed_structure = $this->root . $this->feed_base . '/%feed%'; - - return $this->feed_structure; - } - - function get_comment_feed_permastruct() { - if (isset($this->comment_feed_structure)) { - return $this->comment_feed_structure; - } - - if (empty($this->permalink_structure)) { - $this->comment_feed_structure = ''; - return false; - } - - $this->comment_feed_structure = $this->root . $this->comments_base . '/' . $this->feed_base . '/%feed%'; - - return $this->comment_feed_structure; - } - - function add_rewrite_tag($tag, $pattern, $query) { - // If the tag already exists, replace the existing pattern and query for - // that tag, otherwise add the new tag, pattern, and query to the end of - // the arrays. - $position = array_search($tag, $this->rewritecode); - if (FALSE !== $position && NULL !== $position) { - $this->rewritereplace[$position] = $pattern; - $this->queryreplace[$position] = $query; - } else { - $this->rewritecode[] = $tag; - $this->rewritereplace[] = $pattern; - $this->queryreplace[] = $query; - } - } - - //the main WP_Rewrite function. generate the rules from permalink structure - function generate_rewrite_rules($permalink_structure, $ep_mask = EP_NONE, $paged = true, $feed = true, $forcomments = false, $walk_dirs = true, $endpoints = true) { - //build a regex to match the feed section of URLs, something like (feed|atom|rss|rss2)/? - $feedregex2 = ''; - foreach ($this->feeds as $feed_name) { - $feedregex2 .= $feed_name . '|'; - } - $feedregex2 = '(' . trim($feedregex2, '|') . ')/?$'; - //$feedregex is identical but with /feed/ added on as well, so URLs like /feed/atom - //and /atom are both possible - $feedregex = $this->feed_base . '/' . $feedregex2; - - //build a regex to match the trackback and page/xx parts of URLs - $trackbackregex = 'trackback/?$'; - $pageregex = 'page/?([0-9]{1,})/?$'; - - //build up an array of endpoint regexes to append => queries to append - if ($endpoints) { - $ep_query_append = array (); - foreach ($this->endpoints as $endpoint) { - //match everything after the endpoint name, but allow for nothing to appear there - $epmatch = $endpoint[1] . '(/(.*))?/?$'; - //this will be appended on to the rest of the query for each dir - $epquery = '&' . $endpoint[1] . '='; - $ep_query_append[$epmatch] = array ( $endpoint[0], $epquery ); - } - } - - //get everything up to the first rewrite tag - $front = substr($permalink_structure, 0, strpos($permalink_structure, '%')); - //build an array of the tags (note that said array ends up being in $tokens[0]) - preg_match_all('/%.+?%/', $permalink_structure, $tokens); - - $num_tokens = count($tokens[0]); - - $index = $this->index; //probably 'index.php' - $feedindex = $index; - $trackbackindex = $index; - //build a list from the rewritecode and queryreplace arrays, that will look something like - //tagname=$matches[i] where i is the current $i - for ($i = 0; $i < $num_tokens; ++$i) { - if (0 < $i) { - $queries[$i] = $queries[$i - 1] . '&'; - } - - $query_token = str_replace($this->rewritecode, $this->queryreplace, $tokens[0][$i]) . $this->preg_index($i+1); - $queries[$i] .= $query_token; - } - - //get the structure, minus any cruft (stuff that isn't tags) at the front - $structure = $permalink_structure; - if ($front != '/') { - $structure = str_replace($front, '', $structure); - } - //create a list of dirs to walk over, making rewrite rules for each level - //so for example, a $structure of /%year%/%month%/%postname% would create - //rewrite rules for /%year%/, /%year%/%month%/ and /%year%/%month%/%postname% - $structure = trim($structure, '/'); - if ($walk_dirs) { - $dirs = explode('/', $structure); - } else { - $dirs[] = $structure; - } - $num_dirs = count($dirs); - - //strip slashes from the front of $front - $front = preg_replace('|^/+|', '', $front); - - //the main workhorse loop - $post_rewrite = array(); - $struct = $front; - for ($j = 0; $j < $num_dirs; ++$j) { - //get the struct for this dir, and trim slashes off the front - $struct .= $dirs[$j] . '/'; //accumulate. see comment near explode('/', $structure) above - $struct = ltrim($struct, '/'); - //replace tags with regexes - $match = str_replace($this->rewritecode, $this->rewritereplace, $struct); - //make a list of tags, and store how many there are in $num_toks - $num_toks = preg_match_all('/%.+?%/', $struct, $toks); - //get the 'tagname=$matches[i]' - $query = $queries[$num_toks - 1]; - - //set up $ep_mask_specific which is used to match more specific URL types - switch ($dirs[$j]) { - case '%year%': $ep_mask_specific = EP_YEAR; break; - case '%monthnum%': $ep_mask_specific = EP_MONTH; break; - case '%day%': $ep_mask_specific = EP_DAY; break; - } - - //create query for /page/xx - $pagematch = $match . $pageregex; - $pagequery = $index . '?' . $query . '&paged=' . $this->preg_index($num_toks + 1); - - //create query for /feed/(feed|atom|rss|rss2|rdf) - $feedmatch = $match . $feedregex; - $feedquery = $feedindex . '?' . $query . '&feed=' . $this->preg_index($num_toks + 1); - - //create query for /(feed|atom|rss|rss2|rdf) (see comment near creation of $feedregex) - $feedmatch2 = $match . $feedregex2; - $feedquery2 = $feedindex . '?' . $query . '&feed=' . $this->preg_index($num_toks + 1); - - //if asked to, turn the feed queries into comment feed ones - if ($forcomments) { - $feedquery .= '&withcomments=1'; - $feedquery2 .= '&withcomments=1'; - } - - //start creating the array of rewrites for this dir - $rewrite = array(); - if ($feed) //...adding on /feed/ regexes => queries - $rewrite = array($feedmatch => $feedquery, $feedmatch2 => $feedquery2); - if ($paged) //...and /page/xx ones - $rewrite = array_merge($rewrite, array($pagematch => $pagequery)); - - //do endpoints - if ($endpoints) { - foreach ($ep_query_append as $regex => $ep) { - //add the endpoints on if the mask fits - if ($ep[0] & $ep_mask || $ep[0] & $ep_mask_specific) { - $rewrite[$match . $regex] = $index . '?' . $query . $ep[1] . $this->preg_index($num_toks + 2); - } - } - } - - //if we've got some tags in this dir - if ($num_toks) { - $post = false; - $page = false; - - //check to see if this dir is permalink-level: i.e. the structure specifies an - //individual post. Do this by checking it contains at least one of 1) post name, - //2) post ID, 3) page name, 4) timestamp (year, month, day, hour, second and - //minute all present). Set these flags now as we need them for the endpoints. - if (strpos($struct, '%postname%') !== false || strpos($struct, '%post_id%') !== false - || strpos($struct, '%pagename%') !== false - || (strpos($struct, '%year%') !== false && strpos($struct, '%monthnum%') !== false && strpos($struct, '%day%') !== false && strpos($struct, '%hour%') !== false && strpos($struct, '%minute%') !== false && strpos($struct, '%second%') !== false)) { - $post = true; - if (strpos($struct, '%pagename%') !== false) - $page = true; - } - - //if we're creating rules for a permalink, do all the endpoints like attachments etc - if ($post) { - $post = true; - //create query and regex for trackback - $trackbackmatch = $match . $trackbackregex; - $trackbackquery = $trackbackindex . '?' . $query . '&tb=1'; - //trim slashes from the end of the regex for this dir - $match = rtrim($match, '/'); - //get rid of brackets - $submatchbase = str_replace(array('(',')'),'',$match); - - //add a rule for at attachments, which take the form of /some-text - $sub1 = $submatchbase . '/([^/]+)/'; - $sub1tb = $sub1 . $trackbackregex; //add trackback regex /trackback/... - $sub1feed = $sub1 . $feedregex; //and /feed/(atom|...) - $sub1feed2 = $sub1 . $feedregex2; //and /(feed|atom...) - //add an ? as we don't have to match that last slash, and finally a $ so we - //match to the end of the URL - - //add another rule to match attachments in the explicit form: - ///attachment/some-text - $sub2 = $submatchbase . '/attachment/([^/]+)/'; - $sub2tb = $sub2 . $trackbackregex; //and add trackbacks /attachment/trackback - $sub2feed = $sub2 . $feedregex; //feeds, /attachment/feed/(atom|...) - $sub2feed2 = $sub2 . $feedregex2; //and feeds again on to this /attachment/(feed|atom...) - - //create queries for these extra tag-ons we've just dealt with - $subquery = $index . '?attachment=' . $this->preg_index(1); - $subtbquery = $subquery . '&tb=1'; - $subfeedquery = $subquery . '&feed=' . $this->preg_index(2); - - //do endpoints for attachments - if ($endpoint) { foreach ($ep_query_append as $regex => $ep) { - if ($ep[0] & EP_ATTACHMENT) { - $rewrite[$sub1 . $regex] = $subquery . '?' . $ep[1] . $this->preg_index(2); - $rewrite[$sub2 . $regex] = $subquery . '?' . $ep[1] . $this->preg_index(2); - } - } } - - //now we've finished with endpoints, finish off the $sub1 and $sub2 matches - $sub1 .= '?$'; - $sub2 .= '?$'; - - //allow URLs like /2 for /page/2 - $match = $match . '(/[0-9]+)?/?$'; - $query = $index . '?' . $query . '&page=' . $this->preg_index($num_toks + 1); - } else { //not matching a permalink so this is a lot simpler - //close the match and finalise the query - $match .= '?$'; - $query = $index . '?' . $query; - } - - //create the final array for this dir by joining the $rewrite array (which currently - //only contains rules/queries for trackback, pages etc) to the main regex/query for - //this dir - $rewrite = array_merge($rewrite, array($match => $query)); - - //if we're matching a permalink, add those extras (attachments etc) on - if ($post) { - //add trackback - $rewrite = array_merge(array($trackbackmatch => $trackbackquery), $rewrite); - - //add regexes/queries for attachments, attachment trackbacks and so on - if ( ! $page ) //require /attachment/stuff form for pages because of confusion with subpages - $rewrite = array_merge($rewrite, array($sub1 => $subquery, $sub1tb => $subtbquery, $sub1feed => $subfeedquery, $sub1feed2 => $subfeedquery)); - $rewrite = array_merge($rewrite, array($sub2 => $subquery, $sub2tb => $subtbquery, $sub2feed => $subfeedquery, $sub2feed2 => $subfeedquery)); - } - } //if($num_toks) - //add the rules for this dir to the accumulating $post_rewrite - $post_rewrite = array_merge($rewrite, $post_rewrite); - } //foreach ($dir) - return $post_rewrite; //the finished rules. phew! - } - - function generate_rewrite_rule($permalink_structure, $walk_dirs = false) { - return $this->generate_rewrite_rules($permalink_structure, EP_NONE, false, false, false, $walk_dirs); - } - - /* rewrite_rules - * Construct rewrite matches and queries from permalink structure. - * Returns an associate array of matches and queries. - */ - function rewrite_rules() { - $rewrite = array(); - - if (empty($this->permalink_structure)) { - return $rewrite; - } - - // robots.txt - $robots_rewrite = array('robots.txt$' => $this->index . '?robots=1'); - - //Default Feed rules - These are require to allow for the direct access files to work with permalink structure starting with %category% - $default_feeds = array( 'wp-atom.php$' => $this->index .'?feed=atom', - 'wp-rdf.php$' => $this->index .'?feed=rdf', - 'wp-rss.php$' => $this->index .'?feed=rss', - 'wp-rss2.php$' => $this->index .'?feed=rss2', - 'wp-feed.php$' => $this->index .'?feed=feed', - 'wp-commentsrss2.php$' => $this->index . '?feed=rss2&withcomments=1'); - - // Post - $post_rewrite = $this->generate_rewrite_rules($this->permalink_structure, EP_PERMALINK); - $post_rewrite = apply_filters('post_rewrite_rules', $post_rewrite); - - // Date - $date_rewrite = $this->generate_rewrite_rules($this->get_date_permastruct(), EP_DATE); - $date_rewrite = apply_filters('date_rewrite_rules', $date_rewrite); - - // Root - $root_rewrite = $this->generate_rewrite_rules($this->root . '/', EP_ROOT); - $root_rewrite = apply_filters('root_rewrite_rules', $root_rewrite); - - // Comments - $comments_rewrite = $this->generate_rewrite_rules($this->root . $this->comments_base, EP_COMMENTS, true, true, true, false); - $comments_rewrite = apply_filters('comments_rewrite_rules', $comments_rewrite); - - // Search - $search_structure = $this->get_search_permastruct(); - $search_rewrite = $this->generate_rewrite_rules($search_structure, EP_SEARCH); - $search_rewrite = apply_filters('search_rewrite_rules', $search_rewrite); - - // Categories - $category_rewrite = $this->generate_rewrite_rules($this->get_category_permastruct(), EP_CATEGORIES); - $category_rewrite = apply_filters('category_rewrite_rules', $category_rewrite); - - // Authors - $author_rewrite = $this->generate_rewrite_rules($this->get_author_permastruct(), EP_AUTHORS); - $author_rewrite = apply_filters('author_rewrite_rules', $author_rewrite); - - // Pages - $page_rewrite = $this->page_rewrite_rules(); - $page_rewrite = apply_filters('page_rewrite_rules', $page_rewrite); - - // Put them together. - $this->rules = array_merge($robots_rewrite, $default_feeds, $page_rewrite, $root_rewrite, $comments_rewrite, $search_rewrite, $category_rewrite, $author_rewrite, $date_rewrite, $post_rewrite, $this->extra_rules); - - do_action_ref_array('generate_rewrite_rules', array(&$this)); - $this->rules = apply_filters('rewrite_rules_array', $this->rules); - - return $this->rules; - } - - function wp_rewrite_rules() { - $this->rules = get_option('rewrite_rules'); - if ( empty($this->rules) ) { - $this->matches = 'matches'; - $this->rewrite_rules(); - update_option('rewrite_rules', $this->rules); - } - - return $this->rules; - } - - function mod_rewrite_rules() { - if ( ! $this->using_permalinks()) { - return ''; - } - - $site_root = parse_url(get_option('siteurl')); - $site_root = trailingslashit($site_root['path']); - - $home_root = parse_url(get_option('home')); - $home_root = trailingslashit($home_root['path']); - - $rules = "\n"; - $rules .= "RewriteEngine On\n"; - $rules .= "RewriteBase $home_root\n"; - - //add in the rules that don't redirect to WP's index.php (and thus shouldn't be handled by WP at all) - foreach ($this->non_wp_rules as $match => $query) { - // Apache 1.3 does not support the reluctant (non-greedy) modifier. - $match = str_replace('.+?', '.+', $match); - - // If the match is unanchored and greedy, prepend rewrite conditions - // to avoid infinite redirects and eclipsing of real files. - if ($match == '(.+)/?$' || $match == '([^/]+)/?$' ) { - //nada. - } - - $rules .= 'RewriteRule ^' . $match . ' ' . $home_root . $query . " [QSA,L]\n"; - } - - if ($this->use_verbose_rules) { - $this->matches = ''; - $rewrite = $this->rewrite_rules(); - $num_rules = count($rewrite); - $rules .= "RewriteCond %{REQUEST_FILENAME} -f [OR]\n" . - "RewriteCond %{REQUEST_FILENAME} -d\n" . - "RewriteRule ^.*$ - [S=$num_rules]\n"; - - foreach ($rewrite as $match => $query) { - // Apache 1.3 does not support the reluctant (non-greedy) modifier. - $match = str_replace('.+?', '.+', $match); - - // If the match is unanchored and greedy, prepend rewrite conditions - // to avoid infinite redirects and eclipsing of real files. - if ($match == '(.+)/?$' || $match == '([^/]+)/?$' ) { - //nada. - } - - if (strpos($query, $this->index) !== false) { - $rules .= 'RewriteRule ^' . $match . ' ' . $home_root . $query . " [QSA,L]\n"; - } else { - $rules .= 'RewriteRule ^' . $match . ' ' . $site_root . $query . " [QSA,L]\n"; - } - } - } else { - $rules .= "RewriteCond %{REQUEST_FILENAME} !-f\n" . - "RewriteCond %{REQUEST_FILENAME} !-d\n" . - "RewriteRule . {$home_root}{$this->index} [L]\n"; - } - - $rules .= "\n"; - - $rules = apply_filters('mod_rewrite_rules', $rules); - $rules = apply_filters('rewrite_rules', $rules); // Deprecated - - return $rules; - } - - //Add a straight rewrite rule - function add_rule($regex, $redirect) { - //get everything up to the first ? - $index = (strpos($redirect, '?') == false ? strlen($redirect) : strpos($redirect, '?')); - $front = substr($redirect, 0, $index); - if ($front != $this->index) { //it doesn't redirect to WP's index.php - $this->add_external_rule($regex, $redirect); - } else { - $this->extra_rules[$regex] = $redirect; - } - } - - //add a rule that doesn't redirect to index.php - function add_external_rule($regex, $redirect) { - $this->non_wp_rules[$regex] = $redirect; - } - - //add an endpoint, like /trackback/, to be inserted after certain URL types (specified in $places) - function add_endpoint($name, $places) { - global $wp; - $this->endpoints[] = array ( $places, $name ); - $wp->add_query_var($name); - } - - function flush_rules() { - generate_page_uri_index(); - delete_option('rewrite_rules'); - $this->wp_rewrite_rules(); - if ( function_exists('save_mod_rewrite_rules') ) - save_mod_rewrite_rules(); - } - - function init() { - $this->extra_rules = $this->non_wp_rules = $this->endpoints = array(); - $this->permalink_structure = get_option('permalink_structure'); - $this->front = substr($this->permalink_structure, 0, strpos($this->permalink_structure, '%')); - $this->root = ''; - if ($this->using_index_permalinks()) { - $this->root = $this->index . '/'; - } - $this->category_base = get_option( 'category_base' ); - unset($this->category_structure); - unset($this->author_structure); - unset($this->date_structure); - unset($this->page_structure); - unset($this->search_structure); - unset($this->feed_structure); - unset($this->comment_feed_structure); - $this->use_trailing_slashes = ( substr($this->permalink_structure, -1, 1) == '/' ) ? true : false; - } - - function set_permalink_structure($permalink_structure) { - if ($permalink_structure != $this->permalink_structure) { - update_option('permalink_structure', $permalink_structure); - $this->init(); - } - } - - function set_category_base($category_base) { - if ($category_base != $this->category_base) { - update_option('category_base', $category_base); - $this->init(); - } - } - - function WP_Rewrite() { - $this->init(); - } -} - -?> +add_rule($regex, $redirect); +} + +//Add a new tag (like %postname%) +//warning: you must call this on init or earlier, otherwise the query var addition stuff won't work +function add_rewrite_tag($tagname, $regex) { + //validation + if (strlen($tagname) < 3 || $tagname{0} != '%' || $tagname{strlen($tagname)-1} != '%') { + return; + } + + $qv = trim($tagname, '%'); + + global $wp_rewrite, $wp; + $wp->add_query_var($qv); + $wp_rewrite->add_rewrite_tag($tagname, $regex, $qv . '='); +} + +//Add a new feed type like /atom1/ +function add_feed($feedname, $function) { + global $wp_rewrite; + if (!in_array($feedname, $wp_rewrite->feeds)) { //override the file if it is + $wp_rewrite->feeds[] = $feedname; + } + $hook = 'do_feed_' . $feedname; + remove_action($hook, $function, 10, 1); + add_action($hook, $function, 10, 1); + return $hook; +} + +define('EP_PERMALINK', 1 ); +define('EP_ATTACHMENT', 2 ); +define('EP_DATE', 4 ); +define('EP_YEAR', 8 ); +define('EP_MONTH', 16 ); +define('EP_DAY', 32 ); +define('EP_ROOT', 64 ); +define('EP_COMMENTS', 128 ); +define('EP_SEARCH', 256 ); +define('EP_CATEGORIES', 512 ); +define('EP_AUTHORS', 1024); +define('EP_PAGES', 2048); +//pseudo-places +define('EP_NONE', 0 ); +define('EP_ALL', 4095); + +//and an endpoint, like /trackback/ +function add_rewrite_endpoint($name, $places) { + global $wp_rewrite; + $wp_rewrite->add_endpoint($name, $places); +} + +// examine a url (supposedly from this blog) and try to +// determine the post ID it represents. +function url_to_postid($url) { + global $wp_rewrite; + + $url = apply_filters('url_to_postid', $url); + + // First, check to see if there is a 'p=N' or 'page_id=N' to match against + preg_match('#[?&](p|page_id)=(\d+)#', $url, $values); + $id = intval($values[2]); + if ( $id ) return $id; + + // Check to see if we are using rewrite rules + $rewrite = $wp_rewrite->wp_rewrite_rules(); + + // Not using rewrite rules, and 'p=N' and 'page_id=N' methods failed, so we're out of options + if ( empty($rewrite) ) + return 0; + + // $url cleanup by Mark Jaquith + // This fixes things like #anchors, ?query=strings, missing 'www.', + // added 'www.', or added 'index.php/' that will mess up our WP_Query + // and return a false negative + + // Get rid of the #anchor + $url_split = explode('#', $url); + $url = $url_split[0]; + + // Get rid of URL ?query=string + $url_split = explode('?', $url); + $url = $url_split[0]; + + // Add 'www.' if it is absent and should be there + if ( false !== strpos(get_option('home'), '://www.') && false === strpos($url, '://www.') ) + $url = str_replace('://', '://www.', $url); + + // Strip 'www.' if it is present and shouldn't be + if ( false === strpos(get_option('home'), '://www.') ) + $url = str_replace('://www.', '://', $url); + + // Strip 'index.php/' if we're not using path info permalinks + if ( !$wp_rewrite->using_index_permalinks() ) + $url = str_replace('index.php/', '', $url); + + if ( false !== strpos($url, get_option('home')) ) { + // Chop off http://domain.com + $url = str_replace(get_option('home'), '', $url); + } else { + // Chop off /path/to/blog + $home_path = parse_url(get_option('home')); + $home_path = $home_path['path']; + $url = str_replace($home_path, '', $url); + } + + // Trim leading and lagging slashes + $url = trim($url, '/'); + + $request = $url; + + // Done with cleanup + + // Look for matches. + $request_match = $request; + foreach ($rewrite as $match => $query) { + // If the requesting file is the anchor of the match, prepend it + // to the path info. + if ( (! empty($url)) && (strpos($match, $url) === 0) ) { + $request_match = $url . '/' . $request; + } + + if ( preg_match("!^$match!", $request_match, $matches) ) { + // Got a match. + // Trim the query of everything up to the '?'. + $query = preg_replace("!^.+\?!", '', $query); + + // Substitute the substring matches into the query. + eval("\$query = \"$query\";"); + $query = new WP_Query($query); + if ( $query->is_single || $query->is_page ) + return $query->post->ID; + else + return 0; + } + } + return 0; +} + +/* WP_Rewrite class +*******************************************************************************/ + +class WP_Rewrite { + var $permalink_structure; + var $use_trailing_slashes; + var $category_base; + var $category_structure; + var $author_base = 'author'; + var $author_structure; + var $date_structure; + var $page_structure; + var $search_base = 'search'; + var $search_structure; + var $comments_base = 'comments'; + var $feed_base = 'feed'; + var $comments_feed_structure; + var $feed_structure; + var $front; + var $root = ''; + var $index = 'index.php'; + var $matches = ''; + var $rules; + var $extra_rules; //those not generated by the class, see add_rewrite_rule() + var $non_wp_rules; //rules that don't redirect to WP's index.php + var $endpoints; + var $use_verbose_rules = false; + var $rewritecode = + array( + '%year%', + '%monthnum%', + '%day%', + '%hour%', + '%minute%', + '%second%', + '%postname%', + '%post_id%', + '%category%', + '%author%', + '%pagename%', + '%search%' + ); + + var $rewritereplace = + array( + '([0-9]{4})', + '([0-9]{1,2})', + '([0-9]{1,2})', + '([0-9]{1,2})', + '([0-9]{1,2})', + '([0-9]{1,2})', + '([^/]+)', + '([0-9]+)', + '(.+?)', + '([^/]+)', + '([^/]+)', + '(.+)' + ); + + var $queryreplace = + array ( + 'year=', + 'monthnum=', + 'day=', + 'hour=', + 'minute=', + 'second=', + 'name=', + 'p=', + 'category_name=', + 'author_name=', + 'pagename=', + 's=' + ); + + var $feeds = array ( 'feed', 'rdf', 'rss', 'rss2', 'atom' ); + + function using_permalinks() { + if (empty($this->permalink_structure)) + return false; + else + return true; + } + + function using_index_permalinks() { + if (empty($this->permalink_structure)) { + return false; + } + + // If the index is not in the permalink, we're using mod_rewrite. + if (preg_match('#^/*' . $this->index . '#', $this->permalink_structure)) { + return true; + } + + return false; + } + + function using_mod_rewrite_permalinks() { + if ( $this->using_permalinks() && ! $this->using_index_permalinks()) + return true; + else + return false; + } + + function preg_index($number) { + $match_prefix = '$'; + $match_suffix = ''; + + if (! empty($this->matches)) { + $match_prefix = '$' . $this->matches . '['; + $match_suffix = ']'; + } + + return "$match_prefix$number$match_suffix"; + } + + function page_rewrite_rules() { + $uris = get_option('page_uris'); + $attachment_uris = get_option('page_attachment_uris'); + + $rewrite_rules = array(); + $page_structure = $this->get_page_permastruct(); + if( is_array( $attachment_uris ) ) { + foreach ($attachment_uris as $uri => $pagename) { + $this->add_rewrite_tag('%pagename%', "($uri)", 'attachment='); + $rewrite_rules = array_merge($rewrite_rules, $this->generate_rewrite_rules($page_structure, EP_PAGES)); + } + } + if( is_array( $uris ) ) { + foreach ($uris as $uri => $pagename) { + $this->add_rewrite_tag('%pagename%', "($uri)", 'pagename='); + $rewrite_rules = array_merge($rewrite_rules, $this->generate_rewrite_rules($page_structure, EP_PAGES)); + } + } + + return $rewrite_rules; + } + + function get_date_permastruct() { + if (isset($this->date_structure)) { + return $this->date_structure; + } + + if (empty($this->permalink_structure)) { + $this->date_structure = ''; + return false; + } + + // The date permalink must have year, month, and day separated by slashes. + $endians = array('%year%/%monthnum%/%day%', '%day%/%monthnum%/%year%', '%monthnum%/%day%/%year%'); + + $this->date_structure = ''; + $date_endian = ''; + + foreach ($endians as $endian) { + if (false !== strpos($this->permalink_structure, $endian)) { + $date_endian= $endian; + break; + } + } + + if ( empty($date_endian) ) + $date_endian = '%year%/%monthnum%/%day%'; + + // Do not allow the date tags and %post_id% to overlap in the permalink + // structure. If they do, move the date tags to $front/date/. + $front = $this->front; + preg_match_all('/%.+?%/', $this->permalink_structure, $tokens); + $tok_index = 1; + foreach ($tokens[0] as $token) { + if ( ($token == '%post_id%') && ($tok_index <= 3) ) { + $front = $front . 'date/'; + break; + } + $tok_index++; + } + + $this->date_structure = $front . $date_endian; + + return $this->date_structure; + } + + function get_year_permastruct() { + $structure = $this->get_date_permastruct($this->permalink_structure); + + if (empty($structure)) { + return false; + } + + $structure = str_replace('%monthnum%', '', $structure); + $structure = str_replace('%day%', '', $structure); + + $structure = preg_replace('#/+#', '/', $structure); + + return $structure; + } + + function get_month_permastruct() { + $structure = $this->get_date_permastruct($this->permalink_structure); + + if (empty($structure)) { + return false; + } + + $structure = str_replace('%day%', '', $structure); + + $structure = preg_replace('#/+#', '/', $structure); + + return $structure; + } + + function get_day_permastruct() { + return $this->get_date_permastruct($this->permalink_structure); + } + + function get_category_permastruct() { + if (isset($this->category_structure)) { + return $this->category_structure; + } + + if (empty($this->permalink_structure)) { + $this->category_structure = ''; + return false; + } + + if (empty($this->category_base)) + $this->category_structure = $this->front . 'category/'; + else + $this->category_structure = $this->category_base . '/'; + + $this->category_structure .= '%category%'; + + return $this->category_structure; + } + + function get_author_permastruct() { + if (isset($this->author_structure)) { + return $this->author_structure; + } + + if (empty($this->permalink_structure)) { + $this->author_structure = ''; + return false; + } + + $this->author_structure = $this->front . $this->author_base . '/%author%'; + + return $this->author_structure; + } + + function get_search_permastruct() { + if (isset($this->search_structure)) { + return $this->search_structure; + } + + if (empty($this->permalink_structure)) { + $this->search_structure = ''; + return false; + } + + $this->search_structure = $this->root . $this->search_base . '/%search%'; + + return $this->search_structure; + } + + function get_page_permastruct() { + if (isset($this->page_structure)) { + return $this->page_structure; + } + + if (empty($this->permalink_structure)) { + $this->page_structure = ''; + return false; + } + + $this->page_structure = $this->root . '%pagename%'; + + return $this->page_structure; + } + + function get_feed_permastruct() { + if (isset($this->feed_structure)) { + return $this->feed_structure; + } + + if (empty($this->permalink_structure)) { + $this->feed_structure = ''; + return false; + } + + $this->feed_structure = $this->root . $this->feed_base . '/%feed%'; + + return $this->feed_structure; + } + + function get_comment_feed_permastruct() { + if (isset($this->comment_feed_structure)) { + return $this->comment_feed_structure; + } + + if (empty($this->permalink_structure)) { + $this->comment_feed_structure = ''; + return false; + } + + $this->comment_feed_structure = $this->root . $this->comments_base . '/' . $this->feed_base . '/%feed%'; + + return $this->comment_feed_structure; + } + + function add_rewrite_tag($tag, $pattern, $query) { + // If the tag already exists, replace the existing pattern and query for + // that tag, otherwise add the new tag, pattern, and query to the end of + // the arrays. + $position = array_search($tag, $this->rewritecode); + if (FALSE !== $position && NULL !== $position) { + $this->rewritereplace[$position] = $pattern; + $this->queryreplace[$position] = $query; + } else { + $this->rewritecode[] = $tag; + $this->rewritereplace[] = $pattern; + $this->queryreplace[] = $query; + } + } + + //the main WP_Rewrite function. generate the rules from permalink structure + function generate_rewrite_rules($permalink_structure, $ep_mask = EP_NONE, $paged = true, $feed = true, $forcomments = false, $walk_dirs = true, $endpoints = true) { + //build a regex to match the feed section of URLs, something like (feed|atom|rss|rss2)/? + $feedregex2 = ''; + foreach ($this->feeds as $feed_name) { + $feedregex2 .= $feed_name . '|'; + } + $feedregex2 = '(' . trim($feedregex2, '|') . ')/?$'; + //$feedregex is identical but with /feed/ added on as well, so URLs like /feed/atom + //and /atom are both possible + $feedregex = $this->feed_base . '/' . $feedregex2; + + //build a regex to match the trackback and page/xx parts of URLs + $trackbackregex = 'trackback/?$'; + $pageregex = 'page/?([0-9]{1,})/?$'; + + //build up an array of endpoint regexes to append => queries to append + if ($endpoints) { + $ep_query_append = array (); + foreach ($this->endpoints as $endpoint) { + //match everything after the endpoint name, but allow for nothing to appear there + $epmatch = $endpoint[1] . '(/(.*))?/?$'; + //this will be appended on to the rest of the query for each dir + $epquery = '&' . $endpoint[1] . '='; + $ep_query_append[$epmatch] = array ( $endpoint[0], $epquery ); + } + } + + //get everything up to the first rewrite tag + $front = substr($permalink_structure, 0, strpos($permalink_structure, '%')); + //build an array of the tags (note that said array ends up being in $tokens[0]) + preg_match_all('/%.+?%/', $permalink_structure, $tokens); + + $num_tokens = count($tokens[0]); + + $index = $this->index; //probably 'index.php' + $feedindex = $index; + $trackbackindex = $index; + //build a list from the rewritecode and queryreplace arrays, that will look something like + //tagname=$matches[i] where i is the current $i + for ($i = 0; $i < $num_tokens; ++$i) { + if (0 < $i) { + $queries[$i] = $queries[$i - 1] . '&'; + } + + $query_token = str_replace($this->rewritecode, $this->queryreplace, $tokens[0][$i]) . $this->preg_index($i+1); + $queries[$i] .= $query_token; + } + + //get the structure, minus any cruft (stuff that isn't tags) at the front + $structure = $permalink_structure; + if ($front != '/') { + $structure = str_replace($front, '', $structure); + } + //create a list of dirs to walk over, making rewrite rules for each level + //so for example, a $structure of /%year%/%month%/%postname% would create + //rewrite rules for /%year%/, /%year%/%month%/ and /%year%/%month%/%postname% + $structure = trim($structure, '/'); + if ($walk_dirs) { + $dirs = explode('/', $structure); + } else { + $dirs[] = $structure; + } + $num_dirs = count($dirs); + + //strip slashes from the front of $front + $front = preg_replace('|^/+|', '', $front); + + //the main workhorse loop + $post_rewrite = array(); + $struct = $front; + for ($j = 0; $j < $num_dirs; ++$j) { + //get the struct for this dir, and trim slashes off the front + $struct .= $dirs[$j] . '/'; //accumulate. see comment near explode('/', $structure) above + $struct = ltrim($struct, '/'); + //replace tags with regexes + $match = str_replace($this->rewritecode, $this->rewritereplace, $struct); + //make a list of tags, and store how many there are in $num_toks + $num_toks = preg_match_all('/%.+?%/', $struct, $toks); + //get the 'tagname=$matches[i]' + $query = $queries[$num_toks - 1]; + + //set up $ep_mask_specific which is used to match more specific URL types + switch ($dirs[$j]) { + case '%year%': $ep_mask_specific = EP_YEAR; break; + case '%monthnum%': $ep_mask_specific = EP_MONTH; break; + case '%day%': $ep_mask_specific = EP_DAY; break; + } + + //create query for /page/xx + $pagematch = $match . $pageregex; + $pagequery = $index . '?' . $query . '&paged=' . $this->preg_index($num_toks + 1); + + //create query for /feed/(feed|atom|rss|rss2|rdf) + $feedmatch = $match . $feedregex; + $feedquery = $feedindex . '?' . $query . '&feed=' . $this->preg_index($num_toks + 1); + + //create query for /(feed|atom|rss|rss2|rdf) (see comment near creation of $feedregex) + $feedmatch2 = $match . $feedregex2; + $feedquery2 = $feedindex . '?' . $query . '&feed=' . $this->preg_index($num_toks + 1); + + //if asked to, turn the feed queries into comment feed ones + if ($forcomments) { + $feedquery .= '&withcomments=1'; + $feedquery2 .= '&withcomments=1'; + } + + //start creating the array of rewrites for this dir + $rewrite = array(); + if ($feed) //...adding on /feed/ regexes => queries + $rewrite = array($feedmatch => $feedquery, $feedmatch2 => $feedquery2); + if ($paged) //...and /page/xx ones + $rewrite = array_merge($rewrite, array($pagematch => $pagequery)); + + //do endpoints + if ($endpoints) { + foreach ($ep_query_append as $regex => $ep) { + //add the endpoints on if the mask fits + if ($ep[0] & $ep_mask || $ep[0] & $ep_mask_specific) { + $rewrite[$match . $regex] = $index . '?' . $query . $ep[1] . $this->preg_index($num_toks + 2); + } + } + } + + //if we've got some tags in this dir + if ($num_toks) { + $post = false; + $page = false; + + //check to see if this dir is permalink-level: i.e. the structure specifies an + //individual post. Do this by checking it contains at least one of 1) post name, + //2) post ID, 3) page name, 4) timestamp (year, month, day, hour, second and + //minute all present). Set these flags now as we need them for the endpoints. + if (strpos($struct, '%postname%') !== false || strpos($struct, '%post_id%') !== false + || strpos($struct, '%pagename%') !== false + || (strpos($struct, '%year%') !== false && strpos($struct, '%monthnum%') !== false && strpos($struct, '%day%') !== false && strpos($struct, '%hour%') !== false && strpos($struct, '%minute%') !== false && strpos($struct, '%second%') !== false)) { + $post = true; + if (strpos($struct, '%pagename%') !== false) + $page = true; + } + + //if we're creating rules for a permalink, do all the endpoints like attachments etc + if ($post) { + $post = true; + //create query and regex for trackback + $trackbackmatch = $match . $trackbackregex; + $trackbackquery = $trackbackindex . '?' . $query . '&tb=1'; + //trim slashes from the end of the regex for this dir + $match = rtrim($match, '/'); + //get rid of brackets + $submatchbase = str_replace(array('(',')'),'',$match); + + //add a rule for at attachments, which take the form of /some-text + $sub1 = $submatchbase . '/([^/]+)/'; + $sub1tb = $sub1 . $trackbackregex; //add trackback regex /trackback/... + $sub1feed = $sub1 . $feedregex; //and /feed/(atom|...) + $sub1feed2 = $sub1 . $feedregex2; //and /(feed|atom...) + //add an ? as we don't have to match that last slash, and finally a $ so we + //match to the end of the URL + + //add another rule to match attachments in the explicit form: + ///attachment/some-text + $sub2 = $submatchbase . '/attachment/([^/]+)/'; + $sub2tb = $sub2 . $trackbackregex; //and add trackbacks /attachment/trackback + $sub2feed = $sub2 . $feedregex; //feeds, /attachment/feed/(atom|...) + $sub2feed2 = $sub2 . $feedregex2; //and feeds again on to this /attachment/(feed|atom...) + + //create queries for these extra tag-ons we've just dealt with + $subquery = $index . '?attachment=' . $this->preg_index(1); + $subtbquery = $subquery . '&tb=1'; + $subfeedquery = $subquery . '&feed=' . $this->preg_index(2); + + //do endpoints for attachments + if ($endpoint) { foreach ($ep_query_append as $regex => $ep) { + if ($ep[0] & EP_ATTACHMENT) { + $rewrite[$sub1 . $regex] = $subquery . '?' . $ep[1] . $this->preg_index(2); + $rewrite[$sub2 . $regex] = $subquery . '?' . $ep[1] . $this->preg_index(2); + } + } } + + //now we've finished with endpoints, finish off the $sub1 and $sub2 matches + $sub1 .= '?$'; + $sub2 .= '?$'; + + //allow URLs like /2 for /page/2 + $match = $match . '(/[0-9]+)?/?$'; + $query = $index . '?' . $query . '&page=' . $this->preg_index($num_toks + 1); + } else { //not matching a permalink so this is a lot simpler + //close the match and finalise the query + $match .= '?$'; + $query = $index . '?' . $query; + } + + //create the final array for this dir by joining the $rewrite array (which currently + //only contains rules/queries for trackback, pages etc) to the main regex/query for + //this dir + $rewrite = array_merge($rewrite, array($match => $query)); + + //if we're matching a permalink, add those extras (attachments etc) on + if ($post) { + //add trackback + $rewrite = array_merge(array($trackbackmatch => $trackbackquery), $rewrite); + + //add regexes/queries for attachments, attachment trackbacks and so on + if ( ! $page ) //require /attachment/stuff form for pages because of confusion with subpages + $rewrite = array_merge($rewrite, array($sub1 => $subquery, $sub1tb => $subtbquery, $sub1feed => $subfeedquery, $sub1feed2 => $subfeedquery)); + $rewrite = array_merge($rewrite, array($sub2 => $subquery, $sub2tb => $subtbquery, $sub2feed => $subfeedquery, $sub2feed2 => $subfeedquery)); + } + } //if($num_toks) + //add the rules for this dir to the accumulating $post_rewrite + $post_rewrite = array_merge($rewrite, $post_rewrite); + } //foreach ($dir) + return $post_rewrite; //the finished rules. phew! + } + + function generate_rewrite_rule($permalink_structure, $walk_dirs = false) { + return $this->generate_rewrite_rules($permalink_structure, EP_NONE, false, false, false, $walk_dirs); + } + + /* rewrite_rules + * Construct rewrite matches and queries from permalink structure. + * Returns an associate array of matches and queries. + */ + function rewrite_rules() { + $rewrite = array(); + + if (empty($this->permalink_structure)) { + return $rewrite; + } + + // robots.txt + $robots_rewrite = array('robots.txt$' => $this->index . '?robots=1'); + + //Default Feed rules - These are require to allow for the direct access files to work with permalink structure starting with %category% + $default_feeds = array( 'wp-atom.php$' => $this->index .'?feed=atom', + 'wp-rdf.php$' => $this->index .'?feed=rdf', + 'wp-rss.php$' => $this->index .'?feed=rss', + 'wp-rss2.php$' => $this->index .'?feed=rss2', + 'wp-feed.php$' => $this->index .'?feed=feed', + 'wp-commentsrss2.php$' => $this->index . '?feed=rss2&withcomments=1'); + + // Post + $post_rewrite = $this->generate_rewrite_rules($this->permalink_structure, EP_PERMALINK); + $post_rewrite = apply_filters('post_rewrite_rules', $post_rewrite); + + // Date + $date_rewrite = $this->generate_rewrite_rules($this->get_date_permastruct(), EP_DATE); + $date_rewrite = apply_filters('date_rewrite_rules', $date_rewrite); + + // Root + $root_rewrite = $this->generate_rewrite_rules($this->root . '/', EP_ROOT); + $root_rewrite = apply_filters('root_rewrite_rules', $root_rewrite); + + // Comments + $comments_rewrite = $this->generate_rewrite_rules($this->root . $this->comments_base, EP_COMMENTS, true, true, true, false); + $comments_rewrite = apply_filters('comments_rewrite_rules', $comments_rewrite); + + // Search + $search_structure = $this->get_search_permastruct(); + $search_rewrite = $this->generate_rewrite_rules($search_structure, EP_SEARCH); + $search_rewrite = apply_filters('search_rewrite_rules', $search_rewrite); + + // Categories + $category_rewrite = $this->generate_rewrite_rules($this->get_category_permastruct(), EP_CATEGORIES); + $category_rewrite = apply_filters('category_rewrite_rules', $category_rewrite); + + // Authors + $author_rewrite = $this->generate_rewrite_rules($this->get_author_permastruct(), EP_AUTHORS); + $author_rewrite = apply_filters('author_rewrite_rules', $author_rewrite); + + // Pages + $page_rewrite = $this->page_rewrite_rules(); + $page_rewrite = apply_filters('page_rewrite_rules', $page_rewrite); + + // Put them together. + $this->rules = array_merge($robots_rewrite, $default_feeds, $page_rewrite, $root_rewrite, $comments_rewrite, $search_rewrite, $category_rewrite, $author_rewrite, $date_rewrite, $post_rewrite, $this->extra_rules); + + do_action_ref_array('generate_rewrite_rules', array(&$this)); + $this->rules = apply_filters('rewrite_rules_array', $this->rules); + + return $this->rules; + } + + function wp_rewrite_rules() { + $this->rules = get_option('rewrite_rules'); + if ( empty($this->rules) ) { + $this->matches = 'matches'; + $this->rewrite_rules(); + update_option('rewrite_rules', $this->rules); + } + + return $this->rules; + } + + function mod_rewrite_rules() { + if ( ! $this->using_permalinks()) { + return ''; + } + + $site_root = parse_url(get_option('siteurl')); + $site_root = trailingslashit($site_root['path']); + + $home_root = parse_url(get_option('home')); + $home_root = trailingslashit($home_root['path']); + + $rules = "\n"; + $rules .= "RewriteEngine On\n"; + $rules .= "RewriteBase $home_root\n"; + + //add in the rules that don't redirect to WP's index.php (and thus shouldn't be handled by WP at all) + foreach ($this->non_wp_rules as $match => $query) { + // Apache 1.3 does not support the reluctant (non-greedy) modifier. + $match = str_replace('.+?', '.+', $match); + + // If the match is unanchored and greedy, prepend rewrite conditions + // to avoid infinite redirects and eclipsing of real files. + if ($match == '(.+)/?$' || $match == '([^/]+)/?$' ) { + //nada. + } + + $rules .= 'RewriteRule ^' . $match . ' ' . $home_root . $query . " [QSA,L]\n"; + } + + if ($this->use_verbose_rules) { + $this->matches = ''; + $rewrite = $this->rewrite_rules(); + $num_rules = count($rewrite); + $rules .= "RewriteCond %{REQUEST_FILENAME} -f [OR]\n" . + "RewriteCond %{REQUEST_FILENAME} -d\n" . + "RewriteRule ^.*$ - [S=$num_rules]\n"; + + foreach ($rewrite as $match => $query) { + // Apache 1.3 does not support the reluctant (non-greedy) modifier. + $match = str_replace('.+?', '.+', $match); + + // If the match is unanchored and greedy, prepend rewrite conditions + // to avoid infinite redirects and eclipsing of real files. + if ($match == '(.+)/?$' || $match == '([^/]+)/?$' ) { + //nada. + } + + if (strpos($query, $this->index) !== false) { + $rules .= 'RewriteRule ^' . $match . ' ' . $home_root . $query . " [QSA,L]\n"; + } else { + $rules .= 'RewriteRule ^' . $match . ' ' . $site_root . $query . " [QSA,L]\n"; + } + } + } else { + $rules .= "RewriteCond %{REQUEST_FILENAME} !-f\n" . + "RewriteCond %{REQUEST_FILENAME} !-d\n" . + "RewriteRule . {$home_root}{$this->index} [L]\n"; + } + + $rules .= "\n"; + + $rules = apply_filters('mod_rewrite_rules', $rules); + $rules = apply_filters('rewrite_rules', $rules); // Deprecated + + return $rules; + } + + //Add a straight rewrite rule + function add_rule($regex, $redirect) { + //get everything up to the first ? + $index = (strpos($redirect, '?') == false ? strlen($redirect) : strpos($redirect, '?')); + $front = substr($redirect, 0, $index); + if ($front != $this->index) { //it doesn't redirect to WP's index.php + $this->add_external_rule($regex, $redirect); + } else { + $this->extra_rules[$regex] = $redirect; + } + } + + //add a rule that doesn't redirect to index.php + function add_external_rule($regex, $redirect) { + $this->non_wp_rules[$regex] = $redirect; + } + + //add an endpoint, like /trackback/, to be inserted after certain URL types (specified in $places) + function add_endpoint($name, $places) { + global $wp; + $this->endpoints[] = array ( $places, $name ); + $wp->add_query_var($name); + } + + function flush_rules() { + generate_page_uri_index(); + delete_option('rewrite_rules'); + $this->wp_rewrite_rules(); + if ( function_exists('save_mod_rewrite_rules') ) + save_mod_rewrite_rules(); + } + + function init() { + $this->extra_rules = $this->non_wp_rules = $this->endpoints = array(); + $this->permalink_structure = get_option('permalink_structure'); + $this->front = substr($this->permalink_structure, 0, strpos($this->permalink_structure, '%')); + $this->root = ''; + if ($this->using_index_permalinks()) { + $this->root = $this->index . '/'; + } + $this->category_base = get_option('category_base'); + unset($this->category_structure); + unset($this->author_structure); + unset($this->date_structure); + unset($this->page_structure); + unset($this->search_structure); + unset($this->feed_structure); + unset($this->comment_feed_structure); + $this->use_trailing_slashes = ( substr($this->permalink_structure, -1, 1) == '/' ) ? true : false; + } + + function set_permalink_structure($permalink_structure) { + if ($permalink_structure != $this->permalink_structure) { + update_option('permalink_structure', $permalink_structure); + $this->init(); + } + } + + function set_category_base($category_base) { + if ($category_base != $this->category_base) { + update_option('category_base', $category_base); + $this->init(); + } + } + + function WP_Rewrite() { + $this->init(); + } +} + +?> -- cgit