Allow additional fields to be mapped to the file entitiy

drevops · Apr 2, 2024 · 99e0414 · 99e0414
1 parent 9f1914d
commit 99e0414
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/FileTrait.php b/src/FileTrait.php
@@ -110,7 +110,18 @@ protected function fileCreateEntity(\StdClass $stub): FileInterface {
       }
     }
 
-    return \Drupal::service('file.repository')->writeData(file_get_contents($path), $destination, FileSystemInterface::EXISTS_REPLACE);
+    $entity = \Drupal::service('file.repository')->writeData(file_get_contents($path), $destination, FileSystemInterface::EXISTS_REPLACE);
+
+    foreach ($stub as $property => $value) {
+      // If path or URI has been specified then the value has already been handled.
+      if (in_array($property, ['path', 'uri'])) {
+        continue;
+      }
+      $entity->set($property, $value);
+    }
+
+    $entity->save();
+    return $entity;
   }
 
   /**