//
//	License type: BSD 3-Clause License
//	License copy: https://github.com/Telecominfraproject/wlan-cloud-ucentralgw/blob/master/LICENSE
//
//	Created by Stephane Bourque on 2021-03-04.
//	Arilia Wireless Inc.
//

#include "RESTAPI_blacklist.h"
#include "Poco/JSON/Parser.h"
#include "StorageService.h"
#include "framework/ow_constants.h"
#include "framework/utils.h"

namespace OpenWifi {
	void RESTAPI_blacklist::DoDelete() {
		auto SerialNumber = GetBinding(RESTAPI::Protocol::SERIALNUMBER, "");

		if (!Utils::NormalizeMac(SerialNumber)) {
			return BadRequest(RESTAPI::Errors::MissingSerialNumber);
		}

		poco_debug(Logger(), fmt::format("BLACKLIST-DELETE: {}", SerialNumber));

		GWObjects::BlackListedDevice D;
		if (!StorageService()->GetBlackListDevice(SerialNumber, D)) {
			return NotFound();
		}

		if (StorageService()->DeleteBlackListDevice(SerialNumber)) {
			return OK();
		}
		BadRequest(RESTAPI::Errors::CouldNotBeDeleted);
	}

	void RESTAPI_blacklist::DoGet() {
		auto SerialNumber = GetBinding(RESTAPI::Protocol::SERIALNUMBER, "");

		if (!Utils::NormalizeMac(SerialNumber)) {
			return BadRequest(RESTAPI::Errors::MissingSerialNumber);
		}

		poco_debug(Logger(), fmt::format("BLACKLIST-GET: {}", SerialNumber));
		GWObjects::BlackListedDevice D;
		if (!StorageService()->GetBlackListDevice(SerialNumber, D)) {
			return NotFound();
		}
		return Object(D);
	}

	void RESTAPI_blacklist::DoPost() {

		const auto &Obj = ParsedBody_;
		GWObjects::BlackListedDevice D;
		if (!D.from_json(Obj)) {
			return BadRequest(RESTAPI::Errors::InvalidJSONDocument);
		}

		if (D.serialNumber.empty() || !Utils::NormalizeMac(D.serialNumber)) {
			return BadRequest(RESTAPI::Errors::MissingSerialNumber);
		}

		poco_debug(Logger(), fmt::format("BLACKLIST-POST: {}", D.serialNumber));

		Poco::toLowerInPlace(D.serialNumber);
		if (StorageService()->IsBlackListed(Utils::MACToInt(D.serialNumber))) {
			return BadRequest(RESTAPI::Errors::SerialNumberExists);
		}

		D.author = UserInfo_.userinfo.email;
		D.created = Utils::Now();

		if (StorageService()->AddBlackListDevice(D)) {
			GWObjects::BlackListedDevice CreatedDevice;
			StorageService()->GetBlackListDevice(D.serialNumber, CreatedDevice);
			return Object(CreatedDevice);
		}
		return BadRequest(RESTAPI::Errors::MissingOrInvalidParameters);
	}

	void RESTAPI_blacklist::DoPut() {
		auto SerialNumber = Poco::toLower(GetBinding(RESTAPI::Protocol::SERIALNUMBER, ""));
		if (!Utils::NormalizeMac(SerialNumber)) {
			return BadRequest(RESTAPI::Errors::MissingSerialNumber);
		}

		const auto &Obj = ParsedBody_;
		GWObjects::BlackListedDevice Existing;
		if (!StorageService()->GetBlackListDevice(SerialNumber, Existing)) {
			return BadRequest(RESTAPI::Errors::InvalidJSONDocument);
		}

		poco_debug(Logger(), fmt::format("BLACKLIST-PUT: {}", SerialNumber));

		GWObjects::BlackListedDevice NewDevice;
		if (!NewDevice.from_json(Obj)) {
			return BadRequest(RESTAPI::Errors::InvalidJSONDocument);
		}

		Existing.reason = NewDevice.reason;
		Existing.author = UserInfo_.userinfo.email;

		if (StorageService()->UpdateBlackListDevice(SerialNumber, Existing)) {
			GWObjects::BlackListedDevice CreatedDevice;

			StorageService()->GetBlackListDevice(SerialNumber, CreatedDevice);
			return Object(CreatedDevice);
		}
		return BadRequest(RESTAPI::Errors::MissingOrInvalidParameters);
	}

} // namespace OpenWifi