Changeset 1514 for sites/trunk/wordcamp.org/public_html/wp-content/mu-plugins/wcorg-misc.php

Timestamp:

04/24/2015 04:07:50 AM (10 years ago)

Author:

iandunn

Message:

WordCamp Misc: Add wp-cli command to flush rewrite rules on all sites.

File:

• sites/trunk/wordcamp.org/public_html/wp-content/mu-plugins/wcorg-misc.php (modified) (1 diff)

sites/trunk/wordcamp.org/public_html/wp-content/mu-plugins/wcorg-misc.php

@@ -158 +158 @@
 }
 add_action( 'template_redirect', 'wcorg_subdomactories_redirect' );
+
+/*
+ * Flush the rewrite rules on the current site.
+ *
+ * See WordCamp_Miscellaneous_Commands::flush_rewrite_rules_everywhere() for an explanation.
+ *
+ * Requires authentication because flush_rewrite_rules() is expensive and could be used as a DoS vector.
+ */
+function wcorg_flush_rewrite_rules() {
+    if ( isset( $_GET['nonce'] ) && wp_verify_nonce( $_GET['nonce'], 'flush-rewrite-rules-everywhere-' . get_current_blog_id() ) ) {
+        flush_rewrite_rules();
+        wp_send_json_success( 'Rewrite rules have been flushed.' );
+    } else {
+        wp_send_json_error( 'You are not authorized to flush the rewrite rules.' );
+    }
+}
+add_action( 'wp_ajax_wcorg_flush_rewrite_rules_everywhere',        'wcorg_flush_rewrite_rules' ); // This isn't used by the wp-cli command, but is useful for manual testing
+add_action( 'wp_ajax_nopriv_wcorg_flush_rewrite_rules_everywhere', 'wcorg_flush_rewrite_rules' );
+
+
+/*
+ * WP-CLI Commands
+ */
+if ( defined( 'WP_CLI' ) && WP_CLI ) {
+    /**
+     * WordCamp.org Miscellaneous Commands
+     */
+    class WordCamp_Miscellaneous_Commands extends WP_CLI_Command {
+        /**
+         * Flush rewrite rules on all sites.
+         *
+         * Periodically they break for various reasons and need to be reset on all sites. If we
+         * just called flush_rewrite_rules() inside a switch_to_blog() loop then each site's
+         * plugins wouldn't be loaded and the rewrite rules wouldn't be correct.
+         *
+         * So instead, this issues an HTTP request to wcorg_flush_rewrite_rules() on each site so
+         * that flush_rewrite_rules() will run in the context of the loaded site.
+         *
+         * @subcommand flush-rewrite-rules-everywhere
+         */
+        public function flush_rewrite_rules_everywhere() {
+            $start_timestamp = microtime( true );
+            $error           = '';
+            $sites           = wp_get_sites( array( 'limit' => false ) );
+
+            WP_CLI::line();
+
+            foreach ( $sites as $site ) {
+                $ajax_url    = sprintf( 'http://%s%swp-admin/admin-ajax.php', $site['domain'], $site['path'] );
+                $display_url = $site['domain'] . rtrim( $site['path'], '/' );
+                $nonce       = wp_create_nonce( 'flush-rewrite-rules-everywhere-' . $site['blog_id'] );
+
+                $response = wp_remote_get( esc_url_raw( add_query_arg(
+                    array(
+                        'action' => 'wcorg_flush_rewrite_rules_everywhere',
+                        'nonce'  => $nonce,
+                    ),
+                    $ajax_url
+                ) ) );
+
+                if ( is_wp_error( $response ) ) {
+                    $success = false;
+                    $error   = $response->get_error_message();
+                } else {
+                    $response = json_decode( wp_remote_retrieve_body( $response ) );
+
+                    if ( isset( $response->success ) && $response->success ) {
+                        $success = true;
+                    } else {
+                        $success = false;
+                        $error   = isset( $response->data ) ? $response->data : 'Unknown error';
+                    }
+                }
+
+                if ( $success ) {
+                    WP_CLI::line( sprintf( '%s: Flushed', $display_url ) );
+                } else {
+                    WP_CLI::warning( sprintf( '%s: Failed with error: %s', $display_url, $error ) );
+                }
+            }
+            $execution_time = microtime( true ) - $start_timestamp;
+
+            WP_CLI::line();
+            WP_CLI::line( sprintf(
+                'Flushed all rewrite rules in %d minute(s) and %d second(s).',
+                floor( $execution_time / 60 ),
+                $execution_time % 60
+            ) );
+        }
+    }
+
+    WP_CLI::add_command( 'wcorg-misc', 'WordCamp_Miscellaneous_Commands' );
+}