Remove psychichttp

esp32/include/ap_service.h

@@ -3,8 +3,9 @@
 #include <template/stateful_persistence.h>
 #include <settings/ap_settings.h>
 #include <utils/timing.h>
-#include <WiFi.h>
+#include <utils/http_utils.h>
-#include "esp_timer.h"
+#include <esp_http_server.h>
+#include <esp_timer.h>
 #include <string>
 
 class APService : public StatefulService<APSettings> {
@@ -16,14 +17,13 @@ class APService : public StatefulService<APSettings> {
     void loop();
     void recoveryMode();
 
-    esp_err_t getStatus(PsychicRequest *request);
+    esp_err_t getStatus(httpd_req_t *req);
     void status(JsonObject &root);
     APNetworkStatus getAPNetworkStatus();
 
     StatefulHttpEndpoint<APSettings> endpoint;
 
   private:
-    PsychicHttpServer *_server;
     FSPersistence<APSettings> _persistence;
 
     DNSServer *_dnsServer;

esp32/include/communication/websocket_adapter.h

@@ -1,8 +1,9 @@
 #ifndef Socket_h
 #define Socket_h
 
-#include <PsychicHttp.h>
+#include <esp_http_server.h>
 #include <template/stateful_service.h>
+#include <utils/websocket_server.h>
 #include <list>
 #include <map>
 #include <vector>
@@ -12,7 +13,7 @@
 
 class Websocket : CommAdapterBase {
   public:
-    Websocket(PsychicHttpServer &server, const char *route = "/api/ws");
+    Websocket(httpd_handle_t *server, const char *route = "/api/ws");
 
     void begin() override;
 
@@ -20,17 +21,21 @@ class Websocket : CommAdapterBase {
 
     void emit(const char *event, JsonVariant &payload, const char *originId = "", bool onlyToSameOrigin = false);
 
-  private:
+    httpd_uri_t *getUriHandler() { return &_ws_uri; }
-    PsychicWebSocketHandler _socket;
-    PsychicHttpServer &_server;
-    const char *_route;
 
-    void onWSOpen(PsychicWebSocketClient *client);
+  private:
-    void onWSClose(PsychicWebSocketClient *client);
+    websocket::WebSocketServer _socket;
-    esp_err_t onFrame(PsychicWebSocketRequest *request, httpd_ws_frame *frame);
+    httpd_handle_t *_server;
+    const char *_route;
+    httpd_uri_t _ws_uri;
+
+    void onWSOpen(int fd);
+    void onWSClose(int fd);
+    esp_err_t onFrame(httpd_req_t *req, httpd_ws_frame_t *frame);
 
     void send(const uint8_t *data, size_t len, int cid = -1) override;
+
+    static esp_err_t ws_handler_wrapper(httpd_req_t *req);
 };
 
 #endif
-

esp32/include/features.h

@@ -1,9 +1,8 @@
 #ifndef Features_h
 #define Features_h
 
-#include <WiFi.h>
 #include <ArduinoJson.h>
-#include <PsychicHttp.h>
+#include <esp_http_server.h>
 
 #define FT_ENABLED(feature) feature
 
@@ -83,7 +82,7 @@ void printFeatureConfiguration();
 
 void features(JsonObject &root);
 
-esp_err_t getFeatures(PsychicRequest *request);
+esp_err_t getFeatures(httpd_req_t *req);
 
 } // namespace feature_service
 

esp32/include/mdns_service.h

@@ -1,34 +1,15 @@
 #pragma once
 
-#include <PsychicHttp.h>
+#include <mdns.h>
-#include <ESPmDNS.h>
+#include <esp_http_server.h>
-#include <template/stateful_service.h>
+#include <utils/http_utils.h>
-#include <template/stateful_endpoint.h>
-#include <template/stateful_persistence.h>
-#include <settings/mdns_settings.h>
-#include <utils/timing.h>
 #include <string>
 
-class MDNSService : public StatefulService<MDNSSettings> {
+namespace mdns_service {
-  private:
-    FSPersistence<MDNSSettings> _persistence;
-    bool _started {false};
 
-    void reconfigureMDNS();
+void begin(const char *hostname);
-    void startMDNS();
+void end();
-    void stopMDNS();
+void addService(const char *service, const char *proto, uint16_t port);
-    void addServices();
+void addServiceTxt(const char *service, const char *proto, const char *key, const char *value);
 
-  public:
+} // namespace mdns_service
-    MDNSService();
-    ~MDNSService();
-
-    void begin();
-
-    esp_err_t getStatus(PsychicRequest *request);
-    void getStatus(JsonVariant &root);
-
-    static esp_err_t queryServices(PsychicRequest *request, JsonVariant &json);
-
-    StatefulHttpEndpoint<MDNSSettings> endpoint;
-};

esp32/include/peripherals/camera_service.h

@@ -2,8 +2,8 @@
 #define CameraService_h
 
 #include <ArduinoJson.h>
-#include <PsychicHttp.h>
+#include <esp_http_server.h>
-#include <WiFi.h>
+#include <utils/http_utils.h>
 #include <async_worker.h>
 
 #include <features.h>
@@ -35,8 +35,8 @@ class CameraService : public StatefulService<CameraSettings> {
 
     esp_err_t begin();
 
-    esp_err_t cameraStill(PsychicRequest *request);
+    esp_err_t cameraStill(httpd_req_t *req);
-    esp_err_t cameraStream(PsychicRequest *request);
+    esp_err_t cameraStream(httpd_req_t *req);
 
     StatefulHttpEndpoint<CameraSettings> endpoint;
 

esp32/include/system_service.h

@@ -1,23 +1,21 @@
 #pragma once
 
-#include <ESPmDNS.h>
+#include <esp_http_server.h>
-#include <PsychicHttp.h>
-#include <WiFi.h>
-// #include <communication/websocket_adapter.h>
 #include <filesystem.h>
 #include <global.h>
-#include "esp_timer.h"
+#include <esp_timer.h>
+#include <utils/http_utils.h>
 #include <string>
 
 #define MAX_ESP_ANALYTICS_SIZE 2024
 #define EVENT_ANALYTICS "analytics"
 
 namespace system_service {
-esp_err_t handleReset(PsychicRequest *request);
+esp_err_t handleReset(httpd_req_t *req);
-esp_err_t handleRestart(PsychicRequest *request);
+esp_err_t handleRestart(httpd_req_t *req);
-esp_err_t handleSleep(PsychicRequest *request);
+esp_err_t handleSleep(httpd_req_t *req);
-esp_err_t getStatus(PsychicRequest *request);
+esp_err_t getStatus(httpd_req_t *req);
-esp_err_t getMetrics(PsychicRequest *request);
+esp_err_t getMetrics(httpd_req_t *req);
 
 void reset();
 void restart();

esp32/include/template/stateful_endpoint.h

@@ -1,7 +1,8 @@
 #pragma once
 
-#include <PsychicHttp.h>
+#include <esp_http_server.h>
 #include <template/stateful_service.h>
+#include <utils/http_utils.h>
 
 #include <functional>
 
@@ -20,29 +21,27 @@ class StatefulHttpEndpoint {
                          StatefulService<T> *statefulService)
         : _stateReader(stateReader), _stateUpdater(stateUpdater), _statefulService(statefulService) {}
 
-    esp_err_t handleStateUpdate(PsychicRequest *request, JsonVariant &json) {
+    esp_err_t handleStateUpdate(httpd_req_t *req, JsonVariant &json) {
         JsonVariant jsonObject = json.as<JsonVariant>();
         StateUpdateResult outcome = _statefulService->updateWithoutPropagation(jsonObject, _stateUpdater);
 
-        if (outcome == StateUpdateResult::ERROR)
+        if (outcome == StateUpdateResult::ERROR) {
-            return request->reply(400);
+            return http_utils::send_error(req, 400);
-        else if ((outcome == StateUpdateResult::CHANGED)) {
+        } else if ((outcome == StateUpdateResult::CHANGED)) {
-            // persist the changes to the FS
             _statefulService->callUpdateHandlers(HTTP_ENDPOINT_ORIGIN_ID);
         }
 
-        PsychicJsonResponse response = PsychicJsonResponse(request, false);
+        JsonDocument doc;
-        jsonObject = response.getRoot();
+        jsonObject = doc.to<JsonVariant>();
-
         _statefulService->read(jsonObject, _stateReader);
 
-        return response.send();
+        return http_utils::send_json_response(req, doc);
     }
 
-    esp_err_t getState(PsychicRequest *request) {
+    esp_err_t getState(httpd_req_t *req) {
-        PsychicJsonResponse response = PsychicJsonResponse(request, false);
+        JsonDocument doc;
-        JsonVariant jsonObject = response.getRoot();
+        JsonVariant jsonObject = doc.to<JsonVariant>();
         _statefulService->read(jsonObject, _stateReader);
-        return response.send();
+        return http_utils::send_json_response(req, doc);
     }
 };

esp32/include/utils/http_utils.h

@@ -0,0 +1,24 @@
+#pragma once
+
+#include <esp_http_server.h>
+#include <ArduinoJson.h>
+
+namespace http_utils {
+
+esp_err_t send_json_response(httpd_req_t *req, JsonDocument &doc, int status_code = 200);
+
+esp_err_t send_error(httpd_req_t *req, int status_code, const char *message = nullptr);
+
+esp_err_t send_empty_response(httpd_req_t *req, int status_code = 200);
+
+esp_err_t add_cors_headers(httpd_req_t *req);
+
+esp_err_t add_standard_headers(httpd_req_t *req);
+
+esp_err_t parse_json_body(httpd_req_t *req, JsonDocument &doc);
+
+esp_err_t handle_options_cors(httpd_req_t *req);
+
+const char *get_client_ip(httpd_req_t *req);
+
+} // namespace http_utils

esp32/include/utils/websocket_server.h

@@ -0,0 +1,56 @@
+#pragma once
+
+#include <esp_http_server.h>
+#include <map>
+#include <list>
+#include <functional>
+
+namespace websocket {
+
+struct WebSocketClient {
+    int fd;
+    uint64_t last_seen;
+};
+
+typedef std::function<void(int fd)> ClientCallback;
+typedef std::function<esp_err_t(httpd_req_t *req, httpd_ws_frame_t *frame)> FrameCallback;
+
+class WebSocketServer {
+  public:
+    WebSocketServer();
+    ~WebSocketServer();
+
+    void setOpenCallback(ClientCallback callback);
+    void setCloseCallback(ClientCallback callback);
+    void setFrameCallback(FrameCallback callback);
+
+    esp_err_t handleWebSocket(httpd_req_t *req);
+
+    void addClient(int fd);
+    void removeClient(int fd);
+    WebSocketClient *getClient(int fd);
+
+    esp_err_t sendText(int fd, const char *data, size_t len);
+    esp_err_t sendBinary(int fd, const uint8_t *data, size_t len);
+    esp_err_t sendToAll(httpd_ws_type_t type, const uint8_t *data, size_t len);
+
+    bool hasClients() const { return !_clients.empty(); }
+    size_t clientCount() const { return _clients.size(); }
+
+  public:
+    void setServer(httpd_handle_t server) { _server = server; }
+
+  private:
+    std::map<int, WebSocketClient> _clients;
+    ClientCallback _onOpen;
+    ClientCallback _onClose;
+    FrameCallback _onFrame;
+    SemaphoreHandle_t _mutex;
+    httpd_handle_t _server;
+
+    friend esp_err_t websocket_handler(httpd_req_t *req);
+};
+
+esp_err_t websocket_handler(httpd_req_t *req);
+
+} // namespace websocket

esp32/include/wifi_service.h

@@ -1,12 +1,12 @@
 #pragma once
 
-#include <PsychicHttp.h>
+#include <esp_http_server.h>
-#include <WiFi.h>
+#include <esp_wifi.h>
-#include <ESPmDNS.h>
 #include <string>
 
 #include <filesystem.h>
 #include <utils/timing.h>
+#include <utils/http_utils.h>
 #include <template/stateful_service.h>
 #include <template/stateful_persistence.h>
 #include <template/stateful_endpoint.h>
@@ -43,9 +43,9 @@ class WiFiService : public StatefulService<WiFiSettings> {
 
     const char *getHostname() { return state().hostname.c_str(); }
 
-    static esp_err_t handleScan(PsychicRequest *request);
+    static esp_err_t handleScan(httpd_req_t *req);
-    static esp_err_t getNetworks(PsychicRequest *request);
+    static esp_err_t getNetworks(httpd_req_t *req);
-    static esp_err_t getNetworkStatus(PsychicRequest *request);
+    static esp_err_t getNetworkStatus(httpd_req_t *req);
 
     StatefulHttpEndpoint<WiFiSettings> endpoint;
 };

esp32/include/www_mount.hpp

@@ -1,5 +1,5 @@
 #pragma once
-#include <PsychicHttp.h>
+#include <esp_http_server.h>
 #include "WWWData.h"
 
-void mountStaticAssets(PsychicHttpServer& s);
+void mountStaticAssets(httpd_handle_t server);

esp32/src/ap_service.cpp

@@ -1,4 +1,5 @@
 #include <ap_service.h>
+#include <WiFi.h>
 
 static const char *TAG = "APService";
 
@@ -12,11 +13,11 @@ APService::~APService() {}
 
 void APService::begin() { _persistence.readFromFS(); }
 
-esp_err_t APService::getStatus(PsychicRequest *request) {
+esp_err_t APService::getStatus(httpd_req_t *req) {
-    PsychicJsonResponse response = PsychicJsonResponse(request, false);
+    JsonDocument doc;
-    JsonObject root = response.getRoot();
+    JsonObject root = doc.to<JsonObject>();
     status(root);
-    return response.send();
+    return http_utils::send_json_response(req, doc);
 }
 
 void APService::status(JsonObject &root) {

esp32/src/communication/websocket_adapter.cpp

@@ -3,13 +3,25 @@
 
 static const char *TAG = "Websocket";
 
-Websocket::Websocket(PsychicHttpServer &server, const char *route) : _server(server), _route(route) {
+Websocket::Websocket(httpd_handle_t *server, const char *route) : _server(server), _route(route) {
-    _socket.onOpen((std::bind(&Websocket::onWSOpen, this, std::placeholders::_1)));
+    _socket.setOpenCallback([this](int fd) { this->onWSOpen(fd); });
-    _socket.onClose(std::bind(&Websocket::onWSClose, this, std::placeholders::_1));
+    _socket.setCloseCallback([this](int fd) { this->onWSClose(fd); });
-    _socket.onFrame(std::bind(&Websocket::onFrame, this, std::placeholders::_1, std::placeholders::_2));
+    _socket.setFrameCallback([this](httpd_req_t *req, httpd_ws_frame_t *frame) { return this->onFrame(req, frame); });
+
+    _ws_uri.uri = _route;
+    _ws_uri.method = HTTP_GET;
+    _ws_uri.handler = ws_handler_wrapper;
+    _ws_uri.user_ctx = this;
+    _ws_uri.is_websocket = true;
+    _ws_uri.handle_ws_control_frames = false;
 }
 
-void Websocket::begin() { _server.on(_route, &_socket); }
+void Websocket::begin() {
+    if (_server && *_server) {
+        _socket.setServer(*_server);
+        httpd_register_uri_handler(*_server, &_ws_uri);
+    }
+}
 
 void Websocket::onEvent(std::string event, EventCallback callback) {
     CommAdapterBase::onEvent(std::move(event), std::move(callback));
@@ -19,38 +31,44 @@ void Websocket::emit(const char *event, JsonVariant &payload, const char *origin
     CommAdapterBase::emit(event, payload, originId, onlyToSameOrigin);
 }
 
-void Websocket::onWSOpen(PsychicWebSocketClient *client) {
+void Websocket::onWSOpen(int fd) {
-    ESP_LOGI("EventSocket", "ws[%s][%u] connect", client->remoteIP().toString().c_str(), client->socket());
+    ESP_LOGI(TAG, "ws[%d] connect", fd);
-    ping(client->socket());
+    ping(fd);
 }
 
-void Websocket::onWSClose(PsychicWebSocketClient *client) {
+void Websocket::onWSClose(int fd) {
     xSemaphoreTake(mutex_, portMAX_DELAY);
     for (auto &event_subscriptions : client_subscriptions) {
-        event_subscriptions.second.remove(client->socket());
+        event_subscriptions.second.remove(fd);
     }
     xSemaphoreGive(mutex_);
-    ESP_LOGI("EventSocket", "ws[%s][%u] disconnect", client->remoteIP().toString().c_str(), client->socket());
+    ESP_LOGI(TAG, "ws[%d] disconnect", fd);
 }
 
-esp_err_t Websocket::onFrame(PsychicWebSocketRequest *request, httpd_ws_frame *frame) {
+esp_err_t Websocket::onFrame(httpd_req_t *req, httpd_ws_frame_t *frame) {
-    ESP_LOGV(TAG, "ws[%s][%u] opcode[%d]", request->client()->remoteIP().toString().c_str(),
+    int fd = httpd_req_to_sockfd(req);
-             request->client()->socket(), frame->type);
+    ESP_LOGV(TAG, "ws[%d] opcode[%d]", fd, frame->type);
 
     if (frame->type != HTTPD_WS_TYPE_TEXT && frame->type != HTTPD_WS_TYPE_BINARY) {
         ESP_LOGE(TAG, "Unsupported frame type: %d", frame->type);
         return ESP_OK;
     }
 
-#if USE_MSGPACK
+#if USE_PROTOBUF
     if (frame->type == HTTPD_WS_TYPE_BINARY) {
-        handleIncoming(frame->payload, frame->len, request->client()->socket());
+        handleIncoming(frame->payload, frame->len, fd);
+    } else {
+        ESP_LOGE(TAG, "Expected binary, got text");
+    }
+#elif USE_MSGPACK
+    if (frame->type == HTTPD_WS_TYPE_BINARY) {
+        handleIncoming(frame->payload, frame->len, fd);
     } else {
         ESP_LOGE(TAG, "Expected binary, got text");
     }
 #else
     if (frame->type == HTTPD_WS_TYPE_TEXT) {
-        handleIncoming(frame->payload, frame->len, request->client()->socket());
+        handleIncoming(frame->payload, frame->len, fd);
     } else {
         ESP_LOGE(TAG, "Expected text, got binary");
     }
@@ -60,22 +78,35 @@ esp_err_t Websocket::onFrame(PsychicWebSocketRequest *request, httpd_ws_frame *f
 }
 
 void Websocket::send(const uint8_t *data, size_t len, int cid) {
+    if (!_server || !*_server) {
+        ESP_LOGW(TAG, "Server not initialized, cannot send");
+        return;
+    }
+
+    httpd_ws_type_t type;
+#if USE_PROTOBUF || USE_MSGPACK
+    type = HTTPD_WS_TYPE_BINARY;
+#else
+    type = HTTPD_WS_TYPE_TEXT;
+#endif
+
     if (cid != -1) {
-        auto *client = _socket.getClient(cid);
+        ESP_LOGV(TAG, "Sending to client %d: %.*s", cid, (int)len, data);
-        if (client) {
+
-            ESP_LOGV(TAG, "Sending to client %s: %s", client->remoteIP().toString().c_str(), data);
+        httpd_ws_frame_t ws_pkt;
-#if USE_MSGPACK
+        memset(&ws_pkt, 0, sizeof(httpd_ws_frame_t));
-            client->sendMessage(HTTPD_WS_TYPE_BINARY, data, len);
+        ws_pkt.payload = (uint8_t *)data;
-#else
+        ws_pkt.len = len;
-            client->sendMessage(HTTPD_WS_TYPE_TEXT, data, len);
+        ws_pkt.type = type;
-#endif
+
-        }
+        httpd_ws_send_frame_async(*_server, cid, &ws_pkt);
     } else {
-        ESP_LOGV(TAG, "Sending to all clients: %s", data);
+        ESP_LOGV(TAG, "Sending to all clients: %.*s", (int)len, data);
-#if USE_MSGPACK
+        _socket.sendToAll(type, data, len);
-        _socket.sendAll(HTTPD_WS_TYPE_BINARY, data, len);
-#else
-        _socket.sendAll(HTTPD_WS_TYPE_TEXT, data, len);
-#endif
     }
 }
+
+esp_err_t Websocket::ws_handler_wrapper(httpd_req_t *req) {
+    Websocket *socket = (Websocket *)req->user_ctx;
+    return socket->_socket.handleWebSocket(req);
+}

esp32/src/features.cpp

@@ -1,4 +1,5 @@
 #include <features.h>
+#include <utils/http_utils.h>
 
 namespace feature_service {
 
@@ -45,11 +46,11 @@ void features(JsonObject &root) {
     root["variant"] = KINEMATICS_VARIANT_STR;
 }
 
-esp_err_t getFeatures(PsychicRequest *request) {
+esp_err_t getFeatures(httpd_req_t *req) {
-    PsychicJsonResponse response = PsychicJsonResponse(request, false);
+    JsonDocument doc;
-    JsonObject root = response.getRoot();
+    JsonObject root = doc.to<JsonObject>();
     features(root);
-    return response.send();
+    return http_utils::send_json_response(req, doc);
 }
 
 } // namespace feature_service

esp32/src/main.cpp

@@ -1,8 +1,6 @@
-#include <Arduino.h>
+#include <esp_http_server.h>
-#include <PsychicHttp.h>
+#include <esp_log.h>
-#include <ESPmDNS.h>
+#include <driver/uart.h>
-#include <WiFi.h>
-#include <Wire.h>
 
 #include <filesystem.h>
 #include <peripherals/peripherals.h>
@@ -16,14 +14,13 @@
 #include <ap_service.h>
 #include <mdns_service.h>
 #include <system_service.h>
+#include <utils/http_utils.h>
 
 #include <www_mount.hpp>
 
-// Communication
+httpd_handle_t server = NULL;
-PsychicHttpServer server;
+Websocket socket(&server, "/api/ws");
-Websocket socket {server, "/api/ws"};
 
-// Core
 Peripherals peripherals;
 ServoController servoController;
 MotionService motionService;
@@ -34,47 +31,9 @@ LEDService ledService;
 Camera::CameraService cameraService;
 #endif
 
-// Service
 WiFiService wifiService;
 APService apService;
 
-void setupServer() {
-    server.config.max_uri_handlers = 10 + WWW_ASSETS_COUNT;
-    server.maxUploadSize = 1000000; // 1 MB;
-    server.listen(80);
-    server.serveStatic("/api/config/", ESP_FS, "/config/");
-    server.on("/api/features", feature_service::getFeatures);
-#if USE_CAMERA
-    server.on("/api/camera/still", HTTP_GET,
-              [&](PsychicRequest *request) { return cameraService.cameraStill(request); });
-    server.on("/api/camera/stream", HTTP_GET,
-              [&](PsychicRequest *request) { return cameraService.cameraStream(request); });
-    server.on("/api/camera/settings", HTTP_GET,
-              [&](PsychicRequest *request) { return cameraService.endpoint.getState(request); });
-    server.on("/api/camera/settings", HTTP_POST, [&](PsychicRequest *request, JsonVariant &json) {
-        return cameraService.endpoint.handleStateUpdate(request, json);
-    });
-#endif
-    server.on("/api/servo/config", HTTP_GET,
-              [&](PsychicRequest *request) { return servoController.endpoint.getState(request); });
-    server.on("/api/servo/config", HTTP_POST, [&](PsychicRequest *request, JsonVariant &json) {
-        return servoController.endpoint.handleStateUpdate(request, json);
-    });
-#if EMBED_WEBAPP
-    mountStaticAssets(server);
-#endif
-    server.on("/*", HTTP_OPTIONS, [](PsychicRequest *request) { // CORS handling
-        PsychicResponse response(request);
-        response.setCode(200);
-        return response.send();
-    });
-    DefaultHeaders::Instance().addHeader("Server", APP_NAME);
-    DefaultHeaders::Instance().addHeader("Access-Control-Allow-Origin", "*");
-    DefaultHeaders::Instance().addHeader("Access-Control-Allow-Headers", "Accept, Content-Type, Authorization");
-    DefaultHeaders::Instance().addHeader("Access-Control-Allow-Methods", "GET, POST, OPTIONS");
-    DefaultHeaders::Instance().addHeader("Access-Control-Max-Age", "86400");
-}
-
 #define ANGLES_EVENT "angles"
 #define INPUT_EVENT "input"
 #define MODE_EVENT "mode"
@@ -83,8 +42,103 @@ void setupServer() {
 #define EVENT_SERVO_CONFIGURATION_SETTINGS "servoPWM"
 #define EVENT_SERVO_STATE "servoState"
 
+esp_err_t cors_options_handler(httpd_req_t *req) { return http_utils::handle_options_cors(req); }
+
+esp_err_t servo_config_get_handler(httpd_req_t *req) { return servoController.endpoint.getState(req); }
+
+esp_err_t servo_config_post_handler(httpd_req_t *req) {
+    JsonDocument doc;
+    if (http_utils::parse_json_body(req, doc) != ESP_OK) {
+        return http_utils::send_error(req, 400, "Invalid JSON");
+    }
+    JsonVariant json = doc.as<JsonVariant>();
+    return servoController.endpoint.handleStateUpdate(req, json);
+}
+
+#if USE_CAMERA
+esp_err_t camera_still_handler(httpd_req_t *req) { return cameraService.cameraStill(req); }
+
+esp_err_t camera_stream_handler(httpd_req_t *req) { return cameraService.cameraStream(req); }
+
+esp_err_t camera_settings_get_handler(httpd_req_t *req) { return cameraService.endpoint.getState(req); }
+
+esp_err_t camera_settings_post_handler(httpd_req_t *req) {
+    JsonDocument doc;
+    if (http_utils::parse_json_body(req, doc) != ESP_OK) {
+        return http_utils::send_error(req, 400, "Invalid JSON");
+    }
+    JsonVariant json = doc.as<JsonVariant>();
+    return cameraService.endpoint.handleStateUpdate(req, json);
+}
+#endif
+
+void setupServer() {
+    httpd_config_t config = HTTPD_DEFAULT_CONFIG();
+    config.max_uri_handlers = 20 + WWW_ASSETS_COUNT;
+    config.stack_size = 8192;
+    config.max_open_sockets = 7;
+    config.lru_purge_enable = true;
+
+    if (httpd_start(&server, &config) == ESP_OK) {
+        ESP_LOGI("main", "HTTP server started");
+
+        httpd_uri_t config_static = {.uri = "/api/config/*",
+                                     .method = HTTP_GET,
+                                     .handler = [](httpd_req_t *req) -> esp_err_t {
+                                         return http_utils::send_error(req, 501,
+                                                                       "Static file serving not yet implemented");
+                                     },
+                                     .user_ctx = nullptr};
+        httpd_register_uri_handler(server, &config_static);
+
+        httpd_uri_t features_uri = {
+            .uri = "/api/features", .method = HTTP_GET, .handler = feature_service::getFeatures, .user_ctx = nullptr};
+        httpd_register_uri_handler(server, &features_uri);
+
+#if USE_CAMERA
+        httpd_uri_t camera_still_uri = {
+            .uri = "/api/camera/still", .method = HTTP_GET, .handler = camera_still_handler, .user_ctx = nullptr};
+        httpd_register_uri_handler(server, &camera_still_uri);
+
+        httpd_uri_t camera_stream_uri = {
+            .uri = "/api/camera/stream", .method = HTTP_GET, .handler = camera_stream_handler, .user_ctx = nullptr};
+        httpd_register_uri_handler(server, &camera_stream_uri);
+
+        httpd_uri_t camera_settings_get_uri = {.uri = "/api/camera/settings",
+                                               .method = HTTP_GET,
+                                               .handler = camera_settings_get_handler,
+                                               .user_ctx = nullptr};
+        httpd_register_uri_handler(server, &camera_settings_get_uri);
+
+        httpd_uri_t camera_settings_post_uri = {.uri = "/api/camera/settings",
+                                                .method = HTTP_POST,
+                                                .handler = camera_settings_post_handler,
+                                                .user_ctx = nullptr};
+        httpd_register_uri_handler(server, &camera_settings_post_uri);
+#endif
+
+        httpd_uri_t servo_config_get_uri = {
+            .uri = "/api/servo/config", .method = HTTP_GET, .handler = servo_config_get_handler, .user_ctx = nullptr};
+        httpd_register_uri_handler(server, &servo_config_get_uri);
+
+        httpd_uri_t servo_config_post_uri = {
+            .uri = "/api/servo/config", .method = HTTP_POST, .handler = servo_config_post_handler, .user_ctx = nullptr};
+        httpd_register_uri_handler(server, &servo_config_post_uri);
+
+#if EMBED_WEBAPP
+        mountStaticAssets(server);
+#endif
+
+        httpd_uri_t options_uri = {
+            .uri = "/*", .method = HTTP_OPTIONS, .handler = cors_options_handler, .user_ctx = nullptr};
+        httpd_register_uri_handler(server, &options_uri);
+
+    } else {
+        ESP_LOGE("main", "Failed to start HTTP server");
+    }
+}
+
 void setupEventSocket() {
-    // Motion events
     socket.onEvent(INPUT_EVENT, [&](JsonVariant &root, int originId) { motionService.handleInput(root, originId); });
 
     socket.onEvent(MODE_EVENT, [&](JsonVariant &root, int originId) {
@@ -98,7 +152,6 @@ void setupEventSocket() {
 
     socket.onEvent(ANGLES_EVENT, [&](JsonVariant &root, int originId) { motionService.anglesEvent(root, originId); });
 
-    // Peripherals events
     socket.onEvent(EVENT_I2C_SCAN, [&](JsonVariant &root, int originId) {
         peripherals.scanI2C();
         JsonDocument doc;
@@ -107,15 +160,15 @@ void setupEventSocket() {
         socket.emit(EVENT_I2C_SCAN, results);
     });
 
-    // Servo controller events
     socket.onEvent(EVENT_SERVO_CONFIGURATION_SETTINGS,
                    [&](JsonVariant &root, int originId) { servoController.servoEvent(root, originId); });
+
     socket.onEvent(EVENT_SERVO_STATE,
                    [&](JsonVariant &root, int originId) { servoController.stateUpdate(root, originId); });
 }
 
 void IRAM_ATTR SpotControlLoopEntry(void *) {
-    ESP_LOGI("main", "Setup complete now runing tsk");
+    ESP_LOGI("main", "Setup complete now running control task");
     TickType_t xLastWakeTime = xTaskGetTickCount();
     const TickType_t xFrequency = 5 / portTICK_PERIOD_MS;
 
@@ -140,8 +193,7 @@ void IRAM_ATTR serviceLoopEntry(void *) {
     ESP_LOGI("main", "Service control task starting");
 
     wifiService.begin();
-    MDNS.begin(APP_NAME);
+    mdns_service::begin(APP_NAME);
-    MDNS.setInstanceName(APP_NAME);
     apService.begin();
 
     setupServer();
@@ -159,8 +211,9 @@ void IRAM_ATTR serviceLoopEntry(void *) {
     }
 }
 
-void setup() {
+extern "C" void app_main() {
-    Serial.begin(115200);
+    uart_driver_install(UART_NUM_0, 256, 0, 0, NULL, 0);
+    esp_log_level_set("*", ESP_LOG_INFO);
 
     ESP_FS.begin();
 
@@ -168,11 +221,9 @@ void setup() {
 
     feature_service::printFeatureConfiguration();
 
-    xTaskCreate(serviceLoopEntry, "Service task", 4096, nullptr, 2, nullptr);
+    xTaskCreate(serviceLoopEntry, "Service task", 8192, nullptr, 2, nullptr);
 
-    xTaskCreatePinnedToCore(SpotControlLoopEntry, "Control task", 4096, nullptr, 5, nullptr, 1);
+    xTaskCreatePinnedToCore(SpotControlLoopEntry, "Control task", 8192, nullptr, 5, nullptr, 1);
 
     ESP_LOGI("main", "Finished booting");
 }
-
-void loop() { vTaskDelete(nullptr); }

esp32/src/mdns_service.cpp

@@ -2,99 +2,54 @@
 
 static const char *TAG = "MDNSService";
 
-MDNSService::MDNSService()
+namespace mdns_service {
-    : _persistence(MDNSSettings::read, MDNSSettings::update, this, MDNS_SETTINGS_FILE),
-      endpoint(MDNSSettings::read, MDNSSettings::update, this) {
-    addUpdateHandler([&](const std::string &originId) { reconfigureMDNS(); }, false);
-}
 
-MDNSService::~MDNSService() {
+void begin(const char *hostname) {
-    if (_started) {
+    ESP_LOGI(TAG, "Starting mDNS with hostname: %s", hostname);
-        stopMDNS();
+
+    esp_err_t err = mdns_init();
+    if (err != ESP_OK) {
+        ESP_LOGE(TAG, "mDNS init failed: %d", err);
+        return;
     }
-}
 
-void MDNSService::begin() {
+    err = mdns_hostname_set(hostname);
-    _persistence.readFromFS();
+    if (err != ESP_OK) {
-    startMDNS();
+        ESP_LOGE(TAG, "Failed to set hostname: %d", err);
-}
+        return;
-
-void MDNSService::reconfigureMDNS() {
-    if (_started) {
-        stopMDNS();
     }
-    startMDNS();
+
+    err = mdns_instance_name_set(hostname);
+    if (err != ESP_OK) {
+        ESP_LOGE(TAG, "Failed to set instance name: %d", err);
+        return;
+    }
+
+    addService("http", "_tcp", 80);
+    addService("ws", "_tcp", 80);
+    addServiceTxt("http", "_tcp", "version", APP_VERSION);
+
+    ESP_LOGI(TAG, "mDNS started successfully");
 }
 
-void MDNSService::startMDNS() {
+void end() { mdns_free(); }
-    ESP_LOGV(TAG, "Starting MDNS with hostname: %s", state().hostname.c_str());
 
-    if (MDNS.begin(state().hostname.c_str())) {
+void addService(const char *service, const char *proto, uint16_t port) {
-        _started = true;
+    esp_err_t err = mdns_service_add(NULL, service, proto, port, NULL, 0);
-        MDNS.setInstanceName(state().instance.c_str());
+    if (err != ESP_OK) {
-
+        ESP_LOGE(TAG, "Failed to add service %s.%s: %d", service, proto, err);
-        addServices();
-
-        ESP_LOGI(TAG, "MDNS started successfully with hostname: %s", state().hostname.c_str());
     } else {
-        _started = false;
+        ESP_LOGI(TAG, "Added mDNS service: %s.%s on port %d", service, proto, port);
-        ESP_LOGE(TAG, "Failed to start MDNS");
     }
 }
 
-void MDNSService::stopMDNS() {
+void addServiceTxt(const char *service, const char *proto, const char *key, const char *value) {
-    ESP_LOGV(TAG, "Stopping MDNS");
+    mdns_txt_item_t txt_data[1] = {{(char *)key, (char *)value}};
-    MDNS.end();
-    _started = false;
-}
 
-void MDNSService::addServices() {
+    esp_err_t err = mdns_service_txt_set(service, proto, txt_data, 1);
-    for (const auto &service : state().services) {
+    if (err != ESP_OK) {
-        MDNS.addService(service.service.c_str(), service.protocol.c_str(), service.port);
+        ESP_LOGE(TAG, "Failed to set TXT record for %s.%s: %d", service, proto, err);
-
-        for (const auto &txt : service.txtRecords) {
-            MDNS.addServiceTxt(service.service.c_str(), service.protocol.c_str(), txt.key.c_str(), txt.value.c_str());
-        }
-    }
-
-    for (const auto &txt : state().globalTxtRecords) {
-        for (const auto &service : state().services) {
-            MDNS.addServiceTxt(service.service.c_str(), service.protocol.c_str(), txt.key.c_str(), txt.value.c_str());
-        }
     }
 }
 
-esp_err_t MDNSService::getStatus(PsychicRequest *request) {
+} // namespace mdns_service
-    PsychicJsonResponse response = PsychicJsonResponse(request, false);
-    JsonVariant root = response.getRoot();
-    getStatus(root);
-    return response.send();
-}
-
-void MDNSService::getStatus(JsonVariant &root) {
-    state().read(state(), root);
-    root["started"] = _started;
-}
-
-esp_err_t MDNSService::queryServices(PsychicRequest *request, JsonVariant &json) {
-    std::string service = json["service"].as<std::string>();
-    std::string proto = json["protocol"].as<std::string>();
-
-    PsychicJsonResponse response = PsychicJsonResponse(request, false);
-    JsonVariant root = response.getRoot();
-
-    ESP_LOGI(TAG, "Querying for service: %s, protocol: %s", service.c_str(), proto.c_str());
-
-    int n = MDNS.queryService(service.c_str(), proto.c_str());
-    ESP_LOGI(TAG, "Found %d services", n);
-
-    JsonArray servicesArray = root["services"].to<JsonArray>();
-    for (int i = 0; i < n; i++) {
-        JsonVariant serviceObj = servicesArray.add<JsonVariant>();
-        serviceObj["name"] = MDNS.hostname(i);
-        serviceObj["ip"] = MDNS.IP(i);
-        serviceObj["port"] = MDNS.port(i);
-    }
-
-    return response.send();
-}

esp32/src/peripherals/camera_service.cpp

@@ -85,44 +85,40 @@ esp_err_t CameraService::begin() {
     return err;
 }
 
-esp_err_t CameraService::cameraStill(PsychicRequest *request) {
+esp_err_t CameraService::cameraStill(httpd_req_t *req) {
     camera_fb_t *fb = safe_camera_fb_get();
     if (!fb) {
         ESP_LOGE(TAG, "Camera capture failed");
-        request->reply(500, "text/plain", "Camera capture failed");
+        return http_utils::send_error(req, 500, "Camera capture failed");
-        return ESP_FAIL;
     }
-    PsychicStreamResponse response = PsychicStreamResponse(request, "image/jpeg", "capture.jpg");
+
-    response.beginSend();
+    httpd_resp_set_type(req, "image/jpeg");
-    response.write(fb->buf, fb->len);
+    httpd_resp_set_hdr(req, "Content-Disposition", "inline; filename=capture.jpg");
+
+    esp_err_t res = httpd_resp_send(req, (const char *)fb->buf, fb->len);
     esp_camera_fb_return(fb);
-    return response.endSend();
+    return res;
 }
 
 void streamTask(void *pv) {
     esp_err_t res = ESP_OK;
 
-    PsychicRequest *request = static_cast<PsychicRequest *>(pv);
+    httpd_req_t *req = static_cast<httpd_req_t *>(pv);
 
     httpd_req_t *copy = nullptr;
-    res = httpd_req_async_handler_begin(request->request(), &copy);
+    res = httpd_req_async_handler_begin(req, &copy);
     if (res != ESP_OK) {
         return;
     }
-    PsychicHttpServer *server = request->server();
-    PsychicRequest new_request = PsychicRequest(server, copy);
-    request = &new_request;
 
-    PsychicStreamResponse response = PsychicStreamResponse(request, _STREAM_CONTENT_TYPE);
+    httpd_resp_set_type(copy, _STREAM_CONTENT_TYPE);
     camera_fb_t *fb = NULL;
 
-    char *part_buf[64];
+    char part_buf[64];
     size_t buf_len = 0;
     uint8_t *buf = NULL;
     int64_t fr_start = esp_timer_get_time();
 
-    response.beginSend();
-
     for (;;) {
         fb = safe_camera_fb_get();
         if (!fb) {
@@ -136,26 +132,28 @@ void streamTask(void *pv) {
             buf = fb->buf;
         }
 
-        size_t hlen = snprintf((char *)part_buf, 64, _STREAM_PART, buf_len);
+        size_t hlen = snprintf(part_buf, 64, _STREAM_PART, buf_len);
-        size_t w = response.write((const char *)part_buf, hlen);
+        if (httpd_resp_send_chunk(copy, part_buf, hlen) != ESP_OK) break;
-        w += response.write((const char *)buf, buf_len);
+        if (httpd_resp_send_chunk(copy, (const char *)buf, buf_len) != ESP_OK) break;
-        w += response.write((char *)_STREAM_BOUNDARY, strlen(_STREAM_BOUNDARY));
+        if (httpd_resp_send_chunk(copy, _STREAM_BOUNDARY, strlen(_STREAM_BOUNDARY)) != ESP_OK) break;
-        if (w == 62) break;
+
         esp_camera_fb_return(fb);
         safe_sensor_return();
         buf = NULL;
         taskYIELD();
-        int64_t delay = 30000ll - esp_timer_get_time() - fr_start;
+        int64_t delay = 30000ll - (esp_timer_get_time() - fr_start);
-        if (delay > 0) vTaskDelay(pdMS_TO_TICKS(delay));
+        if (delay > 0) vTaskDelay(pdMS_TO_TICKS(delay / 1000));
+        fr_start = esp_timer_get_time();
     }
+
     ESP_LOGI("Stream", "Stream ended");
-    response.endSend();
+    httpd_resp_send_chunk(copy, nullptr, 0);
     httpd_req_async_handler_complete(copy);
     vTaskDelete(NULL);
 }
 
-esp_err_t CameraService::cameraStream(PsychicRequest *request) {
+esp_err_t CameraService::cameraStream(httpd_req_t *req) {
-    xTaskCreate(streamTask, "Stream client task", 4096, request, 4, nullptr);
+    xTaskCreate(streamTask, "Stream client task", 4096, req, 4, nullptr);
     vTaskDelay(pdMS_TO_TICKS(100));
     return ESP_OK;
 }

esp32/src/system_service.cpp

@@ -4,33 +4,33 @@ namespace system_service {
 
 static const char *TAG = "SystemService";
 
-esp_err_t handleReset(PsychicRequest *request) {
+esp_err_t handleReset(httpd_req_t *req) {
     reset();
-    return request->reply(200);
+    return http_utils::send_empty_response(req, 200);
 }
 
-esp_err_t handleRestart(PsychicRequest *request) {
+esp_err_t handleRestart(httpd_req_t *req) {
     restart();
-    return request->reply(200);
+    return http_utils::send_empty_response(req, 200);
 }
 
-esp_err_t handleSleep(PsychicRequest *request) {
+esp_err_t handleSleep(httpd_req_t *req) {
     sleep();
-    return request->reply(200);
+    return http_utils::send_empty_response(req, 200);
 }
 
-esp_err_t getStatus(PsychicRequest *request) {
+esp_err_t getStatus(httpd_req_t *req) {
-    PsychicJsonResponse response = PsychicJsonResponse(request, false);
+    JsonDocument doc;
-    JsonObject root = response.getRoot();
+    JsonObject root = doc.to<JsonObject>();
     status(root);
-    return response.send();
+    return http_utils::send_json_response(req, doc);
 }
 
-esp_err_t getMetrics(PsychicRequest *request) {
+esp_err_t getMetrics(httpd_req_t *req) {
-    PsychicJsonResponse response = PsychicJsonResponse(request, false);
+    JsonDocument doc;
-    JsonObject root = response.getRoot();
+    JsonObject root = doc.to<JsonObject>();
     metrics(root);
-    return response.send();
+    return http_utils::send_json_response(req, doc);
 }
 
 void reset() {
@@ -49,12 +49,8 @@ void restart() {
     xTaskCreate(
         [](void *pvParameters) {
             for (;;) {
-                vTaskDelay(250 / portTICK_PERIOD_MS);
+                vTaskDelay(1000 / portTICK_PERIOD_MS);
-                MDNS.end();
+                esp_restart();
-                vTaskDelay(100 / portTICK_PERIOD_MS);
-                WiFi.disconnect(true);
-                vTaskDelay(500 / portTICK_PERIOD_MS);
-                ESP.restart();
             }
         },
         "Restart task", 4096, nullptr, 10, nullptr);
@@ -64,11 +60,7 @@ void sleep() {
     xTaskCreate(
         [](void *pvParameters) {
             for (;;) {
-                vTaskDelay(250 / portTICK_PERIOD_MS);
+                vTaskDelay(1000 / portTICK_PERIOD_MS);
-                MDNS.end();
-                vTaskDelay(100 / portTICK_PERIOD_MS);
-                WiFi.disconnect(true);
-                vTaskDelay(500 / portTICK_PERIOD_MS);
 
                 uint64_t bitmask = (uint64_t)1 << (WAKEUP_PIN_NUMBER);
 

esp32/src/utils/http_utils.cpp

@@ -0,0 +1,138 @@
+#include <utils/http_utils.h>
+#include <global.h>
+#include <string.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+
+namespace http_utils {
+
+static const char *TAG = "HttpUtils";
+
+esp_err_t send_json_response(httpd_req_t *req, JsonDocument &doc, int status_code) {
+    add_standard_headers(req);
+    add_cors_headers(req);
+
+    httpd_resp_set_type(req, "application/json");
+
+    if (status_code != 200) {
+        char status_str[4];
+        snprintf(status_str, sizeof(status_str), "%d", status_code);
+        httpd_resp_set_status(req, status_str);
+    }
+
+    String json_str;
+    serializeJson(doc, json_str);
+
+    return httpd_resp_send(req, json_str.c_str(), json_str.length());
+}
+
+esp_err_t send_error(httpd_req_t *req, int status_code, const char *message) {
+    add_standard_headers(req);
+    add_cors_headers(req);
+
+    char status_str[4];
+    snprintf(status_str, sizeof(status_str), "%d", status_code);
+    httpd_resp_set_status(req, status_str);
+
+    if (message) {
+        httpd_resp_set_type(req, "text/plain");
+        return httpd_resp_send(req, message, strlen(message));
+    }
+
+    return httpd_resp_send(req, nullptr, 0);
+}
+
+esp_err_t send_empty_response(httpd_req_t *req, int status_code) {
+    add_standard_headers(req);
+    add_cors_headers(req);
+
+    if (status_code != 200) {
+        char status_str[4];
+        snprintf(status_str, sizeof(status_str), "%d", status_code);
+        httpd_resp_set_status(req, status_str);
+    }
+
+    return httpd_resp_send(req, nullptr, 0);
+}
+
+esp_err_t add_cors_headers(httpd_req_t *req) {
+    httpd_resp_set_hdr(req, "Access-Control-Allow-Origin", "*");
+    httpd_resp_set_hdr(req, "Access-Control-Allow-Headers", "Accept, Content-Type, Authorization");
+    httpd_resp_set_hdr(req, "Access-Control-Allow-Methods", "GET, POST, OPTIONS");
+    httpd_resp_set_hdr(req, "Access-Control-Max-Age", "86400");
+    return ESP_OK;
+}
+
+esp_err_t add_standard_headers(httpd_req_t *req) {
+    httpd_resp_set_hdr(req, "Server", APP_NAME);
+    return ESP_OK;
+}
+
+esp_err_t parse_json_body(httpd_req_t *req, JsonDocument &doc) {
+    size_t content_len = req->content_len;
+
+    if (content_len == 0) {
+        ESP_LOGW(TAG, "Empty request body");
+        return ESP_FAIL;
+    }
+
+    if (content_len > 16384) {
+        ESP_LOGE(TAG, "Request body too large: %d bytes", content_len);
+        return ESP_FAIL;
+    }
+
+    char *buf = (char *)malloc(content_len + 1);
+    if (!buf) {
+        ESP_LOGE(TAG, "Failed to allocate memory for request body");
+        return ESP_FAIL;
+    }
+
+    int ret = httpd_req_recv(req, buf, content_len);
+    if (ret <= 0) {
+        free(buf);
+        if (ret == HTTPD_SOCK_ERR_TIMEOUT) {
+            ESP_LOGE(TAG, "Socket timeout");
+            return ESP_ERR_TIMEOUT;
+        }
+        return ESP_FAIL;
+    }
+
+    buf[ret] = '\0';
+
+    DeserializationError error = deserializeJson(doc, buf, ret);
+    free(buf);
+
+    if (error) {
+        ESP_LOGE(TAG, "JSON parse error: %s", error.c_str());
+        return ESP_FAIL;
+    }
+
+    return ESP_OK;
+}
+
+esp_err_t handle_options_cors(httpd_req_t *req) { return send_empty_response(req, 200); }
+
+const char *get_client_ip(httpd_req_t *req) {
+    int sockfd = httpd_req_to_sockfd(req);
+    struct sockaddr_in6 addr;
+    socklen_t addr_size = sizeof(addr);
+
+    if (getpeername(sockfd, (struct sockaddr *)&addr, &addr_size) < 0) {
+        return "unknown";
+    }
+
+    static char ip_str[INET6_ADDRSTRLEN];
+    if (addr.sin6_family == AF_INET) {
+        struct sockaddr_in *addr_in = (struct sockaddr_in *)&addr;
+        snprintf(ip_str, sizeof(ip_str), "%d.%d.%d.%d", (addr_in->sin_addr.s_addr >> 0) & 0xFF,
+                 (addr_in->sin_addr.s_addr >> 8) & 0xFF, (addr_in->sin_addr.s_addr >> 16) & 0xFF,
+                 (addr_in->sin_addr.s_addr >> 24) & 0xFF);
+    } else {
+        inet_ntop(AF_INET6, &addr.sin6_addr, ip_str, sizeof(ip_str));
+    }
+
+    return ip_str;
+}
+
+} // namespace http_utils

esp32/src/utils/websocket_server.cpp

@@ -0,0 +1,152 @@
+#include <utils/websocket_server.h>
+#include <esp_log.h>
+#include <string.h>
+
+namespace websocket {
+
+static const char *TAG = "WebSocketServer";
+
+WebSocketServer::WebSocketServer() : _server(nullptr) { _mutex = xSemaphoreCreateMutex(); }
+
+WebSocketServer::~WebSocketServer() {
+    if (_mutex) {
+        vSemaphoreDelete(_mutex);
+    }
+}
+
+void WebSocketServer::setOpenCallback(ClientCallback callback) { _onOpen = callback; }
+
+void WebSocketServer::setCloseCallback(ClientCallback callback) { _onClose = callback; }
+
+void WebSocketServer::setFrameCallback(FrameCallback callback) { _onFrame = callback; }
+
+esp_err_t WebSocketServer::handleWebSocket(httpd_req_t *req) {
+    if (req->method == HTTP_GET) {
+        int fd = httpd_req_to_sockfd(req);
+        addClient(fd);
+        ESP_LOGI(TAG, "WebSocket client connected: fd=%d", fd);
+
+        if (_onOpen) {
+            _onOpen(fd);
+        }
+
+        return ESP_OK;
+    }
+
+    httpd_ws_frame_t ws_pkt;
+    memset(&ws_pkt, 0, sizeof(httpd_ws_frame_t));
+
+    esp_err_t ret = httpd_ws_recv_frame(req, &ws_pkt, 0);
+    if (ret != ESP_OK) {
+        ESP_LOGE(TAG, "httpd_ws_recv_frame failed to get frame len with %d", ret);
+        return ret;
+    }
+
+    if (ws_pkt.len) {
+        ws_pkt.payload = (uint8_t *)malloc(ws_pkt.len + 1);
+        if (!ws_pkt.payload) {
+            ESP_LOGE(TAG, "Failed to allocate memory for WebSocket payload");
+            return ESP_ERR_NO_MEM;
+        }
+
+        ret = httpd_ws_recv_frame(req, &ws_pkt, ws_pkt.len);
+        if (ret != ESP_OK) {
+            ESP_LOGE(TAG, "httpd_ws_recv_frame failed with %d", ret);
+            free(ws_pkt.payload);
+            return ret;
+        }
+
+        ((uint8_t *)ws_pkt.payload)[ws_pkt.len] = '\0';
+    }
+
+    if (ws_pkt.type == HTTPD_WS_TYPE_CLOSE) {
+        int fd = httpd_req_to_sockfd(req);
+        ESP_LOGI(TAG, "WebSocket client disconnected: fd=%d", fd);
+        removeClient(fd);
+
+        if (_onClose) {
+            _onClose(fd);
+        }
+
+        if (ws_pkt.payload) {
+            free(ws_pkt.payload);
+        }
+        return ESP_OK;
+    }
+
+    if (_onFrame) {
+        ret = _onFrame(req, &ws_pkt);
+    }
+
+    if (ws_pkt.payload) {
+        free(ws_pkt.payload);
+    }
+
+    return ret;
+}
+
+void WebSocketServer::addClient(int fd) {
+    xSemaphoreTake(_mutex, portMAX_DELAY);
+    WebSocketClient client;
+    client.fd = fd;
+    client.last_seen = esp_timer_get_time();
+    _clients[fd] = client;
+    xSemaphoreGive(_mutex);
+}
+
+void WebSocketServer::removeClient(int fd) {
+    xSemaphoreTake(_mutex, portMAX_DELAY);
+    _clients.erase(fd);
+    xSemaphoreGive(_mutex);
+}
+
+WebSocketClient *WebSocketServer::getClient(int fd) {
+    xSemaphoreTake(_mutex, portMAX_DELAY);
+    auto it = _clients.find(fd);
+    WebSocketClient *client = (it != _clients.end()) ? &it->second : nullptr;
+    xSemaphoreGive(_mutex);
+    return client;
+}
+
+esp_err_t WebSocketServer::sendText(int fd, const char *data, size_t len) {
+    return sendBinary(fd, (const uint8_t *)data, len);
+}
+
+esp_err_t WebSocketServer::sendBinary(int fd, const uint8_t *data, size_t len) {
+    if (!_server) {
+        ESP_LOGE(TAG, "Server handle not set");
+        return ESP_FAIL;
+    }
+
+    httpd_ws_frame_t ws_pkt;
+    memset(&ws_pkt, 0, sizeof(httpd_ws_frame_t));
+    ws_pkt.payload = (uint8_t *)data;
+    ws_pkt.len = len;
+    ws_pkt.type = HTTPD_WS_TYPE_TEXT;
+
+    return httpd_ws_send_frame_async(_server, fd, &ws_pkt);
+}
+
+esp_err_t WebSocketServer::sendToAll(httpd_ws_type_t type, const uint8_t *data, size_t len) {
+    if (!_server) {
+        ESP_LOGE(TAG, "Server handle not set");
+        return ESP_FAIL;
+    }
+
+    httpd_ws_frame_t ws_pkt;
+    memset(&ws_pkt, 0, sizeof(httpd_ws_frame_t));
+    ws_pkt.payload = (uint8_t *)data;
+    ws_pkt.len = len;
+    ws_pkt.type = type;
+
+    xSemaphoreTake(_mutex, portMAX_DELAY);
+    for (auto &client_pair : _clients) {
+        int fd = client_pair.first;
+        httpd_ws_send_frame_async(_server, fd, &ws_pkt);
+    }
+    xSemaphoreGive(_mutex);
+
+    return ESP_OK;
+}
+
+} // namespace websocket

esp32/src/wifi_service.cpp

@@ -1,4 +1,6 @@
 #include <wifi_service.h>
+#include <WiFi.h>
+#include <ESPmDNS.h>
 
 WiFiService::WiFiService()
     : _persistence(WiFiSettings::read, WiFiSettings::update, this, WIFI_SETTINGS_FILE),
@@ -38,25 +40,25 @@ void WiFiService::reconfigureWiFiConnection() {
 
 void WiFiService::loop() { EXECUTE_EVERY_N_MS(reconnectDelay, manageSTA()); }
 
-esp_err_t WiFiService::handleScan(PsychicRequest *request) {
+esp_err_t WiFiService::handleScan(httpd_req_t *req) {
     if (WiFi.scanComplete() != -1) {
         WiFi.scanDelete();
         WiFi.scanNetworks(true);
     }
-    return request->reply(202);
+    return http_utils::send_empty_response(req, 202);
 }
 
-esp_err_t WiFiService::getNetworks(PsychicRequest *request) {
+esp_err_t WiFiService::getNetworks(httpd_req_t *req) {
     int numNetworks = WiFi.scanComplete();
     if (numNetworks == -1)
-        return request->reply(202);
+        return http_utils::send_empty_response(req, 202);
     else if (numNetworks < -1)
-        return handleScan(request);
+        return handleScan(req);
 
-    PsychicJsonResponse response = PsychicJsonResponse(request, false);
+    JsonDocument doc;
-    JsonObject root = response.getRoot();
+    JsonObject root = doc.to<JsonObject>();
     getNetworks(root);
-    return response.send();
+    return http_utils::send_json_response(req, doc);
 }
 
 void WiFiService::setupMDNS(const char *hostname) {
@@ -80,11 +82,11 @@ void WiFiService::getNetworks(JsonObject &root) {
     }
 }
 
-esp_err_t WiFiService::getNetworkStatus(PsychicRequest *request) {
+esp_err_t WiFiService::getNetworkStatus(httpd_req_t *req) {
-    PsychicJsonResponse response = PsychicJsonResponse(request, false);
+    JsonDocument doc;
-    JsonObject root = response.getRoot();
+    JsonObject root = doc.to<JsonObject>();
     getNetworkStatus(root);
-    return response.send();
+    return http_utils::send_json_response(req, doc);
 }
 
 void WiFiService::getNetworkStatus(JsonObject &root) {

esp32/src/www_mount.cpp

@@ -1,34 +1,70 @@
 #include "www_mount.hpp"
+#include <esp_log.h>
+#include <string.h>
+
+static const char *TAG = "WWWMount";
+
+struct asset_handler_ctx {
+    const WebAsset *asset;
+};
+
+static esp_err_t web_send(httpd_req_t *req, const WebAsset &asset) {
+    httpd_resp_set_status(req, "200 OK");
+    httpd_resp_set_type(req, asset.mime);
+
+    if (asset.gz) {
+        httpd_resp_set_hdr(req, "Content-Encoding", "gzip");
+    }
+
+    if (WWW_OPT.add_vary) {
+        httpd_resp_set_hdr(req, "Vary", "Accept-Encoding");
+    }
 
-static esp_err_t web_send(PsychicRequest* req, const WebAsset& asset) {
-    PsychicResponse resp(req);
-    resp.setCode(200);
-    resp.setContentType(asset.mime);
-    if (asset.gz) resp.addHeader("Content-Encoding", "gzip");
-    if (WWW_OPT.add_vary) resp.addHeader("Vary", "Accept-Encoding");
     char cc[64];
     snprintf(cc, sizeof(cc), "public, immutable, max-age=%u", WWW_OPT.max_age);
-    resp.addHeader("Cache-Control", cc);
+    httpd_resp_set_hdr(req, "Cache-Control", cc);
+
     char et[34];
     snprintf(et, sizeof(et), "\"%08x\"", asset.etag);
-    resp.addHeader("ETag", et);
+    httpd_resp_set_hdr(req, "ETag", et);
-    resp.setContent(asset.data, asset.len);
+
-    return resp.send();
+    return httpd_resp_send(req, (const char *)asset.data, asset.len);
 }
 
-void mountStaticAssets(PsychicHttpServer& server) {
+static esp_err_t asset_handler(httpd_req_t *req) {
-    static uint8_t buf[sizeof(PsychicWebHandler) * WWW_ASSETS_COUNT];
+    asset_handler_ctx *ctx = (asset_handler_ctx *)req->user_ctx;
-    for (size_t i = 0; i < WWW_ASSETS_COUNT; i++) {
+    return web_send(req, *ctx->asset);
-        const WebAsset* a = &WWW_ASSETS[i];
+}
-        auto* handle = new (&buf[i * sizeof(PsychicWebHandler)]) PsychicWebHandler();
+
-        handle->onRequest([a](PsychicRequest* req) { return web_send(req, *a); });
+static esp_err_t default_handler(httpd_req_t *req) {
-        server.on(a->uri, HTTP_GET, handle);
-    }
     for (size_t i = 0; i < WWW_ASSETS_COUNT; i++) {
         if (strcmp(WWW_ASSETS[i].uri, WWW_OPT.default_uri) == 0) {
-            server.defaultEndpoint->setHandler(
+            return web_send(req, WWW_ASSETS[i]);
-                reinterpret_cast<PsychicWebHandler*>(&buf[i * sizeof(PsychicWebHandler)]));
-            break;
         }
     }
-}
+    httpd_resp_send_404(req);
+    return ESP_OK;
+}
+
+void mountStaticAssets(httpd_handle_t server) {
+    static asset_handler_ctx contexts[WWW_ASSETS_COUNT];
+
+    for (size_t i = 0; i < WWW_ASSETS_COUNT; i++) {
+        contexts[i].asset = &WWW_ASSETS[i];
+
+        httpd_uri_t uri_handler = {
+            .uri = WWW_ASSETS[i].uri, .method = HTTP_GET, .handler = asset_handler, .user_ctx = &contexts[i]};
+
+        esp_err_t err = httpd_register_uri_handler(server, &uri_handler);
+        if (err != ESP_OK) {
+            ESP_LOGE(TAG, "Failed to register handler for %s: %d", WWW_ASSETS[i].uri, err);
+        } else {
+            ESP_LOGD(TAG, "Registered handler for %s", WWW_ASSETS[i].uri);
+        }
+    }
+
+    httpd_uri_t default_uri_handler = {.uri = "/", .method = HTTP_GET, .handler = default_handler, .user_ctx = nullptr};
+    httpd_register_uri_handler(server, &default_uri_handler);
+
+    ESP_LOGI(TAG, "Mounted %d static assets", WWW_ASSETS_COUNT);
+}

platformio.ini

@@ -105,7 +105,6 @@ build_src_flags =
 test_ignore = test_embedded
 board_build.filesystem = littlefs
 lib_deps = 
-	hoeken/PsychicHttp@^1.2.1
 	ArduinoJson@>=7.0.0
 	teckel12/NewPing@^1.9.7
 	jrowberg/I2Cdevlib-MPU6050@^1.0.0