Changeset 41861 for trunk/src/wp-includes/class-wp-network.php

Timestamp:

10/14/2017 10:41:15 PM (7 years ago)

Author:

flixos90

Message:

Multisite: Take WP_Network::$blog_id into account in get_main_site_id().

When the WP_Network::$blog_id property is set manually, for example in the multisite bootstrap process, get_main_site_id() should use that value instead of running its own logic. The main logic for the function was therefore moved into the internal WP_Network::get_main_site_id() method, which is now being accessed by the function through the magic property handling for WP_Network::$blog_id (and its equivalent WP_Network::$site_id).

Props spacedmonkey, jeremyfelt.
Fixes #41936.

File:

• trunk/src/wp-includes/class-wp-network.php (modified) (2 diffs)

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

@@ -149 +149 @@
                 return (int) $this->id;
             case 'blog_id':
+                return (string) $this->get_main_site_id();
+            case 'site_id':
                 return $this->get_main_site_id();
-            case 'site_id':
-                return (int) $this->get_main_site_id();
         }
 
@@ -209 +209 @@
      *
      * @since 4.9.0
-     * @see get_main_site_id()
-     *
-     * @return string Main site ID as numeric string, for compatibility reasons.
+     *
+     * @return int The ID of the main site.
      */
     private function get_main_site_id() {
-        if ( empty( $this->blog_id ) ) {
-            $this->blog_id = (string) get_main_site_id( $this->id );
-        }
-
-        return $this->blog_id;
+        /**
+         * Filters the main site ID.
+         *
+         * Returning a positive integer will effectively short-circuit the function.
+         *
+         * @since 4.9.0
+         *
+         * @param int|null $main_site_id If a positive integer is returned, it is interpreted as the main site ID.
+         * @param int $network_id The ID of the network for which the main site was detected.
+         */
+        $main_site_id = (int) apply_filters( 'pre_get_main_site_id', null, $this->id );
+        if ( 0 < $main_site_id ) {
+            return $main_site_id;
+        }
+
+        if ( 0 < (int) $this->blog_id ) {
+            return (int) $this->blog_id;
+        }
+
+        if ( ( defined( 'DOMAIN_CURRENT_SITE' ) && defined( 'PATH_CURRENT_SITE' ) && $this->domain === DOMAIN_CURRENT_SITE && $this->path === PATH_CURRENT_SITE )
+            || ( defined( 'SITE_ID_CURRENT_SITE' ) && $this->id == SITE_ID_CURRENT_SITE ) ) {
+            if ( defined( 'BLOG_ID_CURRENT_SITE' ) ) {
+                $this->blog_id = (string) BLOG_ID_CURRENT_SITE;
+
+                return (int) $this->blog_id;
+            }
+
+            if ( defined( 'BLOGID_CURRENT_SITE' ) ) { // deprecated.
+                $this->blog_id = (string) BLOGID_CURRENT_SITE;
+
+                return (int) $this->blog_id;
+            }
+        }
+
+        $site = get_site();
+        if ( $site->domain === $this->domain && $site->path === $this->path ) {
+            $main_site_id = (int) $site->id;
+        } else {
+            $cache_key = 'network:' . $this->id . ':main_site';
+
+            $main_site_id = wp_cache_get( $cache_key, 'site-options' );
+            if ( false === $main_site_id ) {
+                $_sites = get_sites( array(
+                    'fields'     => 'ids',
+                    'number'     => 1,
+                    'domain'     => $this->domain,
+                    'path'       => $this->path,
+                    'network_id' => $this->id,
+                ) );
+                $main_site_id = ! empty( $_sites ) ? array_shift( $_sites ) : 0;
+
+                wp_cache_add( $cache_key, $main_site_id, 'site-options' );
+            }
+        }
+
+        $this->blog_id = (string) $main_site_id;
+
+        return (int) $this->blog_id;
     }