Fix bug in the exception handler

README.md

@@ -68,9 +68,9 @@ class IndexController < Grip::Controllers::Http
 end
 
 class Application < Grip::Application
-  def initialize(environment : String, serve_static : Bool)
+  def initialize(environment : String)
     # By default the environment is set to "development" and serve_static is false.
-    super(environment, serve_static)
+    super(environment)
 
     scope "/api" do
       scope "/v1" do
@@ -84,7 +84,7 @@ class Application < Grip::Application
   end
 end
 
-app = Application.new(environment: "development", serve_static: false)
+app = Application.new(environment: "development")
 app.run
 ```
 

spec/spec_helper.cr

@@ -12,7 +12,7 @@ end
 
 class ErrorApplication < Grip::Application
   def initialize
-    super(environment: "test", serve_static: false)
+    super(environment: "test")
 
     exception Grip::Exceptions::NotFound, ErrorController
   end
@@ -24,7 +24,7 @@ end
 
 class HttpApplication < Grip::Application
   def initialize
-    super(environment: "test", serve_static: false)
+    super(environment: "test")
 
     get "/", ExampleController
     get "/:id", ExampleController, as: :index
@@ -37,7 +37,7 @@ end
 
 class WebSocketApplication < Grip::Application
   def initialize
-    super(environment: "test", serve_static: false)
+    super(environment: "test")
 
     ws "/", MatchController
   end

src/grip/handlers/exception.cr

@@ -29,8 +29,9 @@ module Grip
           context.response.status_code = status_code
           context.exception = exception
 
+          updated_context = @handlers[exception.class.name].call(context)
           context.response.close
-          @handlers[exception.class.name].call(context)
+          updated_context
         else
           if status_code.in?(400..599)
             context.response.status_code = status_code