class MQTT::Homie::Node

def [](id)
  @properties[id]
end

def batch_update(hash)
  mqtt.batch_publish do
    hash.each do |(k, v)|
      self[k].value = v
    end
  end
end

def each(&block)
  @properties.each_value(&block)
end

def initialize(device, id, name, type)
  super(id, name)
  @device = device
  @type = type
  @properties = {}
  @published = false
end

def mqtt
  device.mqtt
end

def property(*args, **kwargs, &block)
  device.init do |prior_state|
    property = Property.new(self, *args, **kwargs, &block)
    raise ArgumentError, "Property '#{property.id}' already exists on '#{id}'" if @properties.key?(property.id)
    @properties[property.id] = property
    property.publish if prior_state == :ready
    property
  end
end

def publish
  mqtt.batch_publish do
    unless @published
      mqtt.publish("#{topic}/$name", name, retain: true, qos: 1)
      mqtt.publish("#{topic}/$type", @type.to_s, retain: true, qos: 1)
      @published = true
    end
    mqtt.publish("#{topic}/$properties", @properties.keys.join(","), retain: true, qos: 1)
    @properties.each_value(&:publish)
  end
end

def remove_property(id)
  return false unless (property = @properties[id])
  init do
    property.unpublish
    @properties.delete(id)
    mqtt.publish("#{topic}/$properties", @properties.keys.join(","), retain: true, qos: 1) if @published
  end
  true
end

def topic
  "#{device.topic}/#{id}"
end

def unpublish
  return unless @published
  @published = false
  mqtt.publish("#{topic}/$name", retain: true, qos: 0)
  mqtt.publish("#{topic}/$type", retain: true, qos: 0)
  mqtt.publish("#{topic}/$properties", retain: true, qos: 0)
  @properties.each_value(&:unpublish)
end