Changeset 34708 for trunk/src/wp-includes/class-wp-rewrite.php

Timestamp:

09/30/2015 01:10:03 AM (11 years ago)

Author:

wonderboymusic

Message:

Rewrite: allow add_rewrite_rule|WP_Rewrite::add_rule() to accept an associative array for the value of $redirect instead of requiring a query string.

Adds unit tests.

Props scribu, DrewAPicture.
Fixes #16840.

File:

• trunk/src/wp-includes/class-wp-rewrite.php (modified) (1 diff)

trunk/src/wp-includes/class-wp-rewrite.php

@@ -1511 +1511 @@
          *
          * @since 2.1.0
-         * @access public
-         *
-         * @param string $regex    Regular expression to match against request.
-         * @param string $redirect URL regex redirects to when regex matches request.
-         * @param string $after    Optional, default is bottom. Location to place rule.
-         */
-        public function add_rule($regex, $redirect, $after = 'bottom') {
-                //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);
+         * @since 4.4.0 Array support was added to the `$redirect` parameter.
+         * @access public
+         *
+         * @param string       $regex    Regular expression to match against request.
+         * @param string|array $redirect URL regex redirects to when regex matches request, or array
+         *                               of query vars and values.
+         * @param string       $after    Optional, default is bottom. Location to place rule.
+         */
+        public function add_rule( $regex, $redirect, $after = 'bottom' ) {
+                if ( is_array( $redirect ) ) {
+                        $external = false;
+                        $redirect = add_query_arg( $redirect, 'index.php' );
                 } else {
-                        if ( 'bottom' == $after)
-                                $this->extra_rules = array_merge($this->extra_rules, array($regex => $redirect));
-                        else
-                                $this->extra_rules_top = array_merge($this->extra_rules_top, array($regex => $redirect));
-                        //$this->extra_rules[$regex] = $redirect;
+                        $index = false === strpos( $redirect, '?' ) ? strlen( $redirect ) : strpos( $redirect, '?' );
+                        $front = substr( $redirect, 0, $index );
+
+                        $external = $front != $this->index;
+                }
+
+                // "external" = it doesn't redirect to index.php
+                if ( $external ) {
+                        $this->add_external_rule( $regex, $redirect );
+                } else {
+                        if ( 'bottom' == $after ) {
+                                $this->extra_rules = array_merge( $this->extra_rules, array( $regex => $redirect ) );
+                        } else {
+                                $this->extra_rules_top = array_merge( $this->extra_rules_top, array( $regex => $redirect ) );
+                        }
                 }
         }