mqtt-homeassistant 0.0.2 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e2701bd3fb3bde99e200213253949dbe73787ada219260463309ea1f27dcd9af
-  data.tar.gz: 3c58be7186fb3aa443e6d1ff33b5282549988422536daf34c430415effece799
+  metadata.gz: 66adc7f6c74477f3b343bdfe1decae41a5a1de06fffd276c3c9cf7bb0bb137c6
+  data.tar.gz: c422275a7683ea94d7f34dd2b880c79901a9f70b15a30c1f928ed548799e3991
 SHA512:
-  metadata.gz: 2257944fc2298287e7c9cd6ffa8b2afd4edd7c22c7a47165817cd8b7ee32084e4e11b7fd30fc794805472cb06c9f608d4d4c587b7685ebdb2e4f1df20ca4777f
-  data.tar.gz: 60a670f138fe6378810cc5b86291a99d92084dfaf509769c67f515132f51ad336fb421cae45f8192113e5e1a01a48cc98fad110218a73071b4c78a3f59f553f8
+  metadata.gz: c41a65d50cb2287f87c7ca9a8c7135596a192c3a81ec5d6d467e511d58de7219028c6af335b8b93dc38f0c4a1c358d070e9695fc1f431c33a73f148cb2c86516
+  data.tar.gz: 801b997f76c306396bec2720465f264de2ce79a7ebef1a97d5f27d99e4d35150944e4e6ecf05c65ed524c0bbea70e6ff5ca7ce693304118f472a158c9711bb8c

data/lib/mqtt/home_assistant/homie/device.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+MQTT::Homie::Device.attr_accessor :home_assistant_device, :home_assistant_discovery_prefix

data/lib/mqtt/home_assistant/homie/node.rb

@@ -0,0 +1,74 @@
+# frozen_string_literal: true
+
+module MQTT
+  module HomeAssistant
+    module Homie
+      module Node
+        {
+          climate: %i[action
+                      aux
+                      away_mode
+                      current_temperature
+                      fan_mode
+                      mode
+                      hold
+                      power
+                      swing_mode
+                      temperature
+                      temperature_high
+                      temperature_low],
+          fan: [nil, :oscillation],
+          humidifier: [nil, :target, :mode],
+          light: [nil, :brightness, :color_mode, :color_temp, :effect, :hs, :rgb, :white, :xy]
+        }.each do |(integration, properties)|
+          has_nil = properties.include?(nil)
+          method_arguments = []
+          method_arguments << "property" if has_nil
+          method_arguments.concat(properties.compact.map { |p| "#{p}_property: nil" })
+          transforms = []
+          transforms << "property = self[property] if property.is_a?(String)"
+          transforms.concat(
+            properties.compact.map do |p|
+              "kwargs[:#{p}_property] = #{p}_property.is_a?(String) ? self[#{p}_property] : #{p}_property"
+            end
+          )
+          args_code = has_nil ? "args = [property]" : "args = []"
+
+          class_eval <<~RUBY, __FILE__, __LINE__ + 1
+            def hass_#{integration}(#{method_arguments.join(", ")}, device: nil, discovery_prefix: nil, **kwargs)
+              #{transforms.join("\n")}
+              discovery_prefix ||= self.device.home_assistant_discovery_prefix
+              device = self.device.home_assistant_device.merge(device || {}) if self.device.home_assistant_device
+
+              kwargs[:device] = device
+              kwargs[:discovery_prefix] = discovery_prefix
+              #{args_code}
+              if published?
+                HomeAssistant.publish_#{integration}(*args, **kwargs)
+              else
+                pending_hass_registrations << [:publish_#{integration}, args, kwargs]
+              end
+            end
+          RUBY
+        end
+
+        def publish
+          super.tap do
+            @pending_hass_registrations&.each do |(method, args, kwargs)|
+              HomeAssistant.public_send(method, *args, **kwargs)
+            end
+            @pending_hass_registrations = nil
+          end
+        end
+
+        private
+
+        def pending_hass_registrations
+          @pending_hass_registrations ||= []
+        end
+      end
+    end
+  end
+end
+
+MQTT::Homie::Node.prepend(MQTT::HomeAssistant::Homie::Node)

data/lib/mqtt/home_assistant/homie/property.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+module MQTT
+  module HomeAssistant
+    module Homie
+      module Property
+        def initialize(*args, hass: nil, **kwargs)
+          super(*args, **kwargs)
+
+          return unless hass
+
+          case hass
+          when Symbol
+            public_send("hass_#{hass}")
+          when Hash
+            raise ArgumentError, "hass must only contain one item" unless hass.length == 1
+
+            public_send("hass_#{hass.first.first}", **hass.first.last)
+          else
+            raise ArgumentError, "hass must be a Symbol or a Hash of HASS device type to additional HASS options"
+          end
+        end
+
+        %i[binary_sensor fan humidifier light number scene select sensor switch].each do |integration|
+          class_eval <<~RUBY, __FILE__, __LINE__ + 1
+            def hass_#{integration}(device: nil, discovery_prefix: nil, **kwargs)
+              discovery_prefix ||= self.device.home_assistant_discovery_prefix
+              device = self.device.home_assistant_device.merge(device || {}) if self.device.home_assistant_device
+
+              kwargs[:device] = device
+              kwargs[:discovery_prefix] = discovery_prefix
+              if published?
+                HomeAssistant.publish_#{integration}(self, **kwargs)
+              else
+                kwargs[:method] = :publish_#{integration}
+                pending_hass_registrations << kwargs
+              end
+            end
+          RUBY
+        end
+
+        def publish
+          super.tap do
+            @pending_hass_registrations&.each do |entity|
+              method = entity.delete(:method)
+              HomeAssistant.public_send(method, self, **entity)
+            end
+            @pending_hass_registrations = nil
+          end
+        end
+
+        private
+
+        def pending_hass_registrations
+          @pending_hass_registrations ||= []
+        end
+      end
+    end
+  end
+end
+
+MQTT::Homie::Property.prepend(MQTT::HomeAssistant::Homie::Property)

data/lib/mqtt/home_assistant/version.rb

@@ -2,6 +2,6 @@
 
 module MQTT
   module HomeAssistant
-    VERSION = "0.0.2"
+    VERSION = "0.1.0"
   end
 end

data/lib/mqtt/home_assistant.rb

@@ -2,6 +2,11 @@
 
 require "json"
 
+require "mqtt/homie"
+require "mqtt/home_assistant/homie/device"
+require "mqtt/home_assistant/homie/node"
+require "mqtt/home_assistant/homie/property"
+
 module MQTT
   module HomeAssistant
     class << self
@@ -211,13 +216,13 @@ module MQTT
                              entity_category: entity_category,
                              icon: icon,
                              templates: {})
