/*
 * =====================================================================================
 *
 *       Filename:  tcp_proxy.cpp
 *
 *    Description:  
 *
 *        Version:  1.0
 *        Created:  2021年05月19日 15时04分15秒
 *       Revision:  none
 *       Compiler:  gcc
 *
 *         Author:  Li Chao (), lichao@aiotlink.com
 *   Organization:  
 *
 * =====================================================================================
 */
#include "tcp_proxy.h"
#include "defs.h"
#include "shm_socket.h"
#include "tcp_connection.h"

TcpProxy::TcpProxy() :
    run_(false) {}

TcpProxy::~TcpProxy() {}

bool TcpProxy::Start(bhome_shm::SharedMemory &shm)
{
	Stop();
	bool cur = false;
	if (!run_.compare_exchange_strong(cur, true)) { return false; }

	auto &mq = GetCenterInfo(shm)->mq_tcp_proxy_;
	local_.reset(new ShmSocket(mq.offset_, shm, mq.id_));
	auto localProc = [this](ShmSocket &sock, MsgI &msg, BHMsgHead &head) {
		auto &dest = head.dest();
		if (dest.ip().empty() || dest.port() == 0) { return; }
		bool r = Send(dest.ip(), dest.port(), msg.content());
		// TODO check send fail.
	};
	local_->Start(1, localProc);

	auto proxyProc = [this]() {
		while (run_) {
			io_context_.run_one_for(std::chrono::milliseconds(100));
		}
	};
	return true;
}

void TcpProxy::Stop()
{
	local_.reset();

	bool cur = true;
	if (run_.compare_exchange_strong(cur, false)) {
		if (worker_.joinable()) {
			worker_.join();
		}
	}
}

bool TcpProxy::Send(const std::string &ip, int port, std::string &&content)
{
	if (content.empty()) { return false; }

	tcp::endpoint dest(ip::address::from_string(ip), port);
	TcpRequest1::Create(io_context_, dest, std::move(content));

	// char tag[sizeof(kBHTcpServerTag)] = {0};

	// int n = read(sock, tag, sizeof(tag));
	// if (n == sizeof(tag) && memcmp(tag, &kBHTcpServerTag, sizeof(tag)) == 0) {
	// 	send(sock, content.data(), content.size(), 0);
	// 	connections_[addr].io_info_.h_ = [this, sock](int events) { OnReply(sock); };
	// 	// success
	// }
}