EMS-ESP32/lib/framework/HttpEndpoint.h

#ifndef HttpEndpoint_h
#define HttpEndpoint_h

#include <functional>

#include <PsychicHttp.h>

#include <SecurityManager.h>
#include <StatefulService.h>

#define HTTP_ENDPOINT_ORIGIN_ID "http"
#define HTTPS_ENDPOINT_ORIGIN_ID "https"

using namespace std::placeholders; // for `_1` etc

template <class T>
class HttpGetEndpoint {
  public:
    HttpGetEndpoint(JsonStateReader<T>      stateReader,
                    StatefulService<T> *    statefulService,
                    PsychicHttpServer *     server,
                    const char *            servicePath,
                    SecurityManager *       securityManager,
                    AuthenticationPredicate authenticationPredicate = AuthenticationPredicates::IS_ADMIN,
                    size_t                  bufferSize              = DEFAULT_BUFFER_SIZE)
        : _stateReader(stateReader)
        , _statefulService(statefulService)
        , _bufferSize(bufferSize)
        , _server(server)
        , _servicePath(servicePath)
        , _securityManager(securityManager)
        , _authenticationPredicate(authenticationPredicate) {
    }

  protected:
    JsonStateReader<T>      _stateReader;
    StatefulService<T> *    _statefulService;
    size_t                  _bufferSize;
    PsychicHttpServer *     _server;
    const char *            _servicePath;
    SecurityManager *       _securityManager;
    AuthenticationPredicate _authenticationPredicate;

    void registerURI() {
#ifdef EMSESP_DEBUG
        ESP_LOGE("HttpGetEndpoint", "Addding GET endpoint %s", _servicePath);
#endif
        _server->on(_servicePath,
                    HTTP_GET,
                    _securityManager->wrapRequest(
                        [this](PsychicRequest * request) {
                            PsychicJsonResponse response   = PsychicJsonResponse(request, false, _bufferSize);
                            JsonObject          jsonObject = response.getRoot();
                            _statefulService->read(jsonObject, _stateReader);
                            return response.send();
                        },
                        _authenticationPredicate));
    }
};

template <class T>
class HttpPostEndpoint {
  public:
    HttpPostEndpoint(JsonStateReader<T>      stateReader,
                     JsonStateUpdater<T>     stateUpdater,
                     StatefulService<T> *    statefulService,
                     PsychicHttpServer *     server,
                     const char *            servicePath,
                     SecurityManager *       securityManager,
                     AuthenticationPredicate authenticationPredicate = AuthenticationPredicates::IS_ADMIN,
                     size_t                  bufferSize              = DEFAULT_BUFFER_SIZE)
        : _stateReader(stateReader)
        , _stateUpdater(stateUpdater)
        , _statefulService(statefulService)
        , _server(server)
        , _servicePath(servicePath)
        , _securityManager(securityManager)
        , _authenticationPredicate(authenticationPredicate)
        , _bufferSize(bufferSize) {
    }


  protected:
    JsonStateReader<T>      _stateReader;
    JsonStateUpdater<T>     _stateUpdater;
    StatefulService<T> *    _statefulService;
    size_t                  _bufferSize;
    SecurityManager *       _securityManager;
    AuthenticationPredicate _authenticationPredicate;
    PsychicHttpServer *     _server;
    const char *            _servicePath;

    void registerURI() {
#ifdef EMSESP_DEBUG
        ESP_LOGE("HttpPostEndpoint", "Addding POST endpoint %s", _servicePath);
#endif
        _server->on(_servicePath,
                    HTTP_POST,
                    _securityManager->wrapCallback(
                        [this](PsychicRequest * request, JsonVariant & json) {
                            if (!json.is<JsonObject>()) {
                                return request->reply(400);
                            }

                            JsonObject        jsonObject = json.as<JsonObject>();
                            StateUpdateResult outcome    = _statefulService->updateWithoutPropagation(jsonObject, _stateUpdater);

                            if (outcome == StateUpdateResult::ERROR) {
                                return request->reply(400);
                            } else if ((outcome == StateUpdateResult::CHANGED) || (outcome == StateUpdateResult::CHANGED_RESTART)) {
                                // TODO see if this works as intended. Before the stat was updated on an onDisconnect
                                // request->onDisconnect([this]() { _statefulService->callUpdateHandlers(HTTP_ENDPOINT_ORIGIN_ID); });
                                // TODO add https
                                _statefulService->callUpdateHandlers(HTTP_ENDPOINT_ORIGIN_ID); // persist the changes to the FS
                            }

                            PsychicJsonResponse response = PsychicJsonResponse(request, false, _bufferSize);
                            jsonObject                   = response.getRoot();

                            _statefulService->read(jsonObject, _stateReader);

                            if (outcome == StateUpdateResult::CHANGED_RESTART) {
                                return request->reply(205); // reboot required
                            }
                            return response.send();
                        },
                        _authenticationPredicate));
    }
};

template <class T>
class HttpEndpoint : public HttpGetEndpoint<T>, public HttpPostEndpoint<T> {
  public:
    HttpEndpoint(JsonStateReader<T>      stateReader,
                 JsonStateUpdater<T>     stateUpdater,
                 StatefulService<T> *    statefulService,
                 PsychicHttpServer *     server,
                 const char *            servicePath,
                 SecurityManager *       securityManager,
                 AuthenticationPredicate authenticationPredicate = AuthenticationPredicates::IS_ADMIN,
                 size_t                  bufferSize              = DEFAULT_BUFFER_SIZE)
        : HttpGetEndpoint<T>(stateReader, statefulService, server, servicePath, securityManager, authenticationPredicate, bufferSize)
        , HttpPostEndpoint<T>(stateReader, stateUpdater, statefulService, server, servicePath, securityManager, authenticationPredicate, bufferSize) {
    }

    // register the web server on() endpoints
    void registerURI() {
        HttpGetEndpoint<T>::registerURI();
        HttpPostEndpoint<T>::registerURI();
    }
};

#endif