batsir 0.1.0

data/lib/batsir/amqp.rb

@@ -0,0 +1,45 @@
+module Batsir
+  module AMQP
+    attr_accessor :queue
+    attr_accessor :host
+    attr_accessor :port
+    attr_accessor :username
+    attr_accessor :password
+    attr_accessor :vhost
+    attr_accessor :exchange
+
+    def bunny_options
+      {
+        :host  => host,
+        :port  => port,
+        :user  => username,
+        :pass  => password,
+        :vhost => vhost
+      }
+    end
+
+    def host
+      @host || 'localhost'
+    end
+
+    def port
+      @port || 5672
+    end
+
+    def username
+      @username || 'guest'
+    end
+
+    def password
+      @password || 'guest'
+    end
+
+    def vhost
+      @vhost || '/'
+    end
+
+    def exchange
+      @exchange || 'amq.direct'
+    end
+  end
+end

data/lib/batsir/chain.rb

@@ -0,0 +1,33 @@
+module Batsir
+  class Chain
+
+    def initialize(options = {})
+      options.each do |attr, value|
+        self.send("#{attr.to_s}=", value)
+      end
+      @stages = []
+    end
+
+    def stages
+      @stages
+    end
+
+    def add_stage(stage)
+      @stages << stage
+    end
+
+    def compile
+      generated = ""
+      stages.each do |stage|
+        generated << stage.compile
+      end
+      generated
+    end
+
+    def start
+      stages.each do | stage |
+        stage.start
+      end
+    end
+  end
+end

data/lib/batsir/config.rb

@@ -0,0 +1,34 @@
+module Batsir
+  class Config
+    def initialize(data={})
+      @data = {}
+      update!(data)
+    end
+
+    def update!(data)
+      data.each do |key, value|
+        self[key] = value
+      end
+    end
+
+    def [](key)
+      @data[key.to_sym]
+    end
+
+    def []=(key, value)
+      if value.class == Hash
+        @data[key.to_sym] = Config.new(value)
+      else
+        @data[key.to_sym] = value
+      end
+    end
+
+    def method_missing(sym, *args)
+      if sym.to_s =~ /(.+)=$/
+        self[$1] = args.first
+      else
+        self[sym]
+      end
+    end
+  end
+end

data/lib/batsir/dsl/dsl_mappings.rb

@@ -0,0 +1,96 @@
+module Batsir
+  module DSL
+    class ChainMapping < ::Blockenspiel::Base
+      def initialize
+        @chain = nil
+      end
+
+      def aggregator_chain(&block)
+        @chain = Batsir::Chain.new
+        ::Blockenspiel.invoke(block, self)
+        @chain
+      end
+
+      def stage(name, &block)
+        new_block = ::Proc.new do
+          stage name, &block
+        end
+        stage = ::Blockenspiel.invoke(new_block, Batsir::DSL::StageMapping.new)
+        stage.chain = @chain
+        @chain.add_stage(stage)
+      end
+    end
+
+    class StageMapping < ::Blockenspiel::Base
+      def initialize
+        @stage = nil
+      end
+
+      def stage(name, &block)
+        @stage = Batsir::Stage.new(:name => name)
+        ::Blockenspiel.invoke(block, self)
+        @stage
+      end
+
+      def filter(operation)
+        @stage.add_filter(operation)
+      end
+
+      def inbound(&block)
+        ::Blockenspiel.invoke(block, Batsir::DSL::InboundMapping.new(@stage))
+      end
+
+      def outbound(&block)
+        ::Blockenspiel.invoke(block, Batsir::DSL::OutboundMapping.new(@stage))
+      end
+    end
+
+    class InboundMapping < ::Blockenspiel::Base
+      def initialize(stage)
+        @stage = stage
+      end
+
+      def transformers(&block)
+        ::Blockenspiel.invoke(block, Batsir::DSL::InboundTransformerMapping.new(@stage))
+      end
+
+      def acceptor(acceptor_class, options = {})
+        @stage.add_acceptor(acceptor_class, options)
+      end
+    end
+
+    class InboundTransformerMapping < ::Blockenspiel::Base
+      def initialize(stage)
+        @stage = stage
+      end
+
+      def transformer(transformer, options = {})
+        @stage.add_acceptor_transformer(transformer, options)
+      end
+    end
+
+    class OutboundMapping < ::Blockenspiel::Base
+      def initialize(stage)
+        @stage = stage
+      end
+
+      def transformers(&block)
+        ::Blockenspiel.invoke(block, Batsir::DSL::OutboundTransformerMapping.new(@stage))
+      end
+
+      def notifier(notifier_class, options = {})
+        @stage.add_notifier(notifier_class, options)
+      end
+    end
+
+    class OutboundTransformerMapping < ::Blockenspiel::Base
+      def initialize(stage)
+        @stage = stage
+      end
+
+      def transformer(transformer, options = {})
+        @stage.add_notifier_transformer(transformer, options)
+      end
+    end
+  end
+end

