// Stl includes
#include <string>
#include <sstream>
#include <algorithm>
#include <exception>

// Build configuration
#include <HyperionConfig.h>

// Leddevice includes
#include <leddevice/LedDeviceFactory.h>
#include <utils/Logger.h>

// Local Leddevice includes
#ifdef ENABLE_SPIDEV
	#include "LedDeviceLpd6803.h"
	#include "LedDeviceLpd8806.h"
	#include "LedDeviceP9813.h"
	#include "LedDeviceWs2801.h"
	#include "LedDeviceWs2812SPI.h"
	#include "LedDeviceSk6812SPI.h"
	#include "LedDeviceAPA102.h"
#endif

#ifdef ENABLE_TINKERFORGE
	#include "LedDeviceTinkerforge.h"
#endif

#include "LedDeviceAdalight.h"
#include "LedDeviceRawHID.h"
#include "LedDeviceLightpack.h"
#include "LedDeviceMultiLightpack.h"
#include "LedDevicePaintpack.h"
#include "LedDevicePiBlaster.h"
#include "LedDeviceSedu.h"
#include "LedDeviceFile.h"
#include "LedDeviceFadeCandy.h"
#include "LedDeviceUdp.h"
#include "LedDeviceUdpRaw.h"
#include "LedDeviceHyperionUsbasp.h"
#include "LedDevicePhilipsHue.h"
#include "LedDeviceTpm2.h"
#include "LedDeviceAtmo.h"
#include "LedDeviceAdalightApa102.h"
#include "LedDeviceAtmoOrb.h"

#ifdef ENABLE_WS2812BPWM
	#include "LedDeviceWS2812b.h"
#endif

#ifdef ENABLE_WS281XPWM
	#include "LedDeviceWS281x.h"
#endif

