Ticket #31138: backup_package_class-wp-upgrader.patch

../../../wp-admin/includes/class-wp-upgrader.php

@@ -410 +410 @@
                 }
 
                 if ( $clear_destination ) {
-                        //We're going to clear the destination if there's something there
-                        $this->skin->feedback('remove_old');
                         $removed = true;
                         if ( $wp_filesystem->exists( $remote_destination ) ) {
+                                // Backup everything.
+                                $this->skin->feedback('backup_old');
+                                $type = $args['hook_extra']['type'];
+                                if ($this->backup_package( $remote_destination, $type, $backup_name, $backup_dir )) {
+                                        $this->backup_set_transient( $remote_destination, $type, $backup_name, $backup_dir );
+                                }
+                                // Remove everything.
+                                $this->skin->feedback('remove_old');
                                 $removed = $wp_filesystem->delete( $remote_destination, true );
                         }
 
@@ -643 +649 @@
                 }
         }
 
-}
+        /**
+         * Backup all files and directories (recursively) in the package_dir.
+         *
+         * @see backup_prefix
+         *
+         * @param string $package_dir The plugin/theme directory.
+         * @param string $type        'plugin' or 'theme'.
+         * @param string $backup_name The unique name of this backup. Used or defaulted.
+         * @param string $backup_dir  The unique directory of this backup. Used or defaulted.
+         *
+         * @return true|WP_Error
+         */
+        protected function backup_package( $package_dir, $type, &$backup_name, &$backup_dir ) {
+                if (is_null($backup_name)) {
+                        $backup_prefix = backup_prefix( $package_dir, $type );
+                        $backup_name = uniqid($backup_prefix);  // eg: 'backup_p_akismet_54c3c6107485f'
+                }
+                if (is_null($backup_dir)) {
+//                      $backup_dir = get_temp_dir() . $backup_name;
+                        $backup_dir = WP_CONTENT_DIR . '/tmp/' . $backup_name;
+                }
+                wp_mkdir_p($backup_dir);
+                $result = copy_dir($package_dir, $backup_dir);
+                return $result;
+        }
 
-/**
+        /**
+         * Set backup data in a transient.
+         *
+         * @see backup_prefix
+         *
+         * @param string $package_dir The plugin/theme directory.
+         * @param string $type        'plugin' or 'theme'.
+         * @param string $backup_name The unique name of this backup.
+         * @param string $backup_dir  The unique directory of this backup.
+         */
+        protected function backup_set_transient( $package_dir, $type, $backup_name, $backup_dir ) {
+                $package_file = $type == 'plugin'
+                                ? $package_dir . basename($package_dir) . '.php'
+                                : $package_dir . 'style.css';
+                $headers = get_file_data( $package_file, array('Version' => 'Version'), $type );
+                $backup_data = array(
+                        'package_dir' => $package_dir,
+                        'package_ver' => $headers['Version'],
+                        'backup_dir'  => $backup_dir,
+                        'backup_date' => date('Y-m-d H:i:s'),
+                );
+                set_transient($backup_name, $backup_data, 1 * WEEK_IN_SECONDS);
+        }
+}
+
+/**
  * Plugin Upgrader class for WordPress Plugins, It is designed to upgrade/install plugins from a local zip, remote zip URL, or uploaded zip file.
  *
  * @package WordPress
@@ -681 +736 @@
                 $this->strings['no_package'] = __('Update package not available.');
                 $this->strings['downloading_package'] = __('Downloading update from <span class="code">%s</span>&#8230;');
                 $this->strings['unpack_package'] = __('Unpacking the update&#8230;');
+                $this->strings['backup_old'] = __('Backing up the old version of the plugin&#8230;');
                 $this->strings['remove_old'] = __('Removing the old version of the plugin&#8230;');
                 $this->strings['remove_old_failed'] = __('Could not remove the old plugin.');
                 $this->strings['process_failed'] = __('Plugin update failed.');
@@ -1115 +1171 @@
                 $this->strings['no_package'] = __('Update package not available.');
                 $this->strings['downloading_package'] = __('Downloading update from <span class="code">%s</span>&#8230;');
                 $this->strings['unpack_package'] = __('Unpacking the update&#8230;');
+                $this->strings['backup_old'] = __('Backing up the old version of the theme&#8230;');
                 $this->strings['remove_old'] = __('Removing the old version of the theme&#8230;');
                 $this->strings['remove_old_failed'] = __('Could not remove the old theme.');
                 $this->strings['process_failed'] = __('Theme update failed.');