// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
// SPDX-License-Identifier: GPL-3.0-only

#pragma once

#include <thread>
#include <atomic>
#include <mutex>
#include <memory>
#include "ImagePuller.h"
#include "../common/Logger.h"
#include "../common/ThreadSafeFIFO.h"

class TCPImagePuller : public ImagePuller {
    int fd = -1;
    std::mutex fd_mutex;

    std::string addr;
    std::string host;
    uint16_t port = 0;
    std::optional<int32_t> receive_buffer_size;
    std::atomic<bool> disconnect{false};

    ThreadSafeFIFO<ImagePullerOutput> cbor_fifo{200};

    std::thread receiver_thread;
    std::thread cbor_thread;

    Logger logger{"TCPImagePuller"};

    bool ReadExact(void *buf, size_t size);
    bool SendAll(const void *buf, size_t len);
    bool EnsureConnected();
    void CloseSocket();
    void ReceiverThread();
    void CBORThread();
public:
    explicit TCPImagePuller(const std::string &tcp_addr, std::optional<int32_t> rcv_buffer_size = {});

    ~TCPImagePuller() override;
    bool SupportsAck() const override { return true; }
    bool SendAck(const PullerAckMessage &ack) override;
    void Disconnect() override;
};