Use Prestissimo to download files if available (#60)

Author: webflo

src/Handler.php

@@ -10,9 +10,9 @@
 use Composer\Composer;
 use Composer\DependencyResolver\Operation\InstallOperation;
 use Composer\DependencyResolver\Operation\UpdateOperation;
+use Composer\EventDispatcher\EventDispatcher;
 use Composer\IO\IOInterface;
 use Composer\Package\PackageInterface;
-use Composer\EventDispatcher\EventDispatcher;
 use Composer\Util\Filesystem;
 use Composer\Util\RemoteFilesystem;
 use Symfony\Component\Filesystem\Filesystem as SymfonyFilesystem;
@@ -113,7 +113,7 @@ public function downloadScaffold() {
 
     $remoteFs = new RemoteFilesystem($this->io);
 
-    $fetcher = new FileFetcher($remoteFs, $options['source'], $files);
+    $fetcher = new PrestissimoFileFetcher($remoteFs, $options['source'], $files, $this->io, $this->composer->getConfig());
     $fetcher->fetch($version, $webroot);
 
     $initialFileFetcher = new InitialFileFetcher($remoteFs, $options['source'], $this->getInitial());

src/InitialFileFetcher.php

@@ -7,10 +7,8 @@
 
 namespace DrupalComposer\DrupalScaffold;
 
-use Composer\Util\Filesystem;
-use Composer\Util\RemoteFilesystem;
-
 class InitialFileFetcher extends FileFetcher {
+
   public function fetch($version, $destination) {
     array_walk($this->filenames, function ($filename, $sourceFilename) use ($version, $destination) {
       $target = "$destination/$filename";
@@ -21,4 +19,5 @@ public function fetch($version, $destination) {
       }
     });
   }
+
 }

src/Plugin.php

@@ -8,10 +8,10 @@
 
 use Composer\Composer;
 use Composer\EventDispatcher\EventSubscriberInterface;
+use Composer\Installer\PackageEvent;
 use Composer\Installer\PackageEvents;
 use Composer\IO\IOInterface;
 use Composer\Plugin\PluginInterface;
-use Composer\Installer\PackageEvent;
 use Composer\Script\ScriptEvents;
 
 /**

src/PrestissimoFileFetcher.php

@@ -0,0 +1,66 @@
+<?php
+
+/**
+ * @file
+ * Contains \DrupalComposer\DrupalScaffold\FileFetcher.
+ */
+
+namespace DrupalComposer\DrupalScaffold;
+
+use Composer\Config;
+use Composer\IO\IOInterface;
+use Hirak\Prestissimo\CopyRequest;
+use Hirak\Prestissimo\CurlMulti;
+
+class PrestissimoFileFetcher extends FileFetcher {
+
+  /**
+   * @var \Composer\IO\IOInterface
+   */
+  protected $io;
+
+  /**
+   * @var \Composer\Config
+   */
+  protected $config;
+
+  public function __construct(\Composer\Util\RemoteFilesystem $remoteFilesystem, $source, array $filenames = [], IOInterface $io, Config $config) {
+    parent::__construct($remoteFilesystem, $source, $filenames);
+    $this->io = $io;
+    $this->config = $config;
+  }
+
+  public function fetch($version, $destination) {
+    if (class_exists(CurlMulti::class)) {
+      $this->fetchWithPrestissimo($version, $destination);
+      return;
+    }
+    parent::fetch($version, $destination);
+  }
+
+  protected function fetchWithPrestissimo($version, $destination) {
+    $requests = [];
+    array_walk($this->filenames, function ($filename) use ($version, $destination, &$requests) {
+      $url = $this->getUri($filename, $version);
+      $this->fs->ensureDirectoryExists($destination . '/' . dirname($filename));
+      $requests[] = new CopyRequest($url, $destination . '/' . $filename, false, $this->io, $this->config);
+    });
+
+    $successCnt = $failureCnt = 0;
+    $totalCnt = count($requests);
+
+    $multi = new CurlMulti;
+    $multi->setRequests($requests);
+    do {
+      $multi->setupEventLoop();
+      $multi->wait();
+      $result = $multi->getFinishedResults();
+      $successCnt += $result['successCnt'];
+      $failureCnt += $result['failureCnt'];
+      foreach ($result['urls'] as $url) {
+        $this->io->writeError("    <comment>$successCnt/$totalCnt</comment>:\t$url", true, \Composer\IO\IOInterface::VERBOSE);
+      }
+    } while ($multi->remain());
+  }
+
+}