sic/src/sicccPersister.H

173 lines
4.6 KiB
C++

#ifndef __SICCCPERSISTER_H_
#define __SICCCPERSISTER_H_
#include <jansson.h>
#include <dirent.h>
#include <sys/types.h>
#include <cstring>
#include <iostream>
#include <vector>
#include "Routerin.H"
#include "sicccobject.H"
static json_t * load_json(const std::string &s)
{
assert(!s.empty());
std::cout << "[DEBUG] joad_json: " << s << std::endl;
json_t *root = nullptr;
json_error_t error;
root = json_load_file(s.c_str(), JSON_DECODE_ANY, &error);
if (!root) {
/* std::cerr << "[read_json] " << error << std::endl; */
return nullptr;
}
return root;
}
static std::string get_string_from_json(const json_t *j)
{
assert(j);
assert(json_is_string(j));
const char *tmp = json_string_value(j);
assert(tmp);
return std::string(tmp);
}
static json_t *get_json_from_string(const std::string &s)
{
json_t* ret = json_string(s.c_str());
assert(ret);
return ret;
}
static unsigned int get_integer_from_json(const json_t *j)
{
assert(j);
assert(json_is_integer(j));
return json_integer_value(j);
}
static json_t *get_json_from_integer(const unsigned int value)
{
assert(value);
json_t* ret = json_integer(value);
assert(ret);
return ret;
}
static std::chrono::time_point<std::chrono::system_clock>
get_time_from_json(const json_t *j)
{
assert(j);
auto tmp = get_string_from_json(j);
time_t t = stol(tmp);
return std::chrono::system_clock::from_time_t(t);
}
static std::string time_to_string(const std::chrono::time_point<std::chrono::system_clock>& t)
{
time_t time = std::chrono::system_clock::to_time_t(t);
std::string ret = std::to_string(time);
return ret;
}
/** TODO
* implement bool operator to take advantages of the use of assert
*/
static json_t *get_json_from_sicccobject(const Sicccobject& so)
{
json_t *root = json_object();
json_object_set_new(root, "filename", get_json_from_string(so.get_filename()));
json_object_set_new(root, "filetype", get_json_from_string(so.get_filetype()));
json_object_set_new(root, "path", get_json_from_string(so.get_path()));
json_object_set_new(root, "size", get_json_from_integer(so.get_size()));
std::string tmp = time_to_string(so.get_uploadtime());
json_object_set_new(root, "upload_time", get_json_from_string(tmp));
assert(root);
return root;
}
static std::string get_json_string_representation(const json_t *j)
{
assert(j);
assert(json_is_object(j));
const char *tmp = json_dumps(j, JSON_INDENT(4));
assert(tmp);
std::string ret(tmp);
free((void*) tmp);
return ret;
}
class SicccPersister
{
public:
SicccPersister(const Routerin & merouter = *Routerin::get_instance()) :
merouter(merouter)
{
root = json_object();
perists_path = "./";
json_file_name = "foo.json"; // not the best default
};
SicccPersister(const Sicccobject& so,const Routerin &merouter = * Routerin::get_instance()):
merouter(merouter)
{
root = get_json_from_sicccobject(so);
perists_path =
json_file_name = so.get_filename();
}
SicccPersister(json_t *j, const Routerin & merouter =* Routerin::get_instance())
{
root = json_object();
json_object_update(root, j);
}
~SicccPersister()
{
if (root)
json_decref(root);
}
int write_sicccfile(const Sicccobject&);
int write_sicccfile();
std::vector<Sicccobject> read_sicccdir();
Sicccobject read_json(const std::string&);
Sicccobject read_json() { return read_json(json_file_name); }
void set_persist_path(std::string&);
void set_json_filename(std::string&);
friend std::ostream& operator << (std::ostream& out, const SicccPersister &p)
{
out << get_json_string_representation(p.root);
return out;
}
friend std::ostream& operator << (std::ostream& out, const json_t *j)
{
assert(j);
out << get_json_string_representation(j);
return out;
}
private:
const Routerin & merouter;
/* std::string perists_path;
std::string json_file_name; */
json_t *root;
std::string
inline beautify_filename()
{
std::string s;
if (json_file_name.empty()) return s;
size_t pos = json_file_name.find_last_of(".");
s = json_file_name.substr(0, pos + 1);
s.append("json");
return s;
}
};
#endif /* __SICCCPERSISTER_H_ */