baton 0.3.6

data/lib/baton/consumer_manager.rb

@@ -0,0 +1,67 @@
+require "baton/logging"
+
+module Baton
+  class ConsumerManager
+    include Baton::Logging
+    attr_accessor :consumer, :channel, :exchange_in, :exchange_out
+
+    # Public: Initialize a ConsumerManager and adds itself as an observer to the consumer.
+    #
+    # consumer - An instance of Baton::Consumer
+    # channel - An AMQP channel
+    # exchange_in - An input exchange
+    # exchange_out - An output exchange
+    def initialize(consumer, channel, exchange_in, exchange_out)
+      @consumer, @channel, @exchange_in, @exchange_out = consumer, channel, exchange_in, exchange_out
+      @consumer.add_observer(self)
+      @consumer.consumer_manager = self
+    end
+
+    # Public: Creates a queue and binds it to the input exchange based on the consumer's
+    # routing key. Also adds handle_message as a callback method to queue.subscribe().
+    #
+    # Returns nothing.
+    def start
+      queue = channel.queue("", :exclusive => true, :auto_delete => true)
+      queue.bind(exchange_in, :routing_key => consumer.routing_key)
+      queue.subscribe(&method(:handle_message))
+      logger.info "Bind queue with routing key '#{consumer.routing_key}' to exchange '#{exchange_in.name}', waiting for messages..."
+    end
+
+    # Public: Triggered whenever a message is received and forwards the message
+    # to the consumer's handle_message.
+    #
+    # metadata - A metadata structure such as OpenStruct
+    # payload - A JSON message
+    #
+    # Examples
+    #
+    #   handle_message(metadata, "{\"message\":\"a message\",\"type\":\"a type\"}")
+    #
+    # Returns nothing.
+    def handle_message(metadata, payload)
+      logger.debug "Received #{payload}, content_type = #{metadata.content_type}"
+      consumer.handle_message(payload)
+    end
+
+    # Public: Method that is triggered when a consumer notifies with a message. It
+    # logs the messages and writes them to the output exchange as json.
+    #
+    # message - A general message (Hash, String, etc)
+    #
+    # Examples
+    #
+    #   update("A message")
+    #
+    # Returns nothing.
+    def update(message)
+      case message.fetch(:type){""}
+      when "error"
+        logger.error message
+      else
+        logger.info message
+      end
+      exchange_out.publish(message.to_json)
+    end
+  end
+end

data/lib/baton/logging.rb

@@ -0,0 +1,18 @@
+require "logger"
+
+module Baton
+  module Logging
+    def logger
+      Baton::Logging.logger
+    end
+
+    def self.logger
+      STDOUT.sync = true
+      @logger ||= Logger.new(STDOUT)
+    end
+
+    def self.logger=(out)
+      @logger = Logger.new(out)
+    end
+  end
+end

data/lib/baton/observer.rb

@@ -0,0 +1,74 @@
+require "observer"
+
+module Baton
+  module Observer
+    include Observable
+
+    # Public: Method that notifies errors to observers.
+    #
+    #  klass - Error class
+    #  message - Error message
+    #
+    # Examples
+    #
+    #   notify_error(Exception, "Error message")
+    #
+    # Returns nothing.
+    def notify_error(klass, message)
+      notify_log({type: "error", error_class: klass, error_message: message})
+    end
+
+    # Public: Method that notifies informations to observers.
+    #
+    #  message - Info message
+    #
+    # Examples
+    #
+    #   notify_info("info message")
+    #
+    # Returns nothing.
+    def notify_info(message)
+      notify_log({:type => "info", :message => message})
+    end
+
+    # Public: Method that notifies success to observers.
+    #
+    #  message - Success message
+    #
+    # Examples
+    #
+    #   notify_success("success message")
+    #
+    # Returns nothing.
+    def notify_success(message)
+      notify_log({:type => "success", :message => message})
+    end
+
+    # Public: Method that merges attributes to be sent as messages and notifies observers.
+    #
+    #  attrs - A number of attributes represented by an Hash
+    #
+    # Examples
+    #
+    #   notify_log({field_1: "text", :field_2: 123})
+    #
+    # Returns nothing.
+    def notify_log(attrs)
+      notify(attributes.merge(attrs))
+    end
+
+    # Public: Method that notifies messages to observers.
+    #
+    #  message - General message
+    #
+    # Examples
+    #
+    #   notify("message")
+    #
+    # Returns nothing.
+    def notify(message)
+      changed
+      notify_observers(message)
+    end
+  end
+end

data/lib/baton/server.rb

@@ -0,0 +1,68 @@
+require "json"
+
+module Baton
+  class Server
+
+    attr_accessor :environment, :fqdn, :app_names
+
+    # Public: Initialize a Server. ALso, configures the server by reading Baton's configuration
+    # file.
+    def initialize
+      configure
+    end
+
+    # Public: Method that configures the server. It sets the fqdn, environment and a list
+    # of app names specified by the ohai config file.
+    #
+    # Returns nothing.
+    def configure
+      @environment = facts.fetch("chef_environment"){"development"}.downcase
+      @fqdn        = facts.fetch("fqdn"){""}
+      @app_names   = facts.fetch("trebuchet"){[]}
+    end
+
+    # Public: Method that reads facts from the file specified by facts_file.
+    #
+    # Examples
+    #
+    #   facts
+    #   # => {"fqdn" => "server.dsci.it", "chef_environment" => "production", "trebuchet" => []}
+    #
+    # Returns a hash with server information.
+    def facts
+      @facts ||= JSON.parse(facts_file.read)
+    end
+
+    # Public: Provides the content of the ohai file.
+    #
+    # Returns content of ohai facts file.
+    def facts_file
+      File.open(facts_path)
+    end
+
+    # Public: Provides the path to the ohai file
+    #
+    # Examples
+    #
+    #   facts_path
+    #   # => "/path/to/file/facts.json"
+    #
+    # Returns the path of the ohai file
+    def facts_path
+      Baton.configuration.ohai_file
+    end
+
+    # Public: Method that provides an hash of attributes for a server.
+    #
+    # Examples
+    #
+    #   attributes
+    #   # => {environment: "production", fqdn: "server.dsci.it", app_names: ["app1","app2"]}
+    #
+    # Returns Output depends on the implementation.
+    def attributes
+      {environment: environment, fqdn: fqdn, app_names: app_names}
+    end
+
+  end
+end

