mqtt-homeassistant 0.1.6 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c9ce071c9aff4a9278fd43c312610aef5a9ef11e99452823067dbf4cf6980029
-  data.tar.gz: 47b50acb51c8280f29f80f0c4f918b744ca291b5484af0119d2a4cbed80cf86f
+  metadata.gz: b52a4f79380d1f0a0aa808fac2328d00a972dbf9c349372a1cd623d836574d9d
+  data.tar.gz: 81747c398e5737eb4e4803a343eeae0b5f293c50a197873dcb436f2b92329eba
 SHA512:
-  metadata.gz: a7521c34bad575547c953530311c9111d4d143579139738a158311595b3996b300d22a5ded21d815712bb7bc50916c6b356ad66c3246fcd82d05fdc02ef26ca1
-  data.tar.gz: c50827bcc8264b573d2f0543f5fb3ef6a94d1b3d04146379251a53681816756e8a731377f98b3f2d8df2f2d42e93cc89ddfa713fd91ca9521711a4c95af69664
+  metadata.gz: 1a35ab09e32a71e6fa9253b6413262bd2cebd922facf6c86e732843f3eb35f6cfa420b82d86d363a4a3b29916d9de52bb0e2a1429eed6ee79d5b52a2288800c9
+  data.tar.gz: e72ca801a8489241d079380059ad7b6b1868e413652cac2eb21f5c8eeccd6dd94068b098701acafeba8e327130a1c40715bf57578d7cfa3f9206c178d2cae3c9

data/lib/mqtt/home_assistant/client.rb

