Add gzip helper to enable/disable HTTP::DeflateHandler

This commit is contained in:
sdogruyol 2016-09-15 19:35:34 +03:00
parent 7c8915bfa4
commit 4cd30839a5
3 changed files with 21 additions and 4 deletions

View file

@ -122,4 +122,15 @@ describe "Macros" do
response.headers["Content-Length"].should eq("6") response.headers["Content-Length"].should eq("6")
end end
end end
describe "#gzip" do
it "adds HTTP::DeflateHandler to handlers" do
gzip true
Kemal.config.handlers.last.is_a?(HTTP::DeflateHandler).should eq true
end
it "doesn't add HTTP::DeflateHandler to handlers by default" do
Kemal.config.handlers.last.is_a?(HTTP::DeflateHandler).should eq false
end
end
end end

View file

@ -50,7 +50,7 @@ end
# send_file env, "./path/to/file", "image/jpeg" # send_file env, "./path/to/file", "image/jpeg"
def send_file(env, path : String, mime_type : String? = nil) def send_file(env, path : String, mime_type : String? = nil)
file_path = File.expand_path(path, Dir.current) file_path = File.expand_path(path, Dir.current)
mime_type ||= "application/octet-stream" mime_type ||= "application/octet-stream"
env.response.content_type = mime_type env.response.content_type = mime_type
env.response.content_length = File.size(file_path) env.response.content_length = File.size(file_path)
File.open(file_path) do |file| File.open(file_path) do |file|
@ -70,4 +70,11 @@ def send_file(env, data : Slice(UInt8), mime_type : String? = nil)
env.response.content_type = mime_type env.response.content_type = mime_type
env.response.content_length = data.bytesize env.response.content_length = data.bytesize
env.response.write data env.response.write data
end end
# Configures an `HTTP::Server::Response` to compress the response
# output, either using gzip or deflate, depending on the `Accept-Encoding` request header.
# It's disabled by default.
def gzip(status : Bool = false)
add_handler HTTP::DeflateHandler.new if status
end

View file

@ -9,7 +9,7 @@ module Kemal
return call_next(context) if context.request.path.not_nil! == "/" return call_next(context) if context.request.path.not_nil! == "/"
super super
end end
def mime_type(path) def mime_type(path)
case File.extname(path) case File.extname(path)
when ".txt" then "text/plain" when ".txt" then "text/plain"
@ -23,6 +23,5 @@ module Kemal
else "application/octet-stream" else "application/octet-stream"
end end
end end
end end
end end