Allow dot notition for getAttribute

Author: jenssegers

src/Jenssegers/Mongodb/Model.php

@@ -195,6 +195,52 @@ public function getTable()
         return parent::getTable();
     }
 
+    /**
+     * Get an attribute from the model.
+     *
+     * @param  string  $key
+     * @return mixed
+     */
+    public function getAttribute($key)
+    {
+        // Check if the key is an array dot notation.
+        if (strstr($key, '.'))
+        {
+            $attributes = array_dot($this->attributes);
+
+            if (array_key_exists($key, $attributes))
+            {
+                return $this->getAttributeValue($key);
+            }
+        }
+
+        return parent::getAttribute($key);
+    }
+
+    /**
+     * Get an attribute from the $attributes array.
+     *
+     * @param  string  $key
+     * @return mixed
+     */
+    protected function getAttributeFromArray($key)
+    {
+        if (array_key_exists($key, $this->attributes))
+        {
+            return $this->attributes[$key];
+        }
+
+        else if (strstr($key, '.'))
+        {
+            $attributes = array_dot($this->attributes);
+
+            if (array_key_exists($key, $attributes))
+            {
+                return $attributes[$key];
+            }
+        }
+    }
+
     /**
      * Set a given attribute on the model.
      *

tests/ModelTest.php

@@ -419,4 +419,19 @@ public function testRaw()
 		$this->assertTrue(is_array($result));
 	}
 
+	public function testDotNotation()
+	{
+		$user = User::create(array(
+			'name' => 'John Doe',
+			'address' => [
+				'city' => 'Paris',
+				'country' => 'France',
+			]
+		));
+
+		$this->assertEquals('Paris', $user->getAttribute('address.city'));
+		$this->assertEquals('Paris', $user['address.city']);
+		$this->assertEquals('Paris', $user->{'address.city'});
+	}
+
 }