le1t0-capistrano 2.5.18.001

data/lib/capistrano/cli/help.txt

@@ -0,0 +1,78 @@
+-----------------------------
+<%= color('Capistrano', :bold) %>
+-----------------------------
+
+Capistrano is a utility for automating the execution of commands across multiple remote machines. It was originally conceived as an aid to deploy Ruby on Rails web applications, but has since evolved to become a much more general-purpose tool.
+
+The command-line interface to Capistrano is via the `cap' command.
+
+	cap [ option ] ... action ...
+
+The following options are understood:
+
+  <%= color '-d, --debug', :bold %>
+	Causes Capistrano to pause and prompt before executing any remote command, giving the user the option to either execute the command, skip the command, or abort execution entirely. This makes it a great way to troubleshoot tasks, or test custom tasks, by executing commands one at a time and checking the server to make sure they worked as expected before moving on to the next command. (Compare this to the --dry-run command.)
+
+  <%= color '-e, --explain TASK', :bold %>
+	Displays the extended description of the given task. Not all tasks will have an extended description, but for those that do, this can provide a wealth of additional usage information, such as describing environment variables or settings that can affect the execution of the task.
+
+  <%= color '-F, --default-config', :bold %>
+	By default, cap will search for a config file named `Capfile' or `capfile' in the current directory, or in any parent directory, and will automatically load it. However, if you specify the -f flag (see below), cap will use that file instead of the default config. If you want to use both the default config, and files loaded via -f, you can specify -F to force cap to search for and load the default config, even if additional files were specified via -f.
+
+  <%= color '-f, --file FILE', :bold %>
+	Causes the named file to be loaded. Capistrano will search both its own recipe directory, as well as the current directory, looking for the named file. An ".rb" extension is optional. The -f option may be given any number of times, but if it is given, it will take the place of the normal `Capfile' or `capfile' detection. Use -F if you want the default capfile to be loaded when you use -f.
+
+  <%= color '-H, --long-help', :bold %>
+	Displays this document and exits.
+
+  <%= color '-h, --help', :bold %>
+	Shows a brief summary of these options and exits.
+
+  <%= color '-l, --logger [STDERR|STDOUT|file]', :bold %>
+	Change the file used to print the output. It offers three options: standard error(stderr), standard output and file. Options are not case sensitive. By default Capistrano uses stderr.
+
+  <%= color '-n, --dry-run', :bold %>
+	Causes Capistrano to simply display each remote command, without executing it. In this sense it is similar to --debug, but without the prompt. Note that commands executed locally are still run--only remote commands are skipped.
+  
+  <%= color '-p, --password', :bold %>
+	Normally, cap will prompt for the password on-demand, the first time it is needed. This can make it hard to walk away from Capistrano, since you might not know if it will prompt for a password down the road. In such cases, you can use the -p option to force cap to prompt for the password immediately.
+
+  <%= color '-q, --quiet', :bold %>
+	Display only critical error messages. All other output is suppressed.
+
+  <%= color '-S, --set-before NAME=VALUE', :bold %>
+	Sets the given variable to the given value, before loading any recipe files. This is useful if you have a recipe file that depends on a certain variable being set, at the time it is loaded.
+
+  <%= color '-s, --set NAME=VALUE', :bold %>
+	Sets the given variable to the given value, after loading all recipe files. This is useful when you want to override the value of a variable which is used in a task. Note that this will set the variables too late for them to affect conditions that are executed as the recipes are loaded.
+
+  <%= color '-T, --tasks PATTERN', :bold %>
+	Displays the list of all tasks (matching optional PATTERN) in all loaded recipe files. If a task has no description, or if the description starts with the [internal] tag, the task will not be listed unless you also specify -v.
+
+  <%= color '-t, --tool', :bold %>
+	Abbreviates the output of -T for integration with other tools. Without -t, -T will list tasks with their summaries, and may include additional instructive text at the bottom. When integrating with other tools (e.g., bash auto-expansion and the like) that additional text can get in the way. This switch makes it easier for those tools to parse the list of tasks. (The -t switch has no effect if the -T switch is not specified.)
+
+  <%= color '-V, --version', :bold %>
+	Shows the current Capistrano version number and exits.
+
+  <%= color '-v, --verbose', :bold %>
+	Increase the verbosity. You can specify this option up to three times to further increase verbosity. By default, cap will use maximum verbosity, but if you specify an explicit verbosity, that will be used instead. See also -q.
+
+  <%= color '-X, --skip-system-config', :bold %>
+	By default, cap will look for and (if it exists) load the global system configuration file located in /etc/capistrano.conf. If you don't want cap to load that file, give this option.
+
+  <%= color '-x, --skip-user-config', :bold %>
+	By default, cap will look for and (if it exists) load the user-specific configuration file located in $HOME/.caprc. If you don't want cap to load that file, give this option.
+
+-----------------------------
+<%= color('Environment Variables', :bold) %>
+-----------------------------
+
+  <%= color 'HOSTS', :bold %>
+	Execute the tasks against this comma-separated list of hosts.  Effectively, this makes the host(s) part of every roles.
+
+  <%= color 'HOSTFILTER', :bold %>
+	Execute tasks against this comma-separated list of host, but only if the host has the proper role for the task.
+
+  <%= color 'ROLES', :bold %>
+	Execute tasks against this comma-separated list of roles.  Hosts which do not have the right roles will be skipped.

