async-websocket 0.13.1 → 0.18.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 237b59048febd538e6112ea06b520c154386d057bd3b58a22bb153d45cd3209f
-  data.tar.gz: '0966a1666e31463adb1c2975f7e72534c7a8003794825ab94c24c3d5d02223e7'
+  metadata.gz: 54e97832ebcebab09dfd5a042aad8532a7a5277cf6609fecae9f73a1826cac7d
+  data.tar.gz: 671da6e463ced51447b8cb49aabbb8b51d644a19adc655e9f955fe4b6772afd4
 SHA512:
-  metadata.gz: 7e25c38a3d25b98209d3999e803f031cb216dd1ccd47f1c2af92fd75cab7859d888ee9fd79873aab8d41d9d6e1ff4bd3bf64addcba74bad619613c948ec17e89
-  data.tar.gz: a7284a36f8525415327177da55e83d35001f329d50b4b1843e4cd0b4d2488b17da26559df0248e493bea86ac163e0f3434113a6b76e4d1c9a16978e3c4118a41
+  metadata.gz: c33efe4ce32ddf75596defe2b3c3f56bb05973d1c13e4b9c39e9fbaed604c9774271b6dd88db11691ba1a498b3c3ee807e6aa3a54447a30ea3ba791d8818388b
+  data.tar.gz: 50b6419551ee9643771cb80dbd7ad48460e265f3e8a168e60f4afa9557802eb4bc2543b5cb678e55b81ccf006679bfa865047e590f5f7b99526f0dd628bc3005

data/lib/async/websocket/adapters/rack.rb

@@ -35,20 +35,26 @@ module Async
 				
 				def self.open(env, headers: [], protocols: [], handler: Connection, **options, &block)
 					if request = env['async.http.request'] and Array(request.protocol).include?(PROTOCOL)
+						env = nil
+						
 						# Select websocket sub-protocol:
 						if requested_protocol = request.headers[SEC_WEBSOCKET_PROTOCOL]
 							protocol = (requested_protocol & protocols).first
 						end
 						
 						response = Response.for(request, headers, protocol: protocol, **options) do |stream|
+							response = nil
+							
 							framer = Protocol::WebSocket::Framer.new(stream)
 							
 							connection = handler.call(framer, protocol)
+							
 							yield connection
 							
 							connection.close unless connection.closed?
 						end
 						
+						request = nil
 						headers = response.headers
 						
 						if protocol = response.protocol
@@ -57,7 +63,7 @@ module Async
 							])
 						end
 						
-						return [response.status, headers, response.body]
+						return [response.status, headers.to_h, response.body]
 					end
 				end
 			end

data/{spec/async/websocket/upgrade.rb → lib/async/websocket/adapters/rails.rb}

@@ -1,3 +1,5 @@
+# frozen_string_literals: true
+#
 # Copyright, 2019, by Samuel G. D. Williams. <http://www.codeotaku.com>
 # 
 # Permission is hereby granted, free of charge, to any person obtaining a copy
@@ -18,26 +20,25 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 # THE SOFTWARE.
 
-require 'async/websocket/adapters/rack'
+require_relative 'rack'
 
-class Upgrade
-	def initialize(app)
-		@app = app
-	end
-	
-	def call(env)
-		Async::WebSocket::Adapters::Rack.open(env, supported_protocols: ['ws']) do |connection|
-			read, write = IO.pipe
-			
-			Process.spawn("ls -lah", :out => write)
-			write.close
-			
-			read.each_line do |line|
-				connection.write({line: line})
+module Async
+	module WebSocket
+		module Adapters
+			module Rails
+				def self.open(request, **options, &block)
+					if response = Rack.open(request.env, **options, &block)
+						response[1]['rack.hijack'] = lambda do |stream|
+							response[2].call(stream)
+						end
+						
+						# Close the response to prevent Rails from... trying to render a view?
+						return ::ActionDispatch::Response.new(response[0], response[1], nil).tap(&:close)
+					end
+					
+					return ::ActionDispatch::Response.new(404, [], [])
+				end
 			end