-        add_property(config, oscillation_property, :oscillation_property, templates)
-        add_property(config, percentage_property, :percentage, templates)
+        add_property(config, oscillation_property, :oscillation_property, templates: templates)
+        add_property(config, percentage_property, :percentage, templates: templates)
         if percentage_property&.range
           config[:speed_range_min] = percentage_property.range.begin
           config[:speed_range_max] = percentage_property.range.end
         end
-        add_property(config, preset_mode_property, :preset, templates)
+        add_property(config, preset_mode_property, :preset, templates: templates)
         add_enum(config, preset_mode_property, :preset)
 
         publish(node.mqtt, "fan", config, discovery_prefix: discovery_prefix)
@@ -301,6 +306,7 @@ module MQTT
                              device: device,
                              entity_category: entity_category,
                              icon: icon)
+        config[:unique_id] = "#{property.device.id}_#{property.node.id}_#{property.id}"
         add_property(config, property)
         case property.datatype
         when :boolean
@@ -311,21 +317,21 @@ module MQTT
         when :float
           config[:payload_off] = "0.0"
         end
-        add_property(config, brightness_property, :brightness, templates)
+        add_property(config, brightness_property, :brightness, templates: templates)
         config[:brightness_scale] = brightness_property.range.end if brightness_property&.range
-        add_property(config, color_mode_property, :color_mode, templates)
-        add_property(config, color_temp_property, :color_temp, templates)
+        add_property(config, color_mode_property, :color_mode, templates: templates)
+        add_property(config, color_temp_property, :color_temp, templates: templates)
         if color_temp_property&.range && color_temp_property.unit == "mired"
           config[:min_mireds] = color_temp_property.range.begin
           config[:max_mireds] = color_temp_property.range.end
         end
-        add_property(config, effect_property, :effect, templates)
+        add_property(config, effect_property, :effect, templates: templates)
         config[:effect_list] = effect_property.range if effect_property&.datatype == :enum
-        add_property(config, hs_property, :hs, templates)
-        add_property(config, rgb_property, :rgb, templates)
-        add_property(config, white_property, :white, templates)
+        add_property(config, hs_property, :hs, templates: templates)
+        add_property(config, rgb_property, :rgb, templates: templates)
+        add_property(config, white_property, :white, templates: templates)
         config[:white_scale] = white_property.range.end if white_property&.range
-        add_property(config, xy_property, :xy, templates)
+        add_property(config, xy_property, :xy, templates: templates)
         config[:on_command_type] = on_command_type if on_command_type
 
         publish(property.mqtt, "light", config, discovery_prefix: discovery_prefix)
@@ -473,7 +479,7 @@ module MQTT
 
       private
 
-      def add_property(config, property, prefix = nil, templates = {}, read_only: false)
+      def add_property(config, property, prefix = nil, templates: {}, read_only: false)
         return unless property
 
         prefix = "#{prefix}_" if prefix

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mqtt-homeassistant
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.1.0
 platform: ruby
 authors:
 - Cody Cutrer
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-12-08 00:00:00.000000000 Z
+date: 2021-12-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: homie-mqtt
@@ -116,6 +116,9 @@ extra_rdoc_files: []
 files:
 - lib/mqtt-homeassistant.rb
 - lib/mqtt/home_assistant.rb
+- lib/mqtt/home_assistant/homie/device.rb
+- lib/mqtt/home_assistant/homie/node.rb
+- lib/mqtt/home_assistant/homie/property.rb
 - lib/mqtt/home_assistant/version.rb
 homepage: https://github.com/ccutrer/ruby-mqtt-homeassistant
 licenses: