web_api.h

Go to the documentation of this file.

#pragma once
 
#include <uwebsockets/App.h>
 
#include <chrono>
#include <functional>
#include <nlohmann/json.hpp>
#include <optional>
#include <string>
#include <string_view>
#include <utility>
 
#include "api/router.h"
 
namespace mm::node {
class DeviceManager;
class MonitoringManager;
}  // namespace mm::node
 
namespace mm::api {
 
struct RouteContext {
  mm::node::DeviceManager& deviceManager;
  mm::node::MonitoringManager& monitoringManager;
  std::string_view corsOrigin;
};
 
using RegisterRoutesFn = std::function<void(Router& router, const RouteContext& ctx)>;
 
template <typename Res>
Res* setCorsOrigin(Res* res, std::string_view corsOrigin) {
  return res->writeHeader("Access-Control-Allow-Origin", corsOrigin);
}
 
template <typename Res>
Res* setWireTime(Res* res, std::chrono::microseconds wireUs) {
  return res->writeHeader("Access-Control-Expose-Headers", "X-Wire-Us")
      ->writeHeader("X-Wire-Us", std::to_string(wireUs.count()));
}
 
template <typename Res>
void sendJson(Res* res, std::string_view corsOrigin, const nlohmann::json& body) {
  setCorsOrigin(res, corsOrigin)
      ->writeHeader("Content-Type", "application/json")
      ->end(body.dump(-1, ' ', false, nlohmann::json::error_handler_t::replace));
}
 
template <typename Res>
void sendBytes(Res* res, std::string_view corsOrigin, std::string_view contentType,
               std::string_view body) {
  setCorsOrigin(res, corsOrigin)->writeHeader("Content-Type", contentType)->end(body);
}
 
template <typename Res>
void sendError(Res* res, std::string_view status, std::string_view corsOrigin,
               std::string_view message,
               std::optional<std::chrono::microseconds> wireUs = std::nullopt) {
  auto* r = setCorsOrigin(res->writeStatus(status), corsOrigin);
  if (wireUs) {
    r = setWireTime(r, *wireUs);
  }
  r->writeHeader("Content-Type", "application/json")
      ->end(nlohmann::json{{"error", std::string(message)}}.dump());
}
 
template <typename Res>
void sendStatus(Res* res, std::string_view status, std::string_view corsOrigin,
                std::optional<std::chrono::microseconds> wireUs = std::nullopt) {
  auto* r = setCorsOrigin(res->writeStatus(status), corsOrigin);
  if (wireUs) {
    r = setWireTime(r, *wireUs);
  }
  r->end();
}
 
}  // namespace mm::api