mycorerepository/MyCore/Services/Devices/DeviceService.cs

using Microsoft.AspNetCore.Mvc;
using Mqtt.Client.AspNetCore.Services;
using MyCore.Interfaces.DTO;
using MyCore.Interfaces.Models;
using MyCore.Interfaces.Models.MyControlPanel;
using MyCore.Services.MyControlPanel;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Net.Http;
using System.Security.Authentication;
using System.Threading.Tasks;

namespace MyCore.Services.Devices
{
    public class DeviceService
    {
        static List<SupportedDevice> supportedDevices = new List<SupportedDevice>() {
            new SupportedDevice { Manufacturer = "Aqara", Model = "RTCGQ11LM", Description = "Aqara human body movement and illuminance sensor", DeviceType = DeviceType.Motion },
            new SupportedDevice { Manufacturer = "Aqara", Model = "DJT11LM", Description = "Aqara vibration sensor", DeviceType = DeviceType.Environment },
            new SupportedDevice { Manufacturer = "Aqara", Model = "SJCGQ11LM", Description = "Aqara water leak sensor", DeviceType = DeviceType.Environment },
            new SupportedDevice { Manufacturer = "Aqara", Model = "WSDCGQ11LM", Description = "Aqara temperature, humidity and pressure sensor", DeviceType = DeviceType.Environment },

            new SupportedDevice { Manufacturer = "Xiaomi", Model = "WXKG01LM", Description = "MiJia wireless switch", DeviceType = DeviceType.Switch },
            new SupportedDevice { Manufacturer = "Xiaomi", Model = "MFKZQ01LM", Description = "Mi/Aqara smart home cube", DeviceType = DeviceType.Switch },
            new SupportedDevice { Manufacturer = "Xiaomi", Model = "MCCGQ01LM", Description = "MiJia door & window contact sensor", DeviceType = DeviceType.Door },
            new SupportedDevice { Manufacturer = "Xiaomi", Model = "JTYJ-GD-01LM/BW", Description = "MiJia Honeywell smoke detector", DeviceType = DeviceType.Environment },
            new SupportedDevice { Manufacturer = "Xiaomi", Model = "ZNCZ04LM", Description = "Mi power plug ZigBee EU", DeviceType = DeviceType.Plug },

            new SupportedDevice { Manufacturer = "Ikea", Model = "E1744", Description = "SYMFONISK Sound Controller", DeviceType = DeviceType.Switch },
            new SupportedDevice { Manufacturer = "Ikea", Model = "LED1836G9", Description = "TRADFRI LED bulb E26/E27 806 lumen, dimmable, warm white", DeviceType = DeviceType.Light },
            new SupportedDevice { Manufacturer = "Ikea", Model = "LED1842G3", Description = "TRADFRI LED bulb E27 WW clear 250 lumen, dimmable", DeviceType = DeviceType.Light },
            new SupportedDevice { Manufacturer = "Ikea", Model = "LED1837R5", Description = "TRADFRI LED bulb GU10 400 lumen, dimmable", DeviceType = DeviceType.Light },
            new SupportedDevice { Manufacturer = "Ikea", Model = "E1743", Description = "TRADFRI ON/OFF switch", DeviceType = DeviceType.Switch },
            new SupportedDevice { Manufacturer = "Ikea", Model = "E1524/E1810", Description = "TRADFRI remote control", DeviceType = DeviceType.Switch },
        };

