cf-autoconfig 0.1.0

data/lib/cfautoconfig.rb

@@ -0,0 +1,3 @@
+require 'rubygems'
+require 'cfautoconfig/configurer'
+require 'cfautoconfig/version'

data/lib/cfautoconfig/blob/aws_s3.rb

@@ -0,0 +1,23 @@
+require 'cfruntime/properties'
+require 'cfruntime/aws_s3'
+
+module AutoReconfiguration
+  SUPPORTED_AWS_S3_VERSION = '0.6.3'
+  module AwsS3
+    def self.included( base )
+      base.send(:alias_method, :original_connect, :connect)
+      base.send(:alias_method, :connect, :connect_with_cf )
+    end
+
+    def connect_with_cf(options = {})
+      service_names = CFRuntime::CloudApp.service_names_of_type('blob')
+      if service_names.length == 1
+        puts "Auto-reconfiguring AWS-S3"
+        options = CFRuntime::AWSS3Client.options_for_svc(service_names[0],options)
+      else
+        puts "Found #{service_names.length} blob services. Skipping auto-reconfiguration."
+      end
+      original_connect options
+    end
+  end
+end

data/lib/cfautoconfig/blob/aws_s3_configurer.rb

@@ -0,0 +1,41 @@
+require 'cfautoconfig/configuration_helper'
+begin
+  require 'aws/s3'
+  require File.join(File.dirname(__FILE__), 'aws_s3')
+  aws_s3_version = Gem.loaded_specs['aws-s3'].version
+  if aws_s3_version >= Gem::Version.new(AutoReconfiguration::SUPPORTED_AWS_S3_VERSION)
+    if AutoReconfiguration::ConfigurationHelper.disabled? :blob
+      puts "Blob auto-reconfiguration disabled."
+      module AWS
+        module S3
+          class << Connection
+            #Remove introduced aliases and methods.
+            #This is mostly for testing, as we don't expect this script
+            #to run twice during a single app startup
+            if method_defined?(:connect_with_cf)
+              undef_method :connect_with_cf
+              alias :connect :original_connect
+            end
+          end
+        end
+      end
+    elsif AWS::S3::Connection.public_methods.index :connect_with_cf
+      #Guard against introducing a method that may already exist
+      puts "AWS-S3 auto-reconfiguration already included."
+    else
+      #Introduce around alias into Connection class
+      module AWS
+        module S3
+          class << Connection
+            include AutoReconfiguration::AwsS3
+          end
+        end
+      end
+    end
+  else
+    puts "Auto-reconfiguration not supported for this AWS-S3 version.  " +
+      "Found: #{aws_s3_version}.  Required: #{AutoReconfiguration::SUPPORTED_AWS_S3_VERSION} or higher."
+  end
+rescue LoadError=>e
+  puts "No AWS-S3 Library Found. Skipping auto-reconfiguration. #{e}"
+end

data/lib/cfautoconfig/configuration_helper.rb

@@ -0,0 +1,19 @@
+module AutoReconfiguration
+  class ConfigurationHelper
+    def self.disabled? (client)
+      disable_auto_config = []
+      if ENV['DISABLE_AUTO_CONFIG']
+        disable_auto_config = ENV['DISABLE_AUTO_CONFIG'].split(/\s*\:\s*/)
+        disable_auto_config.collect! { |element|
+          element.upcase
+        }
+      end
+      if disable_auto_config.include? 'ALL'
+        true
+      else
+        disable_auto_config.include? client.to_s.upcase
+      end
+    end
+  end
+end
+     

data/lib/cfautoconfig/configurer.rb

@@ -0,0 +1,44 @@
+require 'cfruntime/properties'
+
+if CFRuntime::CloudApp.service_props('redis')
+  puts "Loading Redis auto-reconfiguration."
+  require 'cfautoconfig/keyvalue/redis_configurer'
+else
+  puts "No Redis service bound to app.  Skipping auto-reconfiguration."
+end
+
+if CFRuntime::CloudApp.service_props('mongodb')
+  puts "Loading MongoDB auto-reconfiguration."
+  require 'cfautoconfig/document/mongodb_configurer'
+else
+  puts "No Mongo service bound to app.  Skipping auto-reconfiguration."
+end
+
+if CFRuntime::CloudApp.service_props('mysql')
+  puts "Loading MySQL auto-reconfiguration."
+  require 'cfautoconfig/relational/mysql_configurer'
+else
+  puts "No MySQL service bound to app.  Skipping auto-reconfiguration."
+end
+
+if CFRuntime::CloudApp.service_props('postgresql')
+  puts "Loading PostgreSQL auto-reconfiguration."
+  require 'cfautoconfig/relational/postgres_configurer'
+else
+  puts "No PostgreSQL service bound to app.  Skipping auto-reconfiguration."
+end
+
+if CFRuntime::CloudApp.service_props('rabbitmq')
+  puts "Loading RabbitMQ auto-reconfiguration."
+  require 'cfautoconfig/messaging/amqp_configurer'
+  require 'cfautoconfig/messaging/carrot_configurer'
+else
+  puts "No RabbitMQ service bound to app.  Skipping auto-reconfiguration."
+end
+
+if CFRuntime::CloudApp.service_props('blob')
+  puts "Loading Blob auto-reconfiguration."
+  require 'cfautoconfig/blob/aws_s3_configurer'
+else
+  puts "No Blob service bound to app.  Skipping auto-reconfiguration."
+end

