wine 0.1.0

data/README.md

@@ -0,0 +1,38 @@
+Wine for Ruby
+=============
+
+Wine is a protocol for configuring and monitoring agents. Wine for Ruby is an
+implementation of the protocol for Ruby agents, and there is [Wine for Java][]
+for the JVM as well.
+
+  [Wine for Java]: http://github.com/valotrading/wine-java
+
+
+Installation
+------------
+
+Install Wine for Ruby:
+
+    gem install wine
+
+
+Development
+-----------
+
+Install the dependencies:
+
+    bundle install
+
+Run the tests:
+
+    bundle exec rake
+
+Run the test server:
+
+    bundle exec bin/wine-test-server
+
+
+License
+-------
+
+Wine for Ruby is released under the Apache License, Version 2.0.

data/bin/wine-test-server

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+require 'wine'
+
+begin
+  Wine::TestServer.start(ARGV)
+rescue Interrupt
+  exit
+end

data/lib/wine.rb

@@ -0,0 +1,6 @@
+require 'wine/connection'
+require 'wine/error'
+require 'wine/message'
+require 'wine/session'
+require 'wine/test_server'
+require 'wine/version'

data/lib/wine/connection.rb

@@ -0,0 +1,45 @@
+require 'wine/error'
+require 'wine/message'
+
+module Wine
+  class Connection
+
+    DEFAULT_TIMEOUT = 5000
+
+    attr_reader :socket
+
+    def initialize(socket)
+      @socket = socket
+    end
+
+    def send(message)
+      message.write(@socket)
+    end
+
+    def recv(timeout = DEFAULT_TIMEOUT)
+      readable = IO.select([ @socket ], nil, nil, timeout / 1000.0)
+      return nil unless readable
+
+      msg_type = @socket.recv(1, Socket::MSG_PEEK)
+      raise ConnectionClosed unless msg_type.length == 1
+
+      Message.read(@socket)
+    end
+
+    def recv_nonblock
+      begin
+        msg_type = @socket.recv_nonblock(1, Socket::MSG_PEEK)
+        raise ConnectionClosed unless msg_type.length == 1
+
+        Message.read(@socket)
+      rescue IO::WaitReadable
+        nil
+      end
+    end
+
+    def close
+      @socket.close
+    end
+
+  end
+end

data/lib/wine/error.rb

@@ -0,0 +1,18 @@
+module Wine
+
+  class Error < StandardError
+  end
+
+  class ConnectionRefused < Error
+  end
+
+  class ConnectionClosed < Error
+  end
+
+  class ResponseTimeout < Error
+  end
+
+  class ProtocolError < Error
+  end
+
+end

data/lib/wine/message.rb

@@ -0,0 +1,71 @@
+require 'bindata'
+require 'wine/error'
+
+module Wine
+  module Message
+
+    LOGIN          = 'L'
+    LOGIN_ACCEPTED = 'A'
+    LOGIN_REJECTED = 'J'
+    GET            = 'G'
+    VALUE          = 'V'
+    SET            = 'S'
+
+    class Login < BinData::Record
+      string :msg_type, :value  => LOGIN
+      string :username, :length =>  8, :pad_byte => ' '
+      string :password, :length => 20, :pad_byte => ' '
+    end
+
+    class LoginAccepted < BinData::Record
+      string :msg_type, :value => LOGIN_ACCEPTED
+    end
+
+    class LoginRejected < BinData::Record
+      string :msg_type, :value => LOGIN_REJECTED
+    end
+
+    class Get < BinData::Record
+      string  :msg_type,   :value       => GET
+      int32be :key_length, :value       => lambda { key_data.length }
+      string  :key_data,   :read_length => :key_length
+    end
+
+    class Value < BinData::Record
+      string  :msg_type,     :value       => VALUE
+      int32be :key_length,   :value       => lambda { key_data.length }
+      int32be :value_length, :value       => lambda { value_data.length }
+      string  :key_data,     :read_length => :key_length
+      string  :value_data,   :read_length => :value_length
+    end
+
+    class Set < BinData::Record
+      string  :msg_type,     :value       => SET
+      int32be :key_length,   :value       => lambda { key_data.length }
+      int32be :value_length, :value       => lambda { value_data.length }
+      string  :key_data,     :read_length => :key_length
+      string  :value_data,   :read_length => :value_length
+    end
+
+    def self.read(io)
+      begin
+        type = Types[io.read(1)]
+        type.read(io)
+      rescue StandardError
+        raise ProtocolError
+      end
+    end
+
+    private
+
+    Types = {
+      LOGIN          => Login,
+      LOGIN_ACCEPTED => LoginAccepted,
+      LOGIN_REJECTED => LoginRejected,
+      GET            => Get,
+      VALUE          => Value,
+      SET            => Set
+    }
+
+  end
+end

data/lib/wine/session.rb

@@ -0,0 +1,66 @@
+require 'socket'
+require 'wine/connection'
+require 'wine/error'
+require 'wine/message'
+
+module Wine
+  class Session
+
+    def self.connect(host, port)
+      begin
+        socket     = TCPSocket.new(host, port)
+        connection = Connection.new(socket)
+
+        new(connection)
+      rescue Errno::ECONNREFUSED
+        raise ConnectionRefused
+      end
+    end
+
+    def initialize(connection)
+      @connection = connection
+    end
+
+    def login(username, password, timeout = Connection::DEFAULT_TIMEOUT)
+      request = Message::Login.new(:username => username, :password => password)
+      @connection.send(request)
+
+      response = @connection.recv(timeout)
+      raise ResponseTimeout unless response
+
+      case response.msg_type
+      when Message::LOGIN_ACCEPTED
+        true
+      when Message::LOGIN_REJECTED
+        false
+      else
+        raise ProtocolError
+      end
+    end
+
+    def get(key, timeout = Connection::DEFAULT_TIMEOUT)
+      request = Message::Get.new(:key_data => key)
+      @connection.send(request)
+
+      response = @connection.recv(timeout)
+      raise ResponseTimeout unless response
+
+      case response.msg_type
+      when Message::VALUE
+        response.value_data
+      else
+        raise ProtocolError
+      end
+    end
+
+    def set(key, value)
+      request = Message::Set.new(:key_data => key, :value_data => value)
+      @connection.send(request)
+    end
+
+    def close
+      @connection.close
+    end
+
+  end
+end