        public static DeviceDetailDTO CreateOrUpdate(DeviceDatabaseService _DeviceDatabaseService, ProviderDatabaseService _ProviderDatabaseService, LocationDatabaseService _LocationDatabaseService, string userId, DeviceDetailDTO deviceDetailDTO, bool create)
        {
            Device device;
            if (create)
                device = new Device();
            else 
            {
                device = _DeviceDatabaseService.GetById(deviceDetailDTO.Id);
            }

            if (_DeviceDatabaseService.IsAlreadyHere(userId, deviceDetailDTO.ServiceIdentification, deviceDetailDTO.Model) && create)
            {
                return null;
            }

            device.UserId = userId;
            device.Name = deviceDetailDTO.Name;
            device.Description = deviceDetailDTO.Description;
            device.ManufacturerName = deviceDetailDTO.ManufacturerName;

            if (_ProviderDatabaseService.IsExist(deviceDetailDTO.ProviderId))
                device.ProviderId = deviceDetailDTO.ProviderId;
            else
                throw new KeyNotFoundException("Provider does not exist");

            if (device.LocationId == null || _LocationDatabaseService.IsExist(deviceDetailDTO.LocationId))
                device.LocationId = deviceDetailDTO.LocationId;
            else
                throw new KeyNotFoundException("Location does not exist");

            device.Port = deviceDetailDTO.Port;
            device.Model = deviceDetailDTO.Model;
            device.Type = deviceDetailDTO.Type;
            device.FirmwareVersion = deviceDetailDTO.FirmwareVersion;
            device.HardwareVersion = deviceDetailDTO.HardwareVersion;
            device.Status = deviceDetailDTO.Status;
            if (create)
                device.ConnectionStatus = ConnectionStatus.Unknown;
            else
                device.ConnectionStatus = deviceDetailDTO.ConnectionStatus;
            device.Status = deviceDetailDTO.Status;
            device.LocationId = deviceDetailDTO.LocationId;
            device.CreatedDate = DateTime.Now;
            device.UpdatedDate = DateTime.Now;
            device.LastState = deviceDetailDTO.LastState;
            device.LastStateDate = deviceDetailDTO.LastStateDate;

            device.MeansOfCommunications = deviceDetailDTO.MeansOfCommunications;
            device.IpAddress = deviceDetailDTO.IpAddress;
            device.ServiceIdentification = deviceDetailDTO.ServiceIdentification;
            device.Battery = deviceDetailDTO.Battery;
            device.BatteryStatus = deviceDetailDTO.BatteryStatus;
            device.GroupIds = deviceDetailDTO.GroupIds;
            // Todo structure Properties
            device.Properties = deviceDetailDTO.Properties;
            // Todo structure SupportedOperations
            device.SupportedOperations = deviceDetailDTO.SupportedOperations;

            if (create)
                return _DeviceDatabaseService.Create(device).ToDTO();
            else
                return _DeviceDatabaseService.Update(device.Id, device).ToDTO();
        }

        public async static Task<Dictionary<string, List<DeviceDetailDTO>>> CreateFromProvider(DeviceDatabaseService _DeviceDatabaseService, ProviderDatabaseService _ProviderDatabaseService, LocationDatabaseService _LocationDatabaseService, string userId, Provider provider)
        {
            if (!ProviderService.IsProviderSupported(provider.Type))
                throw new KeyNotFoundException("Provider is not yet supported");

            Dictionary<string, List<DeviceDetailDTO>> devices = new Dictionary<string, List<DeviceDetailDTO>>();

            try {
                switch (provider.Type)
                {
                    case "arlo":
                        List<ArloDevice> arloDevices = new ArloService(provider.Username, provider.Password).GetAllDevices();
                        devices = CreateArloDevices(_DeviceDatabaseService, _ProviderDatabaseService, _LocationDatabaseService, userId, arloDevices, provider);
                        break;
                    case "meross":
                        List<MerossDevice> merossDevices = MqttClientMerossService.GetMerossDevices(); // TO TEST IF IT WORKS
                        devices = CreateMerossDevices(_DeviceDatabaseService, _ProviderDatabaseService, _LocationDatabaseService, userId, merossDevices, provider);
                        break;
                    case "yeelight":
                        List<YeelightAPI.Device> yeelightDevices = await YeelightService.GetDevices();
                        devices = CreateYeelightDevices(_DeviceDatabaseService, _ProviderDatabaseService, _LocationDatabaseService, userId, yeelightDevices, provider);
                        break;
                    case "zigbee2mqtt":
                        List<Zigbee2MqttDevice> zigbee2MqttDevices = MqttClientService.devices;
                        devices = await CreateFromZigbeeAsync(_DeviceDatabaseService, _ProviderDatabaseService, _LocationDatabaseService, userId, zigbee2MqttDevices, provider);
                        break;
                }
            }
            catch (AuthenticationException ex) {
                throw new AuthenticationException("Bad username or password for service " + provider.Name + " ex: " + ex.Message);
            }
            catch (HttpRequestException ex) {
                throw new HttpRequestException("Error retrieving devices for " + provider.Name + " ex: " + ex.Message);
            } 
            catch (Exception ex) {

            }
            return devices;
        }

