//
// Created by stephane bourque on 2022-07-26.
//

#include "AP_WS_Connection.h"
#include "AP_WS_Server.h"
#include "StorageService.h"

#include "fmt/format.h"
#include "framework/KafkaManager.h"
#include "framework/utils.h"

namespace OpenWifi {

	void AP_WS_Connection::Process_healthcheck(Poco::JSON::Object::Ptr ParamsObj) {
		if (!State_.Connected) {
			poco_warning(Logger_,
						 fmt::format("INVALID-PROTOCOL({}): Device '{}' is not following protocol",
									 CId_, CN_));
			Errors_++;
			return;
		}
		if (ParamsObj->has(uCentralProtocol::UUID) &&
			ParamsObj->has(uCentralProtocol::SANITY) &&
			ParamsObj->has(uCentralProtocol::DATA)) {

			uint64_t UUID = ParamsObj->get(uCentralProtocol::UUID);
			auto Sanity = ParamsObj->get(uCentralProtocol::SANITY);
			State_.sanity = Sanity;
			auto CheckData = ParamsObj->get(uCentralProtocol::DATA).toString();
			if (CheckData.empty())
				CheckData = uCentralProtocol::EMPTY_JSON_DOC;

			std::string request_uuid;
			if (ParamsObj->has(uCentralProtocol::REQUEST_UUID))
				request_uuid = ParamsObj->get(uCentralProtocol::REQUEST_UUID).toString();

			if (request_uuid.empty()) {
				poco_trace(Logger_, fmt::format("HEALTHCHECK({}): UUID={} Updating.", CId_, UUID));
			} else {
				poco_trace(Logger_, fmt::format("HEALTHCHECK({}): UUID={} Updating for CMD={}.",
												CId_, UUID, request_uuid));
			}

			GWObjects::HealthCheck Check;

			Check.SerialNumber = SerialNumber_;
			Check.Recorded = Utils::Now();
			Check.UUID = UUID;
			Check.Data = CheckData;
			Check.Sanity = Sanity;

			StorageService()->AddHealthCheckData(*DbSession_, Check);

			if (!request_uuid.empty()) {
				StorageService()->SetCommandResult(request_uuid, CheckData);
			}

			SetLastHealthCheck(Check);
			if (KafkaManager()->Enabled() && !AP_WS_Server()->KafkaDisableHealthChecks()) {
				KafkaManager()->PostMessage(KafkaTopics::HEALTHCHECK, SerialNumber_, *ParamsObj);
			}
		} else {
			poco_warning(Logger_, fmt::format("HEALTHCHECK({}): Missing parameter", CId_));
			return;
		}
	}

} // namespace OpenWifi