Multiple albums in MusicRecording

Torann · Jun 15, 2017 · 52d6004 · 52d6004
1 parent 42336c9
commit 52d6004
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/src/ContextTypes/MusicRecording.php b/src/ContextTypes/MusicRecording.php
@@ -26,17 +26,26 @@ class MusicRecording extends MusicAbstractContext
     /**
      * Set in album attribute
      *
-     * @param array|string $item
+     * @param array|string $items
      * @return array
      */
-    protected function setInAlbumAttribute($item)
+    protected function setInAlbumAttribute($items)
     {
-        if ( ! is_array($item))
+        if ( ! is_array($items))
         {
-            return $item;
+            return $items;
+        }
+
+        //Check if not multidimensional array (for backward compatibility)
+        if((count($items) == count($items, COUNT_RECURSIVE)))
+        {
+            return $this->getNestedContext(MusicAlbum::class, $items);
         }
 
-        return $this->getNestedContext(MusicAlbum::class, $item);
+        //multiple albums
+        return array_map(function ($item) {
+            return $this->getNestedContext(MusicAlbum::class, $item);
+        }, $items);
     }
 
     /**