WordPress.org

Make WordPress Core

Ticket #41936: 41936.7.diff

File 41936.7.diff, 6.2 KB (added by jeremyfelt, 3 years ago)
  • src/wp-includes/class-wp-network.php

     
    148148                        case 'id':
    149149                                return (int) $this->id;
    150150                        case 'blog_id':
    151                                 return $this->get_main_site_id();
     151                                return (string) $this->get_main_site_id();
    152152                        case 'site_id':
    153                                 return (int) $this->get_main_site_id();
     153                                return $this->get_main_site_id();
    154154                }
    155155
    156156                return null;
     
    208208         * properties.
    209209         *
    210210         * @since 4.9.0
    211          * @see get_main_site_id()
    212211         *
    213          * @return string Main site ID as numeric string, for compatibility reasons.
     212         * @return int The ID of the main site.
    214213         */
    215214        private function get_main_site_id() {
    216                 if ( empty( $this->blog_id ) ) {
    217                         $this->blog_id = (string) get_main_site_id( $this->id );
     215                /**
     216                 * Filters the main site ID.
     217                 *
     218                 * Returning a positive integer will effectively short-circuit the function.
     219                 *
     220                 * @since 4.9.0
     221                 *
     222                 * @param int|null $main_site_id If a positive integer is returned, it is interpreted as the main site ID.
     223                 * @param int $network_id The ID of the network for which the main site was detected.
     224                 */
     225                $main_site_id = (int) apply_filters( 'pre_get_main_site_id', null, $this->id );
     226                if ( 0 < $main_site_id ) {
     227                        return $main_site_id;
     228                }
     229
     230                if ( 0 < (int) $this->blog_id ) {
     231                        return (int) $this->blog_id;
     232                }
     233
     234                if ( ( defined( 'DOMAIN_CURRENT_SITE' ) && defined( 'PATH_CURRENT_SITE' ) && $this->domain === DOMAIN_CURRENT_SITE && $this->path === PATH_CURRENT_SITE )
     235                        || ( defined( 'SITE_ID_CURRENT_SITE' ) && $this->id == SITE_ID_CURRENT_SITE ) ) {
     236                        if ( defined( 'BLOG_ID_CURRENT_SITE' ) ) {
     237                                $this->blog_id = (string) BLOG_ID_CURRENT_SITE;
     238
     239                                return (int) $this->blog_id;
     240                        }
     241
     242                        if ( defined( 'BLOGID_CURRENT_SITE' ) ) { // deprecated.
     243                                $this->blog_id = (string) BLOGID_CURRENT_SITE;
     244
     245                                return (int) $this->blog_id;
     246                        }
    218247                }
    219248
    220                 return $this->blog_id;
     249                $site = get_site();
     250                if ( $site->domain === $this->domain && $site->path === $this->path ) {
     251                        $main_site_id = (int) $site->id;
     252                } else {
     253                        $cache_key = 'network:' . $this->id . ':main_site';
     254
     255                        $main_site_id = wp_cache_get( $cache_key, 'site-options' );
     256                        if ( false === $main_site_id ) {
     257                                $_sites = get_sites( array(
     258                                        'fields'     => 'ids',
     259                                        'number'     => 1,
     260                                        'domain'     => $this->domain,
     261                                        'path'       => $this->path,
     262                                        'network_id' => $this->id,
     263                                ) );
     264                                $main_site_id = ! empty( $_sites ) ? array_shift( $_sites ) : 0;
     265
     266                                wp_cache_add( $cache_key, $main_site_id, 'site-options' );
     267                        }
     268                }
     269
     270                $this->blog_id = (string) $main_site_id;
     271
     272                return (int) $this->blog_id;
    221273        }
    222274
    223275        /**
  • src/wp-includes/functions.php

     
    44314431                return 0;
    44324432        }
    44334433
    4434         /**
    4435          * Filters the main site ID.
    4436          *
    4437          * Returning anything other than null will effectively short-circuit the function, returning
    4438          * the result parsed as an integer immediately.
    4439          *
    4440          * @since 4.9.0
    4441          *
    4442          * @param int|null $main_site_id If anything other than null is returned, it is interpreted as the main site ID.
    4443          * @param int $network_id The ID of the network for which the main site was detected.
    4444          */
    4445         $main_site_id = apply_filters( 'pre_get_main_site_id', null, $network->id );
    4446         if ( null !== $main_site_id ) {
    4447                 return (int) $main_site_id;
    4448         }
    4449 
    4450         if ( ( defined( 'DOMAIN_CURRENT_SITE' ) && defined( 'PATH_CURRENT_SITE' ) && $network->domain === DOMAIN_CURRENT_SITE && $network->path === PATH_CURRENT_SITE )
    4451              || ( defined( 'SITE_ID_CURRENT_SITE' ) && $network->id == SITE_ID_CURRENT_SITE ) ) {
    4452                 if ( defined( 'BLOG_ID_CURRENT_SITE' ) ) {
    4453                         return BLOG_ID_CURRENT_SITE;
    4454                 } elseif ( defined( 'BLOGID_CURRENT_SITE' ) ) { // deprecated.
    4455                         return BLOGID_CURRENT_SITE;
    4456                 }
    4457         }
    4458 
    4459         $site = get_site();
    4460         if ( $site->domain === $network->domain && $site->path === $network->path ) {
    4461                 $main_site_id = (int) $site->id;
    4462         } else {
    4463                 $main_site_id = wp_cache_get( 'network:' . $network->id . ':main_site', 'site-options' );
    4464                 if ( false === $main_site_id ) {
    4465                         $_sites = get_sites( array(
    4466                                 'fields'     => 'ids',
    4467                                 'number'     => 1,
    4468                                 'domain'     => $network->domain,
    4469                                 'path'       => $network->path,
    4470                                 'network_id' => $network->id,
    4471                         ) );
    4472                         $main_site_id = ! empty( $_sites ) ? array_shift( $_sites ) : 0;
    4473 
    4474                         wp_cache_add( 'network:' . $network->id . ':main_site', $main_site_id, 'site-options' );
    4475                 }
    4476         }
    4477 
    4478         return (int) $main_site_id;
     4434        return $network->site_id;
    44794435}
    44804436
    44814437/**
  • tests/phpunit/tests/multisite/getMainSiteId.php

     
    9898        public function test_get_main_site_id_filtered() {
    9999                add_filter( 'pre_get_main_site_id', array( $this, 'filter_get_main_site_id' ) );
    100100                $result = get_main_site_id();
     101                remove_filter( 'pre_get_main_site_id', array( $this, 'filter_get_main_site_id' ) );
    101102
    102103                $this->assertSame( 333, $result );
    103104        }
     
    105106        public function filter_get_main_site_id() {
    106107                return 333;
    107108        }
     109
     110        /**
     111         * @ticket 41936
     112         */
     113        public function test_get_main_site_id_with_property_value() {
     114                global $current_site;
     115
     116                $original_main_site_id = $current_site->blog_id;
     117                $current_site->blog_id = '123';
     118
     119                $result = get_main_site_id();
     120
     121                $current_site->blog_id = $original_main_site_id;
     122
     123                $this->assertSame( 123, $result );
     124        }
     125
     126        /**
     127         * @ticket 41936
     128         */
     129        public function test_get_main_site_id_filtered_with_property_value() {
     130                global $current_site;
     131
     132                $original_main_site_id = $current_site->blog_id;
     133                $current_site->blog_id = '123';
     134
     135                add_filter( 'pre_get_main_site_id', array( $this, 'filter_get_main_site_id' ) );
     136                $result = get_main_site_id();
     137                remove_filter( 'pre_get_main_site_id', array( $this, 'filter_get_main_site_id' ) );
     138
     139                $current_site->blog_id = $original_main_site_id;
     140
     141                $this->assertSame( 333, $result );
     142        }
    108143}
    109144
    110145endif;