skein 0.3.0

data/lib/skein/handler/async.rb

@@ -0,0 +1,9 @@
+class Skein::Handler::Async < Skein::Handler
+  # == Instance Methods =====================================================
+
+  def delegate(*args)
+    @target.send(*args) do |response, error|
+      yield(response, error)
+    end
+  end
+end

data/lib/skein/handler/threaded.rb

@@ -0,0 +1,7 @@
+class Skein::Handler::Threaded < Skein::Handler
+  # == Instance Methods =====================================================
+
+  def delegate(*args)
+    yield(@target.send(*args))
+  end
+end

data/lib/skein/rabbitmq.rb

@@ -0,0 +1,44 @@
+module Skein::RabbitMQ
+  # == Exceptions ===========================================================
+
+  class MissingDriver < RuntimeError
+  end
+
+  # == Module Methods =======================================================
+
+  def self.force_require!(config = nil)
+    config ||= Skein.config
+
+    case (config.driver.to_s)
+    when 'bunny', 'rubybunny'
+      unless (defined?(Bunny))
+        require 'bunny'
+      end
+    when 'march_hare', 'marchhare'
+      unless (defined?(MarchHare))
+        require 'march_hare'
+      end
+    else
+      raise MissingDriver, 'Missing or invalid configuration for: driver'
+    end
+  end
+
+  def self.connect(config = nil)
+    config ||= Skein.config
+
+    self.force_require!(config)
+
+    case (config.driver.to_s)
+    when 'bunny', 'rubybunny'
+      bunny = Bunny.new(config.to_h)
+
+      bunny.start
+
+      bunny
+    when 'march_hare', 'marchhare'
+      MarchHare.connect(config.to_h)
+    else
+      raise MissingDriver, 'Missing or invalid configuration for: driver'
+    end
+  end
+end

data/lib/skein/reporter.rb

@@ -0,0 +1,11 @@
+class Skein::Reporter
+  # == Instance Methods =====================================================
+  
+  def initialize
+    @logger = Birling.new
+  end
+
+  def exception!(e, *meta)
+    # ...
+  end
+end

data/lib/skein/rpc.rb

@@ -0,0 +1,24 @@
+class Skein::RPC
+  class Exception < ::RuntimeError
+    attr_accessor :request
+
+    def to_error
+      Skein::RPC::Error.new(
+        error: '[%s] %s' % [ self.class, self.to_s ],
+        id: self.request ? self.request.id : nil
+      )
+    end
+  end
+
+  class InvalidPayload < Exception
+  end
+
+  class InvalidMethod < Exception
+  end
+end
+
+require_relative './rpc/base'
+require_relative './rpc/error'
+require_relative './rpc/request'
+require_relative './rpc/response'
+require_relative './rpc/notification'

data/lib/skein/rpc/base.rb

@@ -0,0 +1,23 @@
+require 'json'
+
+class Skein::RPC::Base
+  # == Exceptions ===========================================================
+
+  # == Properties ===========================================================
+  
+  attr_accessor :id
+
+  # == Instance Methods =====================================================
+
+  def to_h
+    {
+      id: self.id
+    }
+  end
+
+  def to_json
+    JSON.dump(
+      self.to_h
+    )
+  end
+end

data/lib/skein/rpc/error.rb

@@ -0,0 +1,34 @@
+class Skein::RPC::Error < Skein::RPC::Base
+  # == Properties ===========================================================
+
+  attr_accessor :error
+
+  # == Instance Methods =====================================================
+
+  def initialize(content = nil)
+    case (content)
+    when String
+      self.error = content
+    when Hash
+      self.assign_from_hash!(content)
+    when nil
+      # Defaults
+    else
+      raise Skein::RPC::Exception, 'Invalid type: %s' % content.class
+    end
+  end
+
+  def self.to_h
+    {
+      result: nil,
+      error: self.error,
+      id: self.id
+    }
+  end
+
+protected
+  def assign_from_hash!(hash)
+    self.error = hash[:error]
+    self.id = hash[:id]
+  end
+end

data/lib/skein/rpc/notification.rb