data/lib/cfautoconfig/document/mongodb.rb

@@ -0,0 +1,56 @@
+require 'cfruntime/properties'
+
+module AutoReconfiguration
+  SUPPORTED_MONGO_VERSION = '1.2.0'
+  module Mongo
+
+    def self.included( base )
+      base.send( :alias_method, :original_initialize, :initialize)
+      base.send( :alias_method, :initialize, :initialize_with_cf )
+      base.send( :alias_method, :original_apply_saved_authentication, :apply_saved_authentication)
+      base.send( :alias_method, :apply_saved_authentication, :apply_saved_authentication_with_cf )
+      base.send( :alias_method, :original_db, :db)
+      base.send( :alias_method, :db, :db_with_cf )
+      base.send( :alias_method, :original_shortcut, :[])
+      base.send( :alias_method, :[], :shortcut_with_cf )
+    end
+
+    def initialize_with_cf(host = nil, port = nil, opts = {})
+      service_names = CFRuntime::CloudApp.service_names_of_type('mongodb')
+      if service_names.length == 1
+        @service_props = CFRuntime::CloudApp.service_props('mongodb')
+        puts "Auto-reconfiguring MongoDB"
+        @auto_config = true
+        original_initialize @service_props[:host], @service_props[:port], opts
+        add_auth(@service_props[:db], @service_props[:username], @service_props[:password])
+      else
+        puts "Found #{service_names.length} mongo services. Skipping auto-reconfiguration."
+        @auto_config = false
+        original_initialize host, port, opts
+      end
+    end
+
+    def apply_saved_authentication_with_cf(opts = {})
+      add_auth(@service_props[:db], @service_props[:username], @service_props[:password])
+      original_apply_saved_authentication opts
+    end
+
+    def db_with_cf(db_name, opts = {})
+      if @auto_config && db_name != 'admin'
+        add_auth(@service_props[:db], @service_props[:username], @service_props[:password])
+        db = original_db @service_props[:db], opts
+      else
+        original_db db_name, opts
+      end
+    end
+
+    def shortcut_with_cf(db_name)
+      if @auto_config && db_name != 'admin'
+        add_auth(@service_props[:db], @service_props[:username], @service_props[:password])
+        db = original_shortcut(@service_props[:db])
+      else
+        db = original_shortcut(db_name)
+      end
+    end
+  end
+end

data/lib/cfautoconfig/document/mongodb_configurer.rb

@@ -0,0 +1,44 @@
+require 'cfautoconfig/configuration_helper'
+begin
+  #Require mongo here is mandatory for configurer to ensure class is loaded before applying OpenClass
+  require "mongo"
+  require File.join(File.dirname(__FILE__), 'mongodb')
+  mongo_version = Gem.loaded_specs['mongo'].version
+  if mongo_version >= Gem::Version.new(AutoReconfiguration::SUPPORTED_MONGO_VERSION)
+    if AutoReconfiguration::ConfigurationHelper.disabled? :mongodb
+       puts "MongoDB auto-reconfiguration disabled."
+       module Mongo
+         class Connection
+           #Remove introduced aliases and methods.
+           #This is mostly for testing, as we don't expect this script
+           #to run twice during a single app startup
+           if method_defined?(:initialize_with_cf)
+             undef_method :initialize_with_cf
+             alias :initialize :original_initialize
+             undef_method :apply_saved_authentication_with_cf
+             alias :apply_saved_authentication :original_apply_saved_authentication
+             undef_method :db_with_cf
+             alias :db :original_db
+             undef_method :shortcut_with_cf
+             alias :[] :original_shortcut
+           end
+         end
+       end
+    elsif Mongo::Connection.public_instance_methods.index :initialize_with_cf
+      puts "MongoDB AutoReconfiguration already included."
+    else
+      module Mongo
+        #Introduce around alias into Mongo Connection class
+        class Connection
+          include AutoReconfiguration::Mongo
+        end
+      end
+    end
+  else
+    puts "Auto-reconfiguration not supported for this Mongo version.  " +
+      "Found: #{mongo_version}.  Required: #{AutoReconfiguration::SUPPORTED_MONGO_VERSION} or higher."
+  end
+rescue LoadError
+  puts "No MongoDB Library Found. Skipping auto-reconfiguration."
+end
+

