#include "Participant.h"

#include <string.h>

namespace RoboidControl {

#pragma region Participant

ParticipantRegistry Participant::registry;

Participant* Participant::LocalParticipant = new Participant();

void Participant::ReplaceLocalParticipant(Participant& newParticipant) {
  LocalParticipant = &newParticipant;
  std::cout << "Replaced local participant" << std::endl;
}

Participant::Participant() {
  std::cout << "P\n";
  //this->root.name = "Isolated";
  this->root = new Thing(this);
  this->root->name = "Root";
  this->Add(this->root);
}

Participant::Participant(const char* ipAddress, int port) {
  // Add the root thing to the list of things, because we could not do that
  // earlier (I think)
  this->Add(this->root);

  // make a copy of the ip address string
  int addressLength = (int)strlen(ipAddress);
  int stringLength = addressLength + 1;
  char* addressString = new char[stringLength];
#if defined(_WIN32) || defined(_WIN64)
  strncpy_s(addressString, stringLength, ipAddress,
            addressLength);  // Leave space for null terminator
#else
  strncpy(addressString, ipAddress, addressLength);
#endif
  addressString[addressLength] = '\0';

  this->ipAddress = addressString;
  this->port = port;
}

Participant::~Participant() {
  // registry.Remove(this);
  delete[] this->ipAddress;
}

void Participant::Update() {
  for (Thing* thing : this->things) {
    if (thing != nullptr)
      thing->Update(true);
  }
}

Thing* Participant::Get(unsigned char thingId) {
  for (Thing* thing : this->things) {
    if (thing->id == thingId)
      return thing;
  }
  // std::cout << "Could not find thing " << this->ipAddress << ":" <<
  // this->port
  //           << "[" << (int)thingId << "]\n";
  return nullptr;
}

void Participant::Add(Thing* thing, bool checkId) {
  if (checkId && thing->id == 0) {
    // allocate a new thing ID
#if defined(NO_STD)
    thing->id = this->thingCount + 1;
    this->things[this->thingCount++] = thing;
#else
    // find highest id
    int highestIx = 0;
    for (Thing* thing : this->things) {
      if (thing == nullptr)
        continue;
      if (thing->id > highestIx)
        highestIx = thing->id;
    }
    thing->id = highestIx + 1;
    this->things.push_back(thing);
#endif
    // std::cout << "Add thing with generated ID " << this->ipAddress << ":"
    //           << this->port << "[" << (int)thing->id << "]\n";
  } else {
    Thing* foundThing = Get(thing->id);
    if (foundThing == nullptr) {
#if defined(NO_STD)
      this->things[this->thingCount++] = thing;
#else
      this->things.push_back(thing);
#endif
      // std::cout << "Add thing " << this->ipAddress << ":" << this->port <<
      // "["
      //           << (int)thing->id << "]\n";
    } else {
      // std::cout << "Did not add, existing thing " << this->ipAddress << ":"
      //           << this->port << "[" << (int)thing->id << "]\n";
    }
  }
}

void Participant::Remove(Thing* thing) {
#if defined(NO_STD)
  for (unsigned char thingIx = 0; thingIx < this->thingCount; thingIx++)
    if (this->things[thingIx] == thing)
      this->things[thingIx] = nullptr;
  // compacting
  unsigned char lastThingIx = 0;
  for (unsigned char thingIx = 0; thingIx < this->thingCount; thingIx++) {
    if (this->things[thingIx] == nullptr)
      continue;
    this->things[lastThingIx] = this->things[thingIx];
    lastThingIx++;
  }
  this->thingCount = lastThingIx;
#else
  this->things.remove_if([thing](Thing* obj) { return obj == thing; });
  // std::cout << "Removing [" << (int)thing->networkId << "/" << (int)thing->id
  //           << "] list size = " << this->things.size() << "\n";
#endif
}

#pragma endregion

#pragma region ParticipantRegistry

Participant* ParticipantRegistry::Get(const char* ipAddress,
                                      unsigned int port) {
#if !defined(NO_STD)
  for (Participant* participant : ParticipantRegistry::participants) {
    if (participant == nullptr)
      continue;
    if (strcmp(participant->ipAddress, ipAddress) == 0 &&
        participant->port == port) {
      // std::cout << "found participant " << participant->ipAddress << ":"
      //           << (int)participant->port << std::endl;
      return participant;
    }
  }
  std::cout << "Could not find participant " << ipAddress << ":" << (int)port
            << std::endl;
#endif
  return nullptr;
}

Participant* ParticipantRegistry::Get(unsigned char participantId) {
#if !defined(NO_STD)
  for (Participant* participant : ParticipantRegistry::participants) {
    if (participant == nullptr)
      continue;
    if (participant->networkId == participantId)
      return participant;
  }
  std::cout << "Could not find participant " << (int)participantId << std::endl;
#endif
  return nullptr;
}

Participant* ParticipantRegistry::Add(const char* ipAddress,
                                      unsigned int port) {
  Participant* participant = new Participant(ipAddress, port);
  Add(participant);
  return participant;
}

void ParticipantRegistry::Add(Participant* participant) {
  Participant* foundParticipant =
      Get(participant->ipAddress, participant->port);

  if (foundParticipant == nullptr) {
#if defined(NO_STD)
    // this->things[this->thingCount++] = thing;
#else
    ParticipantRegistry::participants.push_back(participant);
#endif
    //   std::cout << "Add participant " << participant->ipAddress << ":"
    //             << participant->port << "[" << (int)participant->networkId
    //             << "]\n";
    //   std::cout << "participants " <<
    //   ParticipantRegistry::participants.size()
    //             << "\n";
    // } else {
    //   std::cout << "Did not add, existing participant " <<
    //   participant->ipAddress
    //             << ":" << participant->port << "[" <<
    //             (int)participant->networkId
    //             << "]\n";
  }
}

void ParticipantRegistry::Remove(Participant* participant) {
  // participants.remove(participant);
}

#if defined(NO_STD)
Participant** ParticipantRegistry::GetAll() const {
  return ParticipantRegistry::participants;
}
#else
const std::list<Participant*>& ParticipantRegistry::GetAll() const {
  return ParticipantRegistry::participants;
}
#endif

#pragma endregion ParticipantRegistry

}  // namespace RoboidControl