Middleware.hpp

#pragma once
 
#include "magpie/data/CommonData.hpp"
#include "magpie/transfer/Request.hpp"
#include "magpie/transfer/Response.hpp"
#include <memory>
namespace magpie {
 
template <data::IsCommonData ContextType>
class Middleware;

template <data::IsCommonData ContextType>
struct IMiddlewareProcessor {
    virtual ~IMiddlewareProcessor() = default;
    virtual Middleware<ContextType>* getNext() = 0;
    virtual void invokeRoute() = 0;
};

template <data::IsCommonData ContextType>
class Middleware {
public:
    virtual ~Middleware() = default;

    virtual void onRequest(
        IMiddlewareProcessor<ContextType>* proc,
        ContextType* ctx,
        Request& req,
        Response& res
    ) = 0;

    void next(
        IMiddlewareProcessor<ContextType>* proc,
        ContextType* ctx,
        Request& req,
        Response& res
    ) {
        auto* ptr = proc->getNext();
        if (ptr == nullptr) {
            return;
        }
        ptr->onRequest(
            proc,
            ctx,
            req,
            res
        );
    }
 
};

template <data::IsCommonData ContextType>
struct Middlewares {
    std::vector<std::shared_ptr<Middleware<ContextType>>> middlewares;
};
 
}