powcloud-sprinkle 0.3.0

data/lib/sprinkle/actors/capistrano.rb

@@ -0,0 +1,140 @@
+require 'capistrano/cli'
+
+module Sprinkle
+  module Actors
+    # = Capistrano Delivery Method
+    #
+    # Capistrano is one of the delivery method options available out of the
+    # box with Sprinkle. If you have the capistrano gem install, you may use
+    # this delivery. The only configuration option available, and which is 
+    # mandatory to include is +recipes+. An example:
+    #
+    #   deployment do
+    #     delivery :capistrano do
+    #       recipes 'deploy'
+    #     end
+    #   end
+    #
+    # Recipes is given a list of files which capistrano 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 Capistrano
+      attr_accessor :config, :loaded_recipes #:nodoc:
+
+      def initialize(&block) #:nodoc:
+        @config = ::Capistrano::Configuration.new
+        @config.logger.level = Sprinkle::OPTIONS[:verbose] ? ::Capistrano::Logger::INFO : ::Capistrano::Logger::IMPORTANT
+        @config.set(:password) { ::Capistrano::CLI.password_prompt }
+        
+        @config.set(:_sprinkle_actor, self)
+        
+        def @config.recipes(script)
+          _sprinkle_actor.recipes(script)
+        end
+        
+        if block
+          @config.instance_eval &block
+        else
+          @config.load 'deploy' # normally in the config directory for rails
+        end
+      end
+
+      # Defines a recipe file which will be included by capistrano. Use these
+      # recipe files to set capistrano specific configurations. Default recipe
+      # included is "deploy." But if any other recipe is specified, it will
+      # include that instead. Multiple recipes may be specified through multiple
+      # recipes calls, an example:
+      #
+      #   deployment do
+      #     delivery :capistrano do
+      #       recipes 'deploy'
+      #       recipes 'magic_beans'
+      #     end
+      #   end
+      def recipes(script)
+        @loaded_recipes ||= []
+        @config.load script
+        @loaded_recipes << script
+      end
+
+      def process(name, commands, roles, suppress_and_return_failures = false) #:nodoc:
+        define_task(name, roles) do
+          via = fetch(:run_method, :sudo)
+          commands.each do |command|
+            invoke_command command, :via => via
+          end
+        end
+        
+        begin
+          run(name)
+          return true
+        rescue ::Capistrano::CommandError => e
+          return false if suppress_and_return_failures
+          
+          # Reraise error if we're not suppressing it
+          raise
+        end
+      end
+
+			def transfer(name, source, destination, roles, recursive = true, suppress_and_return_failures = false)
+        define_task(name, roles) do
+          upload source, destination, :via => :scp, :recursive => recursive
+        end
+        
+        begin
+          run(name)
+          return true
+        rescue ::Capistrano::CommandError => e
+          return false if suppress_and_return_failures
+          
+          # Reraise error if we're not suppressing it
+          raise
+        end
+			end
+			
+      private
+
+        # REVISIT: can we set the description somehow?
+        def define_task(name, roles, &block)
+          @config.task task_sym(name), :roles => roles, &block
+        end
+
+        def run(task)
+          @config.send task_sym(task)
+        end
+
+        def task_sym(name)
+          "install_#{name.to_task_name}".to_sym
+        end
+    end
+  end
+end
+
+
+=begin
+
+# channel: the SSH channel object used for this response
+# stream: either :err or :out, for stderr or stdout responses
+# output: the text that the server is sending, might be in chunks
+run "apt-get update" do |channel, stream, output|
+   if output =~ /Are you sure?/
+     answer = Capistrano::CLI.ui.ask("Are you sure: ")
+     channel.send_data(answer + "\n")
+   else
+     # allow the default callback to be processed
+     Capistrano::Configuration.default_io_proc.call[channel, stream, output]
+   end
+ end
+
+
+
+ You can tell subversion to use a different username+password by
+ setting a couple variables:
+    set :svn_username, "my svn username"
+    set :svn_password, "my svn password"
+ If you don't want to set the password explicitly in your recipe like
+ that, you can make capistrano prompt you for it like this:
+    set(:svn_password) { Capistrano::CLI.password_prompt("Subversion
+ password: ") }
+ - Jamis
+=end

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,123 @@
+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?
+        process_direct(name, commands, roles)
+      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)
+          on_gateway do |gateway|
+            Array(roles).each { |role| execute_on_role(commands, role, gateway) }
+          end
+        end
+        
+        def process_direct(name, commands, roles)
+          Array(roles).each { |role| execute_on_role(commands, role) }
+        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]
+          Array(hosts).each { |host| execute_on_host(commands, host, gateway) }
+        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)
+          if gateway # SSH connection via gateway
+            gateway.ssh(host, @options[:user]) do |ssh|
+              execute_on_connection(commands, ssh)
+            end
+          else # direct SSH connection
+            Net::SSH.start(host, @options[:user], :password => @options[:password]) do |ssh|
+              execute_on_connection(commands, ssh)
+            end
+          end
+        end
+        
+        def execute_on_connection(commands, connection)
+          Array(commands).each do |command|
+            connection.exec! command do |ch, stream, data|
+              logger.send((stream == :stderr ? 'error' : 'debug'), data)
+            end
+          end
+        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 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,78 @@
+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 capistrano 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 ||= []
+        self.load name
+        @loaded_recipes << script
+      end
+
+      def process(name, commands, roles, suppress_and_return_failures = false) #:nodoc:
+        commands = commands.join ' && ' if commands.is_a? Array
+        t = remote_task(task_sym(name), :roles => roles) { run commands }
+        
+        begin
+          t.invoke
+          return true
+        rescue ::Vlad::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 ::Vlad::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