dhill-sprinkle 0.3.3.1

data/lib/sprinkle/actors/local.rb

@@ -0,0 +1,37 @@
+module Sprinkle
+  module Actors
+    # = Local Delivery Method
+    #
+    # This actor implementation performs any given commands on your local system, as
+    # opposed to other implementations that generally run commands on a remote system
+    # via the network.
+    #
+    # This is useful if you'd like to use Sprinkle to provision your local machine. 
+    # To enable this actor, in your Sprinkle script specify the :local delivery mechanism. 
+    #
+    #   deployment do
+    #     delivery :local
+    #   end
+    #
+    # Note, your local machine will be assumed to be a member of all roles when applying policies
+    #
+    class Local
+      
+      def process(name, commands, roles, suppress_and_return_failures = false) #:nodoc:
+        commands.each do |command|
+          system command
+          return false if $?.to_i != 0
+        end
+        return true
+      end
+      
+			def transfer(name, source, destination, roles, recursive = true, suppress_and_return_failures = false)
+				if recursive
+					flags = "-R "
+				end
+				
+				system "cp #{flags}#{source} #{destination}"
+			end
+    end
+  end
+end

data/lib/sprinkle/actors/ssh.rb

@@ -0,0 +1,177 @@
+require 'net/ssh/gateway'
+require 'net/scp'
+
+module Sprinkle
+  module Actors
+    class Ssh
+      attr_accessor :options
+
+      def initialize(options = {}, &block) #:nodoc:
+        @options = options.update(:user => 'root')
+        self.instance_eval &block if block
+      end
+
+      def roles(roles)
+        @options[:roles] = roles
+      end
+      
+      def gateway(gateway)
+        @options[:gateway] = gateway
+      end
+      
+      def user(user)
+        @options[:user] = user
+      end
+
+      def password(password)
+        @options[:password] = password
+      end
+
+      def process(name, commands, roles, suppress_and_return_failures = false)
+        return process_with_gateway(name, commands, roles) if gateway_defined?
+        r = process_direct(name, commands, roles)
+        logger.debug green "process returning #{r}"
+        return r
+      end
+
+      def transfer(name, source, destination, roles, recursive = true, suppress_and_return_failures = false)
+        return transfer_with_gateway(name, source, destination, roles, recursive) if gateway_defined?
+        transfer_direct(name, source, destination, roles, recursive)
+      end
+			
+      protected
+      
+        def process_with_gateway(name, commands, roles)
+          res = []
+          on_gateway do |gateway|
+            Array(roles).each { |role| res << execute_on_role(commands, role, gateway) }
+          end
+          !(res.include? false)
+        end
+        
+        def process_direct(name, commands, roles)
+          res = []
+          Array(roles).each { |role| res << execute_on_role(commands, role) }
+          !(res.include? false)
+        end
+        
+        def transfer_with_gateway(name, source, destination, roles, recursive)
+          on_gateway do |gateway|
+            Array(roles).each { |role| transfer_to_role(source, destination, role, recursive, gateway) }
+          end
+        end
+        
+        def transfer_direct(name, source, destination, roles, recursive)
+          Array(roles).each { |role| transfer_to_role(source, destination, role, recursive) }
+        end
+
+        def execute_on_role(commands, role, gateway = nil)
+          hosts = @options[:roles][role]
+          res = []
+          Array(hosts).each { |host| res << execute_on_host(commands, host, gateway) }
+          !(res.include? false)
+        end
+
+        def transfer_to_role(source, destination, role, gateway = nil)
+          hosts = @options[:roles][role]
+          Array(hosts).each { |host| transfer_to_host(source, destination, host, gateway) }
+        end
+        
+        def execute_on_host(commands, host, gateway = nil)
+          res = nil
+          logger.debug(blue "executing #{commands.inspect} on #{host}.")
+          if gateway # SSH connection via gateway
+            gateway.ssh(host, @options[:user]) do |ssh|
+              res = execute_on_connection(commands, ssh)
+              ssh.loop
+            end
+          else # direct SSH connection
+            Net::SSH.start(host, @options[:user], :password => @options[:password]) do |ssh|
+              res = execute_on_connection(commands, ssh)
+              ssh.loop
+            end
+          end
+          res.detect{|x| x!=0}.nil?
+        end
+
+        def execute_on_connection(commands, session)
+          res = []
+          Array(commands).each do |cmd|
+            session.open_channel do |channel|
+              channel.on_data do |ch, data|
+                logger.debug yellow("stdout said-->\n#{data}\n")
+              end
+              channel.on_extended_data do |ch, type, data|
+                next unless type == 1  # only handle stderr
+                logger.debug red("stderr said -->\n#{data}\n")
+              end
+
+              channel.on_request("exit-status") do |ch, data|
+                exit_code = data.read_long
+                if exit_code == 0
+                  logger.debug(green 'success')
+                else
+                  logger.debug(red('failed (%d).'%exit_code))
+                end
+                res << exit_code
+              end
+
+              channel.on_request("exit-signal") do |ch, data|
+                logger.debug red("#{cmd} was signaled!: #{data.read_long}")
+              end
+
+              channel.exec cmd  do  |ch, status|
+                logger.error("couldn't run remote command #{cmd}") unless status
+              end
+            end
+          end
+          res
+        end
+
+        def transfer_to_host(source, destination, host, recursive, gateway = nil)
+          if gateway # SSH connection via gateway
+            gateway.ssh(host, @options[:user]) do |ssh|
+              transfer_on_connection(source, destination, recursive, ssh)
+            end
+          else # direct SSH connection
+            Net::SSH.start(host, @options[:user]) do |ssh|
+              transfer_on_connection(source, destination, recursive, ssh)
+            end
+          end
+        end
+
+        def transfer_on_connection(source, destination, recursive, connection)
+          scp = Net::SCP.new(connection)
+          scp.upload! source, destination, :recursive => recursive
+        end
+
+        private
+        def color(code, s)
+          "\033[%sm%s\033[0m"%[code,s]
+        end
+        def red(s)
+          color(31, s)
+        end
+        def yellow(s)
+          color(33, s)
+        end
+        def green(s)
+          color(32, s)
+        end
+        def blue(s)
+          color(34, s)
+        end
+
+        def gateway_defined?
+          !! @options[:gateway]
+        end
+
+        def on_gateway(&block)
+          gateway = Net::SSH::Gateway.new(@options[:gateway], @options[:user])
+          block.call gateway
+        ensure
+          gateway.shutdown!
+        end
+    end
+  end
+end

