64 lines
1.6 KiB
Crystal
64 lines
1.6 KiB
Crystal
module Kemal
|
|
class WebsocketError < Exception
|
|
property code : Int32 = 1011
|
|
property status_message : String = "websocket error"
|
|
|
|
def initialize(@code, @status_message : String)
|
|
end
|
|
end
|
|
|
|
class UpgradeRequired < WebsocketError
|
|
def initialize
|
|
super(426, "Upgrade required")
|
|
end
|
|
end
|
|
|
|
class BadRequest < WebsocketError
|
|
def initialize(status_message : String)
|
|
super(400, status_message)
|
|
end
|
|
end
|
|
|
|
class WebSocketHandler
|
|
include HTTP::Handler
|
|
|
|
INSTANCE = new
|
|
property routes
|
|
|
|
def initialize
|
|
@routes = Radix::Tree(WebSocket).new
|
|
end
|
|
|
|
def call(context : HTTP::Server::Context)
|
|
return call_next(context) unless context.ws_route_found? && websocket_upgrade_request?(context)
|
|
content = context.websocket.call(context)
|
|
context.response.print(content)
|
|
context
|
|
end
|
|
|
|
def lookup_ws_route(path : String)
|
|
@routes.find "/ws" + path
|
|
end
|
|
|
|
def add_route(path : String, &handler : HTTP::WebSocket, HTTP::Server::Context -> Void)
|
|
add_to_radix_tree path, WebSocket.new(path, &handler)
|
|
end
|
|
|
|
private def add_to_radix_tree(path, websocket)
|
|
node = radix_path "ws", path
|
|
@routes.add node, websocket
|
|
end
|
|
|
|
private def radix_path(method, path)
|
|
'/' + method.downcase + path
|
|
end
|
|
|
|
private def websocket_upgrade_request?(context)
|
|
return unless upgrade = context.request.headers["Upgrade"]?
|
|
return unless upgrade.compare("websocket", case_insensitive: true) == 0
|
|
|
|
context.request.headers.includes_word?("Connection", "Upgrade")
|
|
end
|
|
end
|
|
end
|