data/lib/batsir/filter.rb

@@ -0,0 +1,12 @@
+module Batsir
+  class Filter
+    def initialize(options = {})
+      options.each do |attr, value|
+        self.send("#{attr.to_s}=", value)
+      end
+    end
+
+    def execute(message)
+    end
+  end
+end

data/lib/batsir/filter_queue.rb

@@ -0,0 +1,30 @@
+module Batsir
+  class FilterQueue
+    include Enumerable
+
+    attr_accessor :filters
+    attr_accessor :notifiers
+
+    def initialize
+      @filters = []
+      @notifiers = []
+    end
+
+    def add(operation)
+      @filters << operation
+    end
+
+    def add_notifier(notifier)
+      @notifiers << notifier
+    end
+
+    def each
+      @filters.each {|op| yield op}
+      @notifiers.each {|n| yield n}
+    end
+
+    def empty?
+      !(@notifiers.any? || @filters.any?)
+    end
+  end
+end

data/lib/batsir/logo.rb

@@ -0,0 +1,36 @@
+module Batsir
+  def self.logo
+  <<EOF
+
+
+
+
+
+                                   E        ,.
+                   ,,,agB8@f      ,8v      ,8L      J88&gg,,,
+              .,a88888888@^       |8&,,,,,,88k       ,88888888&g,,
+           ,+88888888888$t        8888888f''9@.       ,888888888888g,
+        ,,88888888888888$;       ,888888&`   8c       ,888888888888888w,
+      ,a88888888888888888&,......|8888888&gg&|8k......,d88888888888888888&,
+     ,888888888888888888888888888888888qp8888|@8888888888888888888888888888y
+    ,888888888888888888888888888888`8` '' `8`|888888888888888888888888888888E
+   ,88888888888888888888888888888888   &&   8|8888888888888888888888888888888c
+   |88888888888888888888888888888888888888888|8888888888888888888888888888888k
+   |88888888888888888888888888888888888888888|8888888888888888888888888888888k
+   ,888888888888888888888888888888888888888888888888888888888888888888888888!
+    j8888888888888888888888888888888888888888888888888888888888888888888888f
+     j8888888888888888@@88888888888888888888888888888888@@888888888888888@f
+      ,9888888888888@:  `988M^  `?88888888888@f:  |988M'  .8888888888888M'
+        ,98888888888F     jf       J8888888@f       jf     ,888888888@f'
+           |98888888&.              ,98888@'               d888888@9"
+              `?9@888&,               j88f               ,d88$@9l'
+                   `|?9t               |f               `9T"`
+
+
+
+
+
+
+EOF
+  end
+end

data/lib/batsir/notifiers/amqp_notifier.rb

@@ -0,0 +1,16 @@
+require 'batsir/amqp'
+
+module Batsir
+  module Notifiers
+    class AMQPNotifier < Notifier
+      include Batsir::AMQP
+
+      def execute(message)
+        Bunny.run( bunny_options )do |bunny|
+          exc = bunny.exchange( exchange )
+          exc.publish( message, :key => queue )
+        end
+      end
+    end
+  end
+end

data/lib/batsir/notifiers/notifier.rb

