Allow Eloquent has one relations to return a default new model

laravel · Oct 31, 2016 · a948c08 · a948c08
1 parent 47480d7
commit a948c08
Showing 1 changed file with 24 additions and 1 deletion.
diff --git a/src/Illuminate/Database/Eloquent/Relations/HasOne.php b/src/Illuminate/Database/Eloquent/Relations/HasOne.php
@@ -6,14 +6,37 @@
 
 class HasOne extends HasOneOrMany
 {
+    protected $withDefault = false;
+
+    /**
+     * Return a new model instance in case the relationship does not exist.
+     *
+     * @return $this
+     */
+    public function withDefault()
+    {
+        $this->withDefault = true;
+
+        return $this;
+    }
+
     /**
      * Get the results of the relationship.
      *
      * @return mixed
      */
     public function getResults()
     {
-        return $this->query->first();
+        if ($result = $this->query->first()) {
+            return $result;
+        }
+
+        if ($this->withDefault) {
+            return $this->related->newInstance()
+                ->setAttribute(
+                    $this->getPlainForeignKey(), $this->getParentKey()
+                );
+        }
     }
 
     /**