449 lines
15 KiB
C++
449 lines
15 KiB
C++
/** @file
|
|
* @author Edouard DUPIN
|
|
* @copyright 2014, Edouard DUPIN, all right reserved
|
|
* @license APACHE v2.0 (see license file)
|
|
*/
|
|
|
|
#include <appl/debug.hpp>
|
|
#include <zeus/Service.hpp>
|
|
#include <zeus/File.hpp>
|
|
#include <etk/etk.hpp>
|
|
#include <zeus/zeus.hpp>
|
|
|
|
#include <mutex>
|
|
#include <ejson/ejson.hpp>
|
|
#include <etk/os/FSNode.hpp>
|
|
#include <sstream>
|
|
|
|
#include <etk/stdTools.hpp>
|
|
namespace appl {
|
|
class User {
|
|
private:
|
|
std::mutex m_mutex;
|
|
std::string m_userName;
|
|
std::string m_basePath;
|
|
ejson::Document m_database;
|
|
std::map<uint64_t,std::string> m_listFile;
|
|
uint64_t m_lastMaxId;
|
|
public:
|
|
User(const std::string& _userName) :
|
|
m_userName(_userName),
|
|
m_lastMaxId(1024) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
APPL_WARNING("new USER: " << m_userName << " [START]");
|
|
m_basePath = std::string("USERDATA:") + m_userName + "/";
|
|
APPL_WARNING("new USER: " << m_userName);
|
|
bool ret = m_database.load(m_basePath + "database.json");
|
|
if (ret == false) {
|
|
APPL_WARNING(" ==> LOAD error");
|
|
}
|
|
// Load all files (image and video ...)
|
|
etk::FSNode node(m_basePath);
|
|
std::vector<etk::FSNode*> tmpList = node.folderGetSubList(false, false, true, false);
|
|
APPL_WARNING("Find " << tmpList.size() << " files");
|
|
for (auto &it : tmpList) {
|
|
if (it == nullptr) {
|
|
continue;
|
|
}
|
|
if ( etk::end_with(it->getNameFile(), ".svg", false) == true
|
|
|| etk::end_with(it->getNameFile(), ".bmp", false) == true
|
|
|| etk::end_with(it->getNameFile(), ".png", false) == true
|
|
|| etk::end_with(it->getNameFile(), ".jpg", false) == true
|
|
|| etk::end_with(it->getNameFile(), ".tga", false) == true
|
|
|| etk::end_with(it->getNameFile(), ".mp4", false) == true
|
|
|| etk::end_with(it->getNameFile(), ".avi", false) == true
|
|
|| etk::end_with(it->getNameFile(), ".mov", false) == true
|
|
|| etk::end_with(it->getNameFile(), ".mkv", false) == true) {
|
|
// TODO : Do it better (proto ..)
|
|
std::string idString = it->getNameFile();
|
|
idString.resize(idString.size()-4);
|
|
uint64_t id = 0;
|
|
std::stringstream ss;
|
|
ss << std::hex << idString;
|
|
ss >> id;
|
|
if (id <= 1024) {
|
|
APPL_WARNING(" ==> REJECTED file " << it->getNameFile() << " with ID = " << id);
|
|
} else {
|
|
m_listFile.insert(std::make_pair(id, it->getNameFile()));
|
|
m_lastMaxId = std::max(m_lastMaxId,id);
|
|
APPL_WARNING(" ==> load file " << it->getNameFile() << " with ID = " << id);
|
|
}
|
|
} else {
|
|
APPL_WARNING(" ==> REJECT file " << it->getNameFile());
|
|
}
|
|
}
|
|
APPL_WARNING("new USER: " << m_userName << " [STOP]");
|
|
}
|
|
~User() {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
APPL_WARNING("delete USER [START]");
|
|
APPL_DEBUG("Store User Info:");
|
|
bool ret = m_database.storeSafe(m_basePath + "database.json");
|
|
if (ret == false) {
|
|
APPL_WARNING(" ==> Store error");
|
|
}
|
|
APPL_WARNING("delete USER [STOP]");
|
|
}
|
|
// Return the list of root albums
|
|
std::vector<std::string> getAlbums() {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
std::vector<std::string> out;
|
|
ejson::Array globalGroups = m_database["group-global"].toArray();
|
|
if (globalGroups.exist() == false) {
|
|
APPL_DEBUG("'group-global' ==> does not exist ==> No album");
|
|
return out;
|
|
}
|
|
ejson::Object groups = m_database["groups"].toObject();
|
|
if (groups.exist() == false) {
|
|
APPL_DEBUG("'group' ==> does not exist ==> No album");
|
|
return out;
|
|
}
|
|
APPL_DEBUG("for element in 'group-global'");
|
|
for (auto it: globalGroups) {
|
|
std::string tmpString = it.toString().get();
|
|
if (tmpString == "") {
|
|
continue;
|
|
}
|
|
APPL_DEBUG(" find emlement:" << tmpString);
|
|
out.push_back(tmpString);
|
|
}
|
|
return out;
|
|
/*
|
|
ejson::Object groups = m_database["groups"].toObject();
|
|
if (groups.exist() == false) {
|
|
return std::vector<std::string>();
|
|
}
|
|
groups
|
|
return getSubAlbums("");
|
|
*/
|
|
}
|
|
// Get the list of sub album
|
|
std::vector<std::string> getSubAlbums(const std::string& _album) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
std::vector<std::string> out;
|
|
ejson::Object groups = m_database["groups"].toObject();
|
|
if (groups.exist() == false) {
|
|
return out;
|
|
}
|
|
// find parrentAlbum ==> to get sub group
|
|
/*
|
|
for (size_t iii=0; iii<groups.size(); ++iii) {
|
|
//ejson::Object group = groups[iii].toObject()["sub"];
|
|
if (groups.getKey(iii) != _parrentAlbum) {
|
|
continue;
|
|
}
|
|
}
|
|
*/
|
|
ejson::Object group = groups[_album].toObject();
|
|
if (group.exist() == false) {
|
|
return out;
|
|
}
|
|
ejson::Array groupSubs = group["sub"].toArray();
|
|
for (auto it: groupSubs) {
|
|
std::string tmpString = it.toString().get();
|
|
if (tmpString == "") {
|
|
continue;
|
|
}
|
|
out.push_back(tmpString);
|
|
}
|
|
// TODO: Check right
|
|
return out;
|
|
}
|
|
uint32_t getAlbumCount(const std::string& _album) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
ejson::Object groups = m_database["groups"].toObject();
|
|
if (groups.exist() == false) {
|
|
// TODO : Throw an error ...
|
|
return 0;
|
|
}
|
|
ejson::Object group = groups[_album].toObject();
|
|
if (group.exist() == false) {
|
|
// TODO : Throw an error ...
|
|
return 0;
|
|
}
|
|
ejson::Array groupSubs = group["files"].toArray();
|
|
// TODO: Check right
|
|
return groupSubs.size();
|
|
/*
|
|
for (auto it: groupSubs) {
|
|
uint64_t id = it.toNumber().getU64();
|
|
if (id == 0) {
|
|
continue;
|
|
}
|
|
out.push_back(id);
|
|
}
|
|
*/
|
|
}
|
|
// Return the list of the album files
|
|
std::vector<std::string> getAlbumListVideo(const std::string& _album) {//, uint32_t _startId, uint32_t _stopId) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
std::vector<std::string> out;
|
|
ejson::Object groups = m_database["groups"].toObject();
|
|
if (groups.exist() == false) {
|
|
// TODO : Throw an error ...
|
|
return out;
|
|
}
|
|
ejson::Object group = groups[_album].toObject();
|
|
if (group.exist() == false) {
|
|
// TODO : Throw an error ...
|
|
return out;
|
|
}
|
|
ejson::Array groupSubs = group["files"].toArray();
|
|
|
|
for (auto it: groupSubs) {
|
|
uint64_t id = it.toNumber().getU64();
|
|
/*
|
|
auto itImage = m_listFile.find(id);
|
|
if (itImage == m_listFile.end()) {
|
|
|
|
}*/
|
|
if (id == 0) {
|
|
continue;
|
|
}
|
|
out.push_back(etk::to_string(id));
|
|
}
|
|
return out;
|
|
}
|
|
// Return a File Data (might be a Video .tiff/.png/.jpg)
|
|
zeus::FileServer getAlbumVideo(const std::string& _VideoName) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
// TODO : Check right ...
|
|
uint64_t id = etk::string_to_uint64_t(_VideoName);
|
|
APPL_WARNING("try to get file : " << _VideoName << " with id=" << id);
|
|
{
|
|
auto it = m_listFile.find(id);
|
|
if (it != m_listFile.end()) {
|
|
return zeus::FileServer(m_basePath + it->second);
|
|
}
|
|
}
|
|
for (auto &it : m_listFile) {
|
|
APPL_WARNING("compare: " << it.first << " with " << id << " " << it.second);
|
|
if (it.first == id) {
|
|
return zeus::FileServer(m_basePath + it.second);
|
|
}
|
|
}
|
|
APPL_ERROR(" ==> Not find ...");
|
|
return zeus::FileServer();
|
|
}
|
|
std::string addFile(const zeus::File& _dataFile) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
// TODO : Check right ...
|
|
APPL_ERROR(" ==> Receive FILE " << _dataFile.getMineType() << " size=" << _dataFile.getData().size());
|
|
uint64_t id = createFileID();
|
|
std::stringstream val;
|
|
val << std::hex << std::setw(16) << std::setfill('0') << id;
|
|
std::string filename = val.str();
|
|
filename += ".";
|
|
filename += zeus::getExtention(_dataFile.getMineType());
|
|
_dataFile.storeIn(m_basePath + filename);
|
|
m_listFile.insert(std::make_pair(id, filename));
|
|
return etk::to_string(id);//zeus::FileServer();
|
|
}
|
|
bool removeFile(const std::string& _file) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
// TODO : Check right ...
|
|
return false;
|
|
}
|
|
|
|
std::string createAlbum(const std::string& _name) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
// TODO : Check right ...
|
|
return "";
|
|
}
|
|
bool removeAlbum(const std::string& _name) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
// TODO : Check right ...
|
|
return false;
|
|
}
|
|
bool setAlbumDescription(const std::string& _name, const std::string& _desc) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
// TODO : Check right ...
|
|
return false;
|
|
}
|
|
std::string getAlbumDescription(const std::string& _name) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
// TODO : Check right ...
|
|
return "";
|
|
}
|
|
|
|
|
|
bool addInAlbum(const std::string& _nameAlbum, const std::string& _nameElement) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
// TODO : Check right ...
|
|
return false;
|
|
}
|
|
bool removeFromAlbum(const std::string& _nameAlbum, const std::string& _nameElement) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
// TODO : Check right ...
|
|
return false;
|
|
}
|
|
/*
|
|
// Return a global UTC time
|
|
zeus::Time getAlbumVideoTime(const std::string& _VideoName) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
// TODO : Check right ...
|
|
return zeus::Time();
|
|
}
|
|
// Return a Geolocalization information (latitude, longitude)
|
|
zeus::Geo getAlbumVideoGeoLocalization(const std::string& _VideoName) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
// TODO : Check right ...
|
|
return zeus::Geo();
|
|
}
|
|
*/
|
|
private:
|
|
uint64_t createFileID() {
|
|
m_lastMaxId++;
|
|
return m_lastMaxId;
|
|
}
|
|
};
|
|
|
|
class UserManager {
|
|
private:
|
|
std::mutex m_mutex;
|
|
std::map<std::string, ememory::SharedPtr<appl::User>> m_listLoaded;
|
|
public:
|
|
UserManager() {
|
|
|
|
}
|
|
ememory::SharedPtr<appl::User> getUser(const std::string& _userName) {
|
|
std::unique_lock<std::mutex> lock(m_mutex);
|
|
auto it = m_listLoaded.find(_userName);
|
|
if (it != m_listLoaded.end()) {
|
|
// User already loaded:
|
|
return it->second;
|
|
}
|
|
// load New User:
|
|
ememory::SharedPtr<appl::User> tmp(new appl::User(_userName));
|
|
m_listLoaded.insert(std::make_pair(_userName, tmp));
|
|
return tmp;
|
|
}
|
|
};
|
|
class VideoService {
|
|
private:
|
|
ememory::SharedPtr<appl::User> m_user;
|
|
private:
|
|
ememory::SharedPtr<zeus::ClientProperty> m_client;
|
|
public:
|
|
VideoService() {
|
|
APPL_WARNING("New VideoService ...");
|
|
}
|
|
VideoService(ememory::SharedPtr<appl::User> _user, ememory::SharedPtr<zeus::ClientProperty> _client) :
|
|
m_user(_user),
|
|
m_client(_client) {
|
|
APPL_WARNING("New VideoService ... for user: ");
|
|
}
|
|
~VideoService() {
|
|
APPL_WARNING("delete VideoService ...");
|
|
}
|
|
public:
|
|
std::vector<std::string> getAlbums() {
|
|
return m_user->getAlbums();
|
|
}
|
|
// Get the list of sub album
|
|
std::vector<std::string> getSubAlbums(std::string _parrentAlbum) {
|
|
return m_user->getSubAlbums(_parrentAlbum);
|
|
}
|
|
uint32_t getAlbumCount(std::string _album) {
|
|
return m_user->getAlbumCount(_album);
|
|
}
|
|
// Return the list of the album files
|
|
std::vector<std::string> getAlbumListVideo(std::string _album) {
|
|
return m_user->getAlbumListVideo(_album);
|
|
}
|
|
// Return a File Data (might be a Video .tiff/.png/.jpg)
|
|
zeus::FileServer getAlbumVideo(std::string _VideoName) {
|
|
return m_user->getAlbumVideo(_VideoName);
|
|
}
|
|
std::string addFile(zeus::File _dataFile) {
|
|
return m_user->addFile(_dataFile);
|
|
}
|
|
/*
|
|
// Return a global UTC time
|
|
zeus::Time getAlbumVideoTime(std::string _VideoName) {
|
|
return m_user->getAlbumVideoTime(_VideoName);
|
|
}
|
|
// Return a Geolocalization information (latitude, longitude)
|
|
zeus::Geo getAlbumVideoGeoLocalization(std::string _VideoName) {
|
|
return m_user->getAlbumVideoGeoLocalization(_VideoName);
|
|
}
|
|
*/
|
|
};
|
|
}
|
|
|
|
|
|
int main(int _argc, const char *_argv[]) {
|
|
etk::init(_argc, _argv);
|
|
zeus::init(_argc, _argv);
|
|
std::string ip;
|
|
uint16_t port = 0;
|
|
for (int32_t iii=0; iii<_argc ; ++iii) {
|
|
std::string data = _argv[iii];
|
|
if (etk::start_with(data, "--ip=") == true) {
|
|
ip = std::string(&data[5]);
|
|
} else if (etk::start_with(data, "--port=") == true) {
|
|
port = etk::string_to_uint16_t(std::string(&data[7]));
|
|
} else if ( data == "-h"
|
|
|| data == "--help") {
|
|
APPL_PRINT(etk::getApplicationName() << " - help : ");
|
|
APPL_PRINT(" " << _argv[0] << " [options]");
|
|
APPL_PRINT(" --ip=XXX Server connection IP (default: 1.7.0.0.1)");
|
|
APPL_PRINT(" --port=XXX Server connection PORT (default: 1983)");
|
|
return -1;
|
|
}
|
|
}
|
|
while (true) {
|
|
APPL_INFO("===========================================================");
|
|
APPL_INFO("== ZEUS instanciate service: " << SERVICE_NAME << " [START]");
|
|
APPL_INFO("===========================================================");
|
|
ememory::SharedPtr<appl::UserManager> userMng = ememory::makeShared<appl::UserManager>();
|
|
zeus::ServiceType<appl::VideoService, appl::UserManager> serviceInterface(userMng);
|
|
if (ip != "") {
|
|
serviceInterface.propertyIp.set(ip);
|
|
}
|
|
if (port != 0) {
|
|
serviceInterface.propertyPort.set(port);
|
|
}
|
|
serviceInterface.propertyNameService.set(SERVICE_NAME);
|
|
serviceInterface.setDescription("Video Private Interface");
|
|
serviceInterface.setVersion("0.1.0");
|
|
serviceInterface.setType("Video", 1);
|
|
serviceInterface.addAuthor("Heero Yui", "yui.heero@gmail.com");
|
|
|
|
serviceInterface.advertise("getAlbums", &appl::VideoService::getAlbums);
|
|
serviceInterface.advertise("getSubAlbums", &appl::VideoService::getSubAlbums);
|
|
serviceInterface.advertise("getAlbumCount", &appl::VideoService::getAlbumCount);
|
|
serviceInterface.advertise("getAlbumListVideo", &appl::VideoService::getAlbumListVideo);
|
|
serviceInterface.advertise("getAlbumVideo", &appl::VideoService::getAlbumVideo);
|
|
serviceInterface.advertise("addFile", &appl::VideoService::addFile);
|
|
APPL_INFO("===========================================================");
|
|
APPL_INFO("== ZEUS service: " << *serviceInterface.propertyNameService << " [service instanciate]");
|
|
APPL_INFO("===========================================================");
|
|
serviceInterface.connect();
|
|
if (serviceInterface.GateWayAlive() == false) {
|
|
APPL_INFO("===========================================================");
|
|
APPL_INFO("== ZEUS service: " << *serviceInterface.propertyNameService << " [STOP] Can not connect to the GateWay");
|
|
APPL_INFO("===========================================================");
|
|
APPL_INFO("wait 5 second ...");
|
|
std::this_thread::sleep_for(std::chrono::seconds(5));
|
|
continue;
|
|
}
|
|
int32_t iii=0;
|
|
while (serviceInterface.GateWayAlive() == true) {
|
|
std::this_thread::sleep_for(std::chrono::seconds(1));
|
|
serviceInterface.pingIsAlive();
|
|
/*
|
|
serviceInterface.store();
|
|
serviceInterface.clean();
|
|
*/
|
|
APPL_INFO("service in waiting ... " << iii << "/inf");
|
|
iii++;
|
|
}
|
|
serviceInterface.disconnect();
|
|
APPL_INFO("===========================================================");
|
|
APPL_INFO("== ZEUS service: " << *serviceInterface.propertyNameService << " [STOP] GateWay Stop");
|
|
APPL_INFO("===========================================================");
|
|
}
|
|
return 0;
|
|
}
|