esphome 0.6.0 → 1.1.0

data/lib/esphome/cli/entities/form.rb

@@ -0,0 +1,111 @@
+# frozen_string_literal: true
+
+require_relative "menu"
+
+module ESPHome
+  module Cli
+    module Entities
+      module FormPrompt
+        private
+
+        def prompt_form(title, initial_value:, suffix:, field_width:)
+          field, form, win = build_form_dialog(title, suffix, field_width)
+          return unless win
+
+          field.set_buffer(0, initial_value) if initial_value
+          form.post
+          win.box(0, 0)
+          win.setpos(0, 2)
+          safe_addstr(win, " #{title} ")
+          if suffix
+            win.setpos(1, field_width + 3)
+            safe_addstr(win, suffix)
+          end
+          win.setpos(1, 2)
+          win.refresh
+
+          loop do
+            case (ch = win.getch)
+            when Curses::KEY_RIGHT
+              form.driver(Curses::REQ_NEXT_CHAR)
+            when Curses::KEY_LEFT
+              form.driver(Curses::REQ_PREV_CHAR)
+            when Curses::KEY_BACKSPACE
+              form.driver(Curses::REQ_DEL_PREV)
+            when Curses::KEY_DC
+              form.driver(Curses::REQ_DEL_CHAR)
+            when "\n".ord
+              form.driver(Curses::REQ_VALIDATION)
+              value = field.buffer(0).strip
+              return nil if value.empty?
+
+              return value
+            when 27, Curses::KEY_RESIZE
+              return nil
+            when nil
+              next
+            else
+              form.driver(ch)
+            end
+          rescue Curses::RequestDeniedError, Curses::UnknownCommandError
+            next
+          end
+        ensure
+          form&.unpost
+          win&.close
+        end
+
+        def build_form_dialog(title, suffix, field_width)
+          width = field_width + 4 + (suffix ? suffix.length + 1 : 0)
+          win = build_dialog_window(3, width, title)
+          return [nil, nil, nil] unless win
+
+          field = Curses::Field.new(1, field_width, 0, 0, 0, 0)
+          field.set_back(Curses::A_UNDERLINE)
+          field.opts_off(Curses::O_AUTOSKIP)
+          form = Curses::Form.new([field])
+          form.set_win(win)
+          form.set_sub(win.derwin(1, field_width, 1, 2))
+          [field, form, win]
+        end
+      end
+
+      class Form < Entity
+        include MenuPrompt
+        include FormPrompt
+
+        def initialize(...)
+          super
+
+          @max_length = length_range&.end || 10
+        end
+
+        def length_range
+          0..10
+        end
+
+        def suffix = nil
+
+        def initial_value = nil
+
+        def activate
+          value = prompt_form(form_title,
+                              initial_value:,
+                              suffix:,
+                              field_width:)
+          command(value) if value
+        end
+
+        private
+
+        def form_title
+          name
+        end
+
+        def field_width
+          @max_length
+        end
+      end
+    end
+  end
+end

data/lib/esphome/cli/entities/lock.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require_relative "menu"
+
+module ESPHome
+  module Cli
+    module Entities
+      class Lock < Menu
+        def options
+          supports_open? ? %w[lock unlock open] : %w[lock unlock]
+        end
+
+        def command(command)
+          cli.info("#{command}ing #{object_id_}")
+          super
+        end
+      end
+    end
+  end
+end

data/lib/esphome/cli/entities/menu.rb

