80 lines
2.1 KiB
C++
80 lines
2.1 KiB
C++
#include "PsychicWebHandler.h"
|
|
|
|
PsychicWebHandler::PsychicWebHandler() : PsychicHandler(),
|
|
_requestCallback(NULL),
|
|
_onOpen(NULL),
|
|
_onClose(NULL)
|
|
{
|
|
}
|
|
PsychicWebHandler::~PsychicWebHandler() {}
|
|
|
|
bool PsychicWebHandler::canHandle(PsychicRequest* request)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
esp_err_t PsychicWebHandler::handleRequest(PsychicRequest* request, PsychicResponse* response)
|
|
{
|
|
// lookup our client
|
|
PsychicClient* client = checkForNewClient(request->client());
|
|
if (client->isNew)
|
|
openCallback(client);
|
|
|
|
/* Request body cannot be larger than a limit */
|
|
if (request->contentLength() > request->server()->maxRequestBodySize)
|
|
{
|
|
ESP_LOGE(PH_TAG, "Request body too large : %d bytes", request->contentLength());
|
|
|
|
/* Respond with 400 Bad Request */
|
|
char error[60];
|
|
sprintf(error, "Request body must be less than %lu bytes!", request->server()->maxRequestBodySize);
|
|
response->send(400, "text/html", error);
|
|
|
|
/* Return failure to close underlying connection else the incoming file content will keep the socket busy */
|
|
return ESP_FAIL;
|
|
}
|
|
|
|
// get our body loaded up.
|
|
esp_err_t err = request->loadBody();
|
|
if (err != ESP_OK)
|
|
return response->send(400, "text/html", "Error loading request body.");
|
|
|
|
// load our params in.
|
|
request->loadParams();
|
|
|
|
// okay, pass on to our callback.
|
|
if (this->_requestCallback != NULL)
|
|
err = this->_requestCallback(request, response);
|
|
|
|
return err;
|
|
}
|
|
|
|
PsychicWebHandler* PsychicWebHandler::onRequest(PsychicHttpRequestCallback fn)
|
|
{
|
|
_requestCallback = fn;
|
|
return this;
|
|
}
|
|
|
|
void PsychicWebHandler::openCallback(PsychicClient* client)
|
|
{
|
|
if (_onOpen != NULL)
|
|
_onOpen(client);
|
|
}
|
|
|
|
void PsychicWebHandler::closeCallback(PsychicClient* client)
|
|
{
|
|
if (_onClose != NULL)
|
|
_onClose(getClient(client));
|
|
}
|
|
|
|
PsychicWebHandler* PsychicWebHandler::onOpen(PsychicClientCallback fn)
|
|
{
|
|
_onOpen = fn;
|
|
return this;
|
|
}
|
|
|
|
PsychicWebHandler* PsychicWebHandler::onClose(PsychicClientCallback fn)
|
|
{
|
|
_onClose = fn;
|
|
return this;
|
|
} |