Add 'with' to allow chaining of methods

src/FactoryStory.php

@@ -2,6 +2,7 @@
 
 namespace FactoryStories;
 
+use Illuminate\Support\Collection;
 use FactoryStories\Contracts\FactoryStoryContract;
 
 /**
@@ -11,6 +12,7 @@ abstract class FactoryStory
 {
 
     protected $times = 1;
+    protected $with = null;
 
     /**
      * Here you can create your complex model factory
@@ -32,7 +34,7 @@ abstract public function build($params = []);
      */
     public function times($amount)
     {
-        $this->times = $amount;
+        $this->times = (int)$amount;
 
         return $this;
     }
@@ -46,12 +48,55 @@ public function times($amount)
      */
     public function create($params = [])
     {
-        if (is_int($this->times) && $this->times > 1) {
-            return collect(range(1, $this->times))
-                ->transform(function ($index) use ($params) {
-                    return $this->build($params);
-                });
+        if ($this->times <= 1) {
+            return $this->forge($params);
         }
+
+        return Collection::times($this->times, function () use ($params) {
+            return $this->forge($params);
+        });
+    }
+
+    /**
+     * Add methods to be ran after initial build
+     *
+     * @param array $methods Array of method names
+     *
+     * @return FactoryStory $this Updated $with
+     */
+
+    public function with($methods = [])
+    {
+        if(!is_null($this->with)) {
+            $this->with->merge(collect($methods));
+        }
+        else {
+            $this->with = collect($methods);
+        }
+
+        return $this;
+    }
+
+    /**
+     * Handles any creation steps that need to be done after building
+     * a story but before returning
+     *
+     * @param  array  $params Array of custom params
+     *
+     * @return Mixed
+     */
+
+    public function forge($params = [])
+    {
+        if(is_null($this->with) || $this->with->count() === 0) {
+            return $this->build($params);
+        }
+
+        $story = $this->build($params);
+
+        $this->with->reduce(function($story, $methodName) {
+            return $this->$methodName($story);
+        }, $story);
 
         return $this->build($params);
     }