-			
-			# Gracefully close the connection:
-			connection.close
-		end or @app.call(env)
+		end
 	end
 end

data/lib/async/websocket/client.rb

@@ -34,8 +34,9 @@ module Async
 		class Client < ::Protocol::HTTP::Middleware
 			include ::Protocol::WebSocket::Headers
 			
-			def self.open(endpoint, *args, &block)
-				client = self.new(HTTP::Client.new(endpoint, *args), mask: endpoint.secure?)
+			# @return [Client] a client which can be used to establish websocket connections to the given endpoint.
+			def self.open(endpoint, **options, &block)
+				client = self.new(HTTP::Client.new(endpoint, **options), mask: true)
 				
 				return client unless block_given?
 				
@@ -46,9 +47,10 @@ module Async
 				end
 			end
 			
+			# @return [Connection] an open websocket connection to the given endpoint.
 			def self.connect(endpoint, *args, **options, &block)
 				self.open(endpoint, *args) do |client|
-					connection = client.connect(endpoint.path, **options)
+					connection = client.connect(endpoint.authority, endpoint.path, **options)
 					
 					return connection unless block_given?
 					
@@ -60,33 +62,53 @@ module Async
 				end
 			end
 			
-			def initialize(delegate, **options)
-				super(delegate)
+			def initialize(client, **options)
+				super(client)
 				
 				@options = options
 			end
 			
-			def connect(path, headers: [], handler: Connection, **options)
-				request = Request.new(nil, nil, path, headers, **options)
+			class Framer < ::Protocol::WebSocket::Framer
+				def initialize(pool, connection, stream)
+					super(stream)
+					@pool = pool
+					@connection = connection
+				end
+				
+				def close
+					super
+					
+					if @pool
+						@pool.release(@connection)
+						@pool = nil
+						@connection = nil
+					end
+				end
+			end
+			
+			def connect(authority, path, headers: nil, handler: Connection, **options, &block)
+				headers = ::Protocol::HTTP::Headers[headers]
+				request = Request.new(nil, authority, path, headers, **options)
+				
+				pool = @delegate.pool
+				connection = pool.acquire
 				
-				response = self.call(request)
+				response = request.call(connection)
 				
 				unless response.stream?
 					raise ProtocolError, "Failed to negotiate connection: #{response.status}"
 				end
 				
 				protocol = response.headers[SEC_WEBSOCKET_PROTOCOL]&.first
-				framer = Protocol::WebSocket::Framer.new(response.stream)
-				
-				connection = handler.call(framer, protocol, **@options)
+				stream = response.stream
+				response = nil
 				
-				return connection unless block_given?
+				framer = Framer.new(pool, connection, stream)
+				connection = nil
 				
-				begin
-					yield connection
-				ensure
-					connection.close
-				end
+				handler.call(framer, protocol, **@options, &block)
+			ensure
+				pool.release(connection) if connection
 			end
 		end
 	end

data/lib/async/websocket/connect_request.rb

@@ -34,17 +34,19 @@ module Async
 			include ::Protocol::WebSocket::Headers
 			
 			class Wrapper
-				def initialize(output, response)
-					@output = output
-					@body = output
+				def initialize(stream, response)
 					@response = response
-					@stream = nil
+					@body = @response.body
+					@stream = stream
 				end
 				
+				attr_accessor :response
+				
 				attr_accessor :body
+				attr_accessor :stream
 				
 				def stream?
-					@response.success?
+					@response.success? and @stream
 				end
 				
 				def status
@@ -62,30 +64,43 @@ module Async
 				def protocol
 					@response.protocol
 				end
+			end
+			
+			class Hijack < Async::HTTP::Body::Readable
+				def initialize(request)
+					@request = request
+					@stream = nil
+				end
 				