@@ -0,0 +1,140 @@
+# frozen_string_literal: true
+
+module MQTT
+  module HomeAssistant
+    module Client
+      KNOWN_ATTRIBUTES.each_key do |platform|
+        class_eval <<~RUBY, __FILE__, __LINE__ + 1
+          def publish_hass_#{platform}(object_id, platform: #{platform.inspect}, **kwargs)
+            raise ArgumentError, "platform must be #{platform.inspect}" unless platform == #{platform.inspect}
+
+            publish_hass_component(object_id, platform: platform, **kwargs)
+          end
+
+          def unpublish_hass_#{platform}(object_id, platform: #{platform.inspect}, **kwargs)
+            raise ArgumentError, "platform must be #{platform.inspect}" unless platform == #{platform.inspect}
+
+            unpublish_hass_component(object_id, platform: platform, **kwargs)
+          end
+        RUBY
+      end
+
+      def unpublish_hass_component(object_id, platform:, discovery_prefix: "homeassistant", node_id: nil)
+        node_and_object_id = node_id ? "#{node_id}/#{object_id}" : object_id
+        unless KNOWN_ATTRIBUTES.key?(platform)
+          raise ArgumentError, "Unknown platform #{platform} for #{node_and_object_id}"
+        end
+
+        publish("#{discovery_prefix || "homeassistant"}/#{platform}/#{node_and_object_id}/config",
+                "",
+                retain: true,
+                qos: 1)
+      end
+
+      def publish_hass_component(object_id, platform:, discovery_prefix: "homeassistant", node_id: nil, **kwargs)
+        node_and_object_id = node_id ? "#{node_id}/#{object_id}" : object_id
+        unless KNOWN_ATTRIBUTES.key?(platform)
+          raise ArgumentError, "Unknown platform #{platform} for #{node_and_object_id}"
+        end
+
+        required_attributes = attributes_for_schema(REQUIRED_ATTRIBUTES, platform, kwargs)
+        if required_attributes
+          missing_attributes = required_attributes - kwargs.keys
+          unless missing_attributes&.empty?
+            raise ArgumentError,
+                  "Missing attribute(s) #{missing_attributes.join(", ")} for #{platform}/#{node_and_object_id}"
+          end
+        end
+
+        known_attributes = attributes_for_schema(KNOWN_ATTRIBUTES, platform, kwargs)
+        unknown_attributes = kwargs.keys - SPECIAL_ATTRIBUTES[:common] - known_attributes
+        unless unknown_attributes.empty?
+          raise ArgumentError,
+                "Unknown attribute(s) #{unknown_attributes.join(", ")} for #{platform}/#{node_and_object_id}"
+        end
+
+        if (availability_list = kwargs[:availability])
+          unless (kwargs.keys & %i[availability_mode
+                                   availability_template
+                                   availability_topic
+                                   payload_available
+                                   payload_not_available]).empty?
+            raise ArgumentError,
+                  "availability cannot be used together with availability topic for #{platform}/#{node_and_object_id}"
+          end
+
+          availability_list = [availability_list] if availability_list.is_a?(Hash)
+          unless availability_list.is_a?(Array)
+            raise ArgumentError, "availability must be an array for #{platform}/#{node_and_object_id}"
+          end
+
+          availability_list.each do |availability|
+            unless availability.key?(:topic)
+              raise ArgumentError, "availability must have a topic for #{platform}/#{node_and_object_id}"
+            end
+
+            unless (extra_keys = availability.keys - SPECIAL_ATTRIBUTES[:availability]).empty?
+              raise ArgumentError,
+                    "Unknown attribute(s) #{extra_keys} for #{platform}/#{node_and_object_id}'s availability"
+            end
+          end
+        end
+
+        if (device = kwargs[:device])
+          raise ArgumentError, "device must be a hash for #{platform}/#{node_and_object_id}" unless device.is_a?(Hash)
+          unless (extra_keys = device.keys - SPECIAL_ATTRIBUTES[:device]).empty?
+            raise ArgumentError, "Unknown attribute(s) #{extra_keys} for #{platform}/#{node_and_object_id}'s device"
+          end
+        end
+
+        INCLUSION_VALIDATIONS[:common].merge(INCLUSION_VALIDATIONS[platform] || {}).each do |attr, valid_values|
+          if (value = kwargs[attr]) && !valid_values.include?(value)
+            raise ArgumentError, "Unrecognized #{attr} #{value} for #{platform}/#{node_and_object_id}"
+          end
+        end
+        SUBSET_VALIDATIONS[platform]&.each do |attr, valid_values|
+          if (values = kwargs[attr]) && !(extra_values = values - valid_values).empty?
+            raise ArgumentError, "Invalid #{attr} #{extra_values.join(", ")} for #{platform}/#{node_and_object_id}"
+          end
+        end
+
+        VALIDATIONS[platform]&.call(**kwargs)
+
+        RANGE_ATTRIBUTES[platform]&.each do |attr, prefix_or_suffix|
+          if (range = kwargs.delete(:"#{attr}_range"))
+            case prefix_or_suffix
+            when :prefix
+              kwargs[:"min_#{attr}"] = range.begin
+              kwargs[:"max_#{attr}"] = range.end
+            when :suffix
+              kwargs[:"#{attr}_min"] = range.begin
+              kwargs[:"#{attr}_max"] = range.end
+            when :singleton
+              kwargs[:min] = range.begin
+              kwargs[:max] = range.end
+            end
+          end
+        end
+
+        publish("#{discovery_prefix || "homeassistant"}/#{platform}/#{node_and_object_id}/config",
+                kwargs.to_json,
+                retain: true,
+                qos: 1)
+      end
+
+      private
+
+      def attributes_for_schema(hash, platform, kwargs)
+        attributes = hash[platform]
+        if attributes.is_a?(Hash)
+          schema = kwargs[:schema] || :default
+          attributes = attributes[schema]
+          raise ArgumentError, "Invalid schema #{schema} for platform #{platfomr}" unless attributes
+        end
+
+        attributes
+      end
+    end
+  end
+  Client.include(HomeAssistant::Client)
+end

data/lib/mqtt/home_assistant/version.rb

@@ -2,6 +2,6 @@
 
 module MQTT
   module HomeAssistant
-    VERSION = "0.1.6"
+    VERSION = "1.0.1"
   end
 end