#include "Session.h"
#include "BoostLog.h"
#include <Wt/Auth/Dbo/AuthInfo.h>
#include <Wt/Auth/Dbo/UserDatabase.h>
#include <Wt/Auth/FacebookService.h>
#include <Wt/Auth/GoogleService.h>
#include <Wt/Auth/HashFunction.h>
#include <Wt/Auth/PasswordService.h>
#include <Wt/Auth/PasswordStrengthValidator.h>
#include <Wt/Auth/PasswordVerifier.h>
#include <Wt/Dbo/backend/Sqlite3.h>

Wt::Auth::AuthService myAuthService;
Wt::Auth::PasswordService myPasswordService(myAuthService);
std::vector<std::unique_ptr<Wt::Auth::OAuthService>> myOAuthServices;

Session::Session(const std::string &sqliteDb) {
    auto connection = std::make_unique<Wt::Dbo::backend::Sqlite3>(sqliteDb);

    connection->setProperty("show-queries", "true");

    setConnection(std::move(connection));

    mapClass<User>("user");
    mapClass<AuthInfo>("auth_info");
    mapClass<AuthInfo::AuthIdentityType>("auth_identity");
    mapClass<AuthInfo::AuthTokenType>("auth_token");

    try {
        createTables();
        LOG(info) << "Created database.";
    } catch (Wt::Dbo::Exception &e) {
        LOG(error) << e.what() << ", using existing database";
    }

    m_users = std::make_unique<UserDatabase>(*this);
}

Session::~Session() {
}

Wt::Auth::AbstractUserDatabase &Session::users() {
    return *m_users;
}

Wt::Auth::Login &Session::login() {
    return m_login;
}

void Session::configureAuth() {
    myAuthService.setAuthTokensEnabled(true, "logincookie");
    // myAuthService.setEmailVerificationEnabled(true);
    // myAuthService.setEmailVerificationRequired(true);

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

    if (Wt::Auth::GoogleService::configured()) {
        myOAuthServices.push_back(std::make_unique<Wt::Auth::GoogleService>(myAuthService));
    }

    if (Wt::Auth::FacebookService::configured()) {
        myOAuthServices.push_back(std::make_unique<Wt::Auth::FacebookService>(myAuthService));
    }

    for (const auto &oAuthService : myOAuthServices) {
        oAuthService->generateRedirectEndpoint();
    }
}

const Wt::Auth::AuthService &Session::auth() {
    return myAuthService;
}

const Wt::Auth::PasswordService &Session::passwordAuth() {
    return myPasswordService;
}