[FEATURE] Add previews and toggles for file references of original records

Author: Bunnyfield

Classes/Domain/Repository/TransfusionRepository.php

@@ -13,6 +13,7 @@
 use TYPO3\CMS\Core\Database\Query\Restriction\WorkspaceRestriction;
 use TYPO3\CMS\Core\Imaging\Icon;
 use TYPO3\CMS\Core\Imaging\IconFactory;
+use TYPO3\CMS\Core\Resource\FileRepository;
 use TYPO3\CMS\Core\Utility\GeneralUtility;
 
 /**
@@ -34,9 +35,15 @@ class TransfusionRepository
      */
     protected IconFactory $iconFactory;
 
+    /**
+     * @var FileRepository
+     */
+    protected FileRepository $fileRepository;
+
     public function __construct()
     {
         $this->iconFactory = GeneralUtility::makeInstance(IconFactory::class);
+        $this->fileRepository = GeneralUtility::makeInstance(FileRepository::class);
     }
 
     /**
@@ -105,7 +112,31 @@ protected function fetchDefaultLanguageRecordsAndConnectionsForTable(
             )
             ->orderBy($transFusionFields['sorting'])
             ->executeQuery();
+        $queryBuilder = GeneralUtility::makeInstance(ConnectionPool::class)->getQueryBuilderForTable('sys_file_reference');
+        $fileFields = $queryBuilder
+            ->select(
+                'fieldname'
+            )
+            ->from('sys_file_reference')
+            ->where(
+                $queryBuilder->expr()->eq('pid', $queryBuilder->createNamedParameter($page, Connection::PARAM_INT)),
+                $queryBuilder->expr()->eq('tablenames', $queryBuilder->createNamedParameter($table)),
+                $queryBuilder->expr()->eq('l10n_parent', 0),
+                $queryBuilder->expr()->eq('sys_language_uid', 0),
+            )
+            ->groupBy('fieldname')
+            ->executeQuery()
+            ->fetchAllAssociativeIndexed();
         while ($record = $defaultLanguageQuery->fetchAssociative()) {
+            $files = [];
+            if (!empty($fileFields)) {
+                foreach (array_keys($fileFields) as $fieldName) {
+                    $fileRecords = $this->fileRepository->findByRelation($table, $fieldName, $record['uid']);
+                    if (!empty($fileRecords)) {
+                        $files[$fieldName] = $fileRecords;
+                    }
+                }
+            }
             $preparedRecord = [
                 'uid' => $record['uid'],
                 'pid' => $record['pid'],
@@ -118,6 +149,9 @@ protected function fetchDefaultLanguageRecordsAndConnectionsForTable(
                 'icon' => $this->getIconForRecord($table, $record, $record[$transFusionFields['label']]),
                 'previewData' => $record
             ];
+            if (!empty($files)) {
+                $preparedRecord['relatedFiles'] = $files;
+            }
             $preparedRecord['column'] = ($table === 'tt_content' ? $record[$transFusionFields['column']] : '');
             $connectedRecords = $this->getConnectedTranslations(
                 $table,
@@ -403,8 +437,37 @@ public function fetchDisconnectedRecordsAndPrepareDataMap(
                 $queryBuilder->expr()->eq($transFusionFields['language'], $language),
             )
             ->executeQuery();
+
+        $queryBuilder = GeneralUtility::makeInstance(ConnectionPool::class)->getQueryBuilderForTable('sys_file_reference');
+        $queryBuilder
+            ->getRestrictions()
+            ->removeAll()
+            ->add(GeneralUtility::makeInstance(DeletedRestriction::class))
+            ->add(GeneralUtility::makeInstance(WorkspaceRestriction::class));
+        $fileFields = $queryBuilder
+            ->select(
+                'fieldname'
+            )
+            ->from('sys_file_reference')
+            ->where(
+                $queryBuilder->expr()->eq('pid', $queryBuilder->createNamedParameter($page, Connection::PARAM_INT)),
+                $queryBuilder->expr()->eq('tablenames', $queryBuilder->createNamedParameter($table)),
+                $queryBuilder->expr()->eq('sys_language_uid', $language),
+            )
+            ->groupBy('fieldname')
+            ->executeQuery()
+            ->fetchAllAssociativeIndexed();
         while ($record = $disconnectedRecords->fetchAssociative()) {
             $fetchPossibleParents = false;
+            $files = [];
+            if (!empty($fileFields)) {
+                foreach (array_keys($fileFields) as $fieldName) {
+                    $fileRecords = $this->fileRepository->findByRelation($table, $fieldName, $record['uid']);
+                    if (!empty($fileRecords)) {
+                        $files[$fieldName] = $fileRecords;
+                    }
+                }
+            }
             $preparedRecord = [
                 'uid' => $record['uid'],
                 'type' => $record[$transFusionFields['type']],
@@ -414,6 +477,9 @@ public function fetchDisconnectedRecordsAndPrepareDataMap(
                 'original' => $record[$transFusionFields['original']],
                 'previewData' => $record
             ];
+            if (!empty($files)) {
+                $preparedRecord['relatedFiles'] = $files;
+            }
             $preparedRecord['column'] = ($table === 'tt_content' ? $record[$transFusionFields['column']] : '');
 
             $fetchFields = $transFusionFields['language'] . ',' . $transFusionFields['type'];