mirror of
https://git.rwth-aachen.de/acs/public/villas/node/
synced 2025-03-16 00:00:02 +01:00
84 lines
2 KiB
C++
84 lines
2 KiB
C++
/** API Response.
|
|
*
|
|
* @file
|
|
* @author Steffen Vogel <post@steffenvogel.de>
|
|
* @copyright 2014-2022, Institute for Automation of Complex Power Systems, EONERC
|
|
* @license Apache 2.0
|
|
*********************************************************************************/
|
|
|
|
#include <villas/api/response.hpp>
|
|
#include <villas/api/request.hpp>
|
|
#include <villas/config.hpp>
|
|
|
|
using namespace villas::node::api;
|
|
|
|
Response::Response(Session *s, int c, const std::string &ct, const Buffer &b) :
|
|
session(s),
|
|
logger(logging.get("api:response")),
|
|
buffer(b),
|
|
code(c),
|
|
contentType(ct),
|
|
headers{
|
|
{ "Server:", HTTP_USER_AGENT },
|
|
{ "Access-Control-Allow-Origin:", "*" },
|
|
{ "Access-Control-Allow-Methods:", "GET, POST, OPTIONS" },
|
|
{ "Access-Control-Allow-Headers:", "Content-Type" },
|
|
{ "Access-Control-Max-Age:", "86400" }
|
|
}
|
|
{ }
|
|
|
|
int Response::writeBody(struct lws *wsi)
|
|
{
|
|
int ret;
|
|
|
|
ret = lws_write(wsi, (unsigned char *) buffer.data(), buffer.size(), LWS_WRITE_HTTP_FINAL);
|
|
if (ret < 0)
|
|
return -1;
|
|
|
|
return 1;
|
|
}
|
|
|
|
int Response::writeHeaders(struct lws *wsi)
|
|
{
|
|
int ret;
|
|
uint8_t headerBuffer[2048], *p = headerBuffer, *end = &headerBuffer[sizeof(headerBuffer) - 1];
|
|
|
|
// We need to encode the buffer here for getting the real content length of the response
|
|
encodeBody();
|
|
|
|
ret = lws_add_http_common_headers(wsi, code, contentType.c_str(),
|
|
buffer.size(),
|
|
&p, end);
|
|
if (ret)
|
|
return 1;
|
|
|
|
for (auto &hdr : headers) {
|
|
ret = lws_add_http_header_by_name (wsi,
|
|
reinterpret_cast<const unsigned char *>(hdr.first.c_str()),
|
|
reinterpret_cast<const unsigned char *>(hdr.second.c_str()),
|
|
hdr.second.size(), &p, end);
|
|
if (ret)
|
|
return -1;
|
|
}
|
|
|
|
ret = lws_finalize_write_http_header(wsi, headerBuffer, &p, end);
|
|
if (ret)
|
|
return 1;
|
|
|
|
/* Do we have a body to send? */
|
|
if (buffer.size() > 0)
|
|
lws_callback_on_writable(wsi);
|
|
|
|
return 0;
|
|
}
|
|
|
|
JsonResponse::~JsonResponse()
|
|
{
|
|
if (response)
|
|
json_decref(response);
|
|
}
|
|
|
|
void JsonResponse::encodeBody()
|
|
{
|
|
buffer.encode(response, JSON_INDENT(4));
|
|
}
|