@@ -0,0 +1,2 @@
+class Skein::RPC::Notification < Skein::RPC::Base
+end

data/lib/skein/rpc/request.rb

@@ -0,0 +1,62 @@
+class Skein::RPC::Request < Skein::RPC::Base
+  # == Properties ===========================================================
+
+  attr_accessor :method
+  attr_accessor :params
+
+  # == Instance Methods =====================================================
+
+  def initialize(content = nil)
+    case (content)
+    when String
+      data = Skein::Support.symbolize_keys(JSON.load(content))
+
+      assign_from_hash!(data)
+    when Hash
+      assign_from_hash!(content)
+    when nil
+      self.id = SecureRandom.uuid
+    else
+      raise Skein::RPC::InvalidPayload, 'Invalid payload type: %s' % content.class
+    end
+  end
+
+  def to_h
+    {
+      method: self.method,
+      params: self.params,
+      id: self.id
+    }
+  end
+
+  def response(result: nil, error: nil)
+    if (result)
+      Skein::RPC::Response.new(
+        result: result,
+        id: self.id
+      )
+    elsif (error)
+      Skein::RPC::Error.new(
+        error: error,
+        id: self.id
+      )
+    end
+  end
+
+protected
+  def assign_from_hash!(hash)
+    self.method = hash[:method]
+    self.params = Skein::Support.arrayify(hash[:params])
+    self.id = hash[:id]
+
+    case (self.method)
+    when String
+      unless (self.method.match(/\A\w+\z/))
+        e = Skein::RPC::InvalidMethod.new('%s is not a valid RPC method name.' % self.method.inspect)
+        e.request = self
+
+        raise e
+      end
+    end
+  end
+end

data/lib/skein/rpc/response.rb

@@ -0,0 +1,38 @@
+class Skein::RPC::Response < Skein::RPC::Base
+  # == Properties ===========================================================
+
+  attr_accessor :result
+  attr_accessor :error
+
+  # == Instance Methods =====================================================
+
+  def initialize(content = nil)
+    case (content)
+    when String
+      data = Skein::Support.symbolize_keys(JSON.load(content))
+
+      assign_from_hash!(data)
+    when Hash
+      assign_from_hash!(content)
+    when nil
+      self.id = SecureRandom.uuid
+    else
+      raise Skein::RPC::InvalidPayload, 'Invalid payload type: %s' % content.class
+    end
+  end
+
+  def to_h
+    {
+      method: self.method,
+      params: self.params,
+      id: self.id
+    }
+  end
+
+protected
+  def assign_from_hash!(hash)
+    self.result = hash[:result]
+    self.error = hash[:error]
+    self.id = hash[:id]
+  end
+end

data/lib/skein/support.rb

@@ -0,0 +1,67 @@
+require 'socket'
+
+module Skein::Support
+  # == Module Methods =======================================================
+
+  def self.hostname
+    Socket.gethostname
+  end
+
+  def self.process_name
+    $0.split(/\s/).first.split('/').last.sub(/\.rb\z/, '')
+  end
+
+  def self.process_id
+    Process.pid
+  end
+
+  def self.arrayify(value)
+    case (value)
+    when nil
+      nil
+    when Array
+      value
+    else
+      [ value ]
+    end
+  end
+
+  def self.hash_format(hash, width: nil)
+    hash = hash.to_h
+
+    width ||= hash.keys.map(&:length).sort[-1].to_i + 1
+
+    template = "%%-%ds %%s" % [ width ]
+
+    hash.map do |pair|
+      template % pair
+    end
+  end
+
+  def self.symbolize_keys(hash)
+    case (hash)
+    when Hash
+      Hash[
+        hash.map do |k,v|
+          [
+            k.to_s.to_sym,
+            case (v)
+            when Hash
+              symbolize_keys(v)
+            when Array
+              v.map { |e| symbolize_keys(e) }
+            else
+              v
+            end
+          ]
+        end
+      ]
+    when Array
+      hash.map do |h|
+        symbolize_keys(h)
+      end
+    else
+      hash
+    end
+  end
+end

data/skein.gemspec

