| @@ -1,304 +1,55 @@ | ||
| 1 | 1 | <?php |
| 2 | 2 | /* |
| 3 | -Plugin Name: WP Reset | |
| 4 | -Plugin URI: https://github.com/chrisberthe/wp-reset | |
| 5 | -Description: A plugin that allows you to reset the database to WordPress's initial state. | |
| 6 | -Version: 1.0 | |
| 7 | -Author: Chris Berthe ☻ | |
| 8 | -Author URI: https://github.com/chrisberthe | |
| 9 | -License: GNU General Public License | |
| 3 | + Plugin Name: WP Database Reset | |
| 4 | + Plugin URI: https://wordpress.org/plugins/wordpress-database-reset/ | |
| 5 | + Description: Reset all or some WP database tables back to their original state. | |
| 6 | + Version: 3.21 | |
| 7 | + Author: WebFactory Ltd | |
| 8 | + Author URI: https://www.webfactoryltd.com/ | |
| 9 | + License: GNU General Public License | |
| 10 | + Text-domain: wordpress-database-reset | |
| 11 | + | |
| 12 | + Copyright 2011 - 2022 WebFactory Ltd (email: support@webfactoryltd.com) | |
| 13 | + | |
| 14 | + This program is free software; you can redistribute it and/or modify | |
| 15 | + it under the terms of the GNU General Public License, version 2, as | |
| 16 | + published by the Free Software Foundation. | |
| 17 | + | |
| 18 | + This program is distributed in the hope that it will be useful, | |
| 19 | + but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 20 | + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 21 | + GNU General Public License for more details. | |
| 22 | + | |
| 23 | + You should have received a copy of the GNU General Public License | |
| 24 | + along with this program; if not, write to the Free Software | |
| 25 | + Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA | |
| 10 | 26 | */ |
| 11 | 27 | |
| 12 | -if ( ! class_exists('WP_Reset') && is_admin() ) : | |
| 28 | +define( 'DB_RESET_VERSION', '3.21' ); | |
| 29 | +define( 'DB_RESET_PATH', dirname( __FILE__ ) ); | |
| 30 | +define( 'DB_RESET_NAME', basename( DB_RESET_PATH ) ); | |
| 31 | +define( 'DB_RESET_FILE', __FILE__ ); | |
| 32 | +define( 'AUTOLOADER', DB_RESET_PATH . '/lib/class-plugin-autoloader.php' ); | |
| 13 | 33 | |
| 14 | - class WP_Reset | |
| 15 | - { | |
| 16 | - /** | |
| 17 | - * Nonce value | |
| 18 | - */ | |
| 19 | - private $_nonce = 'wp-reset-nonce'; | |
| 20 | - | |
| 21 | - /** | |
| 22 | - * Loads default options | |
| 23 | - * | |
| 24 | - * @return void | |
| 25 | - */ | |
| 26 | - function __construct() | |
| 27 | - { | |
| 28 | - add_action('init', array($this, 'init_language')); | |
| 29 | - add_action('admin_init', array($this, 'wp_reset_init')); | |
| 30 | - add_action('admin_init', array($this, '_redirect_user')); | |
| 31 | - add_action('admin_footer', array($this, 'add_admin_javascript')); | |
| 32 | - add_action('admin_menu', array($this, 'add_admin_menu')); | |
| 33 | - add_filter('contextual_help', array($this, 'add_contextual_help'), 10, 2); | |
| 34 | - add_filter('wp_mail', array($this, '_fix_password_mail')); | |
| 35 | - } | |
| 36 | - | |
| 37 | - /** | |
| 38 | - * Handles the admin page functionality | |
| 39 | - * | |
| 40 | - * @access public | |
| 41 | - * @uses wp_install Located in includes/upgrade.php (line 22) | |
| 42 | - */ | |
| 43 | - function wp_reset_init() | |
| 44 | - { | |
| 45 | - global $wpdb, $current_user, $pagenow; | |
| 46 | - | |
| 47 | - if ( isset($_POST['wp-random-value'], $_POST['wp-reset-input']) && $_POST['wp-random-value'] == $_POST['wp-reset-input'] | |
| 48 | - && check_admin_referer('wp-nonce-submit', $this->_nonce) ) | |
| 49 | - { | |
| 50 | - require_once( ABSPATH . '/wp-admin/includes/upgrade.php' ); | |
| 51 | - | |
| 52 | - $blog_title = get_option('blogname'); | |
| 53 | - $public = get_option('blog_public'); | |
| 54 | - | |
| 55 | - $admin_user = get_userdatabylogin('admin'); | |
| 56 | - $user = ( ! $admin_user || $admin_user->wp_user_level < 10 ) ? $current_user : $admin_user; | |
| 57 | - | |
| 58 | - // Run through the database columns and drop all the tables | |
| 59 | - if ( $db_tables = $wpdb->get_col("SHOW TABLES LIKE '{$wpdb->prefix}%'") ) | |
| 60 | - { | |
| 61 | - foreach ( $db_tables as $db_table ) | |
| 62 | - { | |
| 63 | - $wpdb->query("DROP TABLE {$db_table}"); | |
| 64 | - } | |
| 65 | - | |
| 66 | - // Return user keys and import variables | |
| 67 | - $keys = wp_install($blog_title, $user->user_login, $user->user_email, $public); | |
| 68 | - $this->_wp_update_user($user, $keys); | |
| 69 | - | |
| 70 | - // Reactivate the plugin after reinstalling | |
| 71 | - update_option('active_plugins', array(plugin_basename(__FILE__))); | |
| 34 | +require_once( DB_RESET_PATH . '/lib/helpers.php' ); | |
| 72 | 35 | |
| 73 | - wp_redirect(admin_url($pagenow) . '?page=wp-reset&reset=success'); exit(); | |
| 74 | - } | |
| 75 | - } | |
| 76 | - } | |
| 77 | - | |
| 78 | - /** | |
| 79 | - * Displays the admin page | |
| 80 | - * | |
| 81 | - * @access public | |
| 82 | - * @return void | |
| 83 | - */ | |
| 84 | - function show_admin_page() | |
| 85 | - { | |
| 86 | - global $current_user; | |
| 87 | - | |
| 88 | - // Return to see if admin object exists | |
| 89 | - $admin_user = get_userdatabylogin('admin'); | |
| 90 | - | |
| 91 | - // Generate a random value for the input box | |
| 92 | - $random_string = $this->_rand_string(); | |
| 93 | -?> | |
| 94 | - <?php if ( isset($_POST['wp-random-value'], $_POST['wp-reset-input']) && $_POST['wp-random-value'] != $_POST['wp-reset-input'] ) : ?> | |
| 95 | - <div class="error"><p><strong><?php _e('You entered the wrong value - please try again', 'wp-reset') ?>.</strong></p></div> | |
| 96 | - <?php elseif ( isset($_GET['reset']) && $_GET['reset'] == 'success' ) : ?> | |
| 97 | - <div class="updated"><p><strong><?php _e('The WordPress database has been reset successfully', 'wp-reset') ?>.</strong></p></div> | |
| 98 | - <?php endif ?> | |
| 36 | +require_once dirname(__FILE__) . '/wf-flyout/wf-flyout.php'; | |
| 37 | +new wf_flyout(__FILE__); | |
| 99 | 38 | |
| 100 | - <div class="wrap"> | |
| 101 | - <?php screen_icon() ?> | |
| 102 | - <h2><?php _e('Database Reset', 'wp-reset') ?></h2> | |
| 103 | - <p>Please type in (or copy/paste) the generated value into the text box: <strong><?php echo $random_string ?></strong></p> | |
| 104 | - <form action="" method="POST" id="wp-reset-form"> | |
| 105 | - <?php wp_nonce_field('wp-nonce-submit', $this->_nonce) ?> | |
| 106 | - <input type="hidden" name="wp-random-value" value="<?php echo $random_string ?>" id="wp-random-value" /> | |
| 107 | - <input type="text" name="wp-reset-input" value="" id="wp-reset-input" /> | |
| 108 | - <input type="submit" name="wp-reset-submit" value="<?php _e('Reset Database', 'wp-reset') ?>" id="wp-reset-submit" class="button-primary" /> | |
| 109 | - </form> | |
| 110 | - | |
| 111 | - <?php if ( ! $admin_user || $admin_user->wp_user_level < 10 ) : ?> | |
| 112 | - <p style="margin-top: 25px"><?php printf(__('The default user <strong><u>admin</u></strong> was never created for this WordPress install. So <strong><u>%s</u></strong> will be recreated with its current password instead', 'wp-reset'), $current_user->user_login) ?>.</p> | |
| 113 | - <?php else : ?> | |
| 114 | - <p><?php _e('The default user <strong><u>admin</u></strong> will be recreated with its current password upon resetting', 'wp-reset') ?>.</p> | |
| 115 | - <?php endif; ?> | |
| 116 | - | |
| 117 | - <p><?php _e('Note that once you reset the database, all users will be deleted except the initial admin user. The plugin will also reactivate itself after resetting', 'wp-reset') ?>.</p> | |
| 118 | - </div> | |
| 119 | -<?php } | |
| 120 | - | |
| 121 | - /** | |
| 122 | - * Add JavaScript to the bottom of the plugin page | |
| 123 | - * | |
| 124 | - * @access public | |
| 125 | - * @return bool TRUE on reset confirmation | |
| 126 | - */ | |
| 127 | - function add_admin_javascript() | |
| 128 | - { | |
| 129 | -?> | |
| 130 | - <script type="text/javascript"> | |
| 131 | - /* <![CDATA[ */ | |
| 132 | - jQuery('#wp-reset-submit').click(function() { | |
| 133 | - var message = "<?php _e('Clicking OK will result in your database being reset to its initial settings. Continue?', 'wp-reset') ?>"; | |
| 134 | - var reset = confirm(message); | |
| 135 | - | |
| 136 | - if ( reset ) { | |
| 137 | - jQuery('#wp-reset-form').submit(); | |
| 138 | - } else { | |
| 139 | - return false; | |
| 140 | - } | |
| 141 | - }); | |
| 142 | - /* ]]> */ | |
| 143 | - </script> | |
| 144 | -<?php | |
| 145 | - } | |
| 146 | - | |
| 147 | - /** | |
| 148 | - * Adds our submenu item to the Tools menu | |
| 149 | - * | |
| 150 | - * @access public | |
| 151 | - * @return void | |
| 152 | - */ | |
| 153 | - function add_admin_menu() | |
| 154 | - { | |
| 155 | - global $current_user; | |
| 156 | - | |
| 157 | - if ( current_user_can('update_core') && $current_user->wp_user_level == 10) | |
| 158 | - { | |
| 159 | - $this->_hook = add_submenu_page('tools.php', 'Database Reset', 'Database Reset', 'update_core', 'wp-reset', array($this, 'show_admin_page')); | |
| 160 | - } | |
| 161 | - } | |
| 162 | - | |
| 163 | - /** | |
| 164 | - * Adds the contextual help for our plugin page | |
| 165 | - * | |
| 166 | - * @access public | |
| 167 | - * @param $contextual_help Hook text to display | |
| 168 | - * @param $screen_id ID of the current admin screen | |
| 169 | - * @return $contextual_help String The help text | |
| 170 | - */ | |
| 171 | - function add_contextual_help($contextual_help, $screen_id) | |
| 172 | - { | |
| 173 | - if ($screen_id == $this->_hook) | |
| 174 | - { | |
| 175 | - $contextual_help = '<p>' . __('Have any cool ideas for this plugin? Contact me either by <a href="http://twitter.com/#!/chrisberthe">Twitter</a> or by <a href="https://github.com/chrisberthe">GitHub</a>.', 'wp-reset') . '</p>'; | |
| 176 | - $contextual_help .= '<p>' . __('If this plugin becomes non-functional in any way due to WordPress upgrades, rest assured I will update it.', 'wp-reset') . '</p>'; | |
| 177 | - $contextual_help .= '<p>' . __('Goodbye for now.', 'wp-reset') . '</p>'; | |
| 178 | - } | |
| 179 | - | |
| 180 | - return $contextual_help; | |
| 181 | - } | |
| 182 | - | |
| 183 | - /** | |
| 184 | - * Load language path | |
| 185 | - * | |
| 186 | - * @access public | |
| 187 | - * @return void | |
| 188 | - */ | |
| 189 | - function init_language() | |
| 190 | - { | |
| 191 | - $language_dir = basename(dirname(__FILE__)) . '/languages'; | |
| 192 | - load_plugin_textdomain('wp-reset', FALSE, $language_dir); | |
| 193 | - } | |
| 194 | - | |
| 195 | - /** | |
| 196 | - * For activation hook | |
| 197 | - * | |
| 198 | - * @access public | |
| 199 | - * @return void | |
| 200 | - */ | |
| 201 | - function plugin_activate() | |
| 202 | - { | |
| 203 | - add_option('wp-reset-activated', true); | |
| 204 | - } | |
| 205 | - | |
| 206 | - /** | |
| 207 | - * Redirects the user after the plugin is activated | |
| 208 | - * | |
| 209 | - * @access private | |
| 210 | - * @return void | |
| 211 | - */ | |
| 212 | - function _redirect_user() | |
| 213 | - { | |
| 214 | - if ( get_option('wp-reset-activated', false) ) | |
| 215 | - { | |
| 216 | - delete_option('wp-reset-activated'); | |
| 217 | - wp_redirect(admin_url('tools.php') . '?page=wp-reset'); | |
| 218 | - } | |
| 219 | - } | |
| 220 | - | |
| 221 | - /** | |
| 222 | - * Changes the password to a sentence rather than | |
| 223 | - * an auto-generated password that is sent by email | |
| 224 | - * right after the installation is complete | |
| 225 | - * | |
| 226 | - * @access private | |
| 227 | - * @return $mail Version with password changed | |
| 228 | - */ | |
| 229 | - function _fix_password_mail($mail) | |
| 230 | - { | |
| 231 | - $subject = __('WordPress Database Reset', 'wp-reset'); | |
| 232 | - $message = __('The WordPress database has been successfully reset to its default settings:', 'wp-reset'); | |
| 233 | - $password = __('Password: The password you chose during the install.', 'wp-reset'); | |
| 234 | - | |
| 235 | - if ( stristr($mail['message'], 'Your new WordPress site has been successfully set up at:') ) | |
| 236 | - { | |
| 237 | - $mail['subject'] = preg_replace('/New WordPress Site/', $subject, $mail['subject']); | |
| 238 | - $mail['message'] = preg_replace('/Your new WordPress site has been successfully set up at:+/', $message, $mail['message']); | |
| 239 | - $mail['message'] = preg_replace('/Password:\s.+/', $password, $mail['message']); | |
| 240 | - } | |
| 241 | - | |
| 242 | - return $mail; | |
| 243 | - } | |
| 244 | - | |
| 245 | - /** | |
| 246 | - * Updates the user password and clears / sets | |
| 247 | - * the authentication cookie for the user | |
| 248 | - * | |
| 249 | - * @access private | |
| 250 | - * @param $user Current or admin user | |
| 251 | - * @param $keys Array returned by wp_install() | |
| 252 | - * @return TRUE on install success, FALSE otherwise | |
| 253 | - */ | |
| 254 | - function _wp_update_user($user, $keys) | |
| 255 | - { | |
| 256 | - global $wpdb; | |
| 257 | - extract($keys, EXTR_SKIP); | |
| 39 | +register_activation_hook( __FILE__, 'db_reset_activate' ); | |
| 258 | 40 | |
| 259 | - // Set the old password back to the user | |
| 260 | - $query = $wpdb->prepare("UPDATE $wpdb->users SET user_pass = '%s', user_activation_key = '' WHERE ID = '%d'", $user->user_pass, $user_id); | |
| 261 | - | |
| 262 | - if ( $wpdb->query($query) ) | |
| 263 | - { | |
| 264 | - // Set the default_password_nag to nothing | |
| 265 | - // so it doesn't pop up with the password reminder after installing | |
| 266 | - if ( get_user_meta($user_id, 'default_password_nag') ) delete_user_meta($user_id, 'default_password_nag'); | |
| 41 | +load_plugin_textdomain( 'wordpress-database-reset' ); | |
| 267 | 42 | |
| 268 | - wp_clear_auth_cookie(); | |
| 269 | - wp_set_auth_cookie($user_id); | |
| 270 | - | |
| 271 | - return TRUE; | |
| 272 | - } | |
| 273 | - | |
| 274 | - return FALSE; | |
| 275 | - } | |
| 276 | - | |
| 277 | - /** | |
| 278 | - * Generates a random value for our input box | |
| 279 | - * | |
| 280 | - * @access private | |
| 281 | - * @param $length Length of the random string value | |
| 282 | - * @return $random_string | |
| 283 | - */ | |
| 284 | - function _rand_string($length = 5) | |
| 285 | - { | |
| 286 | - $random_string = ''; | |
| 287 | - $chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"; | |
| 288 | - $size = strlen($chars); | |
| 289 | - | |
| 290 | - for ($i = 0; $i < $length; $i++) | |
| 291 | - { | |
| 292 | - $random_string .= $chars[rand(0, $size-1)]; | |
| 293 | - } | |
| 294 | - | |
| 295 | - return $random_string; | |
| 296 | - } | |
| 297 | - | |
| 298 | - } | |
| 43 | +if ( file_exists( AUTOLOADER ) ) { | |
| 44 | + require_once( AUTOLOADER ); | |
| 45 | + new Plugin_Autoloader( DB_RESET_PATH ); | |
| 299 | 46 | |
| 300 | - $wp_reset = new WP_Reset(); | |
| 301 | - | |
| 302 | - register_activation_hook( __FILE__, array('WP_Reset', 'plugin_activate') ); | |
| 47 | + add_action( | |
| 48 | + 'wp_loaded', | |
| 49 | + array ( new DB_Reset_Manager( DB_RESET_VERSION ), 'run' ) | |
| 50 | + ); | |
| 51 | +} | |
| 303 | 52 | |
| 304 | -endif; | |
| 53 | +if ( is_command_line() ) { | |
| 54 | + require_once( __DIR__ . '/class-db-reset-command.php' ); | |
| 55 | +} | |
