PackageDownloader.php in Ludwig 8

Namespace

Drupal\ludwig

File

src/PackageDownloader.php

<?php

namespace Drupal\ludwig;

use Drupal\Core\Archiver\ArchiverManager;
use Drupal\Core\File\FileSystemInterface;
use Drupal\Core\FileTransfer\Local;
use Drupal\Core\Site\Settings;
use GuzzleHttp\ClientInterface;
use GuzzleHttp\Exception\RequestException;

/**
 * Download packages defined in ludwig.json files.
 */
class PackageDownloader implements PackageDownloaderInterface {

  /**
   * The archiver manager.
   *
   * @var \Drupal\Core\Archiver\ArchiverManager
   */
  protected $archiverManager;

  /**
   * The file system.
   *
   * @var \Drupal\Core\File\FileSystemInterface
   */
  protected $fileSystem;

  /**
   * The HTTP client.
   *
   * @var \GuzzleHttp\ClientInterface
   */
  protected $httpClient;

  /**
   * The app root.
   *
   * @var string
   */
  protected $root;

  /**
   * The archive extraction path.
   *
   * @var string
   */
  protected $extractionDir;

  /**
   * The archive download cache path.
   *
   * @var string
   */
  protected $cacheDir;

  /**
   * Constructs a new PackageDownloader object.
   *
   * @param \Drupal\Core\Archiver\ArchiverManager $archiver_manager
   *   The archiver manager.
   * @param \Drupal\Core\File\FileSystemInterface $file_system
   *   The file system.
   * @param \GuzzleHttp\ClientInterface $http_client
   *   The HTTP client.
   * @param string $root
   *   The app root.
   */
  public function __construct(ArchiverManager $archiver_manager, FileSystemInterface $file_system, ClientInterface $http_client, $root) {
    $this->archiverManager = $archiver_manager;
    $this->fileSystem = $file_system;
    $this->httpClient = $http_client;
    $this->root = $root;
    $hash = substr(hash('sha256', Settings::getHashSalt()), 0, 8);
    $this->extractionDir = 'temporary://luwdig-extraction-' . $hash;
    if (!file_exists($this->extractionDir)) {
      mkdir($this->extractionDir);
    }
    $this->cacheDir = 'temporary://ludwig-cache-' . $hash;
    if (!file_exists($this->cacheDir)) {
      mkdir($this->cacheDir);
    }
  }

  /**
   * {@inheritdoc}
   */
  public function download(array $package) {
    $provider_path = $this->root . '/' . $package['provider_path'];
    if (!is_writable($provider_path)) {
      throw new \Exception(sprintf('The extension directory %s is not writable.', $provider_path));
    }
    $archive_path = $this
      ->downloadArchive($package);
    if (!$archive_path) {
      throw new \Exception(sprintf('Unable to retrieve %s from %s.', $package['name'], $package['download_url']));
    }
    $archiver = $this
      ->extractArchive($archive_path);
    $files = $archiver
      ->listContents();
    if (!$files) {
      throw new \Exception(sprintf('The archive downloaded from %s contains no files.', $package['download_url']));
    }

    // The real path the first directory in the extracted archive.
    // @todo Will this work for non-GitHub archives?
    $source_location = $this->fileSystem
      ->realpath($this->extractionDir . '/' . $files[0]);
    $package_destination = $this->root . '/' . $package['path'];
    $file_transfer = new Local($this->root, $this->fileSystem);
    $file_transfer
      ->copyDirectory($source_location, $package_destination);
    $new_perms = substr(sprintf('%o', fileperms($package_destination)), -4, -1) . "5";
    $file_transfer
      ->chmod($package_destination, intval($new_perms, 8), TRUE);
  }

  /**
   * Downloads an archive from the given URL to the temporary directory.
   *
   * Returns the local path if the file has already been downloaded.
   *
   * @param string $package
   *   The package for download.
   *
   * @return string
   *   The path to the local file.
   *
   * @throws \Exception
   */
  protected function downloadArchive($package) {
    $parsed_url = parse_url($package['download_url']);
    $cache_dir = $this->cacheDir . '/' . str_replace('/', '-', $package['name']);
    if (!file_exists($cache_dir)) {
      mkdir($cache_dir);
    }
    $local = $cache_dir . '/' . $this->fileSystem
      ->basename($parsed_url['path']);
    if (!file_exists($local)) {
      $destination = $local;
      try {
        $data = $this->httpClient
          ->request('get', $package['download_url'])
          ->getBody()
          ->getContents();
        $local = $this->fileSystem
          ->saveData($data, $destination, FileSystemInterface::EXISTS_REPLACE);
      } catch (RequestException $exception) {
        throw new \Exception(sprintf('Failed to fetch file due to error "%s". Fix the related "url" record in "%s" module ludwig.json file.', $exception
          ->getMessage(), $package['provider']));
      }
      if (!$local) {
        throw new \Exception(sprintf('%s could not be saved to %s', $package['download_url'], $destination));
      }
      return $local;
    }
    else {
      return $local;
    }
  }

  /**
   * Extracts a downloaded archive file.
   *
   * @param string $file
   *   The filename of the archive.
   *
   * @return \Drupal\Core\Archiver\ArchiverInterface
   *   The used archiver.
   *
   * @throws \Exception
   */
  protected function extractArchive($file) {

    /** @var \Drupal\Core\Archiver\ArchiverInterface $archiver */
    $archiver = $this->archiverManager
      ->getInstance([
      'filepath' => $this->fileSystem
        ->realpath($file),
    ]);
    if (!$archiver) {
      throw new \Exception(sprintf('Cannot extract %file, not a valid archive.', [
        '%file' => $file,
      ]));
    }

    // Unfortunately, we can only use the directory name
    // to determine the package name. Some archivers
    // list the first file as the directory (i.e., MODULE/)
    // and others list an actual file (i.e., MODULE/README.TXT).
    $files = $archiver
      ->listContents();
    $package = strtok($files[0], '/\\');

    // Remove the directory if it exists, otherwise it might contain
    // a mixture of old files mixed with the new files (e.g. in cases
    // where files were removed from a later release).
    $extract_location = $this->extractionDir . '/' . $package;
    if (file_exists($extract_location)) {
      $this->fileSystem
        ->deleteRecursive($extract_location);
    }
    return $archiver
      ->extract($this->extractionDir);
  }

}

Classes

Name	Description
PackageDownloader	Download packages defined in ludwig.json files.