66 lines
2.2 KiB
C++
66 lines
2.2 KiB
C++
#include "Restful.h"
|
|
#include "Application.h"
|
|
#include "Database/Session.h"
|
|
#include <Wt/Auth/AuthService.h>
|
|
#include <Wt/Auth/Identity.h>
|
|
#include <Wt/Dbo/Impl.h>
|
|
#include <Wt/Dbo/Json.h>
|
|
#include <Wt/Dbo/backend/Sqlite3.h>
|
|
#include <Wt/Http/Response.h>
|
|
#include <boost/scope/scope_exit.hpp>
|
|
|
|
DBO_INSTANTIATE_TEMPLATES(MyMessage)
|
|
|
|
DbStruct *m_dbStruct;
|
|
|
|
void AuthenticationResource::handleRequest(const Wt::Http::Request &request, Wt::Http::Response &response) {
|
|
auto app = Amass::Singleton<WebToolkit::Server>::instance();
|
|
auto session = Database::session();
|
|
auto &service = app->authService();
|
|
|
|
auto enabled = service.authTokenUpdateEnabled();
|
|
boost::scope::scope_exit raii([&enabled, &service] { service.setAuthTokenUpdateEnabled(enabled); });
|
|
service.setAuthTokenUpdateEnabled(false);
|
|
Wt::Auth::AuthTokenState state;
|
|
Wt::Auth::User user;
|
|
if (service.authTokensEnabled()) {
|
|
const std::string *token = request.getCookieValue(service.authTokenCookieName());
|
|
if (token != nullptr) {
|
|
Wt::Auth::AuthTokenResult result = service.processAuthToken(*token, session->users());
|
|
state = result.state();
|
|
if (state == Wt::Auth::AuthTokenState::Valid) {
|
|
user = result.user();
|
|
}
|
|
}
|
|
}
|
|
MyMessage message;
|
|
if (user.isValid()) {
|
|
message.user = user.identity(Wt::Auth::Identity::LoginName).toUTF8();
|
|
}
|
|
LOG(info) << "state: " << (int)state << " " << message.user;
|
|
response.setMimeType("application/json");
|
|
response.addHeader("Server", "Wt");
|
|
message.message = "Hello, World!";
|
|
Wt::Dbo::JsonSerializer writer(response.out());
|
|
writer.serialize(message);
|
|
}
|
|
|
|
void PlaintextResource::handleRequest(const Wt::Http::Request &request, Wt::Http::Response &response) {
|
|
response.setMimeType("text/plain");
|
|
response.addHeader("Server", "Wt");
|
|
response.out() << "Hello, World!";
|
|
}
|
|
|
|
int DbStruct::rand() {
|
|
return distribution(rng);
|
|
}
|
|
|
|
AuthenticationResource::AuthenticationResource() {
|
|
}
|
|
|
|
DbStruct::DbStruct(const std::string &db) : rng(clock()), distribution(1, 10000) {
|
|
session.setConnection(std::make_unique<Wt::Dbo::backend::Sqlite3>(db));
|
|
session.mapClass<World>("world");
|
|
session.mapClass<Fortune>("fortune");
|
|
}
|