improve get element

KejawenLab · Jul 20, 2019 · e1d21b9 · e1d21b9
1 parent 0b0fef5
commit e1d21b9
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/Collection.php b/src/Collection.php
@@ -55,11 +55,19 @@ public function add($value, $key = null)
      *
      * @param mixed $key
      *
-     * @return mixed|null
+     * @return int|string|null|Collection
      */
     public function get($key)
     {
-        return $this->elements[$key] ?? null;
+        if (!array_key_exists($key, $this->elements)) {
+            return null;
+        }
+
+        if (!is_array($this->elements[$key])) {
+            return $this->elements[$key];
+        }
+
+        return self::collect($this->elements[$key]);
     }
 
     /**