Ticket #43728: 43728.3.diff

src/wp-includes/class-wp-theme.php

@@ -2057 +2057 @@
                 delete_site_transient( 'wp_theme_files_patterns-' . $this->cache_hash );
         }
 
+        /**
+         * Enables a theme for a site.
+         *
+         * @since x.x.0
+         *
+         * @param string|string[] $stylesheets Stylesheet name or array of stylesheet names.
+         * @param int             $site_id     Site ID.
+         */
+        public static function site_enable_theme( $stylesheets, $site_id = null ) {
+                self::_handle_site_theme_operation(
+                        $stylesheets,
+                        $site_id,
+                        function ( &$allowed_themes, $stylesheets ) {
+                                foreach ( $stylesheets as $stylesheet ) {
+                                        $allowed_themes[ $stylesheet ] = true;
+                                }
+                        }
+                );
+        }
+
+        /**
+         * Disables a theme for a site.
+         *
+         * @since x.x.0
+         *
+         * @param string|string[] $stylesheets Stylesheet name or array of stylesheet names.
+         * @param int             $site_id     Site ID.
+         */
+        public static function site_disable_theme( $stylesheets, $site_id = null ) {
+                self::_handle_site_theme_operation(
+                        $stylesheets,
+                        $site_id,
+                        function ( &$allowed_themes, $stylesheets ) {
+                                foreach ( $stylesheets as $stylesheet ) {
+                                        if ( isset( $allowed_themes[ $stylesheet ] ) ) {
+                                                unset( $allowed_themes[ $stylesheet ] );
+                                        }
+                                }
+                        }
+                );
+        }
+
         /**
          * Enables a theme for all sites on the current network.
          *
@@ -2167 +2209 @@
                 }
                 return true;
         }
+
+        /**
+         * Handles common logic for site theme operations.
+         *
+         * @since x.x.0
+         * @access private
+         *
+         * @param string|string[] $stylesheets Stylesheet name or array of stylesheet names.
+         * @param int             $site_id     Site ID.
+         * @param callable        $operation   Callback function to perform on the allowed themes.
+         * @return void
+         */
+        private static function _handle_site_theme_operation( $stylesheets, $site_id, $operation ) {
+                $site_id = (int) $site_id;
+
+                if ( ! is_multisite() ) {
+                        return;
+                }
+
+                if ( ! is_array( $stylesheets ) ) {
+                        $stylesheets = array( $stylesheets );
+                }
+
+                $current_site_id = get_current_blog_id();
+                if ( ! $site_id ) {
+                        $site_id = $current_site_id;
+                }
+
+                if ( $site_id !== $current_site_id ) {
+                        switch_to_blog( $site_id );
+                }
+
+                $allowed_themes = get_option( 'allowedthemes' );
+                $operation( $allowed_themes, $stylesheets );
+                update_option( 'allowedthemes', $allowed_themes );
+
+                if ( $site_id !== $current_site_id ) {
+                        restore_current_blog();
+                }
+        }
 }