126 lines
2.8 KiB
C++
126 lines
2.8 KiB
C++
#include <Wt/Auth/Dbo/AuthInfo>
|
|
#include <Wt/Auth/Dbo/UserDatabase>
|
|
#include <Wt/Auth/AuthService>
|
|
#include <Wt/Auth/HashFunction>
|
|
#include <Wt/Auth/PasswordService>
|
|
#include <Wt/Auth/PasswordStrengthValidator>
|
|
#include <Wt/Auth/PasswordVerifier>
|
|
|
|
// Db types
|
|
#include "AudioTypes.hpp"
|
|
#include "FileTypes.hpp"
|
|
#include "MediaDirectory.hpp"
|
|
#include "User.hpp"
|
|
|
|
#include "DatabaseHandler.hpp"
|
|
|
|
namespace Database {
|
|
|
|
|
|
namespace {
|
|
Wt::Auth::AuthService authService;
|
|
Wt::Auth::PasswordService passwordService(authService);
|
|
}
|
|
|
|
|
|
void
|
|
Handler::configureAuth(void)
|
|
{
|
|
authService.setEmailVerificationEnabled(true);
|
|
|
|
Wt::Auth::PasswordVerifier *verifier = new Wt::Auth::PasswordVerifier();
|
|
verifier->addHashFunction(new Wt::Auth::BCryptHashFunction(8));
|
|
passwordService.setVerifier(verifier);
|
|
passwordService.setAttemptThrottlingEnabled(true);
|
|
passwordService.setStrengthValidator(new Wt::Auth::PasswordStrengthValidator());
|
|
}
|
|
|
|
const Wt::Auth::AuthService&
|
|
Handler::getAuthService()
|
|
{
|
|
return authService;
|
|
}
|
|
|
|
const Wt::Auth::PasswordService&
|
|
Handler::getPasswordService()
|
|
{
|
|
return passwordService;
|
|
}
|
|
|
|
|
|
Handler::Handler(boost::filesystem::path db)
|
|
:
|
|
_dbBackend( db.string() )
|
|
{
|
|
_session.setConnection(_dbBackend);
|
|
_session.mapClass<Database::Genre>("genre");
|
|
_session.mapClass<Database::Track>("track");
|
|
_session.mapClass<Database::Artist>("artist");
|
|
_session.mapClass<Database::Release>("release");
|
|
_session.mapClass<Database::Release>("release");
|
|
_session.mapClass<Database::Path>("path");
|
|
_session.mapClass<Database::Video>("video");
|
|
_session.mapClass<Database::MediaDirectory>("media_directory");
|
|
_session.mapClass<Database::MediaDirectorySettings>("media_directory_settings");
|
|
|
|
_session.mapClass<Database::User>("user");
|
|
_session.mapClass<Database::AuthInfo>("auth_info");
|
|
_session.mapClass<Database::AuthInfo::AuthIdentityType>("auth_identity");
|
|
_session.mapClass<Database::AuthInfo::AuthTokenType>("auth_token");
|
|
|
|
try {
|
|
_session.createTables();
|
|
}
|
|
catch(std::exception& e) {
|
|
std::cerr << "Cannot create tables: " << e.what() << std::endl;
|
|
}
|
|
|
|
_dbBackend.executeSql("pragma journal_mode=WAL");
|
|
|
|
_users = new UserDatabase(_session);
|
|
}
|
|
|
|
Handler::~Handler()
|
|
{
|
|
delete _users;
|
|
}
|
|
|
|
Wt::Auth::AbstractUserDatabase&
|
|
Handler::getUserDatabase()
|
|
{
|
|
return *_users;
|
|
}
|
|
|
|
User::pointer
|
|
Handler::getCurrentUser()
|
|
{
|
|
if (_login.loggedIn())
|
|
return getUser(_login.user());
|
|
else
|
|
return User::pointer();
|
|
}
|
|
|
|
User::pointer
|
|
Handler::getUser(const Wt::Auth::User& authUser)
|
|
{
|
|
|
|
if (!authUser.isValid()) {
|
|
std::cerr << "Handler::getUser: invalid authUser" << std::endl;
|
|
return User::pointer();
|
|
}
|
|
|
|
Wt::Dbo::ptr<AuthInfo> authInfo = _users->find(authUser);
|
|
|
|
User::pointer user = authInfo->user();
|
|
|
|
if (!user) {
|
|
user = _session.add(new User());
|
|
authInfo.modify()->setUser(user);
|
|
}
|
|
|
|
return user;
|
|
}
|
|
|
|
|
|
} // namespace Database
|