ResourceImporter.php

Go to the documentation of this file.

<?php
 
namespace Drupal\dkan_harvest\Transform;
 
use Drupal\Core\Site\Settings;
use Harvest\ETL\Transform\Transform;
use Drupal\dkan_harvest\Load\FileHelperTrait;
 
class ResourceImporter extends Transform {
 
  use FileHelperTrait;
 
  public function run($dataset) {
    $this->testEnvironment();
    return $this->updateDistributions($dataset);
  }
 
  protected function testEnvironment() {
    $setting = Settings::get('file_public_base_url');
    if (!isset($setting) || empty($setting)) {
      throw new \Exception("file_public_base_url should be set.");
    }
    return TRUE;
  }
 
  protected function updateDistributions($dataset) {
    // Abort if there's no distributions.
    if (empty($dataset->distribution)) {
      return $dataset;
    }
 
    $distributions = [];
 
    // Loop through distributions.
    foreach ($dataset->distribution as $dist) {
      $distributions[] = $this->updateDownloadUrl($dataset, $dist);
    }
 
    // Update distributions.
    $dataset->distribution = $distributions;
 
    return $dataset;
  }
 
  protected function updateDownloadUrl($dataset, $dist) {
    // Abort if there's no downloadURL property.
    if (empty($dist->downloadURL)) {
      return $dist;
    }
 
    // Import distribution file.
    $newUrl = $this->saveFile($dist->downloadURL, $dataset->identifier);
 
    // If successful, update downloadURL.
    if ($newUrl) {
      $dist->downloadURL = $newUrl;
    }
 
    return $dist;
 
  }
 
  public function saveFile($url, $dataset_id) {
 
    $targetDir = 'public://distribution/' . $dataset_id;
 
    $fileHelper = $this->getFileHelper();
 
    $fileHelper->prepareDir($targetDir);
 
    // Abort if file can't be saved locally.
    if (!($path = $fileHelper->retrieveFile($url, $targetDir))) {
      return FALSE;
    }
 
    if (is_object($path)) {
      return file_create_url($path->uri->value);
    }
    else {
      return $fileHelper->fileCreate($path);
    }
  }
 
}