data/lib/wine/test_server.rb

@@ -0,0 +1,137 @@
+require 'logger'
+require 'socket'
+require 'wine/connection'
+require 'wine/error'
+require 'wine/message'
+
+module Wine
+  class TestServer
+
+    Timeout = 100.0 / 1000.0
+
+    def self.start(args)
+      usage unless args.length == 1
+
+      port = args[0].to_i
+      usage if port == 0
+
+      log = Logger.new(STDERR)
+
+      server = new(port, log)
+      server.run
+    end
+
+    def self.usage
+      abort "Usage: wine-test-server <port>"
+    end
+
+    def initialize(port, logger)
+      @port = port
+      @log  = logger
+
+      @stopped = false
+
+      @config = {}
+
+      @server_socket = nil
+      @connections   = {}
+
+      @log.progname = "wine-test-server"
+      @log.formatter = Proc.new do |severity, time, progname, msg|
+        "#{progname}: #{severity.downcase}: #{msg}\n"
+      end
+    end
+
+    def run
+      listen
+
+      @log.info("Listening on port #{@port}")
+
+      until @stopped do
+        readable, writable, erroneous = IO.select(sockets, [], [], Timeout)
+        next unless readable
+
+        readable.each do |socket|
+          if socket == @server_socket
+            accept
+          else
+            recv(@connections[socket])
+          end
+        end
+      end
+
+      sockets.each do |socket|
+        socket.close
+      end
+    end
+
+    def stop
+      @stopped = true
+    end
+
+    private
+
+    def listen
+      @server_socket = Socket.new(:INET, :SOCK_STREAM)
+      @server_socket.setsockopt(:SOL_SOCKET, :SO_REUSEADDR, true)
+      @server_socket.bind(Socket.sockaddr_in(@port, "127.0.0.1"))
+      @server_socket.listen(5)
+    end
+
+    def sockets
+      [ @server_socket ] + client_sockets
+    end
+
+    def client_sockets
+      @connections.keys
+    end
+
+    def accept
+      begin
+        client_socket, client_addrinfo = @server_socket.accept_nonblock
+
+        @connections[client_socket] = Connection.new(client_socket)
+      rescue IO::WaitReadable
+        # Do nothing.
+      end
+    end
+
+    def recv(connection)
+      begin
+        message = connection.recv_nonblock
+        if message
+          @log.info("Received #{message}")
+
+          handle(connection, message)
+        end
+      rescue ConnectionClosed
+        @log.info("Connection closed")
+
+        close(connection)
+      rescue Error
+        @log.warn("Closing connection")
+
+        close(connection)
+      end
+    end
+
+    def handle(connection, message)
+      case message.msg_type
+      when Message::LOGIN
+        connection.send(Message::LoginAccepted.new)
+      when Message::GET
+        value = @config.fetch(message.key_data, '')
+        connection.send(Message::Value.new(:key_data => message.key_data, :value_data => value))
+      when Message::SET
+        @config[message.key_data] = message.value_data
+      end
+    end
+
+    def close(connection)
+      @connections.delete(connection.socket)
+
+      connection.close
+    end
+
+  end
+end

data/lib/wine/version.rb

@@ -0,0 +1,3 @@
+module Wine
+  VERSION = '0.1.0'
+end

metadata

@@ -0,0 +1,95 @@
+--- !ruby/object:Gem::Specification
+name: wine
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Pekka Enberg
+- Jussi Virtanen
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-05-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bindata
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.4'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: ! "Wine for Ruby\n=============\n\nWine is a protocol for configuring
+  and monitoring agents. Wine for Ruby is an\nimplementation of the protocol for Ruby
+  agents, and there is [Wine for Java][]\nfor the JVM as well.\n\n  [Wine for Java]:
+  http://github.com/valotrading/wine-java\n\n\nInstallation\n------------\n\nInstall
+  Wine for Ruby:\n\n    gem install wine\n\n\nDevelopment\n-----------\n\nInstall
+  the dependencies:\n\n    bundle install\n\nRun the tests:\n\n    bundle exec rake\n\nRun
+  the test server:\n\n    bundle exec bin/wine-test-server\n\n\nLicense\n-------\n\nWine
+  for Ruby is released under the Apache License, Version 2.0.\n"
+email: engineering@valotrading.com
+executables:
+- wine-test-server
+extensions: []
+extra_rdoc_files: []
+files:
+- README.md
+- bin/wine-test-server
+- lib/wine.rb
+- lib/wine/version.rb
+- lib/wine/error.rb
+- lib/wine/connection.rb
+- lib/wine/session.rb
+- lib/wine/message.rb
+- lib/wine/test_server.rb
+homepage: http://github.com/valotrading/wine-ruby
+licenses:
+- Apache License, Version 2.0
+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.23
+signing_key: 
+specification_version: 3
+summary: A protocol for configuring and monitoring agents
+test_files: []