-				def stream
-					@stream ||= Async::HTTP::Body::Stream.new(@response.body, @output)
+				def stream?
+					true
+				end
+				
+				attr :stream
+				
+				def call(stream)
+					@stream = stream
 				end
 			end
 			
-			def initialize(request, protocols: [], version: 13)
-				body = Async::HTTP::Body::Writable.new
+			def initialize(request, protocols: [], version: 13, &block)
+				body = Hijack.new(self)
 				
-				headers = []
+				headers = ::Protocol::HTTP::Headers[request.headers]
 				
-				headers << [SEC_WEBSOCKET_VERSION, version]
+				headers.add(SEC_WEBSOCKET_VERSION, String(version))
 				
 				if protocols.any?
-					headers << [SEC_WEBSOCKET_PROTOCOL, protocols.join(',')]
+					headers.add(SEC_WEBSOCKET_PROTOCOL, protocols.join(','))
 				end
 				
-				merged_headers = ::Protocol::HTTP::Headers::Merged.new(request.headers, headers)
-				
-				super(request.scheme, request.authority, ::Protocol::HTTP::Methods::CONNECT, request.path, nil, merged_headers, body, PROTOCOL)
+				super(request.scheme, request.authority, ::Protocol::HTTP::Methods::CONNECT, request.path, nil, headers, body, PROTOCOL)
 			end
 			
 			def call(connection)
-				Wrapper.new(body, super)
+				response = super
+				
+				Wrapper.new(@body.stream, response)
 			end
 		end
 	end

data/lib/async/websocket/connect_response.rb

@@ -30,10 +30,10 @@ module Async
 			include ::Protocol::WebSocket::Headers
 			
 			def initialize(request, headers = nil, protocol: nil, &block)
-				headers = Protocol::HTTP::Headers::Merged.new(headers)
+				headers = ::Protocol::HTTP::Headers[headers]
 				
 				if protocol
-					headers << [[SEC_WEBSOCKET_PROTOCOL, protocol]]
+					headers.add(SEC_WEBSOCKET_PROTOCOL, protocol)
 				end
 				
 				body = Async::HTTP::Body::Hijack.wrap(request, &block)

data/lib/async/websocket/connection.rb

@@ -32,12 +32,31 @@ module Async
 			include ::Protocol::WebSocket::Headers
 			
 			def self.call(framer, protocol = [], **options)
-				return self.new(framer, Array(protocol).first, **options)
+				instance = self.new(framer, Array(protocol).first, **options)
+				
+				return instance unless block_given?
+				
+				begin
+					yield instance
+				ensure
+					instance.close
+				end
 			end
 			
-			def initialize(framer, protocol = nil, **options)
+			def initialize(framer, protocol = nil, response: nil, **options)
 				super(framer, **options)
+				
 				@protocol = protocol
+				@response = response
+			end
+			
+			def close
+				super
+				
+				if @response
+					@response.finish
+					@response = nil
+				end
 			end
 			
 			attr :protocol

data/lib/async/websocket/request.rb

@@ -30,7 +30,7 @@ module Async
 				Array(request.protocol).include?(PROTOCOL)
 			end
 			
-			def initialize(scheme = nil, authority = nil, path = nil, headers = [], **options)
+			def initialize(scheme = nil, authority = nil, path = nil, headers = nil, **options, &block)
 				@scheme = scheme
 				@authority = authority
 				@path = path

data/lib/async/websocket/response.rb

@@ -27,7 +27,7 @@ module Async
 	module WebSocket
 		module Response
 			# Send the request to the given connection.
-			def self.for(request, headers = [], **options, &body)
+			def self.for(request, headers = nil, **options, &body)
 				if request.version =~ /http\/1/i
 					return UpgradeResponse.new(request, headers, **options, &body)
 				elsif request.version =~ /h2/i

data/lib/async/websocket/server.rb

@@ -49,6 +49,8 @@ module Async
 					# Select websocket sub-protocol:
 					protocol = select_protocol(request)
 					
