<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN"
"http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head><meta http-equiv="content-type" content="text/html; charset=utf-8" /><style type="text/css"><!--
#msg dl { border: 1px #006 solid; background: #369; padding: 6px; color: #fff; }
#msg dt { float: left; width: 6em; font-weight: bold; }
#msg dt:after { content:':';}
#msg dl, #msg dt, #msg ul, #msg li, #header, #footer { font-family: verdana,arial,helvetica,sans-serif; font-size: 10pt;  }
#msg dl a { font-weight: bold}
#msg dl a:link    { color:#fc3; }
#msg dl a:active  { color:#ff0; }
#msg dl a:visited { color:#cc6; }
h3 { font-family: verdana,arial,helvetica,sans-serif; font-size: 10pt; font-weight: bold; }
#msg pre { overflow: auto; background: #ffc; border: 1px #fc0 solid; padding: 6px; }
#msg ul, pre { overflow: auto; }
#header, #footer { color: #fff; background: #636; border: 1px #300 solid; padding: 6px; }
#patch { width: 100%; }
#patch h4 {font-family: verdana,arial,helvetica,sans-serif;font-size:10pt;padding:8px;background:#369;color:#fff;margin:0;}
#patch .propset h4, #patch .binary h4 {margin:0;}
#patch pre {padding:0;line-height:1.2em;margin:0;}
#patch .diff {width:100%;background:#eee;padding: 0 0 10px 0;overflow:auto;}
#patch .propset .diff, #patch .binary .diff  {padding:10px 0;}
#patch span {display:block;padding:0 10px;}
#patch .modfile, #patch .addfile, #patch .delfile, #patch .propset, #patch .binary, #patch .copfile {border:1px solid #ccc;margin:10px 0;}
#patch ins {background:#dfd;text-decoration:none;display:block;padding:0 10px;}
#patch del {background:#fdd;text-decoration:none;display:block;padding:0 10px;}
#patch .lines, .info {color:#888;background:#fff;}
--></style>
<title>[12794] trunk/wp-admin/includes/file.php: Optimised unzip_file(),
  reduces is_dir() IO calls.</title>
</head>
<body>

<div id="msg">
<dl>
<dt>Revision</dt> <dd><a href="http://trac.wordpress.org/changeset/12794">12794</a></dd>
<dt>Author</dt> <dd>dd32</dd>
<dt>Date</dt> <dd>2010-01-22 11:02:09 +0000 (Fri, 22 Jan 2010)</dd>
</dl>

<h3>Log Message</h3>
<pre>Optimised unzip_file(), reduces is_dir() IO calls. See <a href="http://trac.wordpress.org/ticket/10779">#10779</a></pre>

<h3>Modified Paths</h3>
<ul>
<li><a href="#trunkwpadminincludesfilephp">trunk/wp-admin/includes/file.php</a></li>
</ul>

