Initial (working?) support for request/response trailers.

Author: ioquatix

async-http.gemspec

@@ -22,7 +22,7 @@ Gem::Specification.new do |spec|
 	spec.add_dependency("async-pool", "~> 0.2")
 
 	spec.add_dependency("protocol-http", "~> 0.16.0")
-	spec.add_dependency("protocol-http1", "~> 0.10.0")
+	spec.add_dependency("protocol-http1", "~> 0.11.0")
 	spec.add_dependency("protocol-http2", "~> 0.13.0")
 
 	# spec.add_dependency("openssl")

bake/async/http.rb

@@ -67,6 +67,15 @@ def fetch(url, method:)
 		end
 
 		response.finish
+		
+		if trailers = response.headers.trailers
+			trailers.each do |key, value|
+				terminal.print_line(
+					:key, key.rjust(align), :reset, ": ", :value, value.inspect
+				)
+			end
+		end
+		
 		internet.close
 	end
 end

lib/async/http/protocol/http1.rb

@@ -33,6 +33,10 @@ def self.bidirectional?
 					true
 				end
 
+				def self.trailers?
+					true
+				end
+				
 				def self.client(peer)
 					stream = IO::Stream.new(peer, sync: false)
 

lib/async/http/protocol/http1/client.rb

@@ -61,7 +61,7 @@ def call(request, task: Task.current)
 									subtask.annotate("Streaming body.")
 
 									# Once we start writing the body, we can't recover if the request fails. That's because the body might be generated dynamically, streaming, etc.
-									write_body(@version, body)
+									write_body(@version, body, false, request.trailers)
 								end
 							end
 						elsif protocol = request.protocol

lib/async/http/protocol/http1/server.rb

@@ -84,11 +84,12 @@ def each(task: Task.current)
 									body.call(stream)
 								else
 									head = request.head?
+									trailers = response.trailers
 
 									request = nil unless body
 									response = nil
 
-									write_body(@version, body, head)
+									write_body(@version, body, head, trailers)
 								end
 							else
 								# If the request failed to generate a response, it was an internal server error:

lib/async/http/protocol/http10.rb

@@ -32,6 +32,10 @@ def self.bidirectional?
 					false
 				end
 
+				def self.trailers?
+					false
+				end
+				
 				def self.client(peer)
 					stream = IO::Stream.new(peer, sync: false)
 

lib/async/http/protocol/http11.rb

@@ -32,6 +32,10 @@ def self.bidirectional?
 					true
 				end
 
+				def self.trailers?
+					true
+				end
+				
 				def self.client(peer)
 					stream = IO::Stream.new(peer, sync: false)
 

lib/async/http/protocol/http2.rb

@@ -33,6 +33,10 @@ def self.bidirectional?
 					true
 				end
 
+				def self.trailers?
+					true
+				end
+				
 				CLIENT_SETTINGS = {
 					::Protocol::HTTP2::Settings::ENABLE_PUSH => 0,
 					::Protocol::HTTP2::Settings::MAXIMUM_FRAME_SIZE => 0x100000,

lib/async/http/protocol/http2/output.rb

@@ -27,22 +27,26 @@ module HTTP
 		module Protocol
 			module HTTP2
 				class Output
-					def self.for(stream, body)
-						output = self.new(stream, body)
+					def self.for(stream, body, trailers)
+						output = self.new(stream, body, trailers)
 
 						output.start
 
 						return output
 					end
 
-					def initialize(stream, body)
+					def initialize(stream, body, trailers = nil)
 						@stream = stream
 						@body = body
+						@trailers = trailers
+						
 						@task = nil
 
 						@window_updated = Async::Condition.new
 					end
 
+					attr :trailers
+					
 					def start(parent: Task.current)
 						raise "Task already started!" if @task
 

lib/async/http/protocol/http2/request.rb

@@ -184,7 +184,7 @@ def send_response(response)
 
 						if body = response.body and !self.head?
 							@stream.send_headers(nil, headers)
-							@stream.send_body(body)
+							@stream.send_body(body, response.trailers)
 						else
 							# Ensure the response body is closed if we are ending the stream:
 							response.close