130 lines
3.7 KiB
C++
130 lines
3.7 KiB
C++
#include "User.h"
|
|
|
|
#include "BlogUserDatabase.h"
|
|
|
|
#include <Wt/Auth/Identity.h>
|
|
|
|
class TransactionImpl : public Wt::Auth::AbstractUserDatabase::Transaction, public Wt::Dbo::Transaction {
|
|
public:
|
|
TransactionImpl(Wt::Dbo::Session &session) : Wt::Dbo::Transaction(session) {
|
|
}
|
|
|
|
virtual ~TransactionImpl() noexcept(false) {
|
|
}
|
|
|
|
virtual void commit() {
|
|
Wt::Dbo::Transaction::commit();
|
|
}
|
|
|
|
virtual void rollback() {
|
|
Wt::Dbo::Transaction::rollback();
|
|
}
|
|
};
|
|
|
|
class InvalidUser : public std::runtime_error {
|
|
public:
|
|
InvalidUser(const std::string &id) : std::runtime_error("Invalid user: " + id) {
|
|
}
|
|
};
|
|
|
|
BlogUserDatabase::BlogUserDatabase(Wt::Dbo::Session &session) : m_session(session) {
|
|
}
|
|
|
|
BlogUserDatabase::~BlogUserDatabase() {
|
|
}
|
|
|
|
Wt::Dbo::ptr<User> BlogUserDatabase::find(const Wt::Auth::User &user) const {
|
|
getUser(user.id());
|
|
return m_user;
|
|
}
|
|
|
|
BlogUserDatabase::Transaction *BlogUserDatabase::startTransaction() {
|
|
return new TransactionImpl(m_session);
|
|
}
|
|
|
|
Wt::Auth::User BlogUserDatabase::findWithId(const std::string &id) const {
|
|
getUser(id);
|
|
|
|
if (m_user)
|
|
return Wt::Auth::User(id, *this);
|
|
else
|
|
return Wt::Auth::User();
|
|
}
|
|
|
|
Wt::Auth::User BlogUserDatabase::findWithIdentity(const std::string &provider, const Wt::WString &identity) const {
|
|
Wt::Dbo::Transaction t(m_session);
|
|
if (provider == Wt::Auth::Identity::LoginName) {
|
|
if (!m_user || m_user->name != identity) m_user = m_session.find<User>().where("name = ?").bind(identity);
|
|
} else
|
|
m_user = m_session.find<User>()
|
|
.where("oauth_id = ?")
|
|
.bind(identity.toUTF8())
|
|
.where("oauth_provider = ?")
|
|
.bind(provider);
|
|
t.commit();
|
|
|
|
if (m_user)
|
|
return Wt::Auth::User(std::to_string(m_user.id()), *this);
|
|
else
|
|
return Wt::Auth::User();
|
|
}
|
|
|
|
void BlogUserDatabase::addIdentity(const Wt::Auth::User &user, const std::string &provider,
|
|
const Wt::WString &identity) {
|
|
WithUser find(*this, user);
|
|
|
|
if (provider == Wt::Auth::Identity::LoginName)
|
|
m_user.modify()->name = identity;
|
|
else {
|
|
m_user.modify()->oAuthProvider = provider;
|
|
m_user.modify()->oAuthId = identity.toUTF8();
|
|
}
|
|
}
|
|
|
|
void BlogUserDatabase::removeIdentity(const Wt::Auth::User &user, const std::string &provider) {
|
|
WithUser find(*this, user);
|
|
|
|
if (provider == Wt::Auth::Identity::LoginName)
|
|
m_user.modify()->name = "";
|
|
else if (provider == m_user->oAuthProvider) {
|
|
m_user.modify()->oAuthProvider = std::string();
|
|
m_user.modify()->oAuthId = std::string();
|
|
}
|
|
}
|
|
|
|
Wt::WString BlogUserDatabase::identity(const Wt::Auth::User &user, const std::string &provider) const {
|
|
WithUser find(*this, user);
|
|
|
|
if (provider == Wt::Auth::Identity::LoginName)
|
|
return m_user->name;
|
|
else if (provider == m_user->oAuthProvider)
|
|
return Wt::WString(m_user->oAuthId);
|
|
else
|
|
return Wt::WString::Empty;
|
|
}
|
|
|
|
void BlogUserDatabase::setLastLoginAttempt(const Wt::Auth::User &user, const Wt::WDateTime &t) {
|
|
WithUser find(*this, user);
|
|
|
|
m_user.modify()->lastLoginAttempt = t;
|
|
}
|
|
|
|
BlogUserDatabase::WithUser::WithUser(const BlogUserDatabase &self, const Wt::Auth::User &user)
|
|
: transaction(self.m_session) {
|
|
self.getUser(user.id());
|
|
|
|
if (!self.m_user) throw InvalidUser(user.id());
|
|
}
|
|
|
|
BlogUserDatabase::WithUser::~WithUser() {
|
|
transaction.commit();
|
|
}
|
|
|
|
void BlogUserDatabase::getUser(const std::string &id) const {
|
|
if (!m_user || std::to_string(m_user.id()) != id) {
|
|
Wt::Dbo::Transaction t(m_session);
|
|
m_user = m_session.find<User>().where("id = ?").bind(User::stringToId(id));
|
|
t.commit();
|
|
}
|
|
}
|