Ticket #21742: 21742.diff

wp-includes/load.php

@@ -583 +583 @@
  * @return bool True if inside WordPress administration pages.
  */
 function is_admin() {
-        if ( defined( 'WP_ADMIN' ) )
+        if ( class_exists( 'WP_Screen' ) && $screen = get_current_screen() )
+                return $screen->is_admin;
+        elseif ( defined( 'WP_ADMIN' ) )
                 return WP_ADMIN;
+
         return false;
 }
 
@@ -599 +602 @@
  * @return bool True if inside WordPress network administration pages.
  */
 function is_blog_admin() {
-        if ( defined( 'WP_BLOG_ADMIN' ) )
+        if ( class_exists( 'WP_Screen' ) && $screen = get_current_screen() )
+                return $screen->is_blog;
+        elseif ( defined( 'WP_BLOG_ADMIN' ) )
                 return WP_BLOG_ADMIN;
+
         return false;
 }
 
@@ -615 +621 @@
  * @return bool True if inside WordPress network administration pages.
  */
 function is_network_admin() {
-        if ( defined( 'WP_NETWORK_ADMIN' ) )
+        if ( class_exists( 'WP_Screen' ) && $screen = get_current_screen() )
+                return $screen->is_network;
+        elseif ( defined( 'WP_NETWORK_ADMIN' ) )
                 return WP_NETWORK_ADMIN;
         return false;
 }
@@ -631 +639 @@
  * @return bool True if inside WordPress user administration pages.
  */
 function is_user_admin() {
-        if ( defined( 'WP_USER_ADMIN' ) )
+        if ( class_exists( 'WP_Screen' ) && $screen = get_current_screen() )
+                return $screen->is_user;
+        elseif ( defined( 'WP_USER_ADMIN' ) )
                 return WP_USER_ADMIN;
         return false;
 }

wp-admin/includes/screen.php

@@ -260 +260 @@
          * @access public
          */
         public $is_user;
+        public $is_blog;
+        public $is_admin;
 
         /**
          * The base menu parent.
@@ -377 +379 @@
                         return $hook_name;
 
                 $post_type = $taxonomy = null;
-                $is_network = $is_user = false;
+                $is_network = $is_user = $is_blog = $is_admin = false;
                 $action = '';
 
                 if ( $hook_name )
@@ -402 +404 @@
                 if ( ! $post_type && $hook_name ) {
                         if ( '-network' == substr( $id, -8 ) ) {
                                 $id = substr( $id, 0, -8 );
-                                $is_network = true;
+                                $is_network = $is_admin = true;
                         } elseif ( '-user' == substr( $id, -5 ) ) {
                                 $id = substr( $id, 0, -5 );
-                                $is_user = true;
+                                $is_user = $is_admin = true;
                         }
 
                         $id = sanitize_key( $id );
@@ -419 +421 @@
                                         $post_type = $maybe;
                                 }
                         }
+
+                        if ( ! $is_network && ! $is_user )
+                                $is_blog = true;
                 } else {
-                        $is_network = is_network_admin();
-                        $is_user = is_user_admin();
+                        $is_network = defined( 'WP_NETWORK_ADMIN' ) ? WP_NETWORK_ADMIN : false;
+                        $is_user = defined( 'WP_USER_ADMIN' ) ? WP_USER_ADMIN : false;
+                        $is_blog = defined( 'WP_BLOG_ADMIN' ) ? WP_BLOG_ADMIN : false;
+                        $is_admin = defined( 'WP_ADMIN' ) ? WP_ADMIN : false;
                 }
 
                 if ( 'index' == $id )
@@ -499 +506 @@
                 $screen->taxonomy   = (string) $taxonomy;
                 $screen->is_user    = $is_user;
                 $screen->is_network = $is_network;
+                $screen->is_blog    = $is_blog;
+                $screen->is_admin = $is_admin;
 
                 self::$_registry[ $id ] = $screen;