434 lines
13 KiB
C++
434 lines
13 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/Object.hpp>
|
|
#include <zeus/File.hpp>
|
|
#include <etk/etk.hpp>
|
|
#include <zeus/zeus.hpp>
|
|
#include <echrono/Time.hpp>
|
|
|
|
#include <mutex>
|
|
#include <ejson/ejson.hpp>
|
|
#include <etk/os/FSNode.hpp>
|
|
#include <sstream>
|
|
|
|
#include <etk/stdTools.hpp>
|
|
|
|
#include <zeus/service/Picture.hpp>
|
|
#include <zeus/service/registerPicture.hpp>
|
|
#include <zeus/ProxyClientProperty.hpp>
|
|
#include <zeus/File.hpp>
|
|
#include <zeus/ProxyFile.hpp>
|
|
|
|
static std::mutex g_mutex;
|
|
static std::string g_basePath;
|
|
static std::string g_baseDBName = std::string(SERVICE_NAME) + "-database.json";
|
|
static ejson::Document g_database;
|
|
class FileProperty {
|
|
public:
|
|
std::string m_fileName; // Sha 512
|
|
std::string m_name;
|
|
std::string m_mineType;
|
|
echrono::Time m_creationData;
|
|
};
|
|
|
|
static std::vector<FileProperty> m_listFile;
|
|
static uint64_t m_lastMaxId = 0;
|
|
static bool g_needToStore = false;
|
|
|
|
static uint64_t createFileID() {
|
|
m_lastMaxId++;
|
|
return m_lastMaxId;
|
|
}
|
|
|
|
namespace appl {
|
|
class PictureService : public zeus::service::Picture {
|
|
private:
|
|
//ememory::SharedPtr<zeus::ClientProperty>& m_client;
|
|
zeus::ProxyClientProperty m_client;
|
|
std::string m_userName;
|
|
public:
|
|
/*
|
|
PictureService(ememory::SharedPtr<zeus::ClientProperty>& _client, const std::string& _userName) :
|
|
m_client(_client),
|
|
m_userName(_userName) {
|
|
APPL_WARNING("New PictureService ... for user: ");
|
|
}
|
|
*/
|
|
PictureService(uint16_t _clientId) {
|
|
APPL_VERBOSE("New PictureService ... for user: " << _clientId);
|
|
}
|
|
~PictureService() {
|
|
APPL_VERBOSE("delete PictureService ...");
|
|
}
|
|
public:
|
|
std::vector<std::string> getAlbums() {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
std::vector<std::string> out;
|
|
ejson::Array globalGroups = g_database["group-global"].toArray();
|
|
if (globalGroups.exist() == false) {
|
|
APPL_DEBUG("'group-global' ==> does not exist ==> No album");
|
|
return out;
|
|
}
|
|
ejson::Object groups = g_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 = g_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(std::string _parrentAlbum) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
std::vector<std::string> out;
|
|
ejson::Object groups = g_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[_parrentAlbum].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(std::string _album) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
ejson::Object groups = g_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();
|
|
}
|
|
// Return the list of the album files
|
|
std::vector<std::string> getAlbumListPicture(std::string _album) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
std::vector<std::string> out;
|
|
ejson::Object groups = g_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 picture .tiff/.png/.jpg)
|
|
ememory::SharedPtr<zeus::File> getAlbumPicture(std::string _mediaName) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
// TODO : Check right ...
|
|
//Check if the file exist:
|
|
bool find = false;
|
|
FileProperty property;
|
|
for (auto &it : m_listFile) {
|
|
if (it.m_fileName == _mediaName) {
|
|
find = true;
|
|
property = it;
|
|
break;
|
|
}
|
|
}
|
|
if (find == false) {
|
|
throw std::invalid_argument("Wrong file name ...");
|
|
}
|
|
return zeus::File::create(g_basePath + property.m_fileName + "." + zeus::getExtention(property.m_mineType), "", property.m_mineType);
|
|
|
|
|
|
|
|
|
|
//uint64_t id = etk::string_to_uint64_t(_pictureName);
|
|
//APPL_WARNING("try to get file : " << _pictureName << " with id=" << id);
|
|
{
|
|
/*
|
|
auto it = m_listFile.find(id);
|
|
if (it != m_listFile.end()) {
|
|
return zeus::File::create(g_basePath + it->second);
|
|
}
|
|
*/
|
|
}
|
|
/*
|
|
for (auto &it : m_listFile) {
|
|
APPL_WARNING("compare: " << it.first << " with " << id << " " << it.second);
|
|
if (it.first == id) {
|
|
return zeus::File::create(g_basePath + it.second);
|
|
}
|
|
}
|
|
*/
|
|
APPL_ERROR(" ==> Not find ...");
|
|
return nullptr;
|
|
}
|
|
std::string addFile(zeus::ProxyFile _dataFile) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
// TODO : Check right ...
|
|
uint64_t id = createFileID();
|
|
|
|
auto futType = _dataFile.getMineType();
|
|
auto futName = _dataFile.getName();
|
|
std::string tmpFileName = g_basePath + "tmpImport_" + etk::to_string(id);
|
|
std::string sha512String = zeus::storeInFile(_dataFile, tmpFileName);
|
|
futType.wait();
|
|
futName.wait();
|
|
// TODO : Get internal data of the file and remove all the meta-data ==> proper files ...
|
|
for (auto &it : m_listFile) {
|
|
if (it.m_fileName == sha512String) {
|
|
APPL_INFO("File already registered at " << it.m_creationData);
|
|
// TODO : Check if data is identical ...
|
|
// remove temporary file
|
|
etk::FSNodeRemove(tmpFileName);
|
|
return sha512String;
|
|
}
|
|
}
|
|
// move the file at the good position:
|
|
APPL_ERROR("move temporay file in : " << g_basePath << sha512String);
|
|
etk::FSNodeMove(tmpFileName, g_basePath + sha512String + "." + zeus::getExtention(futType.get()));
|
|
FileProperty property;
|
|
property.m_fileName = sha512String;
|
|
property.m_name = futName.get();
|
|
property.m_mineType = futType.get();
|
|
property.m_creationData = echrono::Time::now();
|
|
APPL_ERROR("Current Time : " << echrono::Time::now());
|
|
m_listFile.push_back(property);
|
|
g_needToStore = true;
|
|
APPL_ERROR(" filename : " << sha512String);
|
|
return sha512String;
|
|
}
|
|
/*
|
|
// Return a global UTC time
|
|
zeus::Time getAlbumPictureTime(std::string _pictureName) {
|
|
return m_user->getAlbumPictureTime(_pictureName);
|
|
}
|
|
// Return a Geolocalization information (latitude, longitude)
|
|
zeus::Geo getAlbumPictureGeoLocalization(std::string _pictureName) {
|
|
return m_user->getAlbumPictureGeoLocalization(_pictureName);
|
|
}
|
|
*/
|
|
bool removeFile(std::string _file) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
// TODO : Check right ...
|
|
return false;
|
|
}
|
|
|
|
std::string createAlbum(std::string _name) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
// TODO : Check right ...
|
|
return "";
|
|
}
|
|
bool removeAlbum(std::string _name) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
// TODO : Check right ...
|
|
return false;
|
|
}
|
|
bool setAlbumDescription(std::string _name, std::string _desc) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
// TODO : Check right ...
|
|
return false;
|
|
}
|
|
std::string getAlbumDescription(std::string _name) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
// TODO : Check right ...
|
|
return "";
|
|
}
|
|
bool addInAlbum(std::string _nameAlbum, std::string _nameElement) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
// TODO : Check right ...
|
|
return false;
|
|
}
|
|
bool removeFromAlbum(std::string _nameAlbum, std::string _nameElement) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
// TODO : Check right ...
|
|
return false;
|
|
}
|
|
/*
|
|
// Return a global UTC time
|
|
zeus::Time getAlbumPictureTime(std::string _pictureName) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
// TODO : Check right ...
|
|
return zeus::Time();
|
|
}
|
|
// Return a Geolocalization information (latitude, longitude)
|
|
zeus::Geo getAlbumPictureGeoLocalization(std::string _pictureName) {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
// TODO : Check right ...
|
|
return zeus::Geo();
|
|
}
|
|
*/
|
|
};
|
|
}
|
|
|
|
static void store_db() {
|
|
APPL_ERROR("Store database [START]");
|
|
ejson::Document database;
|
|
ejson::Array listFilesArray;
|
|
database.add("list-files", listFilesArray);
|
|
for (auto &it : m_listFile) {
|
|
ejson::Object fileElement;
|
|
fileElement.add("file-name", ejson::String(it.m_fileName));
|
|
fileElement.add("name", ejson::String(it.m_name));
|
|
fileElement.add("mine-type", ejson::String(it.m_mineType));
|
|
fileElement.add("add-date", ejson::Number(it.m_creationData.count()));
|
|
listFilesArray.add(fileElement);
|
|
}
|
|
bool retGenerate = database.storeSafe(g_basePath + g_baseDBName);
|
|
APPL_ERROR("Store database [STOP] : " << (g_basePath + g_baseDBName) << " ret = " << retGenerate);
|
|
g_needToStore = false;
|
|
}
|
|
|
|
static void load_db() {
|
|
ejson::Document database;
|
|
bool ret = database.load(g_basePath + g_baseDBName);
|
|
if (ret == false) {
|
|
APPL_WARNING(" ==> LOAD error");
|
|
}
|
|
ejson::Array listFilesArray = database["list-files"].toArray();
|
|
for (const auto itArray: listFilesArray) {
|
|
ejson::Object fileElement = itArray.toObject();
|
|
FileProperty property;
|
|
|
|
property.m_fileName = fileElement["file-name"].toString().get();
|
|
property.m_name = fileElement["name"].toString().get();
|
|
property.m_mineType = fileElement["mine-type"].toString().get();
|
|
property.m_creationData = echrono::Time(fileElement["add-date"].toNumber().getU64()*1000);
|
|
|
|
if (property.m_fileName == "") {
|
|
APPL_ERROR("Can not access on the file : ... No name ");
|
|
} else {
|
|
m_listFile.push_back(property);
|
|
}
|
|
}
|
|
g_needToStore = false;
|
|
}
|
|
|
|
ETK_EXPORT_API bool SERVICE_IO_init(int _argc, const char *_argv[], std::string _basePath) {
|
|
g_basePath = _basePath;
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
APPL_WARNING("Load USER: " << g_basePath);
|
|
load_db();
|
|
|
|
/*
|
|
|
|
|
|
// Load all files (image and video ...)
|
|
etk::FSNode node(g_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: [STOP]");
|
|
return true;
|
|
}
|
|
|
|
ETK_EXPORT_API bool SERVICE_IO_uninit() {
|
|
std::unique_lock<std::mutex> lock(g_mutex);
|
|
store_db();
|
|
APPL_WARNING("delete USER [STOP]");
|
|
return true;
|
|
}
|
|
|
|
ETK_EXPORT_API void SERVICE_IO_peridic_call() {
|
|
if (g_needToStore == false) {
|
|
return;
|
|
}
|
|
// try lock mutex:
|
|
if (g_mutex.try_lock() == false) {
|
|
return;
|
|
}
|
|
store_db();
|
|
g_mutex.unlock();
|
|
}
|
|
|
|
|
|
ZEUS_SERVICE_PICTURE_DECLARE(appl::PictureService);
|
|
|