rubame 0.0.1

data/lib/rubame.rb

@@ -0,0 +1,177 @@
+require 'websocket'
+require 'socket'
+
+module Rubame
+  class Server
+    def initialize(host, port)
+      Socket.do_not_reverse_lookup
+      @hostname = host
+      @port = port
+
+      @reading = []
+      @writing = []
+
+      @clients = {} # Socket as key, and Client as value
+
+      @socket = TCPServer.new(@hostname, @port)
+      @reading.push @socket
+    end
+
+    def accept
+      socket = @socket.accept_nonblock
+      @reading.push socket
+      handshake = WebSocket::Handshake::Server.new
+      client = Rubame::Client.new(socket, handshake, self)
+      
+      while line = socket.gets
+        client.handshake << line
+        break if client.handshake.finished?
+      end
+      if client.handshake.valid?
+        @clients[socket] = client
+        client.write handshake.to_s
+        client.opened = true
+        return client
+      else
+        close(client)
+      end
+      return nil
+    end
+
+    def read(client)
+
+      pairs = client.socket.recvfrom(2000)
+      messages = []
+
+      if pairs[0].length == 0
+        close(client)
+      else
+        client.frame << pairs[0]
+
+        while f = client.frame.next
+          if (f.type == :close)
+            close(client)
+            return messages
+          else
+            messages.push f
+          end
+        end
+        
+      end
+
+      return messages
+
+    end
+
+    def close(client)
+      puts "Closing socket"
+      @reading.delete client.socket
+      @clients.delete client.socket
+      begin
+        client.socket.close
+      rescue
+      end
+      client.closed = true
+    end
+
+    def run(&blk)
+      readable, writable = IO.select(@reading, @writing, nil, 0)
+
+      if readable
+        readable.each do |socket|
+          client = @clients[socket]
+          if socket == @socket
+            client = accept
+          else
+            msg = read(client)
+            client.messaged = msg
+          end
+
+          blk.call(client) if client and blk
+        end
+      end
+    end
+
+    def stop
+      @socket.close
+    end
+  end
+
+  class Client
+    attr_accessor :socket, :handshake, :frame, :opened, :messaged, :closed
+
+    def initialize(socket, handshake, server)
+      @socket = socket
+      @handshake = handshake
+      @frame = WebSocket::Frame::Incoming::Server.new(:version => @handshake.version)
+      @opened = false
+      @messaged = []
+      @closed = false
+      @server = server
+    end
+
+    def write(data)
+      @socket.write data
+    end
+
+    def send(data)
+      frame = WebSocket::Frame::Outgoing::Server.new(:version => @handshake.version, :data => data, :type => :text)
+      begin
+        @socket.write frame
+        @socket.flush
+      rescue
+        @server.close(self) unless @closed
+      end
+    end
+
+    def onopen(&blk)
+      if @opened
+        begin
+          blk.call
+        ensure
+          @opened = false
+        end
+      end
+    end
+
+    def onmessage(&blk)
+      if @messaged.size > 0
+        begin
+          @messaged.each do |x|
+            blk.call(x.to_s)
+          end
+        ensure
+          @messaged = []
+        end
+      end
+    end
+
+    def onclose(&blk)
+      if @closed
+        begin
+          blk.call
+        ensure
+        end
+      end
+    end
+  end
+
+  line = 0
+end
+
+if __FILE__==$0
+  server = Rubame::Server.new("0.0.0.0", 25222)
+  while (!$quit)
+    server.run do |client|
+      client.onopen do
+        puts "Server reports:  client open"
+      end
+      client.onmessage do |mess|
+        puts "Server reports:  message received: #{mess}"
+      end
+      client.onclose do
+        puts "Server reports:  client closed"
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,45 @@
+--- !ruby/object:Gem::Specification
+name: rubame
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Mark Saward
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-01-06 00:00:00.000000000 Z
+dependencies: []
+description: Ruby Websocket Game Server
+email: me@marksaward.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/rubame.rb
+homepage: http://github.com/saward/Rubame
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Rubame
+test_files: []