From 395e2e040d112ae8be06e7fa814840829676a5ab Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Pluchino?=
 <francois.pluchino@opendisplay.com>
Date: Tue, 10 Jan 2012 21:22:52 +0100
Subject: [PATCH] Add callback download informations

---
 src/Composer/Console/Output/ConsoleOutput.php |  2 +-
 src/Composer/Downloader/FileDownloader.php    | 76 +++++++++++++++++--
 2 files changed, 70 insertions(+), 8 deletions(-)

diff --git a/src/Composer/Console/Output/ConsoleOutput.php b/src/Composer/Console/Output/ConsoleOutput.php
index 0a8a8bd03..a301dd504 100644
--- a/src/Composer/Console/Output/ConsoleOutput.php
+++ b/src/Composer/Console/Output/ConsoleOutput.php
@@ -37,7 +37,7 @@ class ConsoleOutput extends BaseConsoleOutput
 
         $this->write($messages, false, $type);
 
-        for ($place = ($size - strlen($line)); $place > 0; $place--) {
+        for ($place = ($size - strlen($messages)); $place > 0; $place--) {
             $this->write(' ');
         }
 
diff --git a/src/Composer/Downloader/FileDownloader.php b/src/Composer/Downloader/FileDownloader.php
index dc4769408..d7f9575ad 100644
--- a/src/Composer/Downloader/FileDownloader.php
+++ b/src/Composer/Downloader/FileDownloader.php
@@ -26,6 +26,7 @@ abstract class FileDownloader implements DownloaderInterface
 {
     protected $intput;
     protected $output;
+    protected $bytesMax;
 
     /**
      * Constructor.
@@ -52,6 +53,9 @@ abstract class FileDownloader implements DownloaderInterface
      */
     public function download(PackageInterface $package, $path)
     {
+        // init the progress bar
+        $this->bytesMax = 0;
+
         $url = $package->getDistUrl();
         $checksum = $package->getDistSha1Checksum();
 
@@ -67,7 +71,7 @@ abstract class FileDownloader implements DownloaderInterface
         $fileName = rtrim($path.'/'.md5(time().rand()).'.'.pathinfo($url, PATHINFO_EXTENSION), '.');
 
         //echo 'Downloading '.$url.' to '.$fileName.PHP_EOL;
-        $this->output->writeln("  - <info>Downloading</info> <comment>" . $package->getName() . "</comment> (" . $package->getPrettyVersion() . ")");
+        $this->output->writeln("  - Package <comment>" . $package->getName() . "</comment> (<info>" . $package->getPrettyVersion() . "</info>)");
 
         if (!extension_loaded('openssl') && (0 === strpos($url, 'https:') || 0 === strpos($url, 'http://github.com'))) {
             // bypass https for github if openssl is disabled
@@ -79,6 +83,8 @@ abstract class FileDownloader implements DownloaderInterface
         }
 
         // Handle system proxy
+        $ctx = stream_context_create();
+
         if (isset($_SERVER['HTTP_PROXY'])) {
             // http(s):// is not supported in proxy
             $proxy = str_replace(array('http://', 'https://'), array('tcp://', 'ssl://'), $_SERVER['HTTP_PROXY']);
@@ -93,12 +99,14 @@ abstract class FileDownloader implements DownloaderInterface
                     'request_fulluri' => true,
                 ),
             ));
-
-            copy($url, $filename, $ctx);
-        } else {
-            copy($url, $fileName);
         }
 
+        stream_context_set_params($ctx, array("notification" => array($this, 'callbackDownload')));
+
+        copy($url, $fileName, $ctx);
+
+        $this->output->overwrite('', 80);
+
         if (!file_exists($fileName)) {
             throw new \UnexpectedValueException($url.' could not be saved to '.$fileName.', make sure the'
                 .' directory is writable and you have internet connectivity');
@@ -108,11 +116,11 @@ abstract class FileDownloader implements DownloaderInterface
             throw new \UnexpectedValueException('The checksum verification of the archive failed (downloaded from '.$url.')');
         }
 
-        $this->output->writeln('    Unpacking archive');
+        $this->output->overwrite('    Unpacking archive');
         $this->extract($fileName, $path);
 
 
-        $this->output->writeln('    Cleaning up');
+        $this->output->overwrite('    Cleaning up');
         unlink($fileName);
 
         // If we have only a one dir inside it suppose to be a package itself
@@ -126,6 +134,9 @@ abstract class FileDownloader implements DownloaderInterface
             }
             rmdir($contentDir);
         }
+
+        $this->output->overwrite('');
+        $this->output->writeln('');
     }
 
     /**
@@ -147,6 +158,57 @@ abstract class FileDownloader implements DownloaderInterface
         $fs->removeDirectory($path);
     }
 
+    /**
+     * Download notification action.
+     *
+     * @param integer $notificationCode The notification code
+     * @param integer $severity         The severity level
+     * @param string  $message          The message
+     * @param integer $messageCode      The message code
+     * @param integer $bytesTransferred The loaded size
+     * @param integer $bytesMax         The total size
+     */
+    protected function callbackDownload($notificationCode, $severity, $message, $messageCode, $bytesTransferred, $bytesMax)
+    {
+        switch ($notificationCode) {
+            case STREAM_NOTIFY_AUTH_REQUIRED:
+                break;
+
+            case STREAM_NOTIFY_FILE_SIZE_IS:
+                if ($this->bytesMax < $bytesMax) {
+                    $this->bytesMax = $bytesMax;
+                }
+
+                break;
+
+            case STREAM_NOTIFY_PROGRESS:
+                if ($this->bytesMax > 0) {
+                    $progression = 0;
+
+                    if ($this->bytesMax > 0) {
+                        $progression = ($bytesTransferred / $this->bytesMax * 100);
+                    }
+
+                    $levels = array(0, 5, 10, 15, 20, 25, 30, 35, 40, 35, 50, 55, 60,
+                            65, 70, 75, 80, 85, 90, 95, 100);
+
+                    $progression = round($progression, 0);
+
+                    if (in_array($progression, $levels)) {
+                        $this->output->overwrite("    Downloading: <comment>$progression%</comment>", 80);
+                    }
+                }
+
+                break;
+
+            case STREAM_NOTIFY_AUTH_RESULT:
+                break;
+
+            default:
+                break;
+        }
+    }
+
     /**
      * Extract file to directory
      *