LedDevice * LedDeviceFactory::construct(const Json::Value & deviceConfig)
{
	Logger * log = Logger::getInstance("LedDevice");
	std::stringstream ss;
	ss << deviceConfig;
	Info(log, "configuration: %s ", ss.str().c_str());

	std::string type = deviceConfig.get("type", "UNSPECIFIED").asString();
	std::transform(type.begin(), type.end(), type.begin(), ::tolower);

	LedDevice* device = nullptr;
	try
	{
		if (false) {}
		else if (type == "adalight")
		{
			device = new LedDeviceAdalight(
				deviceConfig["output"].asString(),
				deviceConfig["rate"].asInt(),
				deviceConfig.get("delayAfterConnect",500).asInt()
			);
		}
		else if (type == "adalightapa102")
		{
			device = new LedDeviceAdalightApa102(
				deviceConfig["output"].asString(),
				deviceConfig["rate"].asInt(),
				deviceConfig.get("delayAfterConnect",500).asInt()
			);
		}
	#ifdef ENABLE_SPIDEV
		else if (type == "lpd6803" || type == "ldp6803")
		{
			device = new LedDeviceLpd6803(
				deviceConfig["output"].asString(),
				deviceConfig["rate"].asInt()
			);
		}
		else if (type == "lpd8806" || type == "ldp8806")
		{
			device = new LedDeviceLpd8806(
				deviceConfig["output"].asString(),
				deviceConfig["rate"].asInt()
			);
		}
		else if (type == "p9813")
		{
			device = new LedDeviceP9813(
				deviceConfig["output"].asString(),
				deviceConfig["rate"].asInt()
			);
		}
		else if (type == "apa102")
		{
			device = new LedDeviceAPA102(
				deviceConfig["output"].asString(),
				deviceConfig["rate"].asInt()
			);
		}
		else if (type == "ws2801" || type == "lightberry")
		{
			device = new LedDeviceWs2801(
				deviceConfig["output"].asString(),
				deviceConfig["rate"].asInt(),
				deviceConfig.get("latchtime",500000).asInt(),
				deviceConfig.get("spimode",0).asInt(),
				deviceConfig.get("invert",false).asBool()
			);
		}
		else if (type == "ws2812spi")
		{
			device = new LedDeviceWs2812SPI(
				deviceConfig["output"].asString(),
				deviceConfig.get("rate",2857143).asInt(),
				deviceConfig.get("spimode",0).asInt(),
				deviceConfig.get("invert",false).asBool()
			);
		}
		else if (type == "sk6812rgbw-spi")
		{
			device = new LedDeviceSk6812SPI(
				deviceConfig["output"].asString(),
				deviceConfig.get("rate",2857143).asInt(),
				deviceConfig.get("white_algorithm","").asString(),
				deviceConfig.get("spimode",0).asInt(),
				deviceConfig.get("invert",false).asBool()
			);
		}
	#endif
	#ifdef ENABLE_TINKERFORGE
		else if (type=="tinkerforge")
		{
			device = new LedDeviceTinkerforge(
				deviceConfig.get("output", "127.0.0.1").asString(),
				deviceConfig.get("port", 4223).asInt(),
				deviceConfig["uid"].asString(),
				deviceConfig["rate"].asInt()
			);

		}
	#endif
		else if (type == "rawhid")
		{
			const int delay_ms        = deviceConfig["delayAfterConnect"].asInt();
			auto VendorIdString       = deviceConfig.get("VID", "0x2341").asString();
			auto ProductIdString      = deviceConfig.get("PID", "0x8036").asString();

			// Convert HEX values to integer
			auto VendorId = std::stoul(VendorIdString, nullptr, 16);
			auto ProductId = std::stoul(ProductIdString, nullptr, 16);

			device = new LedDeviceRawHID(VendorId, ProductId, delay_ms);
		}
		else if (type == "lightpack")
		{
			device = new LedDeviceLightpack(
				deviceConfig.get("output", "").asString()
			);
		}
		else if (type == "multi-lightpack")
		{
			device = new LedDeviceMultiLightpack();
		}
		else if (type == "paintpack")
		{
			const int delay_ms        = deviceConfig["delayAfterConnect"].asInt();
			auto VendorIdString       = deviceConfig.get("VID", "0x0EBF").asString();
			auto ProductIdString      = deviceConfig.get("PID", "0x0025").asString();

			// Convert HEX values to integer
			auto VendorId = std::stoul(VendorIdString, nullptr, 16);
			auto ProductId = std::stoul(ProductIdString, nullptr, 16);

			device = new LedDevicePaintpack(VendorId, ProductId, delay_ms);
		}
		else if (type == "piblaster")
		{
			const std::string output     = deviceConfig.get("output",     "").asString();
			const std::string assignment = deviceConfig.get("assignment", "").asString();
			const Json::Value gpioMapping = deviceConfig.get("gpiomap", Json::nullValue);

			if (! assignment.empty())
			{
				throw std::runtime_error("Piblaster: The piblaster configuration syntax has changed in this version.");
			}
			if (gpioMapping.isNull())
			{
				throw std::runtime_error("Piblaster: no gpiomap defined.");
			}
			device = new LedDevicePiBlaster(output, gpioMapping);
		}
		else if (type == "sedu")
		{
			device = new LedDeviceSedu(
				deviceConfig["output"].asString(),
				deviceConfig["rate"].asInt()
			);
		}
		else if (type == "hyperion-usbasp-ws2801")
		{
			device = new LedDeviceHyperionUsbasp(LedDeviceHyperionUsbasp::CMD_WRITE_WS2801);
		}
		else if (type == "hyperion-usbasp-ws2812")
		{
			device = new LedDeviceHyperionUsbasp(LedDeviceHyperionUsbasp::CMD_WRITE_WS2812);
		}
		else if (type == "philipshue")
		{
			const std::string output = deviceConfig["output"].asString();
			const std::string username = deviceConfig.get("username", "newdeveloper").asString();
			const bool switchOffOnBlack = deviceConfig.get("switchOffOnBlack", true).asBool();
			const int transitiontime = deviceConfig.get("transitiontime", 1).asInt();
			std::vector<unsigned int> lightIds;
			for (Json::Value::ArrayIndex i = 0; i < deviceConfig["lightIds"].size(); i++) {
				lightIds.push_back(deviceConfig["lightIds"][i].asInt());
			}
			device = new LedDevicePhilipsHue(output, username, switchOffOnBlack, transitiontime, lightIds);
		}
		else if (type == "atmoorb")
		{
			const std::string output = deviceConfig["output"].asString();
			const bool useOrbSmoothing = deviceConfig.get("useOrbSmoothing", false).asBool();
			const int transitiontime = deviceConfig.get("transitiontime", 1).asInt();
			const int skipSmoothingDiff = deviceConfig.get("skipSmoothingDiff", 0).asInt();
			const int port = deviceConfig.get("port", 1).asInt();
			const int numLeds = deviceConfig.get("numLeds", 1).asInt();
			const std::string orbId = deviceConfig["orbIds"].asString();
			std::vector<unsigned int> orbIds;

			// If we find multiple Orb ids separate them and add to list
			const std::string separator (",");
			if (orbId.find(separator) != std::string::npos) {
			std::stringstream ss(orbId);
			std::vector<int> output;
			unsigned int i;
			while (ss >> i) {
				orbIds.push_back(i);
				if (ss.peek() == ',' || ss.peek() == ' ')
					ss.ignore();
			}
			}
			else
			{
			orbIds.push_back(atoi(orbId.c_str()));
			}

			device = new LedDeviceAtmoOrb(output, useOrbSmoothing, transitiontime, skipSmoothingDiff, port, numLeds, orbIds);
		}
		else if (type == "fadecandy")
		{
			device = new LedDeviceFadeCandy(deviceConfig);
		}
		else if (type == "udp")
		{
			device = new LedDeviceUdp(
				deviceConfig["output"].asString(),
				deviceConfig["protocol"].asInt(),
				deviceConfig["maxpacket"].asInt()
			);
		}
		else if (type == "udpraw")
		{
			device = new LedDeviceUdpRaw(
				deviceConfig["output"].asString(),
				deviceConfig.get("latchtime",500000).asInt()
			);
		}
		else if (type == "tpm2")
		{
			device = new LedDeviceTpm2(
				deviceConfig["output"].asString(),
				deviceConfig["rate"].asInt()
			);
		}
		else if (type == "atmo")
		{
			device = new LedDeviceAtmo(
				deviceConfig["output"].asString(),
				38400
			);
		}
	#ifdef ENABLE_WS2812BPWM
		else if (type == "ws2812b")
		{
			device = new LedDeviceWS2812b();
		}
	#endif
	#ifdef ENABLE_WS281XPWM
		else if (type == "ws281x")
		{
			device = new LedDeviceWS281x(
				deviceConfig.get("gpio", 18).asInt(),
				deviceConfig.get("leds", 256).asInt(),
				deviceConfig.get("freq", (Json::UInt)800000ul).asInt(),
				deviceConfig.get("dmanum", 5).asInt(),
				deviceConfig.get("pwmchannel", 0).asInt(),
				deviceConfig.get("invert", 0).asInt(),
				deviceConfig.get("rgbw", 0).asInt(),
				deviceConfig.get("white_algorithm","").asString()
			);
		}
	#endif
		else if (type == "file")
		{
			device = new LedDeviceFile( deviceConfig.get("output", "/dev/null").asString() );
		}
		else
		{
			throw std::runtime_error("unknown device");
		}
	}
	catch(std::exception e)
	{
		
		Error(log, "Dummy device used, because configured device '%s' throws error '%s'", type.c_str(), e.what());
		device = new LedDeviceFile( "/dev/null" );
	}

	device->open();
	
	return device;
}