Adding websocket support :)

src/kemal/config.cr

@@ -4,6 +4,7 @@ module Kemal
   class Config
     INSTANCE = Config.new
     HANDLERS = [] of HTTP::Handler
+    WS_HANDLERS = [] of HTTP::Handler
     property ssl, port, env, workers, public_folder
 
     def initialize
@@ -26,6 +27,10 @@ module Kemal
       HANDLERS << handler
     end
 
+    def add_ws_handler(handler : HTTP::WebSocketHandler)
+      HANDLERS << handler
+    end
+
     # Reads configuration from config.yml. Currently it only supports the public_folder
     # option.
     # config.yml

src/kemal/dsl.cr

@@ -5,3 +5,7 @@ HTTP_METHODS = %w(get post put patch delete)
    Kemal::Handler::INSTANCE.add_route({{method}}.upcase, path, &block)
   end
 {% end %}
+
+def ws(path, &block : HTTP::WebSocketHandler::WebSocketSession -> _)
+  Kemal::WebsocketHandler.new path, &block
+end

src/kemal/websocket_handler.cr

@@ -0,0 +1,13 @@
+class Kemal::WebsocketHandler < HTTP::WebSocketHandler
+  getter handler
+
+  def initialize(@path, &@proc : WebSocketSession ->)
+    @handler = @proc
+    Kemal.config.add_ws_handler self
+  end
+
+  def call(request)
+    return call_next(request) unless request.path.not_nil! == @path
+    super
+  end
+end