@@ -0,0 +1,95 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+# stub: skein 0.3.0 ruby lib
+
+Gem::Specification.new do |s|
+  s.name = "skein".freeze
+  s.version = "0.3.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0".freeze) if s.respond_to? :required_rubygems_version=
+  s.require_paths = ["lib".freeze]
+  s.authors = ["Scott Tadman".freeze]
+  s.date = "2016-12-07"
+  s.description = "Wrapper for RabbitMQ that makes blocking RPC calls and handles pub-sub broadcasts.".freeze
+  s.email = "tadman@postageapp.com".freeze
+  s.executables = ["skein".freeze]
+  s.extra_rdoc_files = [
+    "README.md"
+  ]
+  s.files = [
+    "Gemfile",
+    "Gemfile.lock",
+    "README.md",
+    "RELEASES.md",
+    "Rakefile",
+    "VERSION",
+    "bin/skein",
+    "config/.gitignore",
+    "config/skein.yml.example",
+    "lib/skein.rb",
+    "lib/skein/client.rb",
+    "lib/skein/client/publisher.rb",
+    "lib/skein/client/rpc.rb",
+    "lib/skein/client/subscriber.rb",
+    "lib/skein/client/worker.rb",
+    "lib/skein/config.rb",
+    "lib/skein/connected.rb",
+    "lib/skein/context.rb",
+    "lib/skein/handler.rb",
+    "lib/skein/handler/async.rb",
+    "lib/skein/handler/threaded.rb",
+    "lib/skein/rabbitmq.rb",
+    "lib/skein/reporter.rb",
+    "lib/skein/rpc.rb",
+    "lib/skein/rpc/base.rb",
+    "lib/skein/rpc/error.rb",
+    "lib/skein/rpc/notification.rb",
+    "lib/skein/rpc/request.rb",
+    "lib/skein/rpc/response.rb",
+    "lib/skein/support.rb",
+    "skein.gemspec",
+    "test/data/sample_config.yml",
+    "test/helper.rb",
+    "test/script/em_example",
+    "test/unit/test_skein_client.rb",
+    "test/unit/test_skein_client_publisher.rb",
+    "test/unit/test_skein_client_subscriber.rb",
+    "test/unit/test_skein_client_worker.rb",
+    "test/unit/test_skein_config.rb",
+    "test/unit/test_skein_context.rb",
+    "test/unit/test_skein_rabbitmq.rb",
+    "test/unit/test_skein_reporter.rb",
+    "test/unit/test_skein_rpc_error.rb",
+    "test/unit/test_skein_rpc_request.rb",
+    "test/unit/test_skein_support.rb",
+    "tmp/.gitignore"
+  ]
+  s.homepage = "http://github.com/postageapp/skein".freeze
+  s.licenses = ["closed".freeze]
+  s.rubygems_version = "2.6.8".freeze
+  s.summary = "RabbitMQ RPC/PubSub Library".freeze
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 4
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<birling>.freeze, [">= 0"])
+      s.add_development_dependency(%q<rake>.freeze, [">= 0"])
+      s.add_development_dependency(%q<jeweler>.freeze, [">= 0"])
+      s.add_development_dependency(%q<test-unit>.freeze, [">= 0"])
+    else
+      s.add_dependency(%q<birling>.freeze, [">= 0"])
+      s.add_dependency(%q<rake>.freeze, [">= 0"])
+      s.add_dependency(%q<jeweler>.freeze, [">= 0"])
+      s.add_dependency(%q<test-unit>.freeze, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<birling>.freeze, [">= 0"])
+    s.add_dependency(%q<rake>.freeze, [">= 0"])
+    s.add_dependency(%q<jeweler>.freeze, [">= 0"])
+    s.add_dependency(%q<test-unit>.freeze, [">= 0"])
+  end
+end
+

data/test/data/sample_config.yml

@@ -0,0 +1,13 @@
+development:
+  host: dev.host
+  port: 25672
+  username: dev_user
+  password: dev_password
+  namespace: example
+
+test:
+  host: test.host
+  port: 5670
+  username: test_user
+  password: test_password
+  namespace: test