data/lib/capistrano/cli/options.rb

@@ -0,0 +1,243 @@
+require 'optparse'
+
+module Capistrano
+  class CLI
+    module Options
+      def self.included(base)
+        base.extend(ClassMethods)
+      end
+
+      module ClassMethods
+        # Return a new CLI instance with the given arguments pre-parsed and
+        # ready for execution.
+        def parse(args)
+          cli = new(args)
+          cli.parse_options!
+          cli
+        end
+      end
+
+      # The hash of (parsed) command-line options
+      attr_reader :options
+
+      # Return an OptionParser instance that defines the acceptable command
+      # line switches for Capistrano, and what their corresponding behaviors
+      # are.
+      def option_parser #:nodoc:
+        @logger = Logger.new
+        @option_parser ||= OptionParser.new do |opts|
+          opts.banner = "Usage: #{File.basename($0)} [options] action ..."
+
+          opts.on("-d", "--debug",
+            "Prompts before each remote command execution."
+          ) { |value| options[:debug] = true }
+
+          opts.on("-e", "--explain TASK",
+            "Displays help (if available) for the task."
+          ) { |value| options[:explain] = value }
+
+          opts.on("-F", "--default-config",
+            "Always use default config, even with -f."
+          ) { options[:default_config] = true }
+
+          opts.on("-f", "--file FILE",
+            "A recipe file to load. May be given more than once."
+          ) { |value| options[:recipes] << value }
+
+          opts.on("-H", "--long-help", "Explain these options and environment variables.") do
+            long_help
+            exit
+          end
+
+          opts.on("-h", "--help", "Display this help message.") do
+            puts opts
+            exit
+          end
+
+					opts.on("-l", "--logger [STDERR|STDOUT|file]", 
+						"Choose logger method. STDERR used by default."
+					) do |value| 
+						options[:output] = if value.nil? || value.upcase == 'STDERR'
+																 # Using default logger.
+																 nil
+															 elsif value.upcase == 'STDOUT'
+																 $stdout
+															 else
+																 value
+															 end
+					end
+
+          opts.on("-n", "--dry-run",
+            "Prints out commands without running them."
+          ) { |value| options[:dry_run] = true }
+
+          opts.on("-p", "--password",
+            "Immediately prompt for the password."
+          ) { options[:password] = nil }
+
+          opts.on("-q", "--quiet",
+            "Make the output as quiet as possible."
+          ) { options[:verbose] = 0 }
+
+          opts.on("-r", "--preserve-roles",
+            "Preserve task roles"
+          ) { options[:preserve_roles] = true }
+
+          opts.on("-S", "--set-before NAME=VALUE",
+            "Set a variable before the recipes are loaded."
+          ) do |pair|
+            name, value = pair.split(/=/, 2)
+            options[:pre_vars][name.to_sym] = value
+          end
+
+          opts.on("-s", "--set NAME=VALUE",
+            "Set a variable after the recipes are loaded."
+          ) do |pair|
+            name, value = pair.split(/=/, 2)
+            options[:vars][name.to_sym] = value
+          end
+
+          opts.on("-T", "--tasks [PATTERN]",
+            "List all tasks (matching optional PATTERN) in the loaded recipe files."
+          ) do |value|
+            options[:tasks] = if value
+              value
+            else
+              true
+            end
+            options[:verbose] ||= 0
+          end
+
+          opts.on("-t", "--tool",
+            "Abbreviates the output of -T for tool integration."
+          ) { options[:tool] = true }
+
+          opts.on("-V", "--version",
+            "Display the Capistrano version, and exit."
+          ) do
+            require 'capistrano/version'
+            puts "Capistrano v#{Capistrano::Version}"
+            exit
+          end
+
+          opts.on("-v", "--verbose",
+            "Be more verbose. May be given more than once."
+          ) do
+            options[:verbose] ||= 0
+            options[:verbose] += 1
+          end
+
+          opts.on("-X", "--skip-system-config",
+            "Don't load the system config file (capistrano.conf)"
+          ) { options.delete(:sysconf) }
+
+          opts.on("-x", "--skip-user-config",
+            "Don't load the user config file (.caprc)"
+          ) { options.delete(:dotfile) }
+        end
+      end
+
+      # If the arguments to the command are empty, this will print the
+      # allowed options and exit. Otherwise, it will parse the command
+      # line and set up any default options.
+      def parse_options! #:nodoc:
+        @options = { :recipes => [], :actions => [],
+          :vars => {}, :pre_vars => {},
+          :sysconf => default_sysconf, :dotfile => default_dotfile }
+
+        if args.empty?
+          warn "Please specify at least one action to execute."
+          warn option_parser
+          exit
+        end
+
+        option_parser.parse!(args)
+
+        coerce_variable_types!
+
+        # if no verbosity has been specified, be verbose
+        options[:verbose] = 3 if !options.has_key?(:verbose)
+
+        look_for_default_recipe_file! if options[:default_config] || options[:recipes].empty?
+        extract_environment_variables!
+
+        options[:actions].concat(args)
+
+        password = options.has_key?(:password)
+        options[:password] = Proc.new { self.class.password_prompt }
+        options[:password] = options[:password].call if password
+      end
+
+      # Extracts name=value pairs from the remaining command-line arguments
+      # and assigns them as environment variables.
+      def extract_environment_variables! #:nodoc:
+        args.delete_if do |arg|
+          next unless arg.match(/^(\w+)=(.*)$/)
+          ENV[$1] = $2
+        end
+      end
+
+      # Looks for a default recipe file in the current directory.
+      def look_for_default_recipe_file! #:nodoc:
+        current = Dir.pwd
+
+        loop do
+          %w(Capfile capfile).each do |file|
+            if File.file?(file)
+              options[:recipes] << file
+              @logger.info "Using recipes from #{File.join(current,file)}"
+              return
+            end
+          end
+
+          pwd = Dir.pwd
+          Dir.chdir("..")
+          break if pwd == Dir.pwd # if changing the directory made no difference, then we're at the top
+        end
+
+        Dir.chdir(current)
+      end
+
+      def default_sysconf #:nodoc:
+        File.join(sysconf_directory, "capistrano.conf")
+      end
+      
+      def default_dotfile #:nodoc:
+        File.join(home_directory, ".caprc")
+      end
+
+      def sysconf_directory #:nodoc:
+        # TODO if anyone cares, feel free to submit a patch that uses a more
+        # appropriate location for this file in Windows.
+        ENV["SystemRoot"] || '/etc'
+      end
+      
+      def home_directory #:nodoc:
+        ENV["HOME"] ||
+          (ENV["HOMEPATH"] && "#{ENV["HOMEDRIVE"]}#{ENV["HOMEPATH"]}") ||
+          "/"
+      end
+
+      def coerce_variable_types!
+        [:pre_vars, :vars].each do |collection|
+          options[collection].keys.each do |key|
+            options[collection][key] = coerce_variable(options[collection][key])
+          end
+        end
+      end
+
+      def coerce_variable(value)
+        case value
+        when /^"(.*)"$/ then $1
+        when /^'(.*)'$/ then $1
+        when /^\d+$/ then value.to_i
+        when /^\d+\.\d*$/ then value.to_f
+        when "true" then true
+        when "false" then false
+        when "nil" then nil
+        else value
+        end
+      end
+    end
+  end
+end

