/*
	Created on: Apr 12, 2020

	Copyright 2020 flyinghead

	This file is part of flycast.

    flycast 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 2 of the License, or
    (at your option) any later version.

    flycast 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 flycast.  If not, see <https://www.gnu.org/licenses/>.
 */
#pragma once
#include "types.h"
#include <cstdint>
#include <atomic>
#include <future>
#include <mutex>
#include <vector>
#include "net_platform.h"

class NaomiNetwork
{
public:
	NaomiNetwork() {
#ifdef _WIN32
		server_ip.S_un.S_addr = INADDR_NONE;
#else
		server_ip.s_addr = INADDR_NONE;
#endif
	}
	~NaomiNetwork() { terminate(); }
	std::future<bool> startNetworkAsync();
	void startNow() { start_now = true; }
	bool syncNetwork();
	void pipeSlaves();
	bool receive(u8 *data, u32 size);
	void send(u8 *data, u32 size);
	void shutdown();	// thread-safe
	void terminate();	// thread-safe
	int slotCount() const { return slot_count; }
	int slotId() const { return slot_id; }
	bool hasToken() const { return got_token; }

private:
	bool init();
	bool createServerSocket();
	bool createBeaconSocket();
	bool startNetwork();
	void processBeacon();
	bool findServer();
	sock_t createAndBind(int protocol);
	bool isMaster() const { return slot_id == 0; }
	void closeSocket(sock_t& socket) const { closesocket(socket); socket = INVALID_SOCKET; }

	struct in_addr server_ip;
	std::string server_name;
	// server stuff
	sock_t server_sock = INVALID_SOCKET;
	sock_t beacon_sock = INVALID_SOCKET;
	enum class ClientState { Connected, Waiting, Starting, Ready, Online };
	struct Slave {
		Slave(sock_t socket)
			: state(ClientState::Connected), state_time(std::chrono::steady_clock::now()), socket(socket) {}
		void set_state(ClientState state) { this->state = state; this->state_time = std::chrono::steady_clock::now(); }
		ClientState state;
		std::chrono::steady_clock::time_point state_time;
		sock_t socket;
	};
	std::vector<Slave> slaves;
	bool start_now = false;
	// client stuff
	sock_t client_sock = INVALID_SOCKET;
	// common stuff
	int slot_count = 0;
	int slot_id = 0;
	bool got_token = false;
	std::atomic<bool> network_stopping{ false };
	std::mutex mutex;

	static const uint16_t SERVER_PORT = 37391;
};
extern NaomiNetwork naomiNetwork;

void SetNaomiNetworkConfig(int node);
bool NaomiNetworkSupported();