data/lib/cfautoconfig/keyvalue/redis.rb

@@ -0,0 +1,24 @@
+require 'cfruntime/properties'
+require 'cfruntime/redis'
+
+module AutoReconfiguration
+  SUPPORTED_REDIS_VERSION = '2.0'
+  module Redis
+    def self.included( base )
+      base.send( :alias_method, :original_initialize, :initialize)
+      base.send( :alias_method, :initialize, :initialize_with_cf )
+    end
+
+    def initialize_with_cf(options = {})
+      service_names = CFRuntime::CloudApp.service_names_of_type('redis')
+      if service_names.length == 1
+        puts "Auto-reconfiguring Redis."
+        cfoptions = CFRuntime::RedisClient.options_for_svc(service_names[0],options)
+        original_initialize cfoptions
+      else
+        puts "Found #{service_names.length} redis services. Skipping auto-reconfiguration."
+        original_initialize options
+      end
+    end
+  end
+end

data/lib/cfautoconfig/keyvalue/redis_configurer.rb

@@ -0,0 +1,33 @@
+require 'cfautoconfig/configuration_helper'
+begin
+  require 'redis'
+  require File.join(File.dirname(__FILE__), 'redis')
+  redis_version = Gem.loaded_specs['redis'].version
+  if redis_version >= Gem::Version.new(AutoReconfiguration::SUPPORTED_REDIS_VERSION)
+    if AutoReconfiguration::ConfigurationHelper.disabled? :redis
+      puts "Redis auto-reconfiguration disabled."
+      class Redis
+        #Remove introduced aliases and methods.
+        #This is mostly for testing, as we don't expect this script
+        #to run twice during a single app startup
+        if method_defined?(:initialize_with_cf)
+          undef_method :initialize_with_cf
+          alias :initialize :original_initialize
+        end
+      end
+    elsif Redis.public_instance_methods.index :initialize_with_cf
+      #Guard against introducing a method that may already exist
+      puts "Redis auto-reconfiguration already included."
+    else
+      #Introduce around alias into Redis class
+      class Redis
+        include AutoReconfiguration::Redis
+      end
+    end
+  else
+    puts "Auto-reconfiguration not supported for this Redis version.  " +
+      "Found: #{redis_version}.  Required: #{AutoReconfiguration::SUPPORTED_REDIS_VERSION} or higher."
+  end
+rescue LoadError
+  puts "No Redis Library Found. Skipping auto-reconfiguration."
+end

data/lib/cfautoconfig/messaging/amqp.rb

@@ -0,0 +1,30 @@
+require 'cfruntime/properties'
+require 'cfruntime/amqp'
+
+module AutoReconfiguration
+  SUPPORTED_AMQP_VERSION = '0.8'
+  module AMQP
+    def self.included( base )
+      base.send( :alias_method, :original_connect, :connect)
+      base.send( :alias_method, :connect, :connect_with_cf )
+    end
+
+    def connect_with_cf(connection_options_or_string = {}, other_options = {}, &block)
+      service_names = CFRuntime::CloudApp.service_names_of_type('rabbitmq')
+      if service_names.length == 1
+        puts "Auto-reconfiguring AMQP."
+        case connection_options_or_string
+          when String then
+            cfoptions = {}
+          else
+            cfoptions = connection_options_or_string
+          end
+        original_connect(CFRuntime::AMQPClient.options_for_svc(service_names[0],cfoptions),
+          other_options, &block)
+      else
+        puts "Found #{service_names.length} rabbitmq services. Skipping auto-reconfiguration."
+        original_connect(connection_options_or_string, other_options, &block)
+      end
+    end
+  end
+end

data/lib/cfautoconfig/messaging/amqp_configurer.rb

@@ -0,0 +1,33 @@
+require 'cfautoconfig/configuration_helper'
+begin
+  #Require amqp here is mandatory for configurer to ensure class is loaded before applying OpenClass
+  require "amqp"
+  require File.join(File.dirname(__FILE__), 'amqp')
+  amqp_version = Gem.loaded_specs['amqp'].version
+  if amqp_version >= Gem::Version.new(AutoReconfiguration::SUPPORTED_AMQP_VERSION)
+    if AutoReconfiguration::ConfigurationHelper.disabled? :rabbitmq
+      puts "RabbitMQ auto-reconfiguration disabled."
+      class << AMQP
+        #Remove introduced aliases and methods.
+        #This is mostly for testing, as we don't expect this script
+        #to run twice during a single app startup
+        if method_defined?(:connect_with_cf)
+          undef_method :connect_with_cf
+          alias :connect :original_connect
+        end
+      end
+    elsif AMQP.public_methods.index :connect_with_cf
+      puts "AMQP AutoReconfiguration already included."
+    else
+      #Introduce around alias into AMQP class
+      class << AMQP
+        include AutoReconfiguration::AMQP
+       end
+    end
+  else
+    puts "Auto-reconfiguration not supported for this AMQP version.  " +
+      "Found: #{amqp_version}.  Required: #{AutoReconfiguration::SUPPORTED_AMQP_VERSION} or higher."
+  end
+rescue LoadError
+  puts "No AMQP Library Found. Skipping auto-reconfiguration."
+end

data/lib/cfautoconfig/messaging/carrot.rb

@@ -0,0 +1,24 @@
+require 'cfruntime/properties'
+require 'cfruntime/carrot'
+
+module AutoReconfiguration
+  SUPPORTED_CARROT_VERSION = '1.0'
+  module Carrot
+    def self.included( base )
+      base.send( :alias_method, :original_initialize, :initialize)
+      base.send( :alias_method, :initialize, :initialize_with_cf )
+    end
+
+    def initialize_with_cf(opts = {})
+      service_names = CFRuntime::CloudApp.service_names_of_type('rabbitmq')
+      if service_names.length == 1
+        puts "Auto-reconfiguring Carrot."
+        cfopts = CFRuntime::CarrotClient.options_for_svc(service_names[0],opts)
+        original_initialize cfopts
+      else
+        puts "Found #{service_names.length} rabbitmq services. Skipping auto-reconfiguration."
+        original_initialize opts
+      end
+    end
+  end
+end

data/lib/cfautoconfig/messaging/carrot_configurer.rb

@@ -0,0 +1,33 @@
+require 'cfautoconfig/configuration_helper'
+begin
+  #Require carrot here is mandatory for configurer to ensure class is loaded before applying OpenClass
+  require "carrot"
+  require File.join(File.dirname(__FILE__), 'carrot')
+  carrot_version = Gem.loaded_specs['carrot'].version
+  if carrot_version >= Gem::Version.new(AutoReconfiguration::SUPPORTED_CARROT_VERSION)
+    if AutoReconfiguration::ConfigurationHelper.disabled? :rabbitmq
+      puts "RabbitMQ auto-reconfiguration disabled."
+      class Carrot
+        #Remove introduced aliases and methods.
+        #This is mostly for testing, as we don't expect this script
+        #to run twice during a single app startup
+        if method_defined?(:initialize_with_cf)
+          undef_method :initialize_with_cf
+          alias :initialize :original_initialize
+        end
+      end
+    elsif Carrot.public_instance_methods.index :initialize_with_cf
+      puts "Carrot AutoReconfiguration already included."
+    else
+      #Introduce around alias into Carrot class
+      class Carrot
+        include AutoReconfiguration::Carrot
+       end
+    end
+  else
+    puts "Auto-reconfiguration not supported for this Carrot version.  " +
+      "Found: #{carrot_version}.  Required: #{AutoReconfiguration::SUPPORTED_CARROT_VERSION} or higher."
+  end
+rescue LoadError
+  puts "No Carrot Library Found. Skipping auto-reconfiguration."
+end

data/lib/cfautoconfig/relational/mysql.rb

@@ -0,0 +1,27 @@
+require 'cfruntime/properties'
+require 'cfruntime/mysql'
+
+module AutoReconfiguration
+  SUPPORTED_MYSQL2_VERSION = '0.2.7'
+  module Mysql
+
+    def self.included( base )
+      base.send( :alias_method, :original_initialize, :initialize)
+      base.send( :alias_method, :initialize, :initialize_with_cf )
+    end
+     
+    def initialize_with_cf(opts = {})
+      service_names = CFRuntime::CloudApp.service_names_of_type('mysql')
+      if service_names.length == 1
+        puts "Auto-reconfiguring MySQL"
+        original_initialize(CFRuntime::Mysql2Client.options_for_svc(service_names[0],opts))
+      else
+        puts "Found #{service_names.length} mysql services. Skipping auto-reconfiguration."
+        original_initialize opts
+      end
+    end
+  end
+end
+
+
+