data/lib/capistrano/cli/ui.rb

@@ -0,0 +1,40 @@
+require 'highline'
+
+# work around problem where HighLine detects an eof on $stdin and raises an
+# error.
+HighLine.track_eof = false
+
+module Capistrano
+  class CLI
+    module UI
+      def self.included(base) #:nodoc:
+        base.extend(ClassMethods)
+      end
+
+      module ClassMethods
+        # Return the object that provides UI-specific methods, such as prompts
+        # and more.
+        def ui
+          @ui ||= HighLine.new
+        end
+
+        # Prompt for a password using echo suppression.
+        def password_prompt(prompt="Password: ")
+          ui.ask(prompt) { |q| q.echo = false }
+        end
+        
+        # Debug mode prompt
+        def debug_prompt(cmd)
+          ui.say("Preparing to execute command: #{cmd}")
+          prompt = "Execute ([Yes], No, Abort) "
+          ui.ask("#{prompt}?  ") do |q|
+            q.overwrite = false
+            q.default = 'y'
+            q.validate = /(y(es)?)|(no?)|(a(bort)?|\n)/i
+            q.responses[:not_valid] = prompt
+          end
+        end
+      end
+    end
+  end
+end

data/lib/capistrano/command.rb

@@ -0,0 +1,283 @@
+require 'capistrano/errors'
+require 'capistrano/processable'
+
+module Capistrano
+
+  # This class encapsulates a single command to be executed on a set of remote
+  # machines, in parallel.
+  class Command
+    include Processable
+
+    class Tree
+      attr_reader :configuration
+      attr_reader :branches
+      attr_reader :fallback
+
+      include Enumerable
+
+      class Branch
+        attr_accessor :command, :callback
+        attr_reader :options
+
+        def initialize(command, options, callback)
+          @command = command.strip.gsub(/\r?\n/, "\\\n")
+          @callback = callback || Capistrano::Configuration.default_io_proc
+          @options = options
+          @skip = false
+        end
+
+        def last?
+          options[:last]
+        end
+
+        def skip?
+          @skip
+        end
+
+        def skip!
+          @skip = true
+        end
+
+        def match(server)
+          true
+        end
+
+        def to_s
+          command.inspect
+        end
+      end
+
+      class ConditionBranch < Branch
+        attr_accessor :configuration
+        attr_accessor :condition
+
+        class Evaluator
+          attr_reader :configuration, :condition, :server
+
+          def initialize(config, condition, server)
+            @configuration = config
+            @condition = condition
+            @server = server
+          end
+
+          def in?(role)
+            configuration.roles[role].include?(server)
+          end
+
+          def result
+            eval(condition, binding)
+          end
+
+          def method_missing(sym, *args, &block)
+            if server.respond_to?(sym)
+              server.send(sym, *args, &block)
+            elsif configuration.respond_to?(sym)
+              configuration.send(sym, *args, &block)
+            else
+              super
+            end
+          end
+        end
+
+        def initialize(configuration, condition, command, options, callback)
+          @configuration = configuration
+          @condition = condition
+          super(command, options, callback)
+        end
+
+        def match(server)
+          Evaluator.new(configuration, condition, server).result
+        end
+
+        def to_s
+          "#{condition.inspect} :: #{command.inspect}"
+        end
+      end
+
+      def initialize(config)
+        @configuration = config
+        @branches = []
+        yield self if block_given?
+      end
+
+      def when(condition, command, options={}, &block)
+        branches << ConditionBranch.new(configuration, condition, command, options, block)
+      end
+
+      def else(command, &block)
+        @fallback = Branch.new(command, {}, block)
+      end
+
+      def branches_for(server)
+        seen_last = false
+        matches = branches.select do |branch|
+          success = !seen_last && !branch.skip? && branch.match(server)
+          seen_last = success && branch.last?
+          success
+        end
+
+        matches << fallback if matches.empty? && fallback
+        return matches
+      end
+
+      def each
+        branches.each { |branch| yield branch }
+        yield fallback if fallback
+        return self
+      end
+    end
+
+    attr_reader :tree, :sessions, :options
+
+    def self.process(tree, sessions, options={})
+      new(tree, sessions, options).process!
+    end
+
+    # Instantiates a new command object. The +command+ must be a string
+    # containing the command to execute. +sessions+ is an array of Net::SSH
+    # session instances, and +options+ must be a hash containing any of the
+    # following keys:
+    #
+    # * +logger+: (optional), a Capistrano::Logger instance
+    # * +data+: (optional), a string to be sent to the command via it's stdin
+    # * +env+: (optional), a string or hash to be interpreted as environment
+    #   variables that should be defined for this command invocation.
+    def initialize(tree, sessions, options={}, &block)
+      if String === tree
+        tree = Tree.new(nil) { |t| t.else(tree, &block) }
+      elsif block
+        raise ArgumentError, "block given with tree argument"
+      end
+
+      @tree = tree
+      @sessions = sessions
+      @options = options
+      @channels = open_channels
+    end
+
+    # Processes the command in parallel on all specified hosts. If the command
+    # fails (non-zero return code) on any of the hosts, this will raise a
+    # Capistrano::CommandError.
+    def process!
+      loop do
+        break unless process_iteration { @channels.any? { |ch| !ch[:closed] } }
+      end
+
+      logger.trace "command finished" if logger
+
+      if (failed = @channels.select { |ch| ch[:status] != 0 }).any?
+        commands = failed.inject({}) { |map, ch| (map[ch[:command]] ||= []) << ch[:server]; map }
+        message = commands.map { |command, list| "#{command.inspect} on #{list.join(',')}" }.join("; ")
+        error = CommandError.new("failed: #{message}")
+        error.hosts = commands.values.flatten
+        raise error
+      end
+
+      self
+    end
+
+    # Force the command to stop processing, by closing all open channels
+    # associated with this command.
+    def stop!
+      @channels.each do |ch|
+        ch.close unless ch[:closed]
+      end
+    end
+
+    private
+
+      def logger
+        options[:logger]
+      end
+
+      def open_channels
+        sessions.map do |session|
+          server = session.xserver
+          tree.branches_for(server).map do |branch|
+            session.open_channel do |channel|
+              channel[:server] = server
+              channel[:host] = server.host
+              channel[:options] = options
+              channel[:branch] = branch
+
+              request_pty_if_necessary(channel) do |ch, success|
+                if success
+                  logger.trace "executing command", ch[:server] if logger
+                  cmd = replace_placeholders(channel[:branch].command, ch)
+
+                  if options[:shell] == false
+                    shell = nil
+                  else
+                    shell = "#{options[:shell] || "sh"} -c"
+                    cmd = cmd.gsub(/'/) { |m| "'\\''" }
+                    cmd = "'#{cmd}'"
+                  end
+
+                  command_line = [environment, shell, cmd].compact.join(" ")
+                  ch[:command] = command_line
+
+                  ch.exec(command_line)
+                  ch.send_data(options[:data]) if options[:data]
+                else
+                  # just log it, don't actually raise an exception, since the
+                  # process method will see that the status is not zero and will
+                  # raise an exception then.
+                  logger.important "could not open channel", ch[:server] if logger
+                  ch.close
+                end
+              end
+
+              channel.on_data do |ch, data|
+                ch[:branch].callback[ch, :out, data]
+              end
+
+              channel.on_extended_data do |ch, type, data|
+                ch[:branch].callback[ch, :err, data]
+              end
+
+              channel.on_request("exit-status") do |ch, data|
+                ch[:status] = data.read_long
+              end
+
+              channel.on_close do |ch|
+                ch[:closed] = true
+              end
+            end
+          end
+        end.flatten
+      end
+
+      def request_pty_if_necessary(channel)
+        if options[:pty]
+          channel.request_pty do |ch, success|
+            yield ch, success
+          end
+        else
+          yield channel, true
+        end
+      end
+
+      def replace_placeholders(command, channel)
+        command.gsub(/\$CAPISTRANO:HOST\$/, channel[:host])
+      end
+
+      # prepare a space-separated sequence of variables assignments
+      # intended to be prepended to a command, so the shell sets
+      # the environment before running the command.
+      # i.e.: options[:env] = {'PATH' => '/opt/ruby/bin:$PATH',
+      #                        'TEST' => '( "quoted" )'}
+      # environment returns:
+      # "env TEST=(\ \"quoted\"\ ) PATH=/opt/ruby/bin:$PATH"
+      def environment
+        return if options[:env].nil? || options[:env].empty?
+        @environment ||= if String === options[:env]
+            "env #{options[:env]}"
+          else
+            options[:env].inject("env") do |string, (name, value)|
+              value = value.to_s.gsub(/[ "]/) { |m| "\\#{m}" }
+              string << " #{name}=#{value}"
+            end
+          end
+      end
+  end
+end