austb-tty-prompt 0.13.0

data/lib/tty/prompt/enum_paginator.rb

@@ -0,0 +1,56 @@
+# encoding: utf-8
+
+require_relative 'paginator'
+
+module TTY
+  class Prompt
+    class EnumPaginator < Paginator
+      # Paginate list of choices based on current active choice.
+      # Move entire pages.
+      #
+      # @api public
+      def paginate(list, active, per_page = nil, &block)
+        default_size = (list.size <= DEFAULT_PAGE_SIZE ? list.size : DEFAULT_PAGE_SIZE)
+        @per_page = @per_page || per_page || default_size
+
+        # Don't paginate short lists
+        if list.size <= @per_page
+          @lower_index = 0
+          @upper_index = list.size - 1
+          if block
+            return list.each_with_index(&block)
+          else
+            return list.each_with_index.to_enum
+          end
+        end
+
+        unless active.nil? # User may input index out of range
+          @last_index = active
+        end
+        page  = (@last_index / @per_page.to_f).ceil
+        pages = (list.size / @per_page.to_f).ceil
+        if page == 0
+          @lower_index = 0
+          @upper_index = @lower_index + @per_page - 1
+        elsif page > 0 && page <= pages
+          @lower_index = (page - 1) * @per_page
+          @upper_index = @lower_index + @per_page - 1
+        else
+          @upper_index = list.size - 1
+          @lower_index = @upper_index - @per_page + 1
+        end
+
+        sliced_list = list[@lower_index..@upper_index]
+        indices = (@lower_index..@upper_index)
+
+        if block
+          sliced_list.each_with_index do |item, index|
+            block[item, @lower_index + index]
+          end
+        else
+          sliced_list.zip(indices).to_enum unless block_given?
+        end
+      end
+    end # EnumPaginator
+  end # Prompt
+end # TTY

data/lib/tty/prompt/evaluator.rb

@@ -0,0 +1,29 @@
+# encoding: utf-8
+
+require_relative 'result'
+
+module TTY
+  class Prompt
+    # Evaluates provided parameters and stops if any of them fails
+    # @api private
+    class Evaluator
+      attr_reader :results
+
+      def initialize(question, &block)
+        @question = question
+        @results = []
+        instance_eval(&block) if block
+      end
+
+      def call(initial)
+        seed = Result::Success.new(@question, initial)
+        results.reduce(seed, &:with)
+      end
+
+      def check(proc = nil, &block)
+        results << (proc || block)
+      end
+      alias_method :<<, :check
+    end # Evaluator
+  end # Prompt
+end # TTY

data/lib/tty/prompt/expander.rb

