Changeset 25082

Timestamp:

08/22/2013 04:49:06 AM (13 years ago)

Author:

dd32

Message:

Theme Installer/Updater: Handle custom Theme directories when updating themes, and installing new themes. Fixes #22501

Location:

trunk/src/wp-admin/includes

Files:

• class-wp-filesystem-base.php (modified) (1 diff)
• class-wp-upgrader.php (modified) (9 diffs)

trunk/src/wp-admin/includes/class-wp-filesystem-base.php

@@ -80 +80 @@
          * @access public
          *
-         * @return string The location of the remote path.
-         */
-        function wp_themes_dir() {
-                return $this->wp_content_dir() . 'themes/';
+         * @param string $theme The Theme stylesheet or template for the directory
+         * @return string The location of the remote path.
+         */
+        function wp_themes_dir( $theme = false ) {
+                $theme_root = get_theme_root( $theme );
+
+                // Account for relative theme roots
+                if ( '/themes' == $theme_root || ! is_dir( $theme_root ) )
+                        $theme_root = WP_CONTENT_DIR . $theme_root;
+
+                return $this->find_folder( $theme_root );
         }
         /**

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

@@ -94 +94 @@
                                                 return new WP_Error('fs_no_plugins_dir', $this->strings['fs_no_plugins_dir']);
                                         break;
-                                case WP_CONTENT_DIR . '/themes':
-                                        if ( ! $wp_filesystem->find_folder(WP_CONTENT_DIR . '/themes') )
+                                case get_theme_root():
+                                        if ( ! $wp_filesystem->wp_themes_dir() )
                                                 return new WP_Error('fs_no_themes_dir', $this->strings['fs_no_themes_dir']);
                                         break;
@@ -165 +165 @@
 
         function install_package($args = array()) {
-                global $wp_filesystem;
+                global $wp_filesystem, $wp_theme_directories;
+
                 $defaults = array( 'source' => '', 'destination' => '', //Please always pass these
                                                 'clear_destination' => false, 'clear_working' => false,
@@ -209 +210 @@
                         $source_files = array_keys( $wp_filesystem->dirlist($source) );
 
-                //Protection against deleting files in any important base directories.
-                if ( in_array( $destination, array(ABSPATH, WP_CONTENT_DIR, WP_PLUGIN_DIR, WP_CONTENT_DIR . '/themes') ) ) {
+                // Protection against deleting files in any important base directories.
+                // Theme_Upgrader & Plugin_Upgrader also trigger this, as they pass the destination directory (WP_PLUGIN_DIR / wp-content/themes)
+                // intending to copy the directory into the directory, whilst they pass the source as the actual files to copy.
+                if ( in_array( $destination, array_merge( array( ABSPATH, WP_CONTENT_DIR, WP_PLUGIN_DIR, WP_CONTENT_DIR . '/themes' ), $wp_theme_directories ) ) ) {
                         $remote_destination = trailingslashit($remote_destination) . trailingslashit(basename($source));
                         $destination = trailingslashit($destination) . trailingslashit(basename($source));
@@ -728 +731 @@
                 $parent_result = $this->run( array(
                         'package' => $api->download_link,
-                        'destination' => WP_CONTENT_DIR . '/themes',
+                        'destination' => get_theme_root(),
                         'clear_destination' => false, //Do not overwrite files.
                         'clear_working' => true
@@ -761 +764 @@
 
                 $options = array(
-                                                'package' => $package,
-                                                'destination' => WP_CONTENT_DIR . '/themes',
-                                                'clear_destination' => false, //Do not overwrite files.
-                                                'clear_working' => true
-                                                );
+                        'package' => $package,
+                        'destination' => get_theme_root(),
+                        'clear_destination' => false, //Do not overwrite files.
+                        'clear_working' => true
+                );
 
                 $this->run($options);
@@ -804 +807 @@
 
                 $options = array(
-                                                'package' => $r['package'],
-                                                'destination' => WP_CONTENT_DIR . '/themes',
-                                                'clear_destination' => true,
-                                                'clear_working' => true,
-                                                'hook_extra' => array(
-                                                                                        'theme' => $theme
-                                                                                        )
-                                                );
+                        'package' => $r['package'],
+                        'destination' => get_theme_root( $theme ),
+                        'clear_destination' => true,
+                        'clear_working' => true,
+                        'hook_extra' => array(
+                                'theme' => $theme
+                        ),
+                );
 
                 $this->run($options);
@@ -884 +887 @@
 
                         $options = array(
-                                                        'package' => $r['package'],
-                                                        'destination' => WP_CONTENT_DIR . '/themes',
-                                                        'clear_destination' => true,
-                                                        'clear_working' => true,
-                                                        'hook_extra' => array(
-                                                                                                'theme' => $theme
-                                                                                                )
-                                                        );
+                                'package' => $r['package'],
+                                'destination' => get_theme_root( $theme ),
+                                'clear_destination' => true,
+                                'clear_working' => true,
+                                'hook_extra' => array(
+                                        'theme' => $theme
+                                ),
+                        );
 
                         $result = $this->run($options);
@@ -985 +988 @@
         }
 
-        function delete_old_theme($removed, $local_destination, $remote_destination, $theme) {
+        function delete_old_theme( $removed, $local_destination, $remote_destination, $theme ) {
                 global $wp_filesystem;
 
-                $theme = isset($theme['theme']) ? $theme['theme'] : '';
-
-                if ( is_wp_error($removed) || empty($theme) )
-                        return $removed; //Pass errors through.
-
-                $themes_dir = $wp_filesystem->wp_themes_dir();
-                if ( $wp_filesystem->exists( trailingslashit($themes_dir) . $theme ) )
-                        if ( ! $wp_filesystem->delete( trailingslashit($themes_dir) . $theme, true ) )
+                if ( is_wp_error( $removed ) )
+                        return $removed; // Pass errors through.
+
+                if ( ! isset( $theme['theme'] ) )
+                        return $removed;
+
+                $theme = $theme['theme'];
+                $themes_dir = trailingslashit( $wp_filesystem->wp_themes_dir( $theme ) );
+                if ( $wp_filesystem->exists( $themes_dir . $theme ) ) {
+                        if ( ! $wp_filesystem->delete( $themes_dir . $theme, true ) )
                                 return false;
+                }
+
                 return true;
         }
@@ -1008 +1015 @@
                                 return false;
                 }
-                return wp_get_theme( $theme, WP_CONTENT_DIR . '/themes/' );
+                return wp_get_theme( $theme );
         }