Add support for callable bodies.

socketry · Apr 19, 2021 · 5b73e63 · 5b73e63
1 parent 3ec805b
commit 5b73e63
Show file tree

Hide file tree

Showing 5 changed files with 39 additions and 3 deletions.
diff --git a/.github/workflows/documentation.yml b/.github/workflows/documentation.yml
@@ -0,0 +1,32 @@
+name: Documentation
+
+on:
+  push:
+    branches:
+      - master
+
+jobs:
+  deploy:
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: actions/checkout@v2
+    - uses: ruby/setup-ruby@v1
+      env:
+        BUNDLE_WITH: maintenance
+      with:
+        ruby-version: 2.7
+        bundler-cache: true
+
+    - name: Installing packages
+      run: sudo apt-get install wget
+
+    - name: Generate documentation
+      timeout-minutes: 5
+      run: bundle exec bake utopia:project:static
+
+    - name: Deploy documentation
+      uses: JamesIves/github-pages-deploy-action@4.0.0
+      with:
+        branch: gh-pages
+        folder: docs
diff --git a/falcon.gemspec b/falcon.gemspec
@@ -19,7 +19,7 @@ Gem::Specification.new do |spec|
 
 	spec.add_dependency "async", "~> 1.13"
 	spec.add_dependency "async-container", "~> 0.16.0"
-	spec.add_dependency "async-http", "~> 0.54.0"
+	spec.add_dependency "async-http", "~> 0.55.0"
 	spec.add_dependency "async-http-cache", "~> 0.3.0"
 	spec.add_dependency "async-io", "~> 1.22"
 	spec.add_dependency "build-environment", "~> 1.13"

diff --git a/gems.rb b/gems.rb
@@ -4,6 +4,8 @@
 
 gemspec
 
+gem "rack", git: 'https://github.com/ioquatix/rack', branch: 'streaming'
+
 # gem "async-container", path: "../async-container"
 # gem "async-websocket", path: "../async-websocket"
 # gem "async-http", path: "../async-http"

diff --git a/lib/falcon/adapters/output.rb b/lib/falcon/adapters/output.rb
@@ -35,7 +35,7 @@ class Output < ::Protocol::HTTP::Body::Readable
 			# @parameter status [Integer] The response status.
 			# @parameter headers [Protocol::HTTP::Headers] The response headers.
 			# @parameter body [Object] The `rack` response body.
-			def self.wrap(status, headers, body)
+			def self.wrap(status, headers, body, request = nil)
 				# In no circumstance do we want this header propagating out:
 				if length = headers.delete(CONTENT_LENGTH)
 					# We don't really trust the user to provide the right length to the transport.
@@ -44,6 +44,8 @@ def self.wrap(status, headers, body)
 
 				if body.is_a?(::Protocol::HTTP::Body::Readable)
 					return body
+				elsif body.respond_to?(:call)
+					body = Async::HTTP::Body::Hijack.wrap(request, &body)
 				elsif status == 200 and body.respond_to?(:to_path)
 					# Don't mangle partial responsese (206)
 					return ::Protocol::HTTP::Body::File.open(body.to_path)

diff --git a/lib/falcon/adapters/response.rb b/lib/falcon/adapters/response.rb
@@ -81,7 +81,7 @@ def self.wrap(status, headers, body, request = nil)
 						Async.logger.warn("Ignoring protocol-level headers: #{ignored.inspect}")
 					end
 
-					body = Output.wrap(status, headers, body)
+					body = Output.wrap(status, headers, body, request)
 				end
 
 				if request&.head?