HighLine

# File lib/highline/string_extensions.rb, line 27
def self.String(s)
  HighLine::String.new(s)
end

# File lib/highline/style.rb, line 10
def self.Style(*args)
  args = args.compact.flatten
  if args.size==1
    arg = args.first
    if arg.is_a?(Style)
      Style.list[arg.name] || Style.index(arg)
    elsif arg.is_a?(::String) && arg =~ /^\e\[/ # arg is a code
      if styles = Style.code_index[arg]
        styles.first
      else
        Style.new(:code=>arg)
      end
    elsif style = Style.list[arg]
      style
    elsif HighLine.color_scheme && HighLine.color_scheme[arg]
      HighLine.color_scheme[arg]
    elsif arg.is_a?(Hash)
      Style.new(arg)
    elsif arg.to_s.downcase =~ /^rgb_([a-f0-9]{6})$/
      Style.rgb($1)
    elsif arg.to_s.downcase =~ /^on_rgb_([a-f0-9]{6})$/
      Style.rgb($1).on
    else
      raise NameError, "#{arg.inspect} is not a defined Style"
    end
  else
    name = args
    Style.list[name] || Style.new(:list=>args)
  end
end

# File lib/highline.rb, line 377
def self.color( string, *colors )
  return string unless self.use_color?
  Style(*colors).color(string)
end

# File lib/highline.rb, line 383
def self.color_code(*colors)
  Style(*colors).code
end

# File lib/highline.rb, line 80
def self.color_scheme
  @@color_scheme
end

# File lib/highline.rb, line 75
def self.color_scheme=( setting )
  @@color_scheme = setting
end

# File lib/highline/string_extensions.rb, line 108
def self.colorize_strings
  ::String.send(:include, StringExtensions)
end

# File lib/highline.rb, line 148
def self.const_missing(name)
  if name.to_s =~ /^(ON_)?(RGB_)([A-F0-9]{6})(_STYLE)?$/ # RGB color
    on = $1
    suffix = $4
    if suffix
      code_name = $1.to_s + $2 + $3
    else
      code_name = name.to_s
    end
    style_name = code_name + '_STYLE'
    style = Style.rgb($3)
    style = style.on if on
    const_set(style_name, style)
    const_set(code_name, style.code)
    if suffix
      style
    else
      style.code
    end
  else
    raise NameError, "Bad color or uninitialized constant #{name}"
  end
end

# File lib/highline.rb, line 176
def initialize( input = $stdin, output = $stdout,
                wrap_at = nil, page_at = nil, indent_size=3, indent_level=0 )
  @input   = input
  @output  = output

  @multi_indent = true
  @indent_size = indent_size
  @indent_level = indent_level

  self.wrap_at = wrap_at
  self.page_at = page_at

  @question = nil
  @answer   = nil
  @menu     = nil
  @header   = nil
  @prompt   = nil
  @gather   = nil
  @answers  = nil
  @key      = nil

  initialize_system_extensions if respond_to?(:initialize_system_extensions)
end

# File lib/highline.rb, line 54
def self.supports_rgb_color?
  true
end

# File lib/highline.rb, line 62
def self.track_eof=( setting )
  @@track_eof = setting
end

# File lib/highline.rb, line 67
def self.track_eof?
  @@track_eof
end

# File lib/highline.rb, line 398
def self.uncolor(string)
  Style.uncolor(string)
end

# File lib/highline.rb, line 42
def self.use_color=( setting )
  @@use_color = setting
end

# File lib/highline.rb, line 47
def self.use_color?
  @@use_color
end

# File lib/highline.rb, line 85
def self.using_color_scheme?
  not @@color_scheme.nil?
end

# File lib/highline.rb, line 222
def agree( yes_or_no_question, character = nil )
  ask(yes_or_no_question, lambda { |yn| yn.downcase[0] == yy}) do |q|
    q.validate                 = /\Ay(?:es)?|no?\Z/
    q.responses[:not_valid]    = 'Please enter "yes" or "no".'
    q.responses[:ask_on_error] = :question
    q.character                = character

    yield q if block_given?
  end
end

# File lib/highline.rb, line 247
def ask( question, answer_type = String, &details ) # :yields: question
  @question ||= Question.new(question, answer_type, &details)

  return gather if @question.gather

  # readline() needs to handle its own output, but readline only supports
  # full line reading.  Therefore if @question.echo is anything but true,
  # the prompt will not be issued. And we have to account for that now.
  # Also, JRuby-1.7's ConsoleReader.readLine() needs to be passed the prompt
  # to handle line editing properly.
  say(@question) unless ((JRUBY or @question.readline) and @question.echo == true)

  begin
    @answer = @question.answer_or_default(get_response)
    unless @question.valid_answer?(@answer)
      explain_error(:not_valid)
      raise QuestionError
    end

    @answer = @question.convert(@answer)

    if @question.in_range?(@answer)
      if @question.confirm
        # need to add a layer of scope to ask a question inside a
        # question, without destroying instance data
        context_change = self.class.new(@input, @output, @wrap_at, @page_at, @indent_size, @indent_level)
        if @question.confirm == true
          confirm_question = "Are you sure?  "
        else
          # evaluate ERb under initial scope, so it will have
          # access to @question and @answer
          template  = ERB.new(@question.confirm, nil, "%")
          confirm_question = template.result(binding)
        end
        unless context_change.agree(confirm_question)
          explain_error(nil)
          raise QuestionError
        end
      end

      @answer
    else
      explain_error(:not_in_range)
      raise QuestionError
    end
  rescue QuestionError
    retry
  rescue ArgumentError, NameError => error
    raise if error.is_a?(NoMethodError)
    if error.message =~ /ambiguous/
      # the assumption here is that OptionParser::Completion#complete
      # (used for ambiguity resolution) throws exceptions containing
      # the word 'ambiguous' whenever resolution fails
      explain_error(:ambiguous_completion)
    else
      explain_error(:invalid_type)
    end
    retry
  rescue Question::NoAutoCompleteMatch
    explain_error(:no_completion)
    retry
  ensure
    @question = nil    # Reset Question object.
  end
end

# File lib/highline.rb, line 328
def choose( *items, &details )
  @menu = @question = Menu.new(&details)
  @menu.choices(*items) unless items.empty?

  # Set auto-completion
  @menu.completion = @menu.options
  # Set _answer_type_ so we can double as the Question for ask().
  @menu.answer_type = if @menu.shell
    lambda do |command|    # shell-style selection
      first_word = command.to_s.split.first || ""

      options = @menu.options
      options.extend(OptionParser::Completion)
      answer = options.complete(first_word)

      if answer.nil?
        raise Question::NoAutoCompleteMatch
      end

      [answer.last, command.sub(/^\s*#{first_word}\s*/, "")]
    end
  else
    @menu.options          # normal menu selection, by index or name
  end

  # Provide hooks for ERb layouts.
  @header   = @menu.header
  @prompt   = @menu.prompt

  if @menu.shell
    selected = ask("Ignored", @menu.answer_type)
    @menu.select(self, *selected)
  else
    selected = ask("Ignored", @menu.answer_type)
    @menu.select(self, selected)
  end
end

# File lib/highline.rb, line 393
def color(*args)
  self.class.color(*args)
end

# File lib/highline.rb, line 388
def color_code(*colors)
  self.class.color_code(*colors)
end

# File lib/highline.rb, line 660
def indent(increase=1, statement=nil, multiline=nil)
  @indent_level += increase
  multi = @multi_indent
  @multi_indent = multiline unless multiline.nil?
  begin
      if block_given?
          yield self
      else
          say(statement)
      end
  rescue
      @multi_indent = multi
      @indent_level -= increase
      raise
  end
  @multi_indent = multi
  @indent_level -= increase
end

# File lib/highline.rb, line 653
def indentation
  return ' '*@indent_size*@indent_level
end

# File lib/highline.rb, line 440
def list( items, mode = :rows, option = nil )
  items = items.to_ary.map do |item|
    if item.nil?
      ""
    else
      ERB.new(item, nil, "%").result(binding)
    end
  end

  if items.empty?
    ""
  else
    case mode
    when :inline
      option = " or " if option.nil?

      if items.size == 1
        items.first
      else
        items[0..-2].join(", ") + "#{option}#{items.last}"
      end
    when :columns_across, :columns_down
      max_length = actual_length(
        items.max { |a, b| actual_length(a) <=> actual_length(b) }
      )

      if option.nil?
        limit  = @wrap_at || 80
        option = (limit + 2) / (max_length + 2)
      end

      items = items.map do |item|
        pad = max_length + (item.to_s.length - actual_length(item))
        "%-#{pad}s" % item
      end
      row_count = (items.size / option.to_f).ceil

      if mode == :columns_across
        rows = Array.new(row_count) { Array.new }
        items.each_with_index do |item, index|
          rows[index / option] << item
        end

        rows.map { |row| row.join("  ") + "\n" }.join
      else
        columns = Array.new(option) { Array.new }
        items.each_with_index do |item, index|
          columns[index / row_count] << item
        end

        list = ""
        columns.first.size.times do |index|
          list << columns.map { |column| column[index] }.
                          compact.join("  ") + "\n"
        end
        list
      end
    when :uneven_columns_across
      if option.nil?
        limit = @wrap_at || 80
        items.size.downto(1) do |column_count|
          row_count = (items.size / column_count.to_f).ceil
          rows      = Array.new(row_count) { Array.new }
          items.each_with_index do |item, index|
            rows[index / column_count] << item
          end

          widths = Array.new(column_count, 0)
          rows.each do |row|
            row.each_with_index do |field, column|
              size           = actual_length(field)
              widths[column] = size if size > widths[column]
            end
          end

          if column_count == 1 or
             widths.inject(0) { |sum, n| sum + n + 2 } <= limit + 2
            return rows.map { |row|
              row.zip(widths).map { |field, i|
                "%-#{i + (field.to_s.length - actual_length(field))}s" % field
              }.join("  ") + "\n"
            }.join
          end
        end
      else
        row_count = (items.size / option.to_f).ceil
        rows      = Array.new(row_count) { Array.new }
        items.each_with_index do |item, index|
          rows[index / option] << item
        end

        widths = Array.new(option, 0)
        rows.each do |row|
          row.each_with_index do |field, column|
            size           = actual_length(field)
            widths[column] = size if size > widths[column]
          end
        end

        return rows.map { |row|
          row.zip(widths).map { |field, i|
            "%-#{i + (field.to_s.length - actual_length(field))}s" % field
          }.join("  ") + "\n"
        }.join
      end
    when :uneven_columns_down
      if option.nil?
        limit = @wrap_at || 80
        items.size.downto(1) do |column_count|
          row_count = (items.size / column_count.to_f).ceil
          columns   = Array.new(column_count) { Array.new }
          items.each_with_index do |item, index|
            columns[index / row_count] << item
          end

          widths = Array.new(column_count, 0)
          columns.each_with_index do |column, i|
            column.each do |field|
              size      = actual_length(field)
              widths[i] = size if size > widths[i]
            end
          end

          if column_count == 1 or
             widths.inject(0) { |sum, n| sum + n + 2 } <= limit + 2
            list = ""
            columns.first.size.times do |index|
              list << columns.zip(widths).map { |column, width|
                field = column[index]
                "%-#{width + (field.to_s.length - actual_length(field))}s" %
                field
              }.compact.join("  ").strip + "\n"
            end
            return list
          end
        end
      else
        row_count = (items.size / option.to_f).ceil
        columns   = Array.new(option) { Array.new }
        items.each_with_index do |item, index|
          columns[index / row_count] << item
        end

        widths = Array.new(option, 0)
        columns.each_with_index do |column, i|
          column.each do |field|
            size      = actual_length(field)
            widths[i] = size if size > widths[i]
          end
        end

        list = ""
        columns.first.size.times do |index|
          list << columns.zip(widths).map { |column, width|
            field = column[index]
            "%-#{width + (field.to_s.length - actual_length(field))}s" % field
          }.compact.join("  ").strip + "\n"
        end
        return list
      end
    else
      items.map { |i| "#{i}\n" }.join
    end
  end
end

# File lib/highline.rb, line 682
def newline
  @output.puts
end

# File lib/highline.rb, line 690
def output_cols
  return 80 unless @output.tty?
  terminal_size.first
rescue
  return 80
end

# File lib/highline.rb, line 701
def output_rows
  return 24 unless @output.tty?
  terminal_size.last
rescue
  return 24
end

# File lib/highline.rb, line 646
def page_at=( setting )
  @page_at = setting == :auto ? output_rows - 2 : setting
end

# File lib/highline.rb, line 616
def say( statement )
  statement = format_statement(statement)
  return unless statement.length > 0

    # Don't add a newline if statement ends with whitespace, OR
  # if statement ends with whitespace before a color escape code.
  if /[ \t](\e\[\d+(;\d+)*m)?\Z/ =~ statement
    @output.print(indentation+statement)
    @output.flush
  else
    @output.puts(indentation+statement)
  end
end

# File lib/highline.rb, line 403
def uncolor(string)
  self.class.uncolor(string)
end

# File lib/highline.rb, line 636
def wrap_at=( setting )
  @wrap_at = setting == :auto ? output_cols : setting
end