mirror of
https://github.com/oxan/esphome-stream-server.git
synced 2026-05-16 14:27:03 +00:00
7e5750c9d6
This prevents dropped data if a write to a client is cut short. Move default port number to config schema
74 lines
2.4 KiB
C++
74 lines
2.4 KiB
C++
/* Copyright (C) 2020-2023 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/socket/socket.h"
|
|
#include "esphome/components/uart/uart.h"
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
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 set_buffer_size(size_t size) { this->buf_size_ = size; }
|
|
|
|
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 accept();
|
|
void cleanup();
|
|
void read();
|
|
void flush();
|
|
void write();
|
|
|
|
size_t buf_index(size_t pos) { return pos & (this->buf_size_ - 1); }
|
|
/// Return the number of consecutive elements that are ahead of @p pos in memory.
|
|
size_t buf_ahead(size_t pos) { return (pos | (this->buf_size_ - 1)) - pos + 1; }
|
|
|
|
struct Client {
|
|
Client(std::unique_ptr<esphome::socket::Socket> socket, std::string identifier, size_t position);
|
|
|
|
std::unique_ptr<esphome::socket::Socket> socket{nullptr};
|
|
std::string identifier{};
|
|
bool disconnected{false};
|
|
size_t position{0};
|
|
};
|
|
|
|
esphome::uart::UARTComponent *stream_{nullptr};
|
|
size_t buf_size_;
|
|
|
|
std::unique_ptr<uint8_t[]> buf_{};
|
|
size_t buf_head_{0};
|
|
size_t buf_tail_{0};
|
|
|
|
std::unique_ptr<esphome::socket::Socket> socket_{};
|
|
uint16_t port_{6638};
|
|
std::vector<Client> clients_{};
|
|
};
|