@@ -0,0 +1,83 @@
+# frozen_string_literal: true
+
+module ESPHome
+  module Cli
+    module Entities
+      module MenuPrompt
+        private
+
+        def prompt_menu(title, options, current = nil)
+          content_width = options.map(&:length).max + 4
+          win = build_dialog_window(options.length + 2, content_width, title)
+          return unless win
+
+          index = options.index(current) || 0
+          loop do
+            render_select_options(win, title, options, index)
+
+            case win.getch
+            when Curses::Key::UP
+              index = (index - 1) % options.length
+            when Curses::Key::DOWN
+              index = (index + 1) % options.length
+            when "\n".ord
+              return options[index]
+            when 27, Curses::KEY_RESIZE
+              return nil
+            when nil
+              next
+            end
+          end
+        ensure
+          win&.close
+        end
+
+        def render_select_options(win, title, options, index)
+          win.clear
+          win.box(0, 0)
+          win.setpos(0, 2)
+          safe_addstr(win, " #{title} ")
+          options.each_with_index do |option, idx|
+            win.setpos(idx + 1, 2)
+            if idx == index
+              win.attron(Curses::A_REVERSE) { safe_addstr(win, option) }
+            else
+              safe_addstr(win, option)
+            end
+          end
+          win.refresh
+        end
+
+        def build_dialog_window(height, width, title)
+          root = cli.win
+          return unless root
+
+          width = [width, title.length + 6].max
+          return if width > root.maxx || height > root.maxy
+
+          top = (index + Monitor::HEADER_ROWS).clamp(0, root.maxy - height)
+          left = (cli.name_width + 3).clamp(0, root.maxx - width)
+          win = Curses::Window.new(height, width, top, left)
+          win.keypad = true
+          win.timeout = 100
+          win
+        end
+      end
+
+      class Menu < Entity
+        include MenuPrompt
+
+        def activate
+          choice = prompt_menu(menu_title, options)
+          command(choice) if choice
+        end
+
+        private
+
+        def menu_title
+          name
+        end
+      end
+    end
+  end
+end

data/lib/esphome/cli/entities/number.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+require_relative "form"
+
+module ESPHome
+  module Cli
+    module Entities
+      class Number < Form
+        def length_range
+          max_length = range&.end ? range.end.to_s.length : 10
+          max_length += accuracy_decimals + 1 if accuracy_decimals.positive?
+          1..max_length
+        end
+
+        def suffix
+          unit_of_measurement
+        end
+
+        def initial_value
+          format("%.#{accuracy_decimals}f", state) if state
+        end
+
+        def command(value)
+          begin
+            value = Float(value)
+          rescue ArgumentError
+            cli.error("Invalid number: #{value}")
+            return
+          end
+
+          cli.info("Setting #{object_id_} to #{formatted_state(value)}")
+          set(value)
+        end
+      end
+    end
+  end
+end

data/lib/esphome/cli/entities/select.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+require_relative "menu"
+
+module ESPHome
+  module Cli
+    module Entities
+      class Select < Menu
+        def command(value)
+          cli.info("Setting #{object_id_} to #{value}")
+          super
+        end
+      end
+    end
+  end
+end

data/lib/esphome/cli/entities/subfields.rb

@@ -0,0 +1,63 @@
+# frozen_string_literal: true
+
+require_relative "menu"
+
+module ESPHome
+  module Cli
+    module Entities
+      class Subfields < Entity
+        include MenuPrompt
+
+        def initialize(...)
+          super
+
+          @selected_subfield = 0
+        end
+
+        def move_left
+          return if subfields.size <= 1
+
+          @selected_subfield = (@selected_subfield - 1) % subfields.size
+        end
+
+        def move_right
+          return if subfields.size <= 1
+
+          @selected_subfield = (@selected_subfield + 1) % subfields.size
+        end
+
+        def activate
+          __send__(:"activate_#{active_subfield}")
+        end
+
+        private
+
+        def print_state(win, active:)
+          formatted_segments.each_with_index do |segment, idx|
+            safe_addstr(win, " ") unless idx.zero?
+            if active && idx == highlighted_segment_index
+              prefix, value, suffix = highlighted_segment_parts(idx, segment)
+              safe_addstr(win, prefix)
+              win.attron(Curses::A_REVERSE) { safe_addstr(win, value) }
+              safe_addstr(win, suffix)
+            else
+              safe_addstr(win, segment)
+            end
+          end
+        end
+
+        def active_subfield
+          subfields.fetch(@selected_subfield, subfields.first)
+        end
+
+        def highlighted_segment_index
+          subfield_segment_indexes.fetch(active_subfield)
+        end
+
+        def highlighted_segment_parts(_idx, segment)
+          ["", segment, ""]
+        end
+      end
+    end
+  end
+end

