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") |       response.headers["Content-Type"].should eq("text/plain") | ||||||
|     end |     end | ||||||
|   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 | end | ||||||
|  |  | ||||||
|  | @ -40,3 +40,20 @@ end | ||||||
| def headers(env, additional_headers) | def headers(env, additional_headers) | ||||||
|   env.response.headers.merge!(additional_headers) |   env.response.headers.merge!(additional_headers) | ||||||
| end | 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…
	
	Add table
		Add a link
		
	
		Reference in a new issue