auth_middleware.h

Go to the documentation of this file.

#pragma once

#include <functional>
#include "middleware.h"
#include "entity.h"
#include "auth_strategy.h"

class AuthMiddleware : public Middleware {
    const char * cookieMiddlewareID;
    const char * jsonMiddlewareID;
    Entity * currentUser = nullptr;
    AuthStrategy * strategy = nullptr;
    std::vector<AuthStrategy *> allStrategies;
    std::function<Entity *(std::string &, std::string &)> onLogin = nullptr;
    std::function<std::string(Entity *)> onSerialize = nullptr;
    std::function<Entity *(std::string &)> onDeserialize = nullptr;
public:
    AuthMiddleware(const char * nameID, const char * cookieID, const char * jsonID)
        : Middleware(nameID), cookieMiddlewareID(cookieID), jsonMiddlewareID(jsonID), allStrategies() {}

    ~AuthMiddleware() override;

    void setOnSerialize(std::function<std::string(Entity *)> serialize);

    void setOnDeserialize(std::function<Entity *(std::string &)> deserialize);

    void setOnLogin(std::function<Entity *(std::string &, std::string &)> login);

    bool login(std::string & userName, std::string & password);

    bool autoExec() override;

    void exec() override;

    void clear() override { Middleware::clear(); setUser(nullptr); }

    Entity * getUser();

    void addStrategy(AuthStrategy * strategy);
private:
    void setUser(Entity * user);

    void setStrategy(AuthStrategy * strategy);
};