data/lib/esphome/cli/entities/switch.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+require_relative "menu"
+
+module ESPHome
+  module Cli
+    module Entities
+      class Switch < Menu
+        def options = %w[on off]
+
+        def command(command)
+          cli.info("Turning #{object_id_} #{command}")
+          __send__(command.to_sym)
+        end
+      end
+    end
+  end
+end

data/lib/esphome/cli/entities/text.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+require_relative "form"
+
+module ESPHome
+  module Cli
+    module Entities
+      class Text < Form
+        def command(value)
+          cli.info("Setting #{object_id_} to #{value}")
+          super
+        end
+      end
+    end
+  end
+end

data/lib/esphome/cli/entities/time.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+require_relative "form"
+
+module ESPHome
+  module Cli
+    module Entities
+      class Time < Form
+        def length_range
+          25..25
+        end
+
+        def command(value)
+          if defined?(ActiveSupport::Duration)
+            begin
+              value = ActiveSupport::Duration.parse(value)
+            rescue ArgumentError
+              # ignore; try to parse another way
+            end
+          end
+
+          begin
+            if value.is_a?(String)
+              value = value.split(":").map { |v| Integer(v) }
+              raise ArgumentError if value.length > 3
+
+              value.unshift(0) while value.length < 3
+            end
+          rescue ArgumentError
+            cli.error("Invalid time: #{value}")
+            return
+          end
+
+          cli.info("Setting #{object_id_} to #{value.iso8601}")
+          super
+        end
+      end
+    end
+  end
+end

data/lib/esphome/cli/entity.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+require "delegate"
+
+module ESPHome
+  module Cli
+    class Entity < SimpleDelegator
+      attr_reader :cli, :index
+
+      def initialize(cli, entity, index)
+        super(entity)
+
+        @cli = cli
+        @index = index
+        touch
+      end
+
+      def touch = @last_update = Time.now
+      def move_left = nil
+      def move_right = nil
+
+      def print(win, clear_line: true, active: false)
+        row = @index + Monitor::HEADER_ROWS
+        return if row >= win.maxy
+
+        win.setpos(row, 0)
+        win.clrtoeol if clear_line
+        safe_addstr(win, "#{name.ljust(cli.name_width)} : ")
+        print_state(win, active:)
+
+        return unless __getobj__.is_a?(ESPHome::Entity::HasState)
+
+        timestamp = "[#{@last_update.strftime("%H:%M:%S.%L")}]"
+        return if timestamp.length >= win.maxx
+
+        pos = [win.cury, win.curx]
+        win.setpos(row, win.maxx - timestamp.length)
+        safe_addstr(win, timestamp)
+        win.setpos(*pos) if pos[1] < win.maxx
+      rescue Curses::Error
+        nil
+      end
+
+      private
+
+      def print_state(win, active:)
+        state = formatted_state
+        if active
+          win.attron(Curses::A_REVERSE) { safe_addstr(win, state) }
+        else
+          safe_addstr(win, state)
+        end
+      end
+
+      def safe_addstr(win, string)
+        return if string.nil? || string.empty?
+
+        remaining = win.maxx - win.curx
+        return if remaining <= 0
+
+        win.addstr(string[0, remaining])
+      end
+    end
+  end
+end
+
+require_relative "entities/button"
+require_relative "entities/climate"
+require_relative "entities/cover"
+require_relative "entities/date"
+require_relative "entities/date_time"
+require_relative "entities/lock"
+require_relative "entities/number"
+require_relative "entities/select"
+require_relative "entities/switch"
+require_relative "entities/subfields"
+require_relative "entities/text"
+require_relative "entities/time"