Ticket #43728: 43728.1.patch

wp-includes/class-wp-theme.php

@@ -1482 +1482 @@
         }
 
         /**
+         * Enables a theme for a site.
+         *
+         * @since 5.0.0
+         *
+         * @param int $site_id Site ID.
+         * @param string|string[] $stylesheets Stylesheet name or array of stylesheet names.
+         */
+        public static function site_enable_theme( $site_id = null, $stylesheets ) {
+                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' );
+                foreach ( $stylesheets as $stylesheet ) {
+                        $allowed_themes[ $stylesheet ] = true;
+                }
+                update_option( 'allowedthemes', $allowed_themes );
+
+                if ( $site_id !== $current_site_id ) {
+                        restore_current_blog();
+                }
+        }
+
+        /**
+         * Disables a theme for a site.
+         *
+         * @since 5.0.0
+         *
+         * @param int $site_id Site ID.
+         * @param string|string[] $stylesheets Stylesheet name or array of stylesheet names.
+         */
+        public static function site_disable_theme( $site_id = null, $stylesheets ) {
+                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' );
+                foreach ( $stylesheets as $stylesheet ) {
+                        if ( isset( $allowed_themes[ $stylesheet ] ) ) {
+                                unset( $allowed_themes[ $stylesheet ] );
+                        }
+                }
+                update_option( 'allowedthemes', $allowed_themes );
+
+                if ( $site_id !== $current_site_id ) {
+                        restore_current_blog();
+                }
+        }
+
+        /**
          * Enables a theme for all sites on the current network.
          *
          * @since 4.6.0