pandemonium_engine/modules/users/users/user.cpp

269 lines
8.2 KiB
C++

#include "user.h"
#include "core/class_db.h"
#include "user_module.h"
int User::get_user_id() const {
return _user_id;
}
void User::set_user_id(const int val) {
_user_id = val;
}
String User::get_user_name() const {
return _user_name;
}
void User::set_user_name(const String &val) {
_user_name = val;
}
String User::get_email() const {
return _email;
}
void User::set_email(const String &val) {
_email = val;
}
int User::get_rank() const {
return _rank;
}
void User::set_rank(const int val) {
_rank = val;
}
String User::get_pre_salt() const {
return _pre_salt;
}
void User::set_pre_salt(const String &val) {
_pre_salt = val;
}
String User::get_post_salt() const {
return _post_salt;
}
void User::set_post_salt(const String &val) {
_post_salt = val;
}
String User::get_password_hash() const {
return _password_hash;
}
void User::set_password_hash(const String &val) {
_password_hash = val;
}
bool User::get_banned() const {
return _banned;
}
void User::set_banned(const bool val) {
_banned = val;
}
String User::get_password_reset_token() const {
return _password_reset_token;
}
void User::set_password_reset_token(const String &val) {
_password_reset_token = val;
}
bool User::get_locked() const {
return _locked;
}
void User::set_locked(const bool val) {
_locked = val;
}
void User::add_module(const Ref<UserModule> &module) {
_modules.push_back(module);
if (module.is_valid()) {
Ref<UserModule> m = module;
m->set_module_index(_modules.size() - 1);
m->set_user(this);
}
}
Ref<UserModule> User::get_module(const int index) {
ERR_FAIL_INDEX_V(index, _modules.size(), Ref<UserModule>());
return _modules[index];
}
Ref<UserModule> User::get_module_named(const String &name) {
for (int i = 0; i < _modules.size(); ++i) {
Ref<UserModule> m = _modules[i];
if (m.is_valid() && m->get_module_name() == name) {
return m;
}
}
return Ref<UserModule>();
}
void User::remove_module(const int index) {
ERR_FAIL_INDEX(index, _modules.size());
_modules.remove(index);
}
int User::get_module_count() {
return _modules.size();
}
Vector<Variant> User::get_modules() {
Vector<Variant> r;
for (int i = 0; i < _modules.size(); i++) {
r.push_back(_modules[i].get_ref_ptr());
}
return r;
}
void User::set_modules(const Vector<Variant> &modules) {
_modules.clear();
for (int i = 0; i < modules.size(); i++) {
Ref<UserModule> um = Ref<UserModule>(modules.get(i));
_modules.push_back(um);
if (um.is_valid()) {
um->set_module_index(_modules.size() - 1);
um->set_user(this);
}
}
}
bool User::check_password(const String &p_password) {
return call("_check_password", p_password);
}
void User::create_password(const String &p_password) {
call("_create_password", p_password);
}
String User::hash_password(const String &p_password) {
return call("_hash_password", p_password);
}
bool User::_check_password(const String &p_password) {
return hash_password(p_password) == get_password_hash();
}
void User::_create_password(const String &p_password) {
// todo improve a bit
set_pre_salt(hash_password(get_user_name() + get_email()));
set_post_salt(hash_password(get_email() + get_user_name()));
set_password_hash(hash_password(p_password));
}
String User::_hash_password(const String &p_password) {
String p = get_pre_salt() + p_password + get_post_salt();
return p.sha256_text();
}
void User::save() {
emit_changed();
}
void User::read_lock() {
_rw_lock.read_lock();
}
void User::read_unlock() {
_rw_lock.read_unlock();
}
void User::write_lock() {
_rw_lock.write_lock();
}
void User::write_unlock() {
_rw_lock.write_unlock();
}
User::User() {
_user_id = -1;
_rank = 0;
_banned = false;
_locked = false;
}
User::~User() {
for (int i = 0; i < _modules.size(); ++i) {
Ref<UserModule> um = _modules[i];
if (um.is_valid()) {
um->set_user(nullptr);
}
}
}
void User::_bind_methods() {
ClassDB::bind_method(D_METHOD("get_user_id"), &User::get_user_id);
ClassDB::bind_method(D_METHOD("set_user_id", "val"), &User::set_user_id);
ADD_PROPERTY(PropertyInfo(Variant::INT, "user_id"), "set_user_id", "get_user_id");
ClassDB::bind_method(D_METHOD("get_user_name"), &User::get_user_name);
ClassDB::bind_method(D_METHOD("set_user_name", "val"), &User::set_user_name);
ADD_PROPERTY(PropertyInfo(Variant::STRING, "user_name"), "set_user_name", "get_user_name");
ClassDB::bind_method(D_METHOD("get_email"), &User::get_email);
ClassDB::bind_method(D_METHOD("set_email", "val"), &User::set_email);
ADD_PROPERTY(PropertyInfo(Variant::STRING, "email"), "set_email", "get_email");
ClassDB::bind_method(D_METHOD("get_rank"), &User::get_rank);
ClassDB::bind_method(D_METHOD("set_rank", "val"), &User::set_rank);
ADD_PROPERTY(PropertyInfo(Variant::INT, "rank"), "set_rank", "get_rank");
ClassDB::bind_method(D_METHOD("get_pre_salt"), &User::get_pre_salt);
ClassDB::bind_method(D_METHOD("set_pre_salt", "val"), &User::set_pre_salt);
ADD_PROPERTY(PropertyInfo(Variant::STRING, "pre_salt"), "set_pre_salt", "get_pre_salt");
ClassDB::bind_method(D_METHOD("get_post_salt"), &User::get_post_salt);
ClassDB::bind_method(D_METHOD("set_post_salt", "val"), &User::set_post_salt);
ADD_PROPERTY(PropertyInfo(Variant::STRING, "post_salt"), "set_post_salt", "get_post_salt");
ClassDB::bind_method(D_METHOD("get_password_hash"), &User::get_password_hash);
ClassDB::bind_method(D_METHOD("set_password_hash", "val"), &User::set_password_hash);
ADD_PROPERTY(PropertyInfo(Variant::STRING, "password_hash"), "set_password_hash", "get_password_hash");
ClassDB::bind_method(D_METHOD("get_banned"), &User::get_banned);
ClassDB::bind_method(D_METHOD("set_banned", "val"), &User::set_banned);
ADD_PROPERTY(PropertyInfo(Variant::BOOL, "banned"), "set_banned", "get_banned");
ClassDB::bind_method(D_METHOD("get_password_reset_token"), &User::get_password_reset_token);
ClassDB::bind_method(D_METHOD("set_password_reset_token", "val"), &User::set_password_reset_token);
ADD_PROPERTY(PropertyInfo(Variant::STRING, "password_reset_token"), "set_password_reset_token", "get_password_reset_token");
ClassDB::bind_method(D_METHOD("get_locked"), &User::get_locked);
ClassDB::bind_method(D_METHOD("set_locked", "val"), &User::set_locked);
ADD_PROPERTY(PropertyInfo(Variant::BOOL, "locked"), "set_locked", "get_locked");
ClassDB::bind_method(D_METHOD("add_module", "module"), &User::add_module);
ClassDB::bind_method(D_METHOD("get_module", "index"), &User::get_module);
ClassDB::bind_method(D_METHOD("get_module_named", "name"), &User::get_module_named);
ClassDB::bind_method(D_METHOD("remove_module", "index"), &User::remove_module);
ClassDB::bind_method(D_METHOD("get_module_count"), &User::get_module_count);
ClassDB::bind_method(D_METHOD("get_modules"), &User::get_modules);
ClassDB::bind_method(D_METHOD("set_modules", "modules"), &User::set_modules);
ADD_PROPERTY(PropertyInfo(Variant::ARRAY, "modules", PROPERTY_HINT_NONE, "17/17:UserModule", PROPERTY_USAGE_DEFAULT, "UserModule"), "set_modules", "get_modules");
BIND_VMETHOD(MethodInfo(PropertyInfo(Variant::BOOL, "ret"), "_check_password", PropertyInfo(Variant::STRING, "password")));
BIND_VMETHOD(MethodInfo("_create_password", PropertyInfo(Variant::STRING, "password")));
BIND_VMETHOD(MethodInfo(PropertyInfo(Variant::STRING, "ret"), "_hash_password", PropertyInfo(Variant::STRING, "password")));
ClassDB::bind_method(D_METHOD("check_password", "password"), &User::check_password);
ClassDB::bind_method(D_METHOD("create_password", "password"), &User::create_password);
ClassDB::bind_method(D_METHOD("hash_password", "password"), &User::hash_password);
ClassDB::bind_method(D_METHOD("_check_password", "password"), &User::_check_password);
ClassDB::bind_method(D_METHOD("_create_password", "password"), &User::_create_password);
ClassDB::bind_method(D_METHOD("_hash_password", "password"), &User::_hash_password);
ClassDB::bind_method(D_METHOD("save"), &User::save);
ClassDB::bind_method(D_METHOD("read_lock"), &User::read_lock);
ClassDB::bind_method(D_METHOD("read_unlock"), &User::read_unlock);
ClassDB::bind_method(D_METHOD("write_lock"), &User::write_lock);
ClassDB::bind_method(D_METHOD("write_unlock"), &User::write_unlock);
BIND_ENUM_CONSTANT(PERMISSION_CREATE);
BIND_ENUM_CONSTANT(PERMISSION_READ);
BIND_ENUM_CONSTANT(PERMISSION_UPDATE);
BIND_ENUM_CONSTANT(PERMISSION_DELETE);
BIND_ENUM_CONSTANT(PERMISSION_ALL);
BIND_ENUM_CONSTANT(PERMISSION_NONE);
}