rafaelgaspar-sexyprotocol 0.1.1

data/lib/sexyprotocol.rb

@@ -0,0 +1,6 @@
+module SexyProtocol
+  autoload :NetworkController, "sexyprotocol/network_controller"
+  
+  autoload :Server, "sexyprotocol/server"
+  autoload :Client, "sexyprotocol/client"  
+end

data/lib/sexyprotocol/client.rb

@@ -0,0 +1,37 @@
+module SexyProtocol
+  class Client
+    attr_reader :socket
+    attr_reader :network_controller
+  
+    class << self; attr_accessor :network_controller_class; end
+    self.network_controller_class = SexyProtocol::NetworkController
+  
+    def self.use_network_controller network_controller_class
+      self.network_controller_class = eval(network_controller_class.to_s)
+    end
+  
+    def self.connect host, port, *args
+      new(host, port).connect(*args)
+    end
+
+    def initialize host, port
+     @host = host
+     @port = port
+    end
+
+    def connect *args
+     @socket = TCPSocket.open(@host, @port)   
+     @network_controller = self.class.network_controller_class.new(@socket, *args).start
+ 
+     return self
+    end
+    
+    def close
+      @network_controller.close
+    end
+  
+    def closed?
+      @network_controller.closed?
+    end
+  end
+end

data/lib/sexyprotocol/network_controller.rb

@@ -0,0 +1,53 @@
+require "socket"
+
+module SexyProtocol
+  class NetworkController
+    attr_reader :socket
+  
+    def initialize socket
+      @socket = socket
+    end
+  
+    def start
+      @thread = Thread.new do
+        connected if respond_to?(:connected)
+        if respond_to?(:received_data)
+          until @socket.closed?
+            new_data = receive_data
+            received_data(new_data) unless new_data.nil?
+          end 
+          disconnected if respond_to?(:disconnected)     
+        end        
+      end
+    
+      return self
+    end
+    
+    def received_message message
+      puts message
+    end
+  
+    def close
+      @thread.kill unless @thread.nil?
+      @socket.close unless @socket.closed?
+      disconnected if respond_to?(:disconnected)
+    end
+  
+    def closed?
+      @socket.closed?
+    end
+    
+    def send_data data
+      @socket.write([data.length].pack("I") + data)      
+    end
+    
+    def receive_data
+      length_raw_data = @socket.read(4)
+      if length_raw_data.nil?
+        return nil
+      else         
+        return @socket.read(length_raw_data.unpack("I")[0])
+      end
+    end
+  end
+end

data/lib/sexyprotocol/server.rb

@@ -0,0 +1,54 @@
+module SexyProtocol
+  class Server
+    attr_reader :socket
+    attr_reader :network_controllers
+  
+    class << self; attr_accessor :network_controller_class; end
+    self.network_controller_class = SexyProtocol::NetworkController
+  
+    def self.use_network_controller network_controller_class
+      self.network_controller_class = eval(network_controller_class.to_s)
+    end
+    
+    def self.listen host, port, *args
+      new(host, port).listen(*args)
+    end
+  
+    def initialize host, port
+      @host = host
+      @port = port
+      @network_controllers = []
+    end
+  
+    def listen *args
+      @socket = TCPServer.open(@host, @port)
+    
+      @thread = Thread.new do
+        while true        
+          @network_controllers << self.class.network_controller_class.new(@socket.accept, *args).start
+          collect_garbaged_controllers
+        end
+      end
+    
+      return self
+    end
+  
+    def close
+      @thread.kill
+      @network_controllers.each do |controller|
+        controller.close
+      end
+      collect_garbaged_controllers
+      @socket.close
+    end
+  
+    def closed?
+      @socket.closed?
+    end
+  
+    private
+      def collect_garbaged_controllers
+        @network_controllers.delete_if { |controller| controller.closed? }
+      end
+  end
+end

data/spec/sexyprotocol/client_spec.rb

@@ -0,0 +1,25 @@
+dir = File.dirname(__FILE__)
+lib_path = File.expand_path("#{dir}/../../lib")
+$LOAD_PATH.unshift lib_path unless $LOAD_PATH.include?(lib_path)
+require "sexyprotocol"
+
+describe SexyProtocol::Client do
+  before(:all) do
+    @server = TCPServer.open("0.0.0.0", 8000)    
+    @client = SexyProtocol::Client.connect("localhost", 8000)
+    @server.accept
+  end
+  
+  after(:all) do
+    @server.close
+  end
+  
+  it "should be able to connect at a given host and port" do    
+    @client.socket.should_not be_closed
+  end
+  
+  it "should be able to disconnect" do
+    @client.close
+    @client.socket.should be_closed    
+  end
+end