+					# request.headers = nil
+					
 					Response.for(request, headers, protocol: protocol, **options) do |stream|
 						framer = Protocol::WebSocket::Framer.new(stream)
 						

data/lib/async/websocket/upgrade_request.rb

@@ -42,6 +42,8 @@ module Async
 					@stream = nil
 				end
 				
+				attr_accessor :response
+				
 				def stream?
 					@response.status == 101
 				end
@@ -71,21 +73,19 @@ module Async
 				end
 			end
 			
-			def initialize(request, protocols: [], version: 13)
+			def initialize(request, protocols: [], version: 13, &block)
 				@key = Nounce.generate_key
 				
-				headers = [
-					[SEC_WEBSOCKET_KEY, @key],
-					[SEC_WEBSOCKET_VERSION, version],
-				]
+				headers = ::Protocol::HTTP::Headers[request.headers]
+				
+				headers.add(SEC_WEBSOCKET_KEY, @key)
+				headers.add(SEC_WEBSOCKET_VERSION, String(version))
 				
 				if protocols.any?
-					headers << [SEC_WEBSOCKET_PROTOCOL, protocols.join(',')]
+					headers.add(SEC_WEBSOCKET_PROTOCOL, protocols.join(','))
 				end
 				
-				merged_headers = ::Protocol::HTTP::Headers::Merged.new(request.headers, headers)
-				
-				super(request.scheme, request.authority, ::Protocol::HTTP::Methods::GET, request.path, nil, merged_headers, nil, PROTOCOL)
+				super(request.scheme, request.authority, ::Protocol::HTTP::Methods::GET, request.path, nil, headers, nil, PROTOCOL)
 			end
 			
 			def call(connection)

data/lib/async/websocket/upgrade_response.rb

@@ -31,17 +31,17 @@ module Async
 			include ::Protocol::WebSocket::Headers
 			
 			def initialize(request, headers = nil, protocol: nil, &block)
-				headers = Protocol::HTTP::Headers::Merged.new(headers)
+				headers = ::Protocol::HTTP::Headers[headers]
 				
 				if accept_nounce = request.headers[SEC_WEBSOCKET_KEY]&.first
-					headers << [[SEC_WEBSOCKET_ACCEPT, Nounce.accept_digest(accept_nounce)]]
+					headers.add(SEC_WEBSOCKET_ACCEPT, Nounce.accept_digest(accept_nounce))
 					status = 101
 				else
 					status = 400
 				end
 				
 				if protocol
-					headers << [[SEC_WEBSOCKET_PROTOCOL, protocol]]
+					headers.add(SEC_WEBSOCKET_PROTOCOL, protocol)
 				end
 				
 				body = Async::HTTP::Body::Hijack.wrap(request, &block)

data/lib/async/websocket/version.rb

@@ -20,6 +20,6 @@
 
 module Async
 	module WebSocket
-		VERSION = "0.13.1"
+		VERSION = "0.18.0"
 	end
 end

metadata

@@ -1,43 +1,43 @@
 --- !ruby/object:Gem::Specification
 name: async-websocket
 version: !ruby/object:Gem::Version
-  version: 0.13.1
+  version: 0.18.0
 platform: ruby
 authors:
 - Samuel Williams
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2019-07-04 00:00:00.000000000 Z
+date: 2021-04-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: async-io
+  name: async-http
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.23'
+        version: '0.54'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.23'
+        version: '0.54'
 - !ruby/object:Gem::Dependency
-  name: async-http
+  name: async-io
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.43'
+        version: '1.23'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.43'
+        version: '1.23'
 - !ruby/object:Gem::Dependency
   name: protocol-websocket
   requirement: !ruby/object:Gem::Requirement
@@ -67,21 +67,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: falcon
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '0.32'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: '0.32'
-- !ruby/object:Gem::Dependency
-  name: covered
+  name: bundler
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -95,7 +81,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: bundler
+  name: covered
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -109,21 +95,21 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: rspec
+  name: falcon
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.6'
+        version: '0.34'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.6'
+        version: '0.34'
 - !ruby/object:Gem::Dependency
