package cwmp_handler

import (
	"encoding/json"
	"encoding/xml"
	"log"

	"github.com/OktopUSP/oktopus/backend/services/mtp/adapter/internal/cwmp"
	"github.com/OktopUSP/oktopus/backend/services/mtp/adapter/internal/db"
)

func (h *Handler) HandleDeviceInfo(device string, data []byte, ack func()) {
	defer ack()
	log.Printf("Device %s info", device)
	deviceInfo := parseDeviceInfoMsg(data)
	if deviceExists, _ := h.db.DeviceExists(deviceInfo.SN); !deviceExists {
		fmtDeviceInfo, _ := json.Marshal(deviceInfo)
		h.nc.Publish("device.v1.new", fmtDeviceInfo)
	}
	err := h.db.CreateDevice(deviceInfo)
	if err != nil {
		log.Printf("Failed to create device: %v", err)
	}
}

func parseDeviceInfoMsg(data []byte) db.Device {

	var inform cwmp.CWMPInform
	err := xml.Unmarshal(data, &inform)
	if err != nil {
		log.Println("Error unmarshalling xml:", err)
	}

	var device db.Device

	device.Vendor = inform.DeviceId.Manufacturer
	device.Model = ""
	device.Version = inform.GetSoftwareVersion()
	device.ProductClass = inform.DeviceId.ProductClass
	device.SN = inform.DeviceId.SerialNumber
	device.Cwmp = db.Online
	device.Status = db.Online

	return device
}