Add withContext and withLinkClass to MwStack

Signed-off-by: RJ Garcia <rj@bighead.net>

CHANGELOG.md

@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](http://semver.org/).
 
 ## [Unreleased]
 
+## Added
+
+- Added `withContext` and `withLinkClass` options to MwStack.
+
 ## [0.4.0] - 2017-01-04
 ### Changed
 

doc/api.rst

@@ -197,7 +197,7 @@ array splitArgs(array $args)
                 return $next(...$args);
             };
         }
-    
+
 
 class MwStack implements Countable
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@@ -232,6 +232,12 @@ Closure compose(callable $last = null)
     Composes the stack into a handler.
 Generator getEntries()
     Yields the raw stack entries in the order they were added.
+MwStack withContext(Context $ctx)
+    Creates a clone of the current stack with an updated context
+MwStack withLinkClass($class)
+    Creates a clone of the current stack with an updated link class
+MwStack withEntries($entries)
+    Creates a clone of the current stack with updated entries.
 MwStack static createFromEntries($name, $entries)
     Creates a stack with a set of entries. ``mw\stack`` internally calls this.
 

src/MwStack.php

@@ -167,6 +167,16 @@ public function compose($last = null) {
         return compose($this->normalize(), $this->ctx, $last, $this->link_class);
     }
 
+    public function withContext(Context $ctx) {
+        $stack = clone $this;
+        $stack->ctx = $ctx;
+        return $stack;
+    }
+    public function withLinkClass($link_class) {
+        $stack = clone $this;
+        $stack->link_class = $link_class;
+        return $stack;
+    }
     public function withEntries($entries) {
         return static::createFromEntries($this->name, $entries, $this->ctx, $this->link_class);
     }

test/mw.spec.php

@@ -32,6 +32,8 @@ public function handle($s) {
     }
 }
 
+class MyLink extends Mw\Link {}
+
 describe('Mw', function() {
     describe('#compose', function() {
         it('composes a set of middleware into a handler', function() {
@@ -180,6 +182,21 @@ function() { return 1; },
             $handler = $stack->compose();
             assert($handler(2) === 1);
         });
+        it('allows you to change context', function() {
+            $stack = mw\stack('stack')->withContext(new Mw\Context\StdContext(function() { return 1; }));
+            $stack->push(id());
+            $handler = $stack->compose();
+            assert($handler(2) === 1);
+        });
+        it('allows you to change link class', function() {
+            $stack = mw\stack('stack')->withLinkClass(MyLink::class);
+            $stack->push(function($id, $next) {
+                assert($next instanceof MyLink);
+                return;
+            });
+            $handler = $stack->compose();
+            $handler(1);
+        });
     });
     describe('#stackMerge', function() {
         it('merges stacks together into a new stack', function() {