data/lib/sprinkle/actors/vlad.rb

@@ -0,0 +1,83 @@
+module Sprinkle
+  module Actors
+    # = Vlad Delivery Method
+    #
+    # Vlad is one of the delivery method options available out of the
+    # box with Sprinkle. If you have the vlad the deployer gem install, you 
+    # may use this delivery. The only configuration option available, and 
+    # which is mandatory to include is +script+. An example:
+    #
+    #   deployment do
+    #     delivery :vlad do
+    #       script 'deploy'
+    #     end
+    #   end
+    #
+    # script is given a list of files which vlad will include and load.
+    # These recipes are mainly to set variables such as :user, :password, and to 
+    # set the app domain which will be sprinkled.
+    class Vlad
+      require 'vlad'
+      attr_accessor :loaded_recipes #:nodoc:
+
+      def initialize(&block) #:nodoc:
+        self.instance_eval &block if block
+      end
+
+      # Defines a script file which will be included by vlad. Use these
+      # script files to set vlad specific configurations. Multiple scripts
+      # may be specified through multiple script calls, an example:
+      #
+      #   deployment do
+      #     delivery :vlad do
+      #       script 'deploy'
+      #       script 'magic_beans'
+      #     end
+      #   end
+      def script(name)
+        @loaded_recipes ||= []
+        require name
+        @loaded_recipes << name
+      end
+
+      def process(name, commands, roles, suppress_and_return_failures = false) #:nodoc:
+        commands = Array(commands)
+        if use_sudo
+          commands = commands.map{|x| "sudo #{x}"}
+        end
+        commands = commands.join(' && ')
+        puts "executing #{commands}"
+        t = remote_task(task_sym(name), :roles => roles) { run commands }
+        
+        begin
+          t.invoke
+          return true
+        rescue ::Rake::CommandFailedError => e
+          return false if suppress_and_return_failures
+          
+          # Reraise error if we're not suppressing it
+          raise
+        end
+      end
+
+			# Sorry, all transfers are recursive
+      def transfer(name, source, destination, roles, recursive = true, suppress_and_return_failures = false) #:nodoc:
+        begin
+					rsync source, destination
+          return true
+        rescue ::Rake::CommandFailedError => e
+          return false if suppress_and_return_failures
+          
+          # Reraise error if we're not suppressing it
+          raise
+        end
+      end
+
+      private
+
+        def task_sym(name)
+          "install_#{name.to_task_name}".to_sym
+        end
+    end
+  end
+end

data/lib/sprinkle/configurable.rb

