Example chat client/server.

Author: ioquatix

chat/client.rb

@@ -0,0 +1,41 @@
+#!/usr/bin/env ruby
+
+require 'async/reactor'
+require 'async/io/stream'
+require 'async/http/url_endpoint'
+require 'async/websocket/client'
+
+URL = ARGV.pop
+USER = ARGV.pop
+
+Async::Reactor.run do |task|
+	endpoint = Async::HTTP::URLEndpoint.parse(URL)
+	
+	endpoint.connect do |socket|
+		connection = Async::WebSocket::Client.new(socket, URL)
+		
+		connection.send_message({
+			user: USER,
+			status: "connected",
+		})
+		
+		task.async do
+			stdin = Async::IO::Stream.new(
+				Async::IO::Generic.new($stdin)
+			)
+			
+			puts "Waiting for input..."
+			while line = stdin.read_until("\n")
+				puts "Sending text: #{line}"
+				connection.send_message({
+					user: USER,
+					text: line,
+				})
+			end
+		end
+		
+		while message = connection.next_message
+			puts "Message from server: #{message.inspect}"
+		end
+	end
+end

chat/config.ru

@@ -0,0 +1,21 @@
+#!/usr/bin/env falcon serve --concurrency 1 -c
+
+require 'async/websocket/server'
+
+Message = Struct.new(:user, :contents, :created_at)
+
+$connections = []
+
+run lambda {|env|
+	Async::WebSocket::Server.open(env) do |connection|
+		$connections << connection
+		
+		while message = connection.next_message
+			$connections.each do |connection|
+				connection.send_message(message)
+			end
+		end
+	end
+	
+	[200, {}, ["Hello World"]]
+}

lib/async/websocket/client.rb

@@ -18,71 +18,16 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 # THE SOFTWARE.
 
-require 'websocket/driver'
-require 'json'
+require_relative 'connection'
 
 module Async
 	module WebSocket
 		# This is a basic synchronous websocket client:
-		class Client
-			EVENTS = [:open, :message, :close]
-			
-			def initialize(socket, url: "ws://.")
-				@socket = socket
+		class Client < Connection
+			def initialize(socket, url = "ws://.")
 				@url = url
 
-				@driver = ::WebSocket::Driver.client(self)
-				
-				@queue = []
-				
-				@driver.on(:error) do |error|
-					raise error
-				end
-				
-				EVENTS.each do |event|
-					@driver.on(event) do |data|
-						@queue.push(data)
-					end
-				end
-				
-				@driver.start
-			end
-			
-			attr :driver
-			attr :url
-			
-			def next_event
-				while @queue.empty?
-					data = @socket.read(1024)
-					
-					if data and !data.empty?
-						@driver.parse(data)
-					else
-						return nil
-					end
-				end
-				
-				@queue.shift
-			rescue EOFError
-				return nil
-			end
-			
-			def next_message
-				while event = next_event
-					if event.is_a? ::WebSocket::Driver::MessageEvent
-						return JSON.parse(event.data)
-					elsif event.is_a? ::WebSocket::Driver::CloseEvent
-						return nil
-					end
-				end
-			end
-			
-			def write(data)
-				@socket.write(data)
-			end
-			
-			def close
-				@driver.close
+				super socket, ::WebSocket::Driver.client(self)
 			end
 		end
 	end

lib/async/websocket/connection.rb

@@ -0,0 +1,91 @@
+# Copyright, 2015, by Samuel G. D. Williams. <http://www.codeotaku.com>
+# 
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+# 
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+require 'websocket/driver'
+require 'json'
+
+module Async
+	module WebSocket
+		# This is a basic synchronous websocket client:
+		class Connection
+			EVENTS = [:open, :message, :close]
+			
+			def initialize(socket, driver)
+				@socket = socket
+				@driver = driver
+				
+				@queue = []
+				
+				@driver.on(:error) do |error|
+					raise error
+				end
+				
+				EVENTS.each do |event|
+					@driver.on(event) do |data|
+						@queue.push(data)
+					end
+				end
+				
+				@driver.start
+			end
+			
+			attr :driver
+			attr :url
+			
+			def next_event
+				while @queue.empty?
+					data = @socket.read(1024)
+					
+					if data and !data.empty?
+						@driver.parse(data)
+					else
+						return nil
+					end
+				end
+				
+				@queue.shift
+			rescue EOFError
+				return nil
+			end
+			
+			def next_message
+				while event = next_event
+					if event.is_a? ::WebSocket::Driver::MessageEvent
+						return JSON.parse(event.data)
+					elsif event.is_a? ::WebSocket::Driver::CloseEvent
+						return nil
+					end
+				end
+			end
+			
+			def send_message(message)
+				@driver.text(JSON.dump(message))
+			end
+			
+			def write(data)
+				@socket.write(data)
+			end
+			
+			def close
+				@driver.close
+			end
+		end
+	end
+end

lib/async/websocket/server.rb

@@ -0,0 +1,57 @@
+# Copyright, 2015, by Samuel G. D. Williams. <http://www.codeotaku.com>
+# 
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+# 
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+require_relative 'connection'
+
+module Async
+	module WebSocket
+		class Server < Connection
+			def initialize(env, socket)
+				scheme = env['rack.url_scheme'] == 'https' ? 'wss' : 'ws'
+				@url = "#{scheme}://#{env['HTTP_HOST']}#{env['REQUEST_URI']}"
+				
+				@env = env
+				
+				super socket, ::WebSocket::Driver.rack(self)
+			end
+			
+			attr :env
+			attr :url
+			
+			def self.open(env)
+				if ::WebSocket::Driver.websocket?(env)
+					env['rack.hijack'].call
+					
+					connection = self.new(env, env['rack.hijack_io'])
+					
+					if block_given?
+						begin
+							yield connection
+						ensure
+							connection.close
+						end
+					else
+						return connection
+					end
+				end
+			end
+		end
+	end
+end

lib/async/websocket/version.rb

@@ -20,6 +20,6 @@
 
 module Async
 	module WebSocket
-		VERSION = "0.2.0"
+		VERSION = "0.3.0"
 	end
 end