mirror of
https://github.com/Telecominfraproject/wlan-cloud-ucentralsec.git
synced 2025-10-31 02:37:56 +00:00
90 lines
3.1 KiB
C++
90 lines
3.1 KiB
C++
//
|
|
// Created by stephane bourque on 2021-07-15.
|
|
//
|
|
|
|
#include <fstream>
|
|
#include <iostream>
|
|
|
|
#include "Poco/CountingStream.h"
|
|
#include "Poco/Net/HTMLForm.h"
|
|
#include "RESTAPI_avatar_handler.h"
|
|
#include "StorageService.h"
|
|
#include "framework/MicroServiceFuncs.h"
|
|
|
|
namespace OpenWifi {
|
|
|
|
void AvatarPartHandler::handlePart(const Poco::Net::MessageHeader &Header,
|
|
std::istream &Stream) {
|
|
FileType_ = Header.get(RESTAPI::Protocol::CONTENTTYPE, RESTAPI::Protocol::UNSPECIFIED);
|
|
if (Header.has(RESTAPI::Protocol::CONTENTDISPOSITION)) {
|
|
std::string Disposition;
|
|
Poco::Net::NameValueCollection Parameters;
|
|
Poco::Net::MessageHeader::splitParameters(Header[RESTAPI::Protocol::CONTENTDISPOSITION],
|
|
Disposition, Parameters);
|
|
Name_ = Parameters.get(RESTAPI::Protocol::NAME, RESTAPI::Protocol::UNNAMED);
|
|
}
|
|
Poco::CountingInputStream InputStream(Stream);
|
|
Poco::StreamCopier::copyStream(InputStream, OutputStream_);
|
|
Length_ = OutputStream_.str().size();
|
|
};
|
|
|
|
void RESTAPI_avatar_handler::DoPost() {
|
|
std::string Id = UserInfo_.userinfo.id;
|
|
SecurityObjects::UserInfo UInfo;
|
|
|
|
std::stringstream SS;
|
|
AvatarPartHandler partHandler(Id, Logger_, SS);
|
|
Poco::Net::HTMLForm form(*Request, Request->stream(), partHandler);
|
|
Poco::JSON::Object Answer;
|
|
|
|
if (!partHandler.Name().empty() &&
|
|
partHandler.Length() < MicroServiceConfigGetInt("openwifi.avatar.maxsize", 2000000)) {
|
|
Answer.set(RESTAPI::Protocol::AVATARID, Id);
|
|
Answer.set(RESTAPI::Protocol::ERRORCODE, 0);
|
|
Logger_.information(fmt::format("Uploaded avatar: {} Type: {}", partHandler.Name(),
|
|
partHandler.ContentType()));
|
|
StorageService()->AvatarDB().SetAvatar(UserInfo_.userinfo.email, Id, SS.str(),
|
|
partHandler.ContentType(), partHandler.Name());
|
|
StorageService()->UserDB().SetAvatar(Id, "1");
|
|
Logger().information(fmt::format("Adding avatar for {}", UserInfo_.userinfo.email));
|
|
} else {
|
|
Answer.set(RESTAPI::Protocol::AVATARID, Id);
|
|
Answer.set(RESTAPI::Protocol::ERRORCODE, 13);
|
|
Answer.set(RESTAPI::Protocol::ERRORTEXT, "Avatar upload could not complete.");
|
|
}
|
|
ReturnObject(Answer);
|
|
}
|
|
|
|
void RESTAPI_avatar_handler::DoGet() {
|
|
std::string Id = GetBinding(RESTAPI::Protocol::ID, "");
|
|
if (Id.empty()) {
|
|
return NotFound();
|
|
}
|
|
|
|
std::string Type, Name, AvatarContent;
|
|
if (!StorageService()->AvatarDB().GetAvatar(UserInfo_.userinfo.email, Id, AvatarContent,
|
|
Type, Name)) {
|
|
return NotFound();
|
|
}
|
|
Logger().information(fmt::format("Retrieving avatar for {}, size:{}",
|
|
UserInfo_.userinfo.email, AvatarContent.size()));
|
|
return SendFileContent(AvatarContent, Type, Name);
|
|
}
|
|
|
|
void RESTAPI_avatar_handler::DoDelete() {
|
|
std::string Id = GetBinding(RESTAPI::Protocol::ID, "");
|
|
|
|
if (UserInfo_.userinfo.userRole != SecurityObjects::ROOT && Id != UserInfo_.userinfo.id) {
|
|
return UnAuthorized(RESTAPI::Errors::ACCESS_DENIED);
|
|
}
|
|
|
|
if (!StorageService()->AvatarDB().DeleteAvatar(UserInfo_.userinfo.email, Id)) {
|
|
return NotFound();
|
|
}
|
|
|
|
Logger().information(fmt::format("Deleted avatar for {}", UserInfo_.userinfo.email));
|
|
StorageService()->UserDB().SetAvatar(Id, "");
|
|
OK();
|
|
}
|
|
} // namespace OpenWifi
|