balboa_worldwide_app 1.2.5 → 2.0.2

data/bin/bwa_mqtt_bridge

@@ -1,283 +0,0 @@
-#!/usr/bin/env ruby
-
-require 'mqtt'
-require 'sd_notify'
-require 'set'
-
-require 'bwa/client'
-require 'bwa/discovery'
-
-class MQTTBridge
-  def initialize(mqtt_uri, bwa, device_id: "bwa", base_topic: "homie")
-    @base_topic = "#{base_topic}/#{device_id}"
-    @mqtt = MQTT::Client.new(mqtt_uri)
-    @mqtt.set_will("#{@base_topic}/$state", "lost", true)
-    @mqtt.connect
-    @bwa = bwa
-    @attributes = {}
-    @things = Set.new
-
-    publish_basic_attributes
-
-    # Tell systemd we've started up OK. Ignored if systemd not in use.
-    SdNotify.ready
-
-    bwa_thread = Thread.new do
-      loop do
-        begin
-          message = @bwa.poll
-          next if message.is_a?(BWA::Messages::Ready)
-
-          puts message.inspect unless message.is_a?(BWA::Messages::Status)
-          case message
-          when BWA::Messages::ControlConfiguration
-            publish("spa/$type", message.model)
-          when BWA::Messages::ControlConfiguration2
-            message.pumps.each_with_index do |speed, i|
-              publish_pump(i + 1, speed) if speed != 0
-            end
-            message.lights.each_with_index do |exists, i|
-              publish_thing("light", i + 1) if exists
-            end
-            message.aux.each_with_index do |exists, i|
-              publish_thing("aux", i + 1) if exists
-            end
-            publish_mister if message.mister
-            publish_blower(message.blower) if message.blower != 0
-            publish_circpump if message.circ_pump
-            publish("$state", "ready")
-          when BWA::Messages::Status
-            @bwa.request_control_info unless @bwa.last_control_configuration
-            @bwa.request_control_info2 unless @bwa.last_control_configuration2
-
-            # make sure time is in sync
-            now = Time.now
-            now_minutes = now.hour * 60 + now.min
-            spa_minutes = message.hour * 60 + message.minute
-            # check the difference in both directions
-            diff = [(spa_minutes - now_minutes) % 1440, 1440 - (spa_minutes - now_minutes) % 1440].min
-
-            # allow a skew of 1 minute, since the seconds will always be off
-            if diff > 1
-              @bwa.set_time(now.hour, now.min, message.twenty_four_hour_time)
-            end
-            publish_attribute("spa/priming", message.priming)
-            publish_attribute("spa/heatingmode", message.heating_mode)
-            publish_attribute("spa/temperaturescale", message.temperature_scale)
-            publish_attribute("spa/24htime", message.twenty_four_hour_time)
-            publish_attribute("spa/heating", message.heating)
-            publish_attribute("spa/temperaturerange", message.temperature_range)
-            publish_attribute("spa/currenttemperature", message.current_temperature)
-            publish_attribute("spa/currenttemperature/$unit", "º#{message.temperature_scale.to_s[0].upcase}")
-            publish_attribute("spa/settemperature", message.set_temperature)
-            publish_attribute("spa/settemperature/$unit", "º#{message.temperature_scale.to_s[0].upcase}")
-            if message.temperature_scale == :celsius
-              publish_attribute("spa/currenttemperature/$format", message.temperature_range == :high ? "26:40" : "10:26")
-              publish_attribute("spa/settemperature/$format", message.temperature_range == :high ? "26:40" : "10:26")
-            else
-              publish_attribute("spa/currenttemperature/$format", message.temperature_range == :high ? "80:104" : "26:40")
-              publish_attribute("spa/settemperature/$format", message.temperature_range == :high ? "80:104" : "26:40")
-            end
-            publish_attribute("spa/filter1", message.filter[0])
-            publish_attribute("spa/filter2", message.filter[1])
-
-            publish_attribute("spa/circpump", message.circ_pump) if @bwa.last_control_configuration2&.circ_pump
-            publish_attribute("spa/blower", message.blower) if @bwa.last_control_configuration2&.blower.to_i != 0
-            publish_attribute("spa/mister", message.mister) if @bwa.last_control_configuration2&.mister
-            (0..5).each do |i|
-              publish_attribute("spa/pump#{i + 1}", message.pumps[i]) if @bwa.last_control_configuration2&.pumps&.[](i).to_i != 0
-            end
-            (0..1).each do |i|
-              publish_attribute("spa/light#{i + 1}", message.lights[i]) if @bwa.last_control_configuration2&.lights&.[](i)
-            end
-            (0..1).each do |i|
-              publish_attribute("spa/aux#{i + 1}", message.lights[i]) if @bwa.last_control_configuration2&.aux&.[](i)
-            end
-
-            # Tell systemd we are still alive and kicking. Ignored if systemd not in use.
-            SdNotify.watchdog
-          end
-        end
-      end
-    end
-
-    @mqtt.get do |topic, value|
-      puts "got #{value.inspect} at #{topic}"
-      case topic[@base_topic.length + 1..-1]
-      when "spa/heatingmode/set"
-        next @bwa.toggle_heating_mode if value == 'toggle'
-        next unless %w{ready rest}.include?(value)
-        @bwa.set_heating_mode(value.to_sym)
-      when "spa/temperaturescale/set"
-        next unless %w{fahrenheit celsius}.include?(value)
-        @bwa.set_temperature_scale(value.to_sym)
-      when "spa/24htime/set"
-        next unless %w{true false}.include?(value)
-        now = Time.now
-        @bwa.set_time(now.hour, now.min, value == 'true')
-      when "spa/temperaturerange/set"
-        next @bwa.toggle_temperature_range if value == 'toggle'
-        next unless %w{low high}.include?(value)
-        @bwa.set_temperature_range(value.to_sym)
-      when %r{^spa/pump([1-6])/set$}
-        next @bwa.toggle_pump($1.to_i) if value == 'toggle'
-        @bwa.set_pump($1.to_i, value.to_i)
-      when %r{^spa/(light|aux)([12])/set$}
-        next @bwa.send(:"toggle_#{$1}", $2.to_i) if value == 'toggle'
-        next unless %w{true false}.include?(value)
-        @bwa.send(:"set_#{$1}", $2.to_i, value == 'true')
-      when "spa/mister/set"
-        next @bwa.toggle_mister if value == 'toggle'
-        next unless %w{true false}.include?(value)
-        @bwa.set_mister(value == 'true')
-      when "spa/blower/set"
-        next @bwa.toggle_blower if value == 'toggle'
-        @bwa.set_blower(value.to_i)
-      when "spa/settemperature/set"
-        @bwa.set_temperature(value.to_f)
-      end
-    end
-  end
-
-  def publish(topic, value)
-    @mqtt.publish("#{@base_topic}/#{topic}", value, true)
-  end
-
-  def publish_attribute(attr, value)
-    if !@attributes.key?(attr) || @attributes[attr] != value
-      publish(attr, value.to_s)
-      @attributes[attr] = value
-    end
-  end
-
-  def subscribe(topic)
-    @mqtt.subscribe("#{@base_topic}/#{topic}")
-  end
-
-  def publish_basic_attributes
-    publish("$homie", "4.0.0")
-    publish("$name", "BWA Spa")
-    publish("$state", "init")
-    publish("$nodes", "spa")
-
-    publish("spa/$name", "BWA Spa")
-    publish("spa/$type", "spa")
-    publish_nodes
-
-    publish("spa/priming/$name", "Is the pump priming")
-    publish("spa/priming/$datatype", "boolean")
-
-    publish("spa/heatingmode/$name", "Current heating mode")
-    publish("spa/heatingmode/$datatype", "enum")
-    publish("spa/heatingmode/$format", "ready,rest,ready_in_rest")
-    publish("spa/heatingmode/$settable", "true")
-    subscribe("spa/heatingmode/set")
-
-    publish("spa/temperaturescale/$name", "Temperature scale")
-    publish("spa/temperaturescale/$datatype", "enum")
-    publish("spa/temperaturescale/$format", "fahrenheit,celsius")
-    publish("spa/temperaturescale/$settable", "true")
-    subscribe("spa/temperaturescale/set")
-
-    publish("spa/24htime/$name", "Clock is 24 hour time")
-    publish("spa/24htime/$datatype", "boolean")
-    publish("spa/24htime/$settable", "true")
-    subscribe("spa/24htime/set")
-
-    publish("spa/heating/$name", "Heater is currently running")
-    publish("spa/heating/$datatype", "boolean")
-
-    publish("spa/temperaturerange/$name", "Current temperature range")
-    publish("spa/temperaturerange/$datatype", "enum")
-    publish("spa/temperaturerange/$format", "high,low")
-    publish("spa/temperaturerange/$settable", "true")
-    subscribe("spa/temperaturerange/set")
-
-    publish("spa/currenttemperature/$name", "Current temperature")
-    publish("spa/currenttemperature/$datatype", "float")
-
-    publish("spa/settemperature/$name", "Set Temperature")
-    publish("spa/settemperature/$datatype", "float")
-    publish("spa/settemperature/$settable", "true")
-    subscribe("spa/settemperature/set")
-
-    publish("spa/filter1/$name", "Filter cycle 1 is currently running")
-    publish("spa/filter1/$datatype", "boolean")
-
-    publish("spa/filter2/$name", "Filter cycle 2 is currently running")
-    publish("spa/filter2/$datatype", "boolean")
-  end
-
-  def publish_pump(i, speeds)
-    publish("spa/pump#{i}/$name", "Pump #{i} speed")
-    publish("spa/pump#{i}/$datatype", "integer")
-    publish("spa/pump#{i}/$format", "0:#{speeds}")
-    publish("spa/pump#{i}/$settable", "true")
-    subscribe("spa/pump#{i}/set")
-
-    @things << "pump#{i}"
-    publish_nodes
-  end
-
-  def publish_thing(type, i)
-    publish("spa/#{type}#{i}/$name", "#{type} #{i}")
-    publish("spa/#{type}#{i}/$datatype", "boolean")
-    publish("spa/#{type}#{i}/$settable", "true")
-    subscribe("spa/#{type}#{i}/set")
-
-    @things << "#{type}#{i}"
-    publish_nodes
-  end
-
-  def publish_mister
-    publish("spa/mister/$name", type)
-    publish("spa/mister/$datatype", "boolean")
-    publish("spa/mister/$settable", "true")
-    subscribe("spa/mister/set")
-
-    @things << "mister"
-    publish_nodes
-  end
-
-  def publish_blower(speeds)
-    publish("spa/blower/$name", "Blower speed")
-    publish("spa/blower/$datatype", "integer")
-    publish("spa/blower/$format", "0:#{speeds}")
-    publish("spa/blower/$settable", "true")
-    subscribe("spa/blower/set")
-
-    @things << "blower"
-    publish_nodes
-  end
-
-  def publish_circpump
-    publish("spa/circpump/$name", "Circ pump is currently running")
-    publish("spa/circpump/$datatype", "boolean")
-    @things << "circpump"
-
-    publish_nodes
-  end
-
-  def publish_nodes
-    publish("spa/$properties", (["priming,heatingmode,temperaturescale,24htime,heating,temperaturerange,currenttemperature,settemperature,filter1,filter2"] + @things.to_a).join(','))
-  end
-end
-
-mqtt_uri = ARGV.shift
-
-if ARGV.empty?
-  spas = BWA::Discovery.discover
-  if spas.empty?
-    $stderr.puts "Could not find spa!"
-    exit 1
-  end
-  spa_ip = "tcp://#{spas.first.first}/"
-else
-  spa_ip = ARGV[0]
-end
-
-spa = BWA::Client.new(spa_ip)
-
-spa.request_configuration
-
-MQTTBridge.new(mqtt_uri, spa)