Ticket #15920: 15920-take2.patch

wp-admin/network/themes.php

@@ -37 +37 @@
                         check_admin_referer('enable-theme_' . $_GET['theme']);
                         $allowed_themes[ $_GET['theme'] ] = true;
                         update_site_option( 'allowedthemes', $allowed_themes );
-                        wp_redirect( wp_get_referer() ); // @todo add_query_arg for update message
-                        exit;                   
+                        wp_redirect( add_query_arg( 'update', 'enabled', wp_get_referer() ) );
+                        exit;
                         break;
                 case 'disable':
                         check_admin_referer('disable-theme_' . $_GET['theme']);
                         unset( $allowed_themes[ $_GET['theme'] ] );
                         update_site_option( 'allowedthemes', $allowed_themes );
-                        wp_redirect( wp_get_referer() ); // @todo add_query_arg for update message
+                        wp_redirect( add_query_arg( 'update', 'disabled', wp_get_referer() ) );
                         exit;
                         break;
                 case 'enable-selected':
                         check_admin_referer('bulk-themes');
                         $themes = isset( $_POST['checked'] ) ? (array) $_POST['checked'] : array();
                         if ( empty($themes) ) {
-                                wp_redirect( wp_get_referer() );
+                                wp_redirect( add_query_arg( 'update', 'error', wp_get_referer() ) );
                                 exit;
-                        }                                               
+                        }
                         foreach( (array) $themes as $theme )
                                 $allowed_themes[ $theme ] = true;
                         update_site_option( 'allowedthemes', $allowed_themes );
@@ -62 +62 @@
                         check_admin_referer('bulk-themes');
                         $themes = isset( $_POST['checked'] ) ? (array) $_POST['checked'] : array();
                         if ( empty($themes) ) {
-                                wp_redirect( wp_get_referer() );
+                                wp_redirect( add_query_arg( 'update', 'error', wp_get_referer() ) );
                                 exit;
-                        }                                               
+                        }
                         foreach( (array) $themes as $theme )
                                 unset( $allowed_themes[ $theme ] );
                         update_site_option( 'allowedthemes', $allowed_themes );
@@ -83 +83 @@
 
                         $themes = isset( $_REQUEST['checked'] ) ? (array) $_REQUEST['checked'] : array();
                         if ( empty( $themes ) ) {
-                                wp_redirect( wp_get_referer() );
+                                wp_redirect( add_query_arg( 'update', 'error', wp_get_referer() ) );
                                 exit;
                         }
 
@@ -195 +195 @@
         printf( '<span class="subtitle">' . __('Search results for &#8220;%s&#8221;') . '</span>', esc_html( $s ) ); ?> 
 </h2>
 
+<?php
+if ( isset( $_GET['update'] ) ) {
+        switch ( $_GET['update'] ) {
+        case 'enabled':
+                echo '<div id="message" class="updated"><p>' . __( 'Theme enabled.' ) . '</p></div>';
+                break;
+        case 'disabled':
+                echo '<div id="message" class="updated"><p>' . __( 'Theme disabled.' ) . '</p></div>';
+                break;
+        case 'error':
+                echo '<div id="message" class="error"><p>' . __( 'No theme selected.' ) . '</p></div>';
+                break;
+        }
+}
+?>
+
 <form method="get" action="">
 <?php $wp_list_table->search_box( __( 'Search Installed Themes' ), 'theme' ); ?>
 </form>