mirror of
https://github.com/oxan/esphome-stream-server.git
synced 2026-05-16 22:37:26 +00:00
69 lines
2.1 KiB
C++
69 lines
2.1 KiB
C++
/* Copyright (C) 2020-2021 Oxan van Leeuwen
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/components/uart/uart.h"
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#ifdef ARDUINO_ARCH_ESP8266
|
|
#include <ESPAsyncTCP.h>
|
|
#else
|
|
// AsyncTCP.h includes parts of freertos, which require FreeRTOS.h header to be included first
|
|
#include <freertos/FreeRTOS.h>
|
|
#include <AsyncTCP.h>
|
|
#endif
|
|
|
|
class StreamServerComponent : public esphome::Component {
|
|
public:
|
|
StreamServerComponent() = default;
|
|
explicit StreamServerComponent(esphome::uart::UARTComponent *stream) : stream_{stream} {}
|
|
void set_uart_parent(esphome::uart::UARTComponent *parent) { this->stream_ = parent; }
|
|
|
|
void setup() override;
|
|
void loop() override;
|
|
void dump_config() override;
|
|
void on_shutdown() override;
|
|
|
|
float get_setup_priority() const override { return esphome::setup_priority::AFTER_WIFI; }
|
|
|
|
void set_port(uint16_t port) { this->port_ = port; }
|
|
|
|
protected:
|
|
void cleanup();
|
|
void read();
|
|
void write();
|
|
|
|
struct Client {
|
|
Client(AsyncClient *client, std::vector<uint8_t> &recv_buf);
|
|
~Client();
|
|
|
|
AsyncClient *tcp_client{nullptr};
|
|
std::string identifier{};
|
|
bool disconnected{false};
|
|
};
|
|
|
|
esphome::uart::UARTComponent *stream_{nullptr};
|
|
AsyncServer server_{0};
|
|
uint16_t port_{6638};
|
|
std::vector<uint8_t> recv_buf_{};
|
|
std::vector<std::unique_ptr<Client>> clients_{};
|
|
};
|