        public static async Task<Dictionary<string, List<DeviceDetailDTO>>> CreateFromZigbeeAsync(DeviceDatabaseService _DeviceDatabaseService, ProviderDatabaseService _ProviderDatabaseService, LocationDatabaseService _LocationDatabaseService, string userId, List<Zigbee2MqttDevice> zigbee2MqttDevices, Provider provider)
        {
            List<DeviceDetailDTO> createdZigbeeDevices = new List<DeviceDetailDTO>();
            List<DeviceDetailDTO> notSupportedZigbeeDevices = new List<DeviceDetailDTO>();

            List<Device> existingDevices = _DeviceDatabaseService.GetByProviderId(provider.Id);

            if (zigbee2MqttDevices.Count <= 0)
            {
                zigbee2MqttDevices = await MqttClientService.AskDevicesAsync();
            }

            zigbee2MqttDevices = zigbee2MqttDevices.Where(yd => !existingDevices.Select(ed => ed.ServiceIdentification).ToList().Contains(yd.ieeeAddr)).ToList();

            foreach (var zigbee2MqttDevice in zigbee2MqttDevices)
            {
                DeviceDetailDTO deviceDetailDTO = new DeviceDetailDTO();
                deviceDetailDTO.Name = zigbee2MqttDevice.friendly_name;
                deviceDetailDTO.Description = zigbee2MqttDevice.description;
                deviceDetailDTO.ServiceIdentification = zigbee2MqttDevice.ieeeAddr;
                deviceDetailDTO.ProviderId = provider.Id;
                deviceDetailDTO.ProviderName = provider.Name;

                deviceDetailDTO.Model = zigbee2MqttDevice.type == "Coordinator" ? "Coordinator" : zigbee2MqttDevice.model; // Is the base to understand incoming messages !
                deviceDetailDTO.FirmwareVersion = zigbee2MqttDevice.softwareBuildID;
                deviceDetailDTO.HardwareVersion = zigbee2MqttDevice.hardwareVersion.ToString();

                deviceDetailDTO.Battery = zigbee2MqttDevice.powerSource == null ? false : zigbee2MqttDevice.powerSource.Contains("Battery");
                deviceDetailDTO.ManufacturerName = zigbee2MqttDevice.vendor == null ? provider.Type : zigbee2MqttDevice.vendor.ToLower();
                deviceDetailDTO.MeansOfCommunications = new List<MeansOfCommunication>();
                deviceDetailDTO.MeansOfCommunications.Add(MeansOfCommunication.Zigbee);

                deviceDetailDTO.CreatedDate = DateTime.Now;
                deviceDetailDTO.UpdatedDate = DateTime.Now;
                deviceDetailDTO.LastStateDate = new DateTime(zigbee2MqttDevice.lastSeen);

                deviceDetailDTO.Type = zigbee2MqttDevice.type == "Coordinator" ? DeviceType.Gateway : GetDeviceTypeFromZigbeeModel(zigbee2MqttDevice.model);

                if (deviceDetailDTO.Type != DeviceType.Unknown)
                {
                    // Supoorted device ! 
                    createdZigbeeDevices.Add(CreateOrUpdate(_DeviceDatabaseService, _ProviderDatabaseService, _LocationDatabaseService, userId, deviceDetailDTO, true));
                }
                else {
                    // Not yet supported !
                    notSupportedZigbeeDevices.Add(deviceDetailDTO);
                }
            }

            return new Dictionary<string, List<DeviceDetailDTO>>() {
                { "createdDevices", createdZigbeeDevices },
                { "notSupportedDevices", notSupportedZigbeeDevices }
            };
        }

        private static DeviceType GetDeviceTypeFromZigbeeModel(string zigbeeModel)
        {
            return supportedDevices.Any(sd => sd.Model == zigbeeModel) ?
                // It's supported !
                supportedDevices.Where(sd => sd.Model == zigbeeModel).FirstOrDefault().DeviceType :
                // Not yet supported
                DeviceType.Unknown;
        }

