Feature/add media import

config/data_converters.yml

@@ -6,3 +6,4 @@ data_converters:
     - App\Processor\Converter\ReferenceEntitySingleLinkConverter
     - App\Processor\Converter\ReferenceEntityMultipleLinksConverter
     # Feel free to add your own converters here:
+    - App\Processor\Converter\MediaConverter

src/Command/ImportCommand.php

@@ -94,6 +94,7 @@ protected function configure()
             ->addOption('apiPassword', null, InputOption::VALUE_OPTIONAL, 'The password of the user.', getenv('AKENEO_API_PASSWORD'))
             ->addOption('apiClientId', null, InputOption::VALUE_OPTIONAL, '', getenv('AKENEO_API_CLIENT_ID'))
             ->addOption('apiClientSecret', null, InputOption::VALUE_OPTIONAL, '', getenv('AKENEO_API_CLIENT_SECRET'))
+            ->addOption('importMedia', 'i', InputOption::VALUE_NONE, 'Allow to import media or not. Cell content of the CSV must have an encoded json format with "filePath":"/path/to/folder/picture.png"')
         ;
     }
 
@@ -104,6 +105,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
 
         $referenceEntityCode = $input->getArgument('referenceEntityCode');
         $filePath = $input->getArgument('filePath');
+        $importMedia = (bool)$input->getOption('importMedia');
 
         try {
             $this->reader = new CsvReader(
@@ -152,7 +154,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $this->io->newLine(2);
 
         // Import records
-        $this->importRecords($output, $filePath, $referenceEntityCode, $validValueKeys, $attributes, $channels);
+        $this->importRecords($output, $filePath, $referenceEntityCode, $validValueKeys, $attributes, $channels, $importMedia);
 
         $this->io->newLine(2);
         $this->io->success(sprintf(
@@ -244,7 +246,8 @@ private function importRecords(
         string $referenceEntityCode,
         array $validValueKeys,
         array $attributes,
-        array $channels
+        array $channels,
+        $importMedia = false
     ): void {
         $progressBar = new ProgressBar($output, $this->reader->count());
         $progressBar->start();
@@ -279,7 +282,7 @@ private function importRecords(
             $linesToWrite[] = $line;
 
             if (count($recordsToWrite) === self::BATCH_SIZE) {
-                $this->writeRecords($filePath, $referenceEntityCode, $recordWriter, $recordsToWrite, $linesToWrite);
+                $this->writeRecords($filePath, $referenceEntityCode, $recordWriter, $recordsToWrite, $linesToWrite, $importMedia);
 
                 $recordsToWrite = [];
                 $linesToWrite = [];
@@ -288,7 +291,7 @@ private function importRecords(
         }
 
         if (!empty($recordsToWrite)) {
-            $this->writeRecords($filePath, $referenceEntityCode, $recordWriter, $recordsToWrite, $linesToWrite);
+            $this->writeRecords($filePath, $referenceEntityCode, $recordWriter, $recordsToWrite, $linesToWrite, $importMedia);
             $progressBar->advance(count($recordsToWrite));
         }
 
@@ -300,9 +303,10 @@ private function writeRecords(
         string $referenceEntityCode,
         RecordWriter $recordWriter,
         array $recordsToWrite,
-        array $linesToWrite
+        array $linesToWrite,
+        $importMedia = false
     ): void {
-        $responses = $recordWriter->write($referenceEntityCode, $recordsToWrite);
+        $responses = $recordWriter->write($referenceEntityCode, $recordsToWrite, $importMedia);
 
         $this->logger->logResponses($responses);
         $this->invalidFileGenerator->fromResponses($responses, $linesToWrite, $filePath, $this->reader->getHeaders());

src/Processor/Converter/MediaConverter.php

@@ -0,0 +1,47 @@
+<?php
+/**
+ * Diglin GmbH - Switzerland
+ *
+ * @author      Sylvain Rayé <support at diglin.com>
+ * @category    Diglin
+ * @package     csv_to_reference_entity
+ * @copyright   Copyright (c) Diglin (https://www.diglin.com)
+ */
+
+declare(strict_types=1);
+
+namespace App\Processor\Converter;
+
+class MediaConverter implements DataConverterInterface
+{
+    /**
+     * Does this data converter support the given $attribute
+     *
+     * @param array $attribute
+     *
+     * @return bool
+     */
+    public function support(array $attribute): bool
+    {
+        return isset($attribute['type']) && 'image' === $attribute['type'];
+    }
+
+    /**
+     * Convert the given $data for the given $attribute to the correct format expected by the Akeneo API
+     *
+     * @param array $attribute
+     * @param string $data
+     *
+     * @return string|null
+     */
+    public function convert(array $attribute, string $data): ?string
+    {
+        if (empty($data)) {
+            return null;
+        }
+
+        $jsonArray = json_decode($data, true);
+
+        return (isset($jsonArray['filePath']) ? $jsonArray['filePath'] : null);
+    }
+}

src/Writer/RecordWriter.php

@@ -16,8 +16,39 @@ public function __construct(AkeneoPimEnterpriseClientInterface $client)
         $this->client = $client;
     }
 
-    public function write(string $referenceEntityCode, array $records): array
+    /**
+     * @param string $referenceEntityCode
+     * @param array $records
+     * @param bool $importMedia
+     *
+     * @return array
+     */
+    public function write(string $referenceEntityCode, array $records, $importMedia = false): array
     {
+        if ($importMedia) {
+            // Create MediaFile if not already exist before to import to reference entity record
+            foreach ($records as $key => $record) {
+
+                foreach ($record['values'] as $i => $values) {
+                    foreach ($values as $j => $value) {
+                        if (!empty($value['data'])) {
+
+                            $pathinfo = pathinfo($value['data'], PATHINFO_EXTENSION);
+
+                            if (!empty($pathinfo) && @file_exists($value['data'])) {
+                                $code = $this->client->getReferenceEntityMediaFileApi()->create($value['data']);
+                                $records[$key]['values'][$i][$j] = [
+                                    'channel' => $value['channel'],
+                                    'locale'  => $value['locale'],
+                                    'data'    => $code,
+                                ];
+                            }
+                        }
+                    }
+                }
+            }
+        }
+
         return $this->client->getReferenceEntityRecordApi()->upsertList($referenceEntityCode, $records);
     }
 }