CsvTemplate.php

Go to the documentation of this file.

<?php
namespace Magento\ImportExport\Mtf\Util\Import\File;

use Magento\Mtf\Util\Generate\File\TemplateInterface;

class CsvTemplate implements TemplateInterface
{
    private $config;

    private $csv;

    public function __construct(array $config)
    {
        $this->config = $config;
    }

    public function render()
    {
        if (!isset($this->config['filename'], $this->config['count'])) {
            throw new \InvalidArgumentException(
                'Configuration for file is bad. You must specify "filename" and "count" in configuration.'
            );
        }

        $filename = MTF_TESTS_PATH . $this->config['filename'] . '.php';

        if (!file_exists($filename)) {
            throw new \Exception('File "' . $filename . '" does not exist.');
        }

        $fh = fopen('php://temp', 'rw');
        $fh = $this->addEntitiesData($fh);
        rewind($fh);
        $this->csv = stream_get_contents($fh);
        fclose($fh);

        return $this->csv;
    }

    public function getName()
    {
        return sprintf(
            '%x_%s.csv',
            crc32(time()),
            basename($this->config['filename'])
        );
    }

    private function addEntitiesData($stream)
    {
        $filename = MTF_TESTS_PATH . $this->config['filename'] . '.php';
        $entitiesData = include $filename;

        $placeholders = [];
        if (!empty($this->config['placeholders'])) {
            $placeholders = $this->config['placeholders'];
        }

        fputcsv($stream, array_keys($entitiesData['entity_0']['data_0']));
        foreach ($placeholders as $entityKey => $entityData) {
            foreach ($entityData as $dataKey => $dataValue) {
                $row = array_map(
                    function ($value) use ($placeholders, $entityKey, $dataKey, $dataValue) {
                        if (is_string($value) && isset($placeholders[$entityKey][$dataKey])) {
                            return strtr($value, $dataValue);
                        }
                        return $value;
                    },
                    $entitiesData[$entityKey][$dataKey]
                );
                fputcsv($stream, $row);
            }
        }
        return $stream;
    }

    public function getCsv()
    {
        return $this->csv;
    }
}