data/lib/baton/service.rb

@@ -0,0 +1,55 @@
+require "baton"
+require "amqp"
+require "eventmachine"
+require "em-http-request"
+require "json"
+require "baton/channel"
+require "baton/server"
+require "baton/consumer_manager"
+
+module Baton
+  class Service
+    include Baton::Logging
+
+    attr_accessor :server, :channel
+
+    # Public: Initialize a Service. Sets up a new server for this service.
+    def initialize
+      @server = Baton::Server.new
+    end
+
+    # Public: Method that starts the service.
+    #
+    # Returns nothing.
+    def run
+      logger.info "Starting Baton v#{Baton::VERSION}"
+      EM.run do
+        @channel = Baton::Channel.new
+        setup_consumers
+      end
+    end
+
+    # Public: Method that allows implementations to setup new consumers
+    # depending on their needs. An example would be add a deploy consumer
+    # which will listen to deploy messages. For each consumer, add_consumer
+    # should be called to attach the consumers to the AMQP channel.
+    #
+    # Returns Output depends on the implementation.
+    def setup_consumers
+    end
+
+    # Public: Adds a given consumer to the AMQP channel.
+    #
+    # consumer - An instance of Baton::Consumer.
+    #
+    # Examples
+    #
+    #   add_consumer(Baton::DeployConsumer.new("consumer_name", Baton::Server.new))
+    #
+    # Returns nothing..
+    def add_consumer(consumer)
+      channel.add_consumer(consumer)
+    end
+
+  end
+end

data/lib/baton/templates/gem/Gemfile.tt

@@ -0,0 +1,5 @@
+source 'https://rubygems.org'
+source 'http://gems.dsci.it'
+
+# Specify your gem's dependencies in <%=config[:name]%>.gemspec
+gemspec

data/lib/baton/templates/gem/LICENSE.tt

@@ -0,0 +1,22 @@
+Copyright (c) <%=Time.now.year%> <%=config[:author]%>
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/lib/baton/templates/gem/README.md.tt

@@ -0,0 +1,29 @@
+# <%=config[:constant_name]%>
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem '<%=config[:name]%>'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install <%=config[:name]%>
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/lib/baton/templates/gem/Rakefile.tt

@@ -0,0 +1,6 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"
+
+task :console do
+  sh "irb -rubygems -I lib/baton -r <%= config[:name] %>.rb"
+end

data/lib/baton/templates/gem/bin/gem-monitor.tt

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+
+require "baton/<%= config[:name] %>/<%= config[:name] %>-monitor"
+
+Baton.configure do |c|
+  c.config_path  = "<%= File.join(Dir.pwd, config[:name], 'config', config[:name]) %>.cfg"
+  c.exchange_out = "<%= config[:name] %>_out"
+end
+
+Baton::<%= config[:constant_array].last %>Monitor.run

data/lib/baton/templates/gem/bin/gem.tt

@@ -0,0 +1,12 @@
+#!/usr/bin/env ruby
+
+require 'baton/<%= config[:name] %>'
+
+Baton.configure do |c|
+  c.config_path  = "<%= File.join(Dir.pwd, config[:name], 'config', config[:name]) %>.cfg"
+  c.ohai_file    = "/var/lib/ohai/facts.json"
+  c.exchange     = "<%= config[:name] %>_in"
+  c.exchange_out = "<%= config[:name] %>_out"
+end
+
+Baton::<%= config[:constant_array].last %>Service.new.run

data/lib/baton/templates/gem/config/gem.cfg.tt

@@ -0,0 +1,4 @@
+RABBIT_HOST: localhost
+RABBIT_VHOST: 
+RABBIT_USER: guest
+RABBIT_PASS: guest

data/lib/baton/templates/gem/gem.gemspec.tt

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/baton/<%=config[:name]%>/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = [<%=config[:author].inspect%>]
+  gem.email         = [<%=config[:email].inspect%>]
+  gem.description   = %q{TODO: Write a gem description}
+  gem.summary       = %q{TODO: Write a gem summary}
+  gem.homepage      = ""
+
+  gem.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  gem.files         = `git ls-files`.split("\n")
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = <%=config[:name].inspect%>
+  gem.require_paths = ["lib"]
+  gem.version       = <%=config[:constant_name]%>::VERSION
+
+  gem.add_runtime_dependency "baton"
+
+  gem.add_development_dependency "rake"
+end

data/lib/baton/templates/gem/gitignore.tt

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/lib/baton/templates/gem/lib/baton/gem.rb.tt

@@ -0,0 +1,15 @@
+require "baton"
+require "baton/service"
+require "baton/<%=config[:name]%>/version"
+require "baton/<%=config[:name]%>/<%=config[:name]%>-consumer"
+
+module Baton
+  class <%= config[:constant_array].last %>Service < Baton::Service
+
+    def setup_consumers
+      srv = <%=config[:constant_name]%>Consumer.new(server.fqdn, server)
+      add_consumer(srv)
+    end
+  end
+end
+