-  name: rake
+  name: rack-test
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -136,64 +122,29 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: 
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.6'
+description:
 email:
-- samuel.williams@oriontransfer.co.nz
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".editorconfig"
-- ".gitignore"
-- ".rspec"
-- ".travis.yml"
-- Gemfile
-- README.md
-- Rakefile
-- async-websocket.gemspec
-- examples/chat/README.md
-- examples/chat/client.rb
-- examples/chat/config.ru
-- examples/chat/multi-client.rb
-- examples/mud/client.rb
-- examples/mud/config.ru
-- examples/rack/client.rb
-- examples/rack/config.ru
-- examples/utopia/.bowerrc
-- examples/utopia/.gitignore
-- examples/utopia/.rspec
-- examples/utopia/Gemfile
-- examples/utopia/Guardfile
-- examples/utopia/README.md
-- examples/utopia/Rakefile
-- examples/utopia/config.ru
-- examples/utopia/config/README.md
-- examples/utopia/config/environment.rb
-- examples/utopia/lib/readme.txt
-- examples/utopia/pages/_heading.xnode
-- examples/utopia/pages/_page.xnode
-- examples/utopia/pages/client/client.js
-- examples/utopia/pages/client/index.xnode
-- examples/utopia/pages/errors/exception.xnode
-- examples/utopia/pages/errors/file-not-found.xnode
-- examples/utopia/pages/links.yaml
-- examples/utopia/pages/server/controller.rb
-- examples/utopia/public/_static/icon.png
-- examples/utopia/public/_static/site.css
-- examples/utopia/public/_static/utopia-background.svg
-- examples/utopia/public/_static/utopia.svg
-- examples/utopia/public/readme.txt
-- examples/utopia/spec/spec_helper.rb
-- examples/utopia/spec/website_context.rb
-- examples/utopia/spec/website_spec.rb
-- examples/utopia/tasks/bower.rake
-- examples/utopia/tasks/deploy.rake
-- examples/utopia/tasks/development.rake
-- examples/utopia/tasks/environment.rake
-- examples/utopia/tasks/log.rake
-- examples/utopia/tasks/static.rake
 - lib/async/websocket.rb
 - lib/async/websocket/adapters/rack.rb
+- lib/async/websocket/adapters/rails.rb
 - lib/async/websocket/client.rb
 - lib/async/websocket/connect_request.rb
 - lib/async/websocket/connect_response.rb
@@ -206,20 +157,11 @@ files:
 - lib/async/websocket/upgrade_request.rb
 - lib/async/websocket/upgrade_response.rb
 - lib/async/websocket/version.rb
-- spec/async/websocket/adapters/rack/client.rb
-- spec/async/websocket/adapters/rack/config.ru
-- spec/async/websocket/adapters/rack_spec.rb
-- spec/async/websocket/client_spec.rb
-- spec/async/websocket/connection_spec.rb
-- spec/async/websocket/server_examples.rb
-- spec/async/websocket/server_spec.rb
-- spec/async/websocket/upgrade.rb
-- spec/spec_helper.rb
-homepage: ''
+homepage: https://github.com/socketry/async-websocket
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -234,17 +176,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
-signing_key: 
+rubygems_version: 3.1.2
+signing_key:
 specification_version: 4
 summary: An async websocket library on top of websocket-driver.
-test_files:
-- spec/async/websocket/adapters/rack/client.rb
-- spec/async/websocket/adapters/rack/config.ru
-- spec/async/websocket/adapters/rack_spec.rb
-- spec/async/websocket/client_spec.rb
-- spec/async/websocket/connection_spec.rb
-- spec/async/websocket/server_examples.rb
-- spec/async/websocket/server_spec.rb
-- spec/async/websocket/upgrade.rb
-- spec/spec_helper.rb
+test_files: []