#include "WebApplication.h"
#include "BoostLog.h"
#include "Hello.h"
#include "Restful.h"
#include "Session.h"
#include <Wt/Auth/AuthService.h>
#include <Wt/Auth/HashFunction.h>
#include <Wt/Auth/PasswordService.h>
#include <Wt/Auth/PasswordStrengthValidator.h>
#include <Wt/Auth/PasswordVerifier.h>
#include <Wt/Dbo/SqlConnectionPool.h>
#include <Wt/WServer.h>

static std::unique_ptr<Wt::WApplication> createApplication(const Wt::WEnvironment &env) {
    return std::make_unique<Hello>(env, false);
}

static std::unique_ptr<Wt::WApplication> createWidgetSet(const Wt::WEnvironment &env) {
    return std::make_unique<Hello>(env, true);
}

WebApplication::WebApplication() {
    try {
        std::vector<std::string> args;
        args.push_back("--approot=./build");
        args.push_back("--docroot=./build");
        args.push_back("--http-listen=127.0.0.1:8855");
        // --docroot=. --no-compression --http-listen 127.0.0.1:8855
        initializeAuthenticationService();
        m_server = std::make_unique<Wt::WServer>("./build", args);
        m_server->addEntryPoint(Wt::EntryPointType::Application, createApplication, "/hello");
        m_server->addEntryPoint(Wt::EntryPointType::WidgetSet, createWidgetSet, "/gui/hello.js");
        m_server->addResource(std::make_shared<JsonResource>(), "/json");
        m_server->addResource(std::make_shared<PlaintextResource>(), "/plaintext");
        m_server->addResource(std::make_shared<DbResource>("database.sqlite"), "/db");

        m_server->start();
    } catch (const std::exception &e) {
        LOG(error) << e.what();
    }
}

WebApplication::~WebApplication() {
}

void WebApplication::initializeAuthenticationService() {
    m_authService = std::make_unique<Wt::Auth::AuthService>();
    m_authService->setAuthTokensEnabled(true, "logincookie");
    m_passwordService = std::make_unique<Wt::Auth::PasswordService>(*m_authService);

    auto verifier = std::make_unique<Wt::Auth::PasswordVerifier>();
    verifier->addHashFunction(std::make_unique<Wt::Auth::BCryptHashFunction>(7));
    m_passwordService->setVerifier(std::move(verifier));
    m_passwordService->setPasswordThrottle(std::make_unique<Wt::Auth::AuthThrottle>());
    m_passwordService->setStrengthValidator(std::make_unique<Wt::Auth::PasswordStrengthValidator>());
}

const Wt::Auth::AuthService &WebApplication::authService() {
    return *m_authService;
}

const Wt::Auth::PasswordService &WebApplication::passwordService() {
    return *m_passwordService;
}