        public static Dictionary<string, List<DeviceDetailDTO>> CreateArloDevices(DeviceDatabaseService _DeviceDatabaseService, ProviderDatabaseService _ProviderDatabaseService, LocationDatabaseService _LocationDatabaseService, string userId, List<ArloDevice> arloDevices, Provider provider)
        {
            List<DeviceDetailDTO> createdArloDevices = new List<DeviceDetailDTO>();

            List<Device> existingDevices = _DeviceDatabaseService.GetByProviderId(provider.Id);

            arloDevices = arloDevices.Where(yd => !existingDevices.Select(ed => ed.ServiceIdentification).ToList().Contains(yd.deviceId)).ToList();

            foreach (var arlo in arloDevices)
            {
                DeviceDetailDTO deviceDetailDTO = new DeviceDetailDTO();
                deviceDetailDTO.ManufacturerName = provider.Type;
                deviceDetailDTO.Name = arlo.deviceName;
                deviceDetailDTO.Description = arlo.deviceName; // As description not exist, put name in description
                deviceDetailDTO.ServiceIdentification = arlo.deviceId;
                deviceDetailDTO.ProviderId = provider.Id;
                deviceDetailDTO.ProviderName = provider.Name;
                if (arlo.connectivity != null)
                {
                    deviceDetailDTO.ConnectionStatus = arlo.connectivity.connected ? ConnectionStatus.Connected : ConnectionStatus.Unknown;
                    deviceDetailDTO.Status = arlo.connectivity.connected ? true : false; // TODO STATE
                }
                else
                    deviceDetailDTO.ConnectionStatus = ConnectionStatus.Unknown;        
                
                deviceDetailDTO.Model = arlo.modelId;
                deviceDetailDTO.FirmwareVersion = arlo.interfaceVersion; // TODO
                switch (arlo.deviceType)
                {
                    case "camera":
                        deviceDetailDTO.Type = DeviceType.Camera;
                        Dictionary<string, object> properties = new Dictionary<string, object>();
                        properties.Add("lastImageUploaded", arlo.lastImageUploaded);
                        properties.Add("presignedLastImageUrl", arlo.presignedLastImageUrl);
                        properties.Add("presignedFullFrameSnapshotUrl", arlo.presignedFullFrameSnapshotUrl);
                        properties.Add("presignedSnapshotUrl", arlo.presignedSnapshotUrl);
                        deviceDetailDTO.Properties = properties;
                        break;
                    case "siren":
                        deviceDetailDTO.Type = DeviceType.Sound;
                        break;
                    case "basestation":
                        deviceDetailDTO.Type = DeviceType.Gateway;
                        break;
                }
                
                deviceDetailDTO.MeansOfCommunications = new List<MeansOfCommunication>();
                deviceDetailDTO.MeansOfCommunications.Add(MeansOfCommunication.Wifi); // To check

                deviceDetailDTO.CreatedDate = DateTime.Now;
                deviceDetailDTO.UpdatedDate = DateTime.Now;
                createdArloDevices.Add(CreateOrUpdate(_DeviceDatabaseService, _ProviderDatabaseService, _LocationDatabaseService, userId, deviceDetailDTO, true));
            }

            return new Dictionary<string, List<DeviceDetailDTO>>() { { "createdDevices", createdArloDevices } }; // TODO Check if exist not supported devices
        }

        public static Dictionary<string, List<DeviceDetailDTO>> CreateMerossDevices(DeviceDatabaseService _DeviceDatabaseService, ProviderDatabaseService _ProviderDatabaseService, LocationDatabaseService _LocationDatabaseService, string userId, List<MerossDevice> merossDevices, Provider provider)
        {
            List<DeviceDetailDTO> createdMerossDevices = new List<DeviceDetailDTO>();

            List<Device> existingDevices = _DeviceDatabaseService.GetByProviderId(provider.Id);

            merossDevices = merossDevices.Where(yd => !existingDevices.Select(ed => ed.ServiceIdentification).ToList().Contains(yd.uuid)).ToList();

            foreach (var meross in merossDevices)
            {
                DeviceDetailDTO deviceDetailDTO = new DeviceDetailDTO();
                deviceDetailDTO.ManufacturerName = provider.Type;
                deviceDetailDTO.HardwareVersion = meross.hdwareVersion;
                deviceDetailDTO.Name = meross.devName;
                deviceDetailDTO.Description = meross.devName; // As description not exist, put name in description
                deviceDetailDTO.ServiceIdentification = meross.uuid;
                deviceDetailDTO.ProviderId = provider.Id;
                deviceDetailDTO.ProviderName = provider.Name;
                deviceDetailDTO.ConnectionStatus = meross.onlineStatus == 1 ? ConnectionStatus.Connected : ConnectionStatus.Disconnected;
                // deviceDetailDTO.Status = meross. ? true : false; // TODO STATE
                deviceDetailDTO.Model = meross.deviceType;
                switch (deviceDetailDTO.Model) // TODO
                {
                    case "mss425f":
                        deviceDetailDTO.Type = DeviceType.Plug;
                        break;
                    case "mss310":
                        deviceDetailDTO.Type = DeviceType.Actuator;
                        break;
                    case "msh300":
                        deviceDetailDTO.Type = DeviceType.Gateway;
                        break;
                    default:
                        deviceDetailDTO.Type = DeviceType.Actuator;
                        break;
                }
                deviceDetailDTO.FirmwareVersion = meross.firmwareVersion; // TODO
                Dictionary<string, object> properties = new Dictionary<string, object>();
                foreach (var property in meross.channels)
                {
                    if (property.type != null) 
                        properties.Add(property.devName, property.type);
                }
                // deviceDetailDTO.SupportedOperations = supportedOperationsDTO; TODO

                deviceDetailDTO.MeansOfCommunications = new List<MeansOfCommunication>();
                deviceDetailDTO.MeansOfCommunications.Add(MeansOfCommunication.Wifi); // TO CHECK
                deviceDetailDTO.CreatedDate = DateTime.Now;
                deviceDetailDTO.UpdatedDate = DateTime.Now;
                createdMerossDevices.Add(CreateOrUpdate(_DeviceDatabaseService, _ProviderDatabaseService, _LocationDatabaseService, userId, deviceDetailDTO, true));
            }

            return new Dictionary<string, List<DeviceDetailDTO>>() { { "createdDevices", createdMerossDevices } }; // TODO Check if exist not supported devices
        }

