Refactoring project layout

2025-11-08 14:43:21 +00:00 · 2021-10-19 19:45:50 -07:00
parent 816f0a6e7c
commit 2a0b45fde1
110 changed files with 1151 additions and 230 deletions
--- a/src/RESTAPI/RESTAPI_device_handler.cpp
+++ b/src/RESTAPI/RESTAPI_device_handler.cpp
@@ -0,0 +1,155 @@
+//
+//	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_device_handler.h"
+#include "CentralConfig.h"
+#include "ConfigurationCache.h"
+#include "ConfigurationValidator.h"
+#include "Poco/JSON/Parser.h"
+#include "StorageService.h"
+#include "framework/RESTAPI_errors.h"
+#include "framework/RESTAPI_protocol.h"
+#include "framework/Utils.h"
+
+namespace OpenWifi {
+	void RESTAPI_device_handler::DoGet() {
+		std::string SerialNumber = GetBinding(RESTAPI::Protocol::SERIALNUMBER, "");
+
+		if(SerialNumber.empty()) {
+			return BadRequest(RESTAPI::Errors::MissingSerialNumber);
+		}
+
+		GWObjects::Device Device;
+
+		if (Storage()->GetDevice(SerialNumber, Device)) {
+			Poco::JSON::Object Obj;
+			Device.to_json(Obj);
+			return ReturnObject(Obj);
+		}
+		NotFound();
+	}
+
+	void RESTAPI_device_handler::DoDelete() {
+		std::string SerialNumber = GetBinding(RESTAPI::Protocol::SERIALNUMBER, "");
+
+		if(SerialNumber.empty()) {
+			return BadRequest(RESTAPI::Errors::MissingSerialNumber);
+		}
+
+		if (Storage()->DeleteDevice(SerialNumber)) {
+			return OK();
+		}
+		NotFound();
+	}
+
+	void RESTAPI_device_handler::DoPost() {
+
+		std::string SerialNumber = GetBinding(RESTAPI::Protocol::SERIALNUMBER, "");
+		if(SerialNumber.empty()) {
+			return BadRequest(RESTAPI::Errors::MissingSerialNumber);
+		}
+
+		std::string Arg;
+		if(HasParameter("validateOnly",Arg) && Arg=="true") {
+			auto Body = ParseStream();
+			if(!Body->has("configuration")) {
+				return BadRequest("Must have 'configuration' element.");
+			}
+			auto Config=Body->get("configuration").toString();
+			Poco::JSON::Object  Answer;
+			auto Res = ValidateUCentralConfiguration(Config);
+			Answer.set("valid",Res);
+			return ReturnObject(Answer);
+		}
+
+		if (!Utils::ValidSerialNumber(SerialNumber)) {
+			Logger_.warning(Poco::format("CREATE-DEVICE(%s): Illegal serial number.", SerialNumber));
+			return BadRequest( RESTAPI::Errors::InvalidSerialNumber);
+		}
+
+		auto Obj = ParseStream();
+		GWObjects::Device Device;
+		if (!Device.from_json(Obj)) {
+			return BadRequest(RESTAPI::Errors::InvalidJSONDocument);
+		}
+
+		if(SerialNumber!=Device.SerialNumber) {
+			return BadRequest(RESTAPI::Errors::SerialNumberMismatch);
+		}
+
+		if(Device.Configuration.empty() || (!Device.Configuration.empty() && !ValidateUCentralConfiguration(Device.Configuration))) {
+			return BadRequest(RESTAPI::Errors::ConfigBlockInvalid);
+		}
+
+		for(auto &i:Device.Notes)
+			i.createdBy = UserInfo_.userinfo.email;
+
+		Config::Config NewConfig(Device.Configuration);
+		Device.UUID = std::time(nullptr);
+		NewConfig.SetUUID(Device.UUID);
+		Device.Configuration = NewConfig.get();
+
+		Poco::toLowerInPlace(Device.SerialNumber);
+
+		if (Storage()->CreateDevice(Device)) {
+			SetCurrentConfigurationID(SerialNumber, Device.UUID);
+			Poco::JSON::Object DevObj;
+			Device.to_json(DevObj);
+			return ReturnObject(DevObj);
+		}
+		InternalError(RESTAPI::Errors::RecordNotCreated);
+	}
+
+	void RESTAPI_device_handler::DoPut() {
+		std::string SerialNumber = GetBinding(RESTAPI::Protocol::SERIALNUMBER, "");
+
+		if(SerialNumber.empty()) {
+			return BadRequest(RESTAPI::Errors::MissingSerialNumber);
+		}
+
+		auto Obj = ParseStream();
+		GWObjects::Device NewDevice;
+		if (!NewDevice.from_json(Obj)) {
+			return BadRequest(RESTAPI::Errors::InvalidJSONDocument);
+		}
+
+		GWObjects::Device	Existing;
+		if(!Storage()->GetDevice(SerialNumber, Existing)) {
+			return NotFound();
+		}
+
+		if(!NewDevice.Configuration.empty()) {
+			if (!ValidateUCentralConfiguration(NewDevice.Configuration)) {
+				return BadRequest(RESTAPI::Errors::ConfigBlockInvalid);
+			}
+			Config::Config NewConfig(NewDevice.Configuration);
+			uint64_t NewConfigUUID = std::time(nullptr);
+			NewConfig.SetUUID(NewConfigUUID);
+			Existing.Configuration = NewConfig.get();
+			Existing.UUID = NewConfigUUID;
+		}
+
+		AssignIfPresent(Obj, "venue", Existing.Venue);
+		AssignIfPresent(Obj, "owner", Existing.Owner);
+		AssignIfPresent(Obj, "location", Existing.Location);
+
+		for(auto &i:NewDevice.Notes) {
+			i.createdBy = UserInfo_.userinfo.email;
+			Existing.Notes.push_back(i);
+		}
+
+		Existing.LastConfigurationChange = std::time(nullptr);
+		if (Storage()->UpdateDevice(Existing)) {
+			SetCurrentConfigurationID(SerialNumber, Existing.UUID);
+			Poco::JSON::Object DevObj;
+			NewDevice.to_json(DevObj);
+			return ReturnObject(DevObj);
+		}
+		InternalError(RESTAPI::Errors::RecordNotUpdated);
+	}
+}