natty-ui 0.12.0 → 0.25.0

data/lib/natty-ui/choice.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+require_relative 'element'
+
+module NattyUI
+  class Choice < Element
+    def select
+      yield(self) if block_given?
+      pin_line = NattyUI.lines_written
+      draw(current = 0)
+      while (key = Terminal.read_key)
+        case key
+        when 'Esc', 'Ctrl+c'
+          break nil if @abortable
+        when 'Enter', ' '
+          break @ret[current]
+        when 'Up', 'Left', 'Back', 'Shift+Tab'
+          current = @texts.size - 1 if (current -= 1) < 0
+          pin_line = NattyUI.back_to_line(pin_line, erase: false)
+          draw(current)
+        when 'Down', 'Right', 'Tab'
+          current = 0 if (current += 1) == @texts.size
+          pin_line = NattyUI.back_to_line(pin_line, erase: false)
+          draw(current)
+        end
+      end
+    ensure
+      NattyUI.back_to_line(@start_line)
+    end
+
+    private
+
+    def initialize(parent, args, kwargs, abortable)
+      super(parent)
+      @start_line = NattyUI.lines_written
+      @texts = args + kwargs.values
+      @ret = Array.new(args.size, &:itself) + kwargs.keys
+      @abortable = abortable
+      theme = Theme.current
+      @mark = [theme.mark(:choice), theme.choice_style]
+      @mark_current = [theme.mark(:current_choice), theme.choice_current_style]
+    end
+
+    def draw(current)
+      @texts.each_with_index do |str, idx|
+        mark, decor = idx == current ? @mark_current : @mark
+        @parent.puts(
+          "#{decor}#{str}",
+          first_line_prefix: mark,
+          first_line_prefix_width: mark.width
+        )
+      end
+    end
+  end
+  private_constant :Choice
+end

data/lib/natty-ui/dumb_choice.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require_relative 'element'
+
+module NattyUI
+  class DumbChoice < Element
+    def select
+      yield(self) if block_given?
+      draw
+      while (code = Terminal.read_key)
+        return if @abortable && (code == 'Esc' || code == 'Ctrl+c')
+        next if code.size > 1
+        code = code[0].upcase
+        if @ret.size <= 9 && ('1'..'9').include?(code)
+          code = @ret[code.ord - 49] and break code
+        elsif ('A'..'Z').include?(code)
+          code = @ret[code.ord - 65] and break code
+        end
+      end
+    end
+
+    private
+
+    def draw
+      glyph = @ret.size <= 9 ? 1 : 'A'
+      @texts.each do |str|
+        @parent.puts(
+          str,
+          first_line_prefix: "[\\#{glyph}] ",
+          first_line_prefix_width: 4
+        )
+        glyph = glyph.succ
+      end
+      @texts = nil
+    end
+
+    def initialize(parent, args, kwargs, abortable)
+      super(parent)
+      @ret = Array.new(args.size, &:itself) + kwargs.keys
+      @texts = args + kwargs.values
+      @abortable = abortable
+    end
+  end
+  private_constant :DumbChoice
+end

data/lib/natty-ui/element.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+require_relative 'features'
+
+module NattyUI
+  # Base element class supporting all {Features}.
+  #
+  class Element
+    include Features
+
+    # @!visibility private
+    def columns = @parent.columns - @prefix_width - @suffix_width
+
+    # @!visibility private
+    def puts(*objects, **options)
+      if @prefix
+        options[:prefix] = "#{@prefix}#{options[:prefix]}"
+        options[:prefix_width] = options[:prefix_width].to_i + @prefix_width
+        if (first_line = options[:first_line_prefix])
+          options[:first_line_prefix] = "#{@prefix}#{first_line}"
+          if (flw = options[:first_line_prefix_width])
+            options[:first_line_prefix_width] = flw + @prefix_width
+          end
+        end
+      end
+      if @suffix
+        options[:suffix] = "#{options[:suffix]}#{@suffix}"
+        options[:suffix_width] = options[:suffix_width].to_i + @suffix_width
+      end
+      # options[:max_width] = @max_width if @max_width
+      @parent.puts(*objects, **options)
+      self
+    end
+
+    alias _to_s to_s
+
+    # @!visibility private
+    alias to_s inspect
+    private :_to_s
+
+    private
+
+    def initialize(parent)
+      @parent = parent
+      @prefix_width = @suffix_width = 0
+    end
+  end
+
+  module StateMixin
+    attr_reader :state
+
+    def closed? = @state ? true : false
+    def ok? = @state == :ok
+    def failed? = @state == :failed
+
+    # @return [Element] itself
+    def done(*text) = @state ? self : finish_ok(text)
+    alias ok done
+
+    # @!visibility private
+    def failed(title, *msg)
+      return if @state
+      super
+      finish_failed
+    end
+
+    # @!visibility private
+    def inspect = "#{_to_s.chop} state=#{@state.inspect}>"
+
+    protected
+
+    def finish_failed
+      @state = :failed
+      self
+    end
+  end
+  private_constant :StateMixin
+end