        public static Dictionary<string, List<DeviceDetailDTO>> CreateYeelightDevices(DeviceDatabaseService _DeviceDatabaseService, ProviderDatabaseService _ProviderDatabaseService, LocationDatabaseService _LocationDatabaseService, string userId, List<YeelightAPI.Device> yeelightDevices, Provider provider)
        {
            List<DeviceDetailDTO> createdYeelightDevices = new List<DeviceDetailDTO>();

            List<Device> existingDevices = _DeviceDatabaseService.GetByProviderId(provider.Id);

            yeelightDevices = yeelightDevices.Where(yd => !existingDevices.Select(ed => ed.ServiceIdentification).ToList().Contains(yd.Id)).ToList();

            foreach (var light in yeelightDevices)
            {
                DeviceDetailDTO deviceDetailDTO = new DeviceDetailDTO();
                deviceDetailDTO.ManufacturerName = provider.Type;
                deviceDetailDTO.Name = light.Name;
                deviceDetailDTO.Description = light.Name; // As description not exist, put name in description
                deviceDetailDTO.IpAddress = light.Hostname;
                deviceDetailDTO.ServiceIdentification = light.Id;
                deviceDetailDTO.ProviderId = provider.Id;
                deviceDetailDTO.ProviderName = provider.Name;
                deviceDetailDTO.ConnectionStatus = ConnectionStatus.Connected;
                deviceDetailDTO.Status = false;
                deviceDetailDTO.Model = light.Model.ToString();
                deviceDetailDTO.Type = DeviceType.Light; // TO CHECK
                deviceDetailDTO.Port = light.Port;
                deviceDetailDTO.FirmwareVersion = light.FirmwareVersion;
                Dictionary<string, object> properties = new Dictionary<string, object>();
                foreach (var property in light.Properties)
                {
                    properties.Add(property.Key, property.Value);
                }
                deviceDetailDTO.Properties = properties;
                List<string> supportedOperationsDTO = new List<string>();
                foreach (var supportedOperation in light.SupportedOperations)
                {
                    supportedOperationsDTO.Add(supportedOperation.ToString());
                }
                deviceDetailDTO.SupportedOperations = supportedOperationsDTO;
                deviceDetailDTO.MeansOfCommunications = new List<MeansOfCommunication>();
                deviceDetailDTO.MeansOfCommunications.Add(MeansOfCommunication.Wifi);
                deviceDetailDTO.CreatedDate = DateTime.Now;
                deviceDetailDTO.UpdatedDate = DateTime.Now;
                createdYeelightDevices.Add(CreateOrUpdate(_DeviceDatabaseService, _ProviderDatabaseService, _LocationDatabaseService, userId, deviceDetailDTO, true));
            }

            return new Dictionary<string, List<DeviceDetailDTO>>() { { "createdDevices", createdYeelightDevices } }; // TODO Check if exist not supported devices
        }
    }
}