From 050676f4bd8e52ebfca758b3c5dd30444ce0b6e9 Mon Sep 17 00:00:00 2001
From: Keegan Rankin <keegan@agaric.coop>
Date: Thu, 9 Nov 2023 17:42:58 -0800
Subject: [PATCH] Refactor to account for migration of translated entities

---
 src/MigrationHelperFieldTransformations.php | 54 +++++++++++++--------
 1 file changed, 34 insertions(+), 20 deletions(-)

diff --git a/src/MigrationHelperFieldTransformations.php b/src/MigrationHelperFieldTransformations.php
index e1ac020..1f100d7 100644
--- a/src/MigrationHelperFieldTransformations.php
+++ b/src/MigrationHelperFieldTransformations.php
@@ -50,29 +50,43 @@ class MigrationHelperFieldTransformations {
 
     foreach ($entities as $entity) {
       $media_target_ids = [];
-
-      foreach ($source_field_names as $source_field_name) {
-        foreach ($entity->$source_field_name as $fieldItem) {
-          $entity_data = [
-            'bundle' => $media_entity_bundle,
-            'uid' => $entity->getOwnerId(),
-            'langcode' => $entity->language()->getId(),
-            'status' => 1,
-          ];
-
-          foreach (\array_keys($fieldItem->getValue()) as $subfield) {
-            $entity_data[$media_target_field][$subfield] = $fieldItem->get($subfield)->getValue();
+      $langcodes = $entity->getTranslationLanguages();
+      $translations = [];
+      if ($langcodes > 0) {
+        foreach($langcodes as $langcode) {
+          $check_translation = $entity->getTranslation($langcode->getId());
+          if ($check_translation !== NULL) {
+            $translations[] = $entity->getTranslation($langcode->getId());
           }
-
-          $media_entity = $media_entity_storage->create($entity_data);
-          $media_entity->save();
-          $media_target_ids[] = $media_entity->id();
         }
       }
 
-      $entity->set($media_field_name, $media_target_ids);
-      $entity->save();
-    }
-  }
+      foreach ($translations as $translation) {
 
+        foreach ($source_field_names as $source_field_name) {
+          foreach ($translation->$source_field_name as $fieldItem) {
+            $entity_data = [
+              'bundle' => $media_entity_bundle,
+              'langcode' => $fieldItem->getLangcode(),
+              'status' => 1,
+            ];
+            // Some entities (like taxonomy terms) do not have a an owner id
+            if(method_exists($entity, 'getOwnerId')) {
+              $entity_data['uid']  = $entity->getOwnerId();
+            }
+
+            foreach (\array_keys($fieldItem->getValue()) as $subfield) {
+              $entity_data[$media_target_field][$subfield] = $fieldItem->get($subfield)->getValue();
+            }
+
+            $media_entity = $media_entity_storage->create($entity_data);
+            $media_entity->save();
+            $media_target_ids[] = $media_entity->id();
+          }
+        }
+        $translation->set($media_field_name, $media_target_ids);
+      }
+      $entity->save();
+   }
+  }
 }