import asyncdispatch, strutils, json, tables, options, times
import ../types, ../vars, ../utils
import nmqtt

proc setZigbee2MqttRelayState* (deviceName: string, value: bool) {.async.} =
  let config    = server.config.devices[deviceName]
  var sendState = "OFF"

  if config.type != Zigbee2MqttRelay:
    return 

  if value == true:
    sendState = "ON"

  await mqttContext.publish("zigbee2mqtt/" & config.deviceName.get & "/set", "{\"state\": \"" & sendState & "\"}")

proc toggleZigbee2MqttRelayState* (deviceName: string) {.async.} =
  let config = server.config.devices[deviceName]

  if config.type != Zigbee2MqttRelay:
    return 

  await mqttContext.publish("zigbee2mqtt/" & config.deviceName.get & "/set", "{\"state\": \"TOGGLE\"}")

proc updateRelay (topic: string, message: string) =
  try:
    let deviceName = zigbee2mqttDevices[topic]
    let recivedData = parseJson(message)
  
    server.state[deviceName].lastUpdated = some(toUnix(getTime()))

    if recivedData.hasKey("linkquality"):
      server.state[deviceName].relayLinkquality = recivedData["linkquality"].getInt

    if recivedData.hasKey("state"):
      if recivedData["state"].getStr != "ON":
        server.state[deviceName].relayState = false
      else:
        server.state[deviceName].relayState = true

    broadcastServerState()
  except:
    echo "Error[updateRelay]:\n", getCurrentExceptionMsg()

proc initZigbee2MqttRelays* () {.async.} =
  for key, device in server.config.devices.pairs():
    if device.type != Zigbee2MqttRelay: continue

    zigbee2mqttDevices["zigbee2mqtt/" & device.deviceName.get] = key
    server.state[key] = DeviceState(type: Zigbee2MqttRelay)

    await mqttContext.subscribe("zigbee2mqtt/" & device.deviceName.get, 2, updateRelay)