dvf-2.x-dev/dvf_csv/src/Plugin/Visualisation/Source/CsvFile.php

<?php

namespace Drupal\dvf_csv\Plugin\Visualisation\Source;

use Drupal\Core\Cache\CacheBackendInterface;
use Drupal\Core\Extension\ModuleHandlerInterface;
use Drupal\Core\File\FileUrlGeneratorInterface;
use Drupal\Core\Form\FormStateInterface;
use Drupal\Core\Plugin\ContainerFactoryPluginInterface;
use Drupal\dvf\Plugin\VisualisationInterface;
use Drupal\dvf\Plugin\Visualisation\Source\VisualisationSourceBase;
use GuzzleHttp\Client;
use Psr\Log\LoggerInterface;
use Symfony\Component\DependencyInjection\ContainerInterface;

/**
 * Plugin implementation of the 'dvf_csv_file' visualisation source.
 *
 * @VisualisationSource(
 *   id = "dvf_csv_file",
 *   label = @Translation("CSV file"),
 *   category = @Translation("CSV"),
 *   visualisation_types = {
 *     "dvf_file",
 *     "dvf_url"
 *   }
 * )
 */
class CsvFile extends VisualisationSourceBase implements ContainerFactoryPluginInterface {

  /**
   * The cache backend.
   *
   * @var \Drupal\Core\Cache\CacheBackendInterface
   */
  protected $cache;

  /**
   * The CSV file fields.
   *
   * @var array
   */
  protected $fields;

  /**
   * The CSV file records.
   *
   * @var array
   */
  protected $records;

  /**
   * Constructs a new CsvFile.
   *
   * @param array $configuration
   *   A configuration array containing information about the plugin instance.
   * @param string $plugin_id
   *   The plugin_id for the plugin instance.
   * @param mixed $plugin_definition
   *   The plugin implementation definition.
   * @param \Drupal\dvf\Plugin\VisualisationInterface $visualisation
   *   The visualisation context in which the plugin will run.
   * @param \Drupal\Core\Extension\ModuleHandlerInterface $module_handler
   *   The module handler.
   * @param \Psr\Log\LoggerInterface $logger
   *   Instance of the logger object.
   * @param \GuzzleHttp\Client $http_client
   *   The HTTP client.
   * @param \Drupal\Core\Cache\CacheBackendInterface $cache
   *   The cache backend.
   * @param \Drupal\Core\File\FileUrlGeneratorInterface $file_url_generator
   *   The file URL generator.
   */
  public function __construct(
    array $configuration,
    $plugin_id,
    $plugin_definition,
    VisualisationInterface $visualisation = NULL,
    ModuleHandlerInterface $module_handler,
    LoggerInterface $logger,
    Client $http_client,
    CacheBackendInterface $cache,
    FileUrlGeneratorInterface $file_url_generator
  ) {
    parent::__construct($configuration, $plugin_id, $plugin_definition, $visualisation, $module_handler, $logger, $http_client, $file_url_generator);
    $this->cache = $cache;
  }

  /**
   * Creates an instance of the plugin.
   *
   * @param \Symfony\Component\DependencyInjection\ContainerInterface $container
   *   The container to pull out services used in the plugin.
   * @param array $configuration
   *   A configuration array containing information about the plugin instance.
   * @param string $plugin_id
   *   The plugin ID for the plugin instance.
   * @param mixed $plugin_definition
   *   The plugin implementation definition.
   * @param \Drupal\dvf\Plugin\VisualisationInterface $visualisation
   *   The visualisation context in which the plugin will run.
   *
   * @return static
   *   Returns an instance of this plugin.
   */
  public static function create(ContainerInterface $container, array $configuration, $plugin_id, $plugin_definition, VisualisationInterface $visualisation = NULL) {
    return new static(
      $configuration,
      $plugin_id,
      $plugin_definition,
      $visualisation,
      $container->get('module_handler'),
      $container->get('logger.channel.dvf'),
      $container->get('http_client'),
      $container->get('cache.dvf_csv'),
      $container->get('file_url_generator')
    );
  }

