// Copyright (c) 2019-present Anonymous275.
// BeamMP Server code is not in the public domain and is not free software.
// One must be granted explicit permission by the copyright holder in order to modify or distribute any part of the source or binaries.
// Anything else is prohibited. Modified works may not be published and have be upstreamed to the official repository.
///
/// Created by Anonymous275 on 5/8/2020
///

#pragma once
#ifdef WIN32
#include <WS2tcpip.h>
#else
#include <arpa/inet.h>
#define SOCKET int
#endif
#include "CustomAssert.h"
#include <algorithm>
#include <chrono>
#include <set>
#include <string>
#include <vector>

struct VData {
    int ID = -1;
    std::string Data;
};

class Client {
private:
    std::set<std::unique_ptr<VData>> VehicleData; //ID and Data;
    std::string Name = "Unknown Client";
    SOCKET SOCK[2] { SOCKET(-1) };
    sockaddr_in UDPADDR;
    std::string Role;
    std::string DID;
    int Status = 0;
    int ID = -1;

public:
    bool isConnected = false;
    bool isSynced = false;
    bool isGuest = false;

    void AddNewCar(int ident, const std::string& Data);
    void SetCarData(int ident, const std::string& Data);
    std::set<std::unique_ptr<VData>>& GetAllCars();
    void SetName(const std::string& name) { Name = name; }
    void SetRoles(const std::string& role) { Role = role; }
    std::string GetCarData(int ident);
    void SetUDPAddr(sockaddr_in Addr) { UDPADDR = Addr; }
    void SetDownSock(SOCKET CSock) { SOCK[1] = CSock; }
    void SetTCPSock(SOCKET CSock) { SOCK[0] = CSock; }
    void SetStatus(int status) { Status = status; }
    void DeleteCar(int ident);
    sockaddr_in GetUDPAddr() { return UDPADDR; }
    std::string GetRoles() { return Role; }
    std::string GetName() { return Name; }
    SOCKET GetDownSock() { return SOCK[1]; }
    SOCKET GetTCPSock() { return SOCK[0]; }
    void SetID(int ID) { this->ID = ID; }
    int GetOpenCarID();
    int GetCarCount();
    void ClearCars();
    int GetStatus() { return Status; }
    int GetID() { return ID; }
};
struct ClientInterface {
    std::set<std::unique_ptr<Client>> Clients;
    void RemoveClient(Client*& c) {
        Assert(c);
        c->ClearCars();
        auto Iter = std::find_if(Clients.begin(), Clients.end(), [&](auto& ptr) {
            return c == ptr.get();
        });
        Assert(Iter != Clients.end());
        if (Iter == Clients.end()) {
            return;
        }
        Clients.erase(Iter);
        c = nullptr;
    }
    void AddClient(Client*&& c) {
        Assert(c);
        Clients.insert(std::unique_ptr<Client>(c));
    }
    int Size() {
        return int(Clients.size());
    }
};

extern std::unique_ptr<ClientInterface> CI;