package zigbee

import (
	"automation/device"
	"automation/home"
	"context"
	"encoding/json"
	"fmt"
	"log"

	paho "github.com/eclipse/paho.mqtt.golang"
)

func DevicesHandler(client paho.Client, prefix string, home *home.Home) {
	var handler paho.MessageHandler = func(client paho.Client, msg paho.Message) {
		var devices []Info
		json.Unmarshal(msg.Payload(), &devices)

		for name, d := range device.GetDevices[Device](&home.Devices) {
			d.Delete(client)
			// Delete all zigbee devices from the device list
			delete(home.Devices, name)
		}

		for _, d := range devices {
			d.MQTTAddress = fmt.Sprintf("%s/%s", prefix, d.FriendlyName.String())

			switch d.Description {
			case "Kettle":
				kettle := NewKettle(d, client, home.Service)
				home.AddDevice(kettle)
			case "LightSensor":
				lightSensor := NewLightSensor(d, client)
				home.AddDevice(lightSensor)
			}
		}

		// Send sync request
		// @TODO Instead of sending a sync request we should do something like home.sync <- interface{}
		// This will then restart a timer, that way the sync will only trigger once everything has settled from multiple locations
		home.Service.RequestSync(context.Background(), home.Username)

		// Unsubscribe again, otherwise updates here will re-add all the devices which causes issues with the light sensor
		if token := client.Unsubscribe(fmt.Sprintf("%s/bridge/devices", prefix)); token.Wait() && token.Error() != nil {
			log.Println(token.Error())
		}
	}

	if token := client.Subscribe(fmt.Sprintf("%s/bridge/devices", prefix), 1, handler); token.Wait() && token.Error() != nil {
		log.Println(token.Error())
	}
}