/*
    This file is part of the Arduino_RouterBridge library.

    Copyright (c) 2025 Arduino SA

    This Source Code Form is subject to the terms of the Mozilla Public
    License, v. 2.0. If a copy of the MPL was not distributed with this
    file, You can obtain one at http://mozilla.org/MPL/2.0/.

*/

#pragma once

#ifndef BRIDGE_TCP_CLIENT_H
#define BRIDGE_TCP_CLIENT_H

#define TCP_CONNECT_METHOD          "tcp/connect"
#define TCP_CONNECT_SSL_METHOD      "tcp/connectSSL"
#define TCP_CLOSE_METHOD            "tcp/close"
#define TCP_WRITE_METHOD            "tcp/write"
#define TCP_READ_METHOD             "tcp/read"

#include <api/RingBuffer.h>
#include <api/Client.h>
#include "bridge.h"

#define DEFAULT_TCP_CLIENT_BUF_SIZE    512


template<size_t BufferSize=DEFAULT_TCP_CLIENT_BUF_SIZE>
class BridgeTCPClient : public Client {

    BridgeClass* bridge;
    uint32_t connection_id{};
    uint32_t read_timeout = 0;
    RingBufferN<BufferSize> temp_buffer;
    struct k_mutex client_mutex{};
    bool _connected = false;

public:
    explicit BridgeTCPClient(BridgeClass& bridge): bridge(&bridge) {}

    BridgeTCPClient(BridgeClass& bridge, uint32_t connection_id, bool connected=true): bridge(&bridge), connection_id(connection_id), _connected {connected} {}

    bool begin() {
        k_mutex_init(&client_mutex);
        if (!(*bridge)) {
            return bridge->begin();
        }
        return true;
    }

    void setTimeout(const uint32_t ms) {
        k_mutex_lock(&client_mutex, K_FOREVER);
        read_timeout = ms;
        k_mutex_unlock(&client_mutex);
    }

    int connect(IPAddress ip, uint16_t port) override {
        return connect(ip.toString().c_str(), port);
    }

    int connect(const char *host, uint16_t port) override {

        k_mutex_lock(&client_mutex, K_FOREVER);

        String hostname = host;
        const bool ok = _connected || bridge->call(TCP_CONNECT_METHOD, hostname, port).result(connection_id);
        _connected = ok;

        k_mutex_unlock(&client_mutex);

        return ok? 0 : -1;
    }

    int connectSSL(const char *host, uint16_t port, const char *ca_cert) {

        k_mutex_lock(&client_mutex, K_FOREVER);

        String hostname = host;
        String ca_cert_str = ca_cert;

        const bool ok = _connected || bridge->call(TCP_CONNECT_SSL_METHOD, hostname, port, ca_cert_str).result(connection_id);
        _connected = ok;
        k_mutex_unlock(&client_mutex);

        return ok? 0 : -1;
    }

    uint32_t getId() {
        k_mutex_lock(&client_mutex, K_FOREVER);
        const uint32_t out = connection_id;
        k_mutex_unlock(&client_mutex);
        return out;
    }

    size_t write(uint8_t c) override {
        return write(&c, 1);
    }

    size_t write(const uint8_t *buffer, size_t size) override {

        if (!connected()) return 0;

        MsgPack::arr_t<uint8_t> payload;

        for (size_t i = 0; i < size; ++i) {
            payload.push_back(buffer[i]);
        }

        size_t written;
        k_mutex_lock(&client_mutex, K_FOREVER);
        const bool ok = bridge->call(TCP_WRITE_METHOD, connection_id, payload).result(written);
        k_mutex_unlock(&client_mutex);
        return ok? written : 0;
    }

    int available() override {
        k_mutex_lock(&client_mutex, K_FOREVER);
        const int size = temp_buffer.availableForStore();
        if (size > 0) _read(size);
        const int _available = temp_buffer.available();
        k_mutex_unlock(&client_mutex);
        return _available;
    }

    int read() override {
        uint8_t c;
        read(&c, 1);
        return c;
    }

    int read(uint8_t *buf, size_t size) override {
        k_mutex_lock(&client_mutex, K_FOREVER);
        size_t i = 0;
        while (temp_buffer.available() && i < size) {
            buf[i++] = temp_buffer.read_char();
        }
        k_mutex_unlock(&client_mutex);
        return (int)i;
    }

    int peek() override {
        k_mutex_lock(&client_mutex, K_FOREVER);
        int out = -1;
        if (temp_buffer.available()) {
            out = temp_buffer.peek();
        }
        k_mutex_unlock(&client_mutex);
        return out;
    }

    void flush() override {
        // No-op: flush is implemented for Client subclasses using an output buffer
    }

    void close() {
        stop();
    }

    void stop() override {
        k_mutex_lock(&client_mutex, K_FOREVER);
        String msg;
        if (_connected) {
            _connected = !bridge->call(TCP_CLOSE_METHOD, connection_id).result(msg);
        }
        k_mutex_unlock(&client_mutex);
    }

    uint8_t connected() override {
        k_mutex_lock(&client_mutex, K_FOREVER);
        const uint8_t out = _connected? 1 : 0;
        k_mutex_unlock(&client_mutex);
        return out;
    }

    operator bool() override {
        return available() || connected();
    }

    using Print::write;

private:
    void _read(size_t size) {

        if (size == 0) return;

        k_mutex_lock(&client_mutex, K_FOREVER);

        if (!_connected) {
            k_mutex_unlock(&client_mutex);
            return;
        }

        MsgPack::arr_t<uint8_t> message;
        bool ret;
        int err;

        if (read_timeout > 0) {
            RpcCall async_rpc_timeout = bridge->call(TCP_READ_METHOD, connection_id, size, read_timeout);
            ret = async_rpc_timeout.result(message);
            err = async_rpc_timeout.getErrorCode();
        } else {
            RpcCall async_rpc = bridge->call(TCP_READ_METHOD, connection_id, size);
            ret = async_rpc.result(message);
            err = async_rpc.getErrorCode();
        }

        if (ret) {
            for (size_t i = 0; i < message.size(); ++i) {
                temp_buffer.store_char(static_cast<char>(message[i]));
            }
        }

        if (err > NO_ERR) {
            _connected = false;
        }

        k_mutex_unlock(&client_mutex);
    }

};


#endif //BRIDGE_TCP_CLIENT_H