#!/usr/bin/ruby

# Example script using MQTT to access adafruit.io
# - Nicholas Humfrey has an excellent MQTT library, which is very easy to use.

begin
  require 'mqtt' # gem install mqtt ;  https://github.com/njh/ruby-mqtt
rescue LoadError
  fatal 'The mqtt gem is missing ; try the following command: gem install mqtt'
end

require 'date'
require 'json'
require 'securerandom'
require 'socket'

require './refidtracker.rb'

# ---
# Define MQTT config, mostly from the environment

begin
  require 'dotenv' # If a .env file is present, it will be loaded
  Dotenv.load('mqtt.env') #   into the environment, which is a handy way to do it.
rescue LoadError
  warn 'Warning: dotenv not found - make sure ENV contains proper variables'
end

# Required
MQTT_USER = ENV['ADAFRUIT_USER'].freeze
MQTT_KEY = ENV['ADAFRUIT_IO_KEY'].freeze

# Optional
MQTT_HOST   = (ENV['ADAFRUIT_HOST'] || 'jophiel.wisslanding.com').freeze
MQTT_PORT   = (ENV['ADAFRUIT_PORT'] || 1883).freeze

ADAFRUIT_FORMAT = ENV['ADAFRUIT_FORMAT'].freeze
CLIENT	= Socket.gethostname

# ---
# Allow filtering to a specific format

# ADAFRUIT_DOCUMENTED_FORMATS = %w( csv json xml ).freeze
# Adafruit-MQTT doesn't support XML 160619
ADAFRUIT_MQTT_FORMATS       = %w[csv json].freeze

FORMAT_REGEX_PATTERN        = %r{/(csv|json)$}

FILTER_FORMAT = if ADAFRUIT_FORMAT.nil?
                  nil
                elsif ADAFRUIT_MQTT_FORMATS.include?(ADAFRUIT_FORMAT)
                  "/#{ADAFRUIT_FORMAT}".freeze
                else
                  $stderr.puts("Unsupported format (#{ADAFRUIT_FORMAT})")
                  exit 1
                end

CONNECT_INFO = {
  username: MQTT_USER,
  password: MQTT_KEY,
  host:     MQTT_HOST,
  port:     MQTT_PORT,
  client_id:	CLIENT,
  clean_session: true
}.freeze

TOPIC = 'mcr/f/report'.freeze
$stderr.puts "Connecting to #{MQTT_HOST} as #{MQTT_USER} for #{TOPIC}"

def handleClientStartup(client)
  send_time_sync(client)
  Time.new.to_i # return time for recording last publish time
end

def send_time_sync(client)
  t = Time.new
  time_str = "#{t.to_i} "

  time_sync_hash = {
    version: '1', cmd: 'time.sync', mtime: time_str.to_s, key: '0xAA'
  }

  time_sync_json = JSON.generate(time_sync_hash)

  client.publish('mcr/f/command', time_sync_json)
end

def human_ms(msec)
  unit = 'ms'
  val = msec.round(2)

  if msec > 1000
    unit = 's'
    val = msec / 1000
    val = val.round(2)
  end

  "#{val}#{unit}"
end

def human_us(usec)
  unit = 'us'
  val = usec

  if usec > 1000
    unit = 'ms'
    val = usec / 1000
    val = val.round(2)
  end

  "#{val}#{unit}"
end

cmd_tracker = CmdTracker.new

MQTT::Client.connect(CONNECT_INFO).connect do |client|
  log = File.open('/tmp/merc.log', 'a+')

  client.subscribe(TOPIC => 2)

  last_publish_time = Time.new
  last_heartbeat_time = Time.new
  last_led_flash = Time.new

  led_state = true
  loop do
    if client.queue_length > 0
      feed, data_str = client.get(TOPIC => 2)

      if data_str.include?('{')
        # $stderr.puts "json=#{data_str}"
        datum = JSON.parse(data_str)

        msg_time = if datum.key?('mtime')
                     Time.at(datum['mtime'])
                   else
                     Time.now
                   end

        # msg1 = "#{msg_time} host=#{datum["host"]} "
				msg0 = " #{data_str}"
        msg1 = "#{msg_time}  "
        # msg2 = " " * (feed.length - 1)
        msg2 = ' '

        if datum.key?('startup')
          last_publish_time = handleClientStartup(client)
          msg2 += "startup=#{datum['startup']}"
        end

        if datum.key?('device')
          # time_diff = Time.new.to_i - datum['mtime'].to_i

          # msg1 += "device=#{datum["device"]} type=#{datum["type"]} time_diff=#{time_diff}"
          msg1 += "#{datum['device']} #{datum['type']} "
					msg2 = "#{datum["pio"]}"

          if datum.key?('cmdack')
            ref_id = datum['refid']
            rt_latency = cmd_tracker.untrack(ref_id)
            msg1 += "cmdack latency=#{human_us(datum['latency'])} "
            msg1 += "rt_latency=#{human_ms(rt_latency)}"

            log.puts msg1
            log.fsync
          end

        end


				# $stderr.puts msg0
        $stderr.puts msg1

				if msg2.length > 10
        	$stderr.puts "   #{msg2}"
				end
        $stderr.puts ' '
      end

    else
      # $stderr.puts "#{TOPIC} queue empty, sleeping...."
      sleep(0.01)
    end

    if (Time.new.to_i - last_publish_time.to_i) > 10
      send_time_sync(client)
      last_publish_time = Time.new.to_i
    end

    heartbeat_time = Time.new
    if (heartbeat_time.to_i - last_heartbeat_time.to_i) > 10
      heartbeat_hash = {
        version: '1', cmd: 'heartbeat', mtime: heartbeat_time.to_i.to_s, key: '0xAA',
        master: MQTT_HOST
      }

      heartbeat_json = JSON.generate(heartbeat_hash)

      client.publish('mcr/f/command', heartbeat_json, qos = 0)

      last_heartbeat_time = heartbeat_time
    end

    led = 'ds/291d1823000000'
    #	switch = "ds/124c8421000000"
    # buzzer = 'ds/12128521000000'
    sysX = 'ds/12838421000000'
    sysY = 'ds/12398521000000'
    sysZ = 'ds/124c8421000000'
    pio = 0

    devs = [led, sysZ, sysY, sysX]
    states = [true, false]

    led_flash = Time.new
    next unless (led_flash.to_i - last_led_flash.to_i) >= 3

    # hash1 = {
    #	:version => "1", :cmd => "set.switch", :mtime => "#{led_flash.to_i}", :key => "0xAA",
    #	:switch => buzzer, :pio => ["#{pio}" => true], :pio_count => 1 }

    for d in devs
      for s in states
        cid = SecureRandom.uuid
        hash = {
          version: '1', cmd: 'set.switch', mtime: led_flash.to_i.to_s, key: '0xAA',
          switch: d, pio: [pio.to_s => s], pios: 1, refid: cid
        }

        json = JSON.generate(hash)
        cmd_tracker.track(cid)
        client.publish('mcr/f/command', json, qos = 0)
        #		sleep(0.05)
      end
    end

    led_state = !led_state
    last_led_flash = led_flash
  end
end

# (auto-disconnects when #connect is given a block)
exit 0