</div>
<div id="patch">
<h3>Diff</h3>
<a id="trunkwpadminincludesfilephp"></a>
<div class="modfile"><h4>Modified: trunk/wp-admin/includes/file.php (12793 => 12794)</h4>
<pre class="diff"><span>
<span class="info">--- trunk/wp-admin/includes/file.php        2010-01-22 07:23:14 UTC (rev 12793)
+++ trunk/wp-admin/includes/file.php        2010-01-22 11:02:09 UTC (rev 12794)
</span><span class="lines">@@ -504,8 +504,6 @@
</span><span class="cx">         // Unzip uses a lot of memory, but not this much hopefully
</span><span class="cx">         @ini_set('memory_limit', '256M');
</span><span class="cx"> 
</span><del>-        $fs =&amp; $wp_filesystem;
-
</del><span class="cx">         require_once(ABSPATH . 'wp-admin/includes/class-pclzip.php');
</span><span class="cx"> 
</span><span class="cx">         $archive = new PclZip($file);
</span><span class="lines">@@ -517,43 +515,48 @@
</span><span class="cx">         if ( 0 == count($archive_files) )
</span><span class="cx">                 return new WP_Error('empty_archive', __('Empty archive.'));
</span><span class="cx"> 
</span><del>-        $path = explode('/', untrailingslashit($to));
-        for ( $i = count($path); $i &gt; 0; $i-- ) { //&gt;0 = first element is empty allways for paths starting with '/'
-                $tmppath = implode('/', array_slice($path, 0, $i) );
-                if ( $fs-&gt;is_dir($tmppath) ) { //Found the highest folder that exists, Create from here(ie +1)
-                        for ( $i = $i + 1; $i &lt;= count($path); $i++ ) {
-                                $tmppath = implode('/', array_slice($path, 0, $i) );
-                                if ( ! $fs-&gt;mkdir($tmppath, FS_CHMOD_DIR) )
-                                        return new WP_Error('mkdir_failed', __('Could not create directory.'), $tmppath);
-                        }
-                        break; //Exit main for loop
-                }
-        }
</del><ins>+        $needed_dirs = array();
+        $to = trailingslashit($to);
</ins><span class="cx"> 
</span><del>-        $to = trailingslashit($to);
-        foreach ($archive_files as $file) {
-                $path = $file['folder'] ? $file['filename'] : dirname($file['filename']);
-                $path = explode('/', $path);
-                for ( $i = count($path); $i &gt;= 0; $i-- ) { //&gt;=0 as the first element contains data
</del><ins>+        // Determine any parent dir's needed (of the upgrade directory)
+        if ( ! $wp_filesystem-&gt;is_dir($to) ) { //Only do parents if no children exist
+                $path = preg_split('![/\\\]!', untrailingslashit($to));
+                for ( $i = count($path); $i &gt;= 0; $i-- ) {
</ins><span class="cx">                         if ( empty($path[$i]) )
</span><span class="cx">                                 continue;
</span><del>-                        $tmppath = $to . implode('/', array_slice($path, 0, $i) );
-                        if ( $fs-&gt;is_dir($tmppath) ) {//Found the highest folder that exists, Create from here
-                                for ( $i = $i + 1; $i &lt;= count($path); $i++ ) { //&lt; count() no file component please.
-                                        $tmppath = $to . implode('/', array_slice($path, 0, $i) );
-                                        if ( ! $fs-&gt;is_dir($tmppath) &amp;&amp; ! $fs-&gt;mkdir($tmppath, FS_CHMOD_DIR) )
-                                                return new WP_Error('mkdir_failed', __('Could not create directory.'), $tmppath);
-                                }
-                                break; //Exit main for loop
-                        }
-                }
</del><span class="cx"> 
</span><del>-                // We've made sure the folders are there, so let's extract the file now:
-                if ( ! $file['folder'] ) {
-                        if ( !$fs-&gt;put_contents( $to . $file['filename'], $file['content'], FS_CHMOD_FILE) )
-                                return new WP_Error('copy_failed', __('Could not copy file.'), $to . $file['filename']);
</del><ins>+                        $dir = implode('/', array_slice($path, 0, $i+1) );
+                        if ( preg_match('!^[a-z]:$!i', $dir) ) // Skip it if it looks like a Windows Drive letter only.
+                                continue;
+
+                        if ( ! $wp_filesystem-&gt;is_dir($dir) )
+                                $needed_dirs[] = $dir;
+                        else
+                                break; // A folder exists, therefor, we dont need the check the levels below this
</ins><span class="cx">                 }
</span><span class="cx">         }
</span><ins>+
+        // Determine any children directories needed (From within the archive)
+        foreach ( $archive_files as $file )
+                $needed_dirs[] = $to . untrailingslashit( $file['folder'] ? $file['filename'] : dirname($file['filename']) );
+
+        $needed_dirs = array_unique($needed_dirs);
+        asort($needed_dirs);
+
+        // Create those directories if need be:
+        foreach ( $needed_dirs as $_dir ) {
+                if ( ! $wp_filesystem-&gt;mkdir($_dir, FS_CHMOD_DIR) &amp;&amp; ! $wp_filesystem-&gt;is_dir($_dir) ) // Only check to see if the Dir exists upon creation failure. Less I/O this way.
+                        return new WP_Error('mkdir_failed', __('Could not create directory.'), $_dir);
+        }
+
+        // Extract the files from the zip
+        foreach ( $archive_files as $file ) {
+                if ( $file['folder'] )
+                        continue;
+
+                if ( ! $wp_filesystem-&gt;put_contents( $to . $file['filename'], $file['content'], FS_CHMOD_FILE) )
+                        return new WP_Error('copy_failed', __('Could not copy file.'), $to . $file['filename']);
+        }
</ins><span class="cx">         return true;
</span><span class="cx"> }
</span><span class="cx"> 
</span></span></pre>
</div>
</div>

</body>
</html>