  /**
   * {@inheritdoc}
   */
  public function defaultConfiguration() {
    return [
      'csv' => [
        'delimiter' => ',',
        'enclosure' => '"',
        'escape' => '\\',
      ],
    ];
  }

  /**
   * {@inheritdoc}
   */
  public function settingsForm(array $form, FormStateInterface $form_state) {
    $form['csv'] = [
      '#type' => 'details',
      '#title' => $this->t('CSV file settings'),
      '#open' => TRUE,
      '#tree' => TRUE,
    ];

    $form['csv']['delimiter'] = [
      '#type' => 'textfield',
      '#title' => $this->t('Delimiter'),
      '#description' => $this->t('Set the field delimiter (one character only).'),
      '#default_value' => $this->config('csv', 'delimiter'),
    ];

    $form['csv']['enclosure'] = [
      '#type' => 'textfield',
      '#title' => $this->t('Enclosure'),
      '#description' => $this->t('Set the field enclosure character (one character only).'),
      '#default_value' => $this->config('csv', 'enclosure'),
    ];

    $form['csv']['escape'] = [
      '#type' => 'textfield',
      '#title' => $this->t('Escape'),
      '#description' => $this->t('Set the escape character (one character only).'),
      '#default_value' => $this->config('csv', 'escape'),
    ];

    return $form;
  }

  /**
   * {@inheritdoc}
   */
  public function getFields() {
    $fields = $this->getHeaders();
    $fields = array_combine($fields, $fields);

    return $fields;
  }

  /**
   * {@inheritdoc}
   */
  public function getRecords() {
    $records = [];

    foreach ($this->getHeaders() as $header_index => $header_label) {
      foreach ($this->getRows() as $row_id => $row) {
        if (!isset($records[$row_id])) {
          $records[$row_id] = new \stdClass();
        }
        $records[$row_id]->{$header_label} = array_key_exists($header_index, $row) ? $row[$header_index] : '';
      }
    }

    return $records;
  }

  /**
   * Returns the CSV headers.
   *
   * @return array
   *   An array of CSV headers.
   */
  protected function getHeaders() {
    $data = $this->getData();
    return !empty($data) ? array_shift($data) : [];
  }

  /**
   * Returns the CSV rows.
   *
   * @return array
   *   An array of CSV rows, excluding the headers.
   */
  protected function getRows() {
    $rows = $this->getData();

    // Remove headers.
    array_shift($rows);

    return $rows;
  }

  /**
   * Returns the CSV data.
   *
   * @return array
   *   An array of CSV rows, including the headers.
   */
  public function getData() {
    $cache_key = $this->getCacheKey();
    $cache_object = $this->cache->get($cache_key);

    if (is_object($cache_object)) {
      $data = $cache_object->data;
    }
    else {
      $data = $this->fetchData();
      $this->cache->set($cache_key, $data, $this->getCacheExpiry());
    }

    return $data;
  }

  /**
   * Fetches the CSV data.
   *
   * @return array
   *   An array containing the CSV file data.
   */
  protected function fetchData() {
    try {
      $uri = $this->config('uri');
      $response = $this->getContentFromUri($uri);
    }
    catch (\Exception $e) {
      $this->messenger()->addError('Unable to read CSV');
      $this->logger->error($this->t('Error reading CSV file :error',
        [':error' => $e->getMessage()]));
      $response = NULL;
    }

    $data = [];

    if ($response) {
      $delimiter = $this->config('csv', 'delimiter');
      $enclosure = $this->config('csv', 'enclosure');
      $escape = $this->config('csv', 'escape');

      foreach (preg_split('/\r\n|\r|\n/', $response) as $line) {
        $data[] = str_getcsv($line, $delimiter, $enclosure, $escape);
      }
    }

    return $data;
  }

  /**
   * Gets a cache key for this plugin.
   *
   * @return string
   *   The cache key.
   */
  protected function getCacheKey() {
    $plugin_id = hash('sha256', $this->getPluginId());
    $uri = $this->config('uri');

    return $plugin_id . ':' . $uri;
  }

}