@@ -0,0 +1,39 @@
+module Batsir
+  module Notifiers
+    class Notifier
+      attr_accessor :field_mapping
+      attr_accessor :transformer_queue
+
+      def initialize(options = {})
+        fields = options.delete(:fields)
+
+        options.each do |option, value|
+          self.send("#{option}=", value)
+        end
+        @transformer_queue = []
+        if fields
+          add_transformer(Batsir::Transformers::FieldTransformer.new(:fields => fields))
+        end
+      end
+
+      def add_transformer(transformer)
+        @transformer_queue << transformer
+      end
+
+      def notify(message)
+        execute(transform(message))
+      end
+
+      def execute(message)
+
+      end
+
+      def transform(message)
+        transformer_queue.each do |transformer|
+          message = transformer.transform(message)
+        end
+        message
+      end
+    end
+  end
+end

data/lib/batsir/registry.rb

@@ -0,0 +1,15 @@
+module Batsir
+  class Registry
+    def self.register(name, klass)
+      registry[name] = klass
+    end
+
+    def self.registry
+      @registry ||= {}
+    end
+
+    def self.get(name)
+      registry.fetch(name)
+    end
+  end
+end

data/lib/batsir/stage.rb

@@ -0,0 +1,94 @@
+module Batsir
+  class Stage
+    include Celluloid
+
+    FilterDeclaration       = Struct.new(:filter, :options)
+    TransformerDeclaration  = Struct.new(:transformer, :options)
+
+    attr_accessor :name
+    attr_accessor :chain
+    attr_accessor :cancellators
+    attr_reader   :filter_declarations
+    attr_reader   :notifiers
+    attr_reader   :acceptors
+    attr_reader   :running_acceptors
+    attr_reader   :notifier_transformers
+    attr_reader   :acceptor_transformers
+
+    def initialize(options = {})
+      options.each do |attr, value|
+        self.send("#{attr.to_s}=", value)
+      end
+      @cancellators           = []
+      @acceptor_transformers  = []
+      @running_acceptors      = []
+      @acceptors              = {}
+      @filter_declarations    = []
+      @notifiers              = {}
+      @notifier_transformers  = []
+      @built                  = false
+    end
+
+    def built?
+      @built
+    end
+
+    def add_acceptor_transformer(transformer, options = {})
+      @acceptor_transformers << TransformerDeclaration.new(transformer, options)
+    end
+
+    def add_acceptor(acceptor, options = {})
+      @acceptors[acceptor] ||= Set.new
+      @acceptors[acceptor] << options
+    end
+
+    def add_filter(filter, options = {})
+      @filter_declarations << FilterDeclaration.new(filter, options)
+    end
+
+    def filters
+      @filter_declarations.map{ |filter_declaration| filter_declaration.filter }
+    end
+
+    def add_notifier(notifier, options = {})
+      @notifiers[notifier] ||= Set.new
+      @notifiers[notifier] << options
+    end
+
+    def add_notifier_transformer(transformer, options = {})
+      @notifier_transformers << TransformerDeclaration.new(transformer, options)
+    end
+
+    def compile
+      Batsir::StageWorker.compile_from(self)
+    end
+
+    def finalize
+      @cancellators.each do |cancellator|
+        cancellator.write "STOP"
+      end
+    end
+
+    def start
+      acceptors.each do |acceptor_class, options|
+        options.each do |acceptor_options|
+          cancellator_reader, cancellator_writer = ::IO.pipe
+          acceptor_options.merge!(:stage_name => self.name, :cancellator => cancellator_reader)
+          @cancellators << cancellator_writer
+          acceptor = acceptor_class.new(acceptor_options)
+          if acceptor_transformers.any?
+            acceptor_transformers.each do |transformer_declaration|
+              transformer = transformer_declaration.transformer.new(transformer_declaration.options)
+              acceptor.add_transformer(transformer)
+            end
+          else
+            acceptor.add_transformer(Batsir::Transformers::JSONInputTransformer.new)
+          end
+          @running_acceptors << acceptor
+          acceptor.start!
+        end
+      end
+      true
+    end
+  end
+end