Make data handler generic and working on localization

Author: hoogi91

Classes/Hooks/DataHandlerHook.php

@@ -4,6 +4,7 @@
 
 namespace Hoogi91\Spreadsheets\Hooks;
 
+use Hoogi91\Spreadsheets\Domain\ValueObject\DsnValueObject;
 use TYPO3\CMS\Backend\Utility\BackendUtility;
 use TYPO3\CMS\Core\Database\ConnectionPool;
 use TYPO3\CMS\Core\DataHandling\DataHandler;
@@ -17,10 +18,38 @@ class DataHandlerHook
      */
     private array $records = [];
 
+    /**
+     * @var array<string, array<string, array<string, array<string>>>>
+     */
+    private array $activationTypes = [];
+
     public function __construct(
         private readonly FileRepository $fileRepository,
         private readonly ConnectionPool $connectionPool
     ) {
+        foreach ($GLOBALS['TCA'] as $table => $tca) {
+            $table = (string)$table;
+            foreach ($tca['columns'] ?? [] as $column => $conf) {
+                if (
+                    isset($conf['config']['renderType'], $conf['config']['uploadField'])
+                    && $conf['config']['renderType'] === 'spreadsheetInput'
+                ) {
+                    $this->activationTypes[$table]['*'][(string)$conf['config']['uploadField']][] = $column;
+                }
+            }
+
+            foreach ($GLOBALS['TCA'][$table]['types'] as $CType => $type) {
+                $CType = (string)$CType;
+                foreach ($type['columnsOverrides'] as $column => $conf) {
+                    if (
+                        isset($conf['config']['renderType'], $conf['config']['uploadField'])
+                        && $conf['config']['renderType'] === 'spreadsheetInput'
+                    ) {
+                        $this->activationTypes[$table][$CType][(string)$conf['config']['uploadField']][] = $column;
+                    }
+                }
+            }
+        }
     }
 
     /**
@@ -37,56 +66,107 @@ public function processDatamap_afterDatabaseOperations( // @codingStandardsIgnor
         array $fieldArray,
         DataHandler $dataHandler
     ): void {
-        // skip processing for unknown uid, wrong table, status or not updated assets
+        // skip processing for not found uid or irrelevant status
         $uid = $dataHandler->substNEWwithIDs[$id] ?? (is_int($id) ? $id : null);
-        if (
-            $uid === null
-            || $table !== 'tt_content'
-            || !array_key_exists('tx_spreadsheets_assets', $fieldArray)
-            || !in_array($status, ['new', 'update'], true)
-        ) {
+        if ($uid === null || !is_string($table) || !in_array($status, ['new', 'update'], true)) {
             return;
         }
 
-        // skip if not spreadsheet table or bodytext is already filled
-        $CType = $fieldArray['CType'] ?? $this->getBackendRecordField($uid, 'CType');
-        if (!in_array($CType, ['spreadsheets_table', 'spreadsheets_tabs'], true)) {
+        // ignore if handler should not process for table and/or CType
+        $CType = $fieldArray['CType'] ?? $this->getBackendRecordField($uid, $table, 'CType');
+        if (!isset($this->activationTypes[$table]['*']) && !isset($this->activationTypes[$table][$CType])) {
             return;
         }
 
-        // truncate bodytext after update if assets have been removed
-        if ($fieldArray['tx_spreadsheets_assets'] === 0) {
-            if ($status === 'update') {
-                $this->connectionPool
-                    ->getConnectionForTable('tt_content')
-                    ->update('tt_content', ['bodytext' => ''], ['uid' => $uid]);
+        $activationConfig = $this->activationTypes[$table][$CType] ?? $this->activationTypes[$table]['*'];
+        foreach ($activationConfig as $uploadField => $renderFields) {
+            // truncate render fields after update if assets have been removed
+            if ($fieldArray[$uploadField] === 0) {
+                if ($status === 'update') {
+                    $this->connectionPool
+                        ->getConnectionForTable($table)
+                        ->update($table, array_fill_keys($renderFields, ''), ['uid' => $uid]);
+                }
+
+                continue;
             }
 
-            return;
+            // if upload fields was filled we get it's relations and start to update all render fields if required
+            /** @var array<FileReference> $relations */
+            $relations = $this->fileRepository->findByRelation($table, $uploadField, $uid);
+            foreach ($renderFields as $renderField) {
+                $this->setSpreadsheetValue($uid, $table, $status, $renderField, $relations);
+            }
         }
+    }
 
-        /** @var array<FileReference> $relations */
-        $relations = $this->fileRepository->findByRelation('tt_content', 'tx_spreadsheets_assets', $uid);
+    /**
+     * @param int $uid UID of chart record
+     * @param string $table Table to update
+     * @param string $status Status of current record update
+     * @param string $field Field to update spreadsheet value
+     * @param array<FileReference> $relations File relations found
+     *
+     */
+    private function setSpreadsheetValue(
+        int $uid,
+        string $table,
+        string $status,
+        string $field,
+        array $relations
+    ): void {
         if (empty($relations)) {
             return;
         }
 
-        // update bodytext to default file selection
-        if (empty($this->getBackendRecordField($uid, 'bodytext')) === true) {
+        // if backend record field is currently empty we pre-select with first relation
+        $fieldValue = $this->getBackendRecordField($uid, $table, $field);
+        if (empty($fieldValue) === true) {
             $this->connectionPool
-                ->getConnectionForTable('tt_content')
-                ->update('tt_content', ['bodytext' => 'spreadsheet://' . $relations[0]->getUid()], ['uid' => $uid]);
+                ->getConnectionForTable($table)
+                ->update($table, [$field => 'spreadsheet://' . $relations[0]->getUid()], ['uid' => $uid]);
+        } elseif ($status === 'new' && is_string($fieldValue)) {
+            $dsn = $this->getTranslatedSpreadsheetDsn(
+                DsnValueObject::createFromDSN($fieldValue),
+                $relations
+            );
+            if ($dsn !== null) {
+                $this->connectionPool
+                    ->getConnectionForTable($table)
+                    ->update($table, [$field => $dsn], ['uid' => $uid]);
+            }
+        }
+    }
+
+    /**
+     * @param DsnValueObject $dsn Original DSN
+     * @param array<FileReference> $references File relations found
+     *
+     */
+    private function getTranslatedSpreadsheetDsn(DsnValueObject $dsn, array $references): ?string
+    {
+        foreach ($references as $reference) {
+            if ($reference->getReferenceProperty('l10n_parent') === $dsn->getFileReference()) {
+                return str_replace(
+                    'spreadsheet://' . $dsn->getFileReference(),
+                    'spreadsheet://' . $reference->getUid(),
+                    $dsn->getDsn()
+                );
+            }
         }
+
+        return null;
     }
 
     /**
-     * @param int $uid UID of tt_content record
+     * @param int $uid UID of record
+     * @param string $table Table to get record from
      * @param string $field Field to extract
      */
-    private function getBackendRecordField(int $uid, string $field): mixed
+    private function getBackendRecordField(int $uid, string $table, string $field): mixed
     {
         if (!isset($this->records[$uid])) {
-            $this->records[$uid] = BackendUtility::getRecord('tt_content', $uid); // @codeCoverageIgnore
+            $this->records[$uid] = BackendUtility::getRecord($table, $uid); // @codeCoverageIgnore
         }
 
         return $this->records[$uid][$field] ?? null;