data/spec/sexyprotocol/network_controller_spec.rb

@@ -0,0 +1,71 @@
+dir = File.dirname(__FILE__)
+lib_path = File.expand_path("#{dir}/../../lib")
+$LOAD_PATH.unshift lib_path unless $LOAD_PATH.include?(lib_path)
+require "sexyprotocol"
+
+describe SexyProtocol::NetworkController do  
+  before(:all) do
+    @server = TCPServer.open("0.0.0.0", 8000)
+    @client = SexyProtocol::NetworkController.new(TCPSocket.open("localhost", 8000))
+    @session = @server.accept
+  end
+  
+  after(:all) do
+    @server.close
+  end
+  
+  it "should be able to get the socket" do
+    @client.socket.should be_instance_of(TCPSocket)
+  end
+  
+  it "should be able to send a message" do
+    @client.send_data("Hello World!")
+    @session.read(16).should ==([12].pack("I") + "Hello World!")
+  end
+  
+  it "should be able to receive a message" do
+    @session.write([12].pack("I") + "Hello World!")
+    @client.receive_data.should == "Hello World!"
+  end
+  
+end
+
+describe SexyProtocol do
+  before do
+    @server = TCPServer.open("0.0.0.0", 8000)
+    @client = SexyProtocol::NetworkController.new(TCPSocket.open("localhost", 8000))
+    @session = @server.accept
+  end
+  
+  after do
+    @server.close
+  end
+  
+  it "should be sent connected message when it connects" do
+    @client.stub!(:connected)
+    @client.should_receive(:connected)  
+    
+    @client.start
+    sleep 0.01
+  end  
+    
+  it "should be sent received_message when message is received" do
+    @client.stub!(:received_data)    
+    @client.should_receive(:received_data).with("Hello World!")
+    @session.write([12].pack("I") + "Hello World!")
+    
+    @client.start
+    sleep 0.01
+  end
+  
+  it "should be sent disconnected message when it disconnects" do
+    @client.stub!(:disconnected)
+    @client.should_receive(:disconnected)
+    
+    @client.start
+    sleep 0.01
+    
+    @client.close
+    sleep 0.01
+  end
+end

data/spec/sexyprotocol/server_spec.rb

@@ -0,0 +1,24 @@
+dir = File.dirname(__FILE__)
+lib_path = File.expand_path("#{dir}/../../lib")
+$LOAD_PATH.unshift lib_path unless $LOAD_PATH.include?(lib_path)
+require "sexyprotocol"
+
+describe SexyProtocol::Server do
+  it "should be able to receive connections at a given host and port" do
+    server = SexyProtocol::Server.listen("0.0.0.0", 8000)
+    client = TCPSocket.open("localhost", 8000)
+    sleep 0.1
+    server.network_controllers[0].should be_instance_of(SexyProtocol::NetworkController)
+    server.network_controllers[0].should_not be_closed
+  
+    server.close
+  end
+end
+
+describe SexyProtocol do  
+  it "should be able to get the socket" do
+    server = SexyProtocol::Server.listen("0.0.0.0", 8000)    
+    server.socket.should be_instance_of(TCPServer)
+    server.close
+  end
+end

metadata

@@ -0,0 +1,60 @@
+--- !ruby/object:Gem::Specification 
+name: rafaelgaspar-sexyprotocol
+version: !ruby/object:Gem::Version 
+  version: 0.1.1
+platform: ruby
+authors: 
+- Rafael Gaspar
+- Fernando Cezar
+autorequire: sexyprotocol
+bindir: bin
+cert_chain: []
+
+date: 2009-05-16 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: 
+email: rafael.gaspar@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/sexyprotocol.rb
+- lib/sexyprotocol
+- lib/sexyprotocol/client.rb
+- lib/sexyprotocol/server.rb
+- lib/sexyprotocol/network_controller.rb
+has_rdoc: false
+homepage: http://www.github.com/rafaelgaspar/sexyprotocol
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: An Event-Driven Network Framework for Rapid Network Client and Server Protocol Development.
+test_files: 
+- spec/sexyprotocol/client_spec.rb
+- spec/sexyprotocol/server_spec.rb
+- spec/sexyprotocol/network_controller_spec.rb