@@ -0,0 +1,292 @@
+# encoding: utf-8
+
+require_relative 'choices'
+
+module TTY
+  class Prompt
+    # A class responsible for rendering expanding options
+    # Used by {Prompt} to display key options question.
+    #
+    # @api private
+    class Expander
+      HELP_CHOICE = {
+        key: 'h',
+        name: 'print help',
+        value: :help
+      }
+
+      # Create instance of Expander
+      #
+      # @api public
+      def initialize(prompt, options = {})
+        @prompt       = prompt
+        @prefix       = options.fetch(:prefix) { @prompt.prefix }
+        @default      = options.fetch(:default) { 1 }
+        @active_color = options.fetch(:active_color) { @prompt.active_color }
+        @help_color   = options.fetch(:help_color) { @prompt.help_color }
+        @choices      = Choices.new
+        @selected     = nil
+        @done         = false
+        @status       = :collapsed
+        @hint         = nil
+        @default_key  = false
+
+        @prompt.subscribe(self)
+      end
+
+      def expanded?
+        @status == :expanded
+      end
+
+      def collapsed?
+        @status == :collapsed
+      end
+
+      def expand
+        @status = :expanded
+      end
+
+      # Respond to submit event
+      #
+      # @api public
+      def keyenter(_)
+        if @input.nil? || @input.empty?
+          @input = @choices[@default - 1].key
+          @default_key = true
+        end
+
+        selected = select_choice(@input)
+
+        if selected && selected.key.to_s == 'h'
+          expand
+          @selected = nil
+          @input = ''
+        elsif selected
+          @done = true
+          @selected = selected
+        else
+          @input = ''
+        end
+      end
+      alias_method :keyreturn, :keyenter
+
+      # Respond to key press event
+      #
+      # @api public
+      def keypress(event)
+        if [:backspace, :delete].include?(event.key.name)
+          @input.chop! unless @input.empty?
+        elsif event.value =~ /^[^\e\n\r]/
+          @input += event.value
+        end
+        @selected = select_choice(@input)
+        if @selected && !@default_key && collapsed?
+          @hint = @selected.name
+        end
+      end
+
+      # Select choice by given key
+      #
+      # @return [Choice]
+      #
+      # @api private
+      def select_choice(key)
+        @choices.find_by(:key, key)
+      end
+
+      # Set default value.
+      #
+      # @api public
+      def default(value = (not_set = true))
+        return @default if not_set
+        @default = value
+      end
+
+      # Add a single choice
+      #
+      # @api public
+      def choice(value, &block)
+        if block
+          @choices << value.update(value: block)
+        else
+          @choices << value
+        end
+      end
+
+      # Add multiple choices
+      #
+      # @param [Array[Object]] values
+      #   the values to add as choices
+      #
+      # @api public
+      def choices(values)
+        values.each { |val| choice(val) }
+      end
+
+      # Execute this prompt
+      #
+      # @api public
+      def call(message, possibilities, &block)
+        choices(possibilities)
+        @message = message
+        block.call(self) if block
+        setup_defaults
+        choice(HELP_CHOICE)
+        render
+      end
+
+      private
+
+      # Create possible keys with current choice highlighted
+      #
+      # @return [String]
+      #
+      # @api private
+      def possible_keys
+        keys = @choices.pluck(:key)
+        default_key = keys[@default - 1]
+        if @selected
+          index = keys.index(@selected.key)
+          keys[index] = @prompt.decorate(keys[index], @active_color)
+        elsif @input.to_s.empty? && default_key
+          keys[@default - 1] = @prompt.decorate(default_key, @active_color)
+        end
+        keys.join(',')
+      end
+
+      # @api private
+      def render
+        @input = ''
+        until @done
+          question = render_question
+          @prompt.print(question)
+          read_input
+          @prompt.print(refresh(question.lines.count))
+        end
+        @prompt.print(render_question)
+        answer
+      end
+
+      # @api private
+      def answer
+        @selected.value
+      end
+
+      # Render message with options
+      #
+      # @return [String]
+      #
+      # @api private
+      def render_header
+        header = "#{@prefix}#{@message} "
+        if @done
+          selected_item = "#{@selected.name}"
+          header << @prompt.decorate(selected_item, @active_color)
+        elsif collapsed?
+          header << %[(enter "h" for help) ]
+          header << "[#{possible_keys}] "
+          header << @input
+        end
+        header
+      end
+
+      # Show hint for selected option key
+      #
+      # return [String]
+      #
+      # @api private
+      def render_hint
+        hint = "\n"
+        hint << @prompt.decorate('>> ', @active_color)
+        hint << @hint
+        hint << @prompt.cursor.prev_line
+        hint << @prompt.cursor.forward(@prompt.strip(render_header).size)
+      end
+
+      # Render question with menu
+      #
+      # @return [String]
+      #
+      # @api private
+      def render_question
+        header = render_header
+        header << render_hint if @hint
+        header << "\n" if @done
+
+        if !@done && expanded?
+          header << render_menu
+          header << render_footer
+        end
+        header
+      end
+
+      def render_footer
+        "  Choice [#{@choices[@default - 1].key}]: #{@input}"
+      end
+
+      def read_input
+        @prompt.read_keypress
+      end
+
+      # Refresh the current input
+      #
+      # @param [Integer] lines
+      #
+      # @return [String]
+      #
+      # @api private
+      def refresh(lines)
+        if @hint && (!@selected || @done)
+          @hint = nil
+          @prompt.clear_lines(lines, :down) +
+            @prompt.cursor.prev_line
+        elsif expanded?
+          @prompt.clear_lines(lines)
+        else
+          @prompt.clear_line
+        end
+      end
+
+      # Render help menu
+      #
+      # @api private
+      def render_menu
+        output = "\n"
+        @choices.each do |choice|
+          chosen = %(#{choice.key} - #{choice.name})
+          if @selected && @selected.key == choice.key
+            chosen = @prompt.decorate(chosen, @active_color)
+          end
+          output << '  ' + chosen + "\n"
+        end
+        output
+      end
+
+      def setup_defaults
+        validate_choices
+      end
+
+      def validate_choices
+        errors = []
+        keys = []
+        @choices.each do |choice|
+          if choice.key.nil?
+            errors <<  "Choice #{choice.name} is missing a :key attribute"
+            next
+          end
+          if choice.key.length != 1
+            errors << "Choice key `#{choice.key}` is more than one character long."
+          end
+          if choice.key.to_s == 'h'
+            errors << "Choice key `#{choice.key}` is reserved for help menu."
+          end
+          if keys.include?(choice.key)
+            errors << "Choice key `#{choice.key}` is a duplicate."
+          end
+          keys << choice.key if choice.key
+        end
+        errors.each { |err| fail ConfigurationError, err }
+      end
+    end # Expander
+  end # Prompt
+end # TTY

data/lib/tty/prompt/keypress.rb

@@ -0,0 +1,94 @@
+# encoding: utf-8
+
+require_relative 'question'
+require_relative 'symbols'
+require_relative 'timeout'
+
+module TTY
+  class Prompt
+    class Keypress < Question
+      # Create keypress question
+      #
+      # @param [Prompt] prompt
+      # @param [Hash] options
+      #
+      # @api public
+      def initialize(prompt, options = {})
+        super
+        @echo    = options.fetch(:echo) { false }
+        @keys    = options.fetch(:keys) { UndefinedSetting }
+        @timeout = options.fetch(:timeout) { UndefinedSetting }
+        @interval = options.fetch(:interval) {
+          (@timeout != UndefinedSetting && @timeout < 1) ? @timeout : 1
+        }
+        @pause   = true
+        @countdown = @timeout
+        @interval_handler = proc { |time|
+          question = render_question
+          @prompt.print(refresh(question.lines.count))
+          countdown(time)
+          @prompt.print(render_question)
+        }
+
+        @prompt.subscribe(self)
+      end
+
+      def countdown(value = (not_set = true))
+        return @countdown if not_set
+        @countdown = value
+      end
+
+      # Check if any specific keys are set
+      def any_key?
+        @keys == UndefinedSetting
+      end
+
+      # Check if timeout is set
+      def timeout?
+        @timeout != UndefinedSetting
+      end
+
+      def keypress(event)
+        if any_key?
+          @pause = false
+        elsif @keys.is_a?(Array) && @keys.include?(event.key.name)
+          @pause = false
+        else
+          @pause = true
+        end
+      end
+
+      def render_question
+        header = super
+        header.gsub!(/:countdown/, countdown.to_s)
+        header
+      end
+
+      def process_input(question)
+        time do
+          while @pause
+            @input = @prompt.read_keypress
+          end
+          @pause
+        end
+        @evaluator.(@input)
+      end
+
+      def refresh(lines)
+        @prompt.clear_lines(lines)
+      end
+
+      def time(&block)
+        if timeout?
+          time = Float(@timeout)
+          interval = Float(@interval)
+          scheduler = Timeout.new(interval_handler: @interval_handler)
+          scheduler.timeout(time, interval, &block)
+        else
+          block.()
+        end
+      rescue Timeout::Error
+      end
+    end # Keypress
+  end # Prompt
+end # TTY