mirror of
https://gitea.invidious.io/iv-org/shard-kemal.git
synced 2024-08-15 00:53:36 +00:00
Add send_file to helpers
This commit is contained in:
parent
b00bc7a202
commit
5e297d86f8
2 changed files with 43 additions and 0 deletions
|
@ -84,4 +84,30 @@ describe "Macros" do
|
|||
response.headers["Content-Type"].should eq("text/plain")
|
||||
end
|
||||
end
|
||||
|
||||
describe "#send_file" do
|
||||
it "sends file with given path and default mime-type" do
|
||||
get "/" do |env|
|
||||
send_file env, "./spec/asset/hello.ecr"
|
||||
end
|
||||
|
||||
request = HTTP::Request.new("GET", "/")
|
||||
response = call_request_on_app(request)
|
||||
response.status_code.should eq(200)
|
||||
response.headers["Content-Type"].should eq("application/octet-stream")
|
||||
response.headers["Content-Length"].should eq("20")
|
||||
end
|
||||
|
||||
it "sends file with given path and given mime-type" do
|
||||
get "/" do |env|
|
||||
send_file env, "./spec/asset/hello.ecr", "image/jpeg"
|
||||
end
|
||||
|
||||
request = HTTP::Request.new("GET", "/")
|
||||
response = call_request_on_app(request)
|
||||
response.status_code.should eq(200)
|
||||
response.headers["Content-Type"].should eq("image/jpeg")
|
||||
response.headers["Content-Length"].should eq("20")
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -40,3 +40,20 @@ end
|
|||
def headers(env, additional_headers)
|
||||
env.response.headers.merge!(additional_headers)
|
||||
end
|
||||
|
||||
# Send a file with given path and default `application/octet-stream` mime_type.
|
||||
#
|
||||
# send_file env, "./path/to/file"
|
||||
#
|
||||
# Optionally you can override the mime_type
|
||||
#
|
||||
# send_file env, "./path/to/file", "image/jpeg"
|
||||
def send_file(env, path : String, mime_type : String? = nil)
|
||||
file_path = File.expand_path(path, Dir.current)
|
||||
mime_type = "application/octet-stream" unless mime_type
|
||||
env.response.headers.add "Content-Type", mime_type
|
||||
env.response.content_length = File.size(file_path)
|
||||
File.open(file_path) do |file|
|
||||
IO.copy(file, env.response)
|
||||
end
|
||||
end
|
||||
|
|
Loading…
Reference in a new issue