@@ -0,0 +1,31 @@
+module Sprinkle
+  #--
+  # TODO: Possible documentation?
+  #++
+  module Configurable #:nodoc:
+    attr_accessor :delivery
+    
+    def defaults(deployment)
+      defaults = deployment.defaults[self.class.name.split(/::/).last.downcase.to_sym]
+      self.instance_eval(&defaults) if defaults
+      @delivery = deployment.style
+    end
+    
+    def assert_delivery
+      raise 'Unknown command delivery target' unless @delivery
+    end
+    
+    def method_missing(sym, *args, &block)
+      unless args.empty? # mutate if not set
+        @options ||= {}
+        @options[sym] = *args unless @options[sym]
+      end
+
+      @options[sym] || @package.send(sym, *args, &block) # try the parents options if unknown
+    end
+    
+    def option?(sym)
+      !@options[sym].nil?
+    end
+  end
+end

data/lib/sprinkle/deployment.rb

@@ -0,0 +1,73 @@
+module Sprinkle
+  # Deployment blocks specify deployment specific information about a 
+  # sprinkle script. An example:
+  #
+  #   deployment do
+  #     # mechanism for deployment
+  #     delivery :capistrano do
+  #       recipes 'deploy'
+  #     end
+  # 
+  #     # source based package installer defaults
+  #     source do
+  #       prefix   '/usr/local'
+  #       archives '/usr/local/sources'
+  #       builds   '/usr/local/build'
+  #     end
+  #   end
+  #
+  # What the above example does is tell sprinkle that we will be using
+  # *capistrano* (Sprinkle::Actors::Capistrano) for deployment and
+  # everything within the block is capistrano specific configuration.
+  # For more information on what options are available, check the corresponding
+  # Sprinkle::Actors doc page.
+  #
+  # In addition to what delivery mechanism we're using, we specify some
+  # configuration options for the "source" command. The only things
+  # configurable, at this time, in the deployment block other than
+  # the delivery method are installers. If installers are configurable,
+  # they will say so on their corresponding documentation page. See
+  # Sprinkle::Installers
+  #
+  # <b>Only one deployment block is on any given sprinkle script</b>
+  module Deployment
+    # The method outlined above which specifies deployment specific information
+    # for a sprinkle script. For more information, read the header of this module.
+    def deployment(&block)
+      @deployment = Deployment.new(&block)
+    end
+
+    class Deployment
+      attr_accessor :style, :defaults #:nodoc:
+
+      def initialize(&block) #:nodoc:
+        @defaults = {}
+        self.instance_eval(&block)
+        raise 'No delivery mechanism defined' unless @style
+      end
+
+      # Specifies which Sprinkle::Actors to use for delivery. Although all
+      # actors jobs are the same: to run remote commands on a server, you
+      # may have a personal preference. The block you pass is used to configure
+      # the actor. For more information on what configuration options are
+      # available, view the corresponding Sprinkle::Actors page.
+      def delivery(type, &block) #:doc:
+        @style = ("Sprinkle::Actors::" + type.to_s.titleize).constantize.new &block
+      end
+
+      def method_missing(sym, *args, &block) #:nodoc:
+        @defaults[sym] = block
+      end
+
+      def respond_to?(sym) #:nodoc:
+        !!@defaults[sym]
+      end
+
+      def process #:nodoc:
+        POLICIES.each do |policy|
+          policy.process(self)
+        end
+      end
+    end
+  end
+end

data/lib/sprinkle/extensions/arbitrary_options.rb

@@ -0,0 +1,10 @@
+module ArbitraryOptions #:nodoc:
+  def self.included(base)
+    base.alias_method_chain :method_missing, :arbitrary_options
+  end
+  
+  def method_missing_with_arbitrary_options(sym, *args, &block)
+    self.class.dsl_accessor sym
+    send(sym, *args, &block)
+  end
+end

data/lib/sprinkle/extensions/array.rb

@@ -0,0 +1,5 @@
+class Array #:nodoc:
+  def to_task_name
+    collect(&:to_task_name).join('_')
+  end
+end

data/lib/sprinkle/extensions/blank_slate.rb

@@ -0,0 +1,5 @@
+class BlankSlate #:nodoc:
+  instance_methods.each do |m| 
+    undef_method(m) unless %w( __send__ __id__ send class inspect instance_eval instance_variables ).include?(m)
+  end
+end

data/lib/sprinkle/extensions/dsl_accessor.rb

@@ -0,0 +1,15 @@
+class Module #:nodoc:
+  def dsl_accessor(*symbols)
+    symbols.each do |sym|
+      class_eval %{
+        def #{sym}(*val)
+          if val.empty?
+            @#{sym}
+          else
+            @#{sym} = val.size == 1 ? val[0] : val
+          end
+        end
+      }
+    end
+  end
+end