Added new Fix's Application.Run signature

markrendle · Sep 29, 2013 · 021ce04 · 021ce04
1 parent f590f92
commit 021ce04
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/packaging/nuspec/Simple.Web.nuspec b/packaging/nuspec/Simple.Web.nuspec
@@ -16,6 +16,7 @@
   <files>
     <file src="..\..\build\Simple.Web.dll" target="lib\net40" />
     <file src="..\..\build\Simple.Web.*db" target="lib\net40" />
+    <file src="..\..\build\Simple.Web.xml" target="lib\net40" />
     <file src="..\..\src\Simple.Web\**\*.cs" target="src" />
   </files>
 </package> 
diff --git a/src/Simple.Web/Application.cs b/src/Simple.Web/Application.cs
@@ -54,6 +54,20 @@ public static Task Run(IDictionary<string, object> env)
                 .ContinueWith(t => WriteResponse(t, context, env)).Unwrap();
         }
 
+        public static Task Run(IDictionary<string, object> env, Func<Task> next)
+        {
+            var context = new OwinContext(env);
+            var task = Run(context);
+
+            if (task == null)
+            {
+                return next();
+            }
+
+            return task
+                .ContinueWith(t => WriteResponse(t, context, env)).Unwrap();
+        }
+
         private static Func<Stream, Task> ErrorHandler(string message)
         {
             return stream =>