#pragma once

#include "Messages.h"
#include "Messages/ClientMsg.h"
#include "Messages/CustomMsg.h"
#include "Messages/InvestigateMsg.h"
#include "Messages/ModelUrlMsg.h"
#include "Messages/NameMsg.h"
#include "Messages/NetworkIdMsg.h"
#include "Messages/PoseMsg.h"
#include "Messages/ThingMsg.h"

#include <list>

#if defined(_WIN32) || defined(_WIN64)
#include <winsock2.h>
#elif defined(__unix__) || defined(__APPLE__)
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <unistd.h>
#elif defined(ARDUINO)
#include <WiFiUdp.h>
#endif

namespace Passer {
namespace RoboidControl {

/// @brief A participant is device which can communicate with other participants
class Participant {
public:
  char buffer[1024];
  long publishInterval = 3000; // 3 seconds
  unsigned char networkId = 0;

  const char *name = "Participant";

  const char *ipAddress = "0.0.0.0";
  int port = 0;
  int localPort = 0;

#if defined(ARDUINO)
  const char *remoteIpAddress = nullptr;
  unsigned short remotePort = 0;
  char *broadcastIpAddress = nullptr;

  WiFiUDP udp;
#else

#if defined(_WIN32) || defined(_WIN64)
  SOCKET sock;
#elif defined(__unix__) || defined(__APPLE__)
  int sock;
#endif
  sockaddr_in remote_addr;
  sockaddr_in server_addr;
  sockaddr_in broadcast_addr;

#endif

  Participant();
  Participant(int port);
  Participant(const char *ipAddress, int port);

  void begin();
  bool connected = false;

  virtual void Update(unsigned long currentTimeMs = 0);

  std::list<Thing *> things;
  Thing *Get(unsigned char networkId, unsigned char thingId);
  int Add(Thing *thing);
  void Remove(Thing *thing);
  void UpdateAll(unsigned long currentTimeMs);

  void SendThingInfo(Thing *thing);
  void PublishThingInfo(Thing *thing);

  bool Send(IMessage *msg);
  bool Publish(IMessage *msg);

  void ReceiveData(unsigned char bufferSize, Participant *remoteParticipant);

protected:
  std::list<Participant *> participants;

  unsigned long nextPublishMe = 0;

  void SetupUDP(int localPort, const char *remoteIpAddress, int remotePort);

  Participant *GetParticipant(const char *ipAddress, int port);
  Participant *AddParticipant(const char *ipAddress, int port);

  void ReceiveUDP();

  virtual void Process(Participant *sender, ClientMsg *msg);
  virtual void Process(Participant *sender, NetworkIdMsg *msg);
  virtual void Process(InvestigateMsg *msg);
  virtual void Process(ThingMsg *msg);
  virtual void Process(NameMsg *msg);
  virtual void Process(PoseMsg *msg);
  virtual void Process(CustomMsg *msg);
};

} // namespace Control
} // namespace Passer
using namespace Passer::RoboidControl;