//
//  AgentList.h
//  hifi
//
//  Created by Stephen Birarda on 2/15/13.
//  Copyright (c) 2013 High Fidelity, Inc. All rights reserved.
//

#ifndef __hifi__AgentList__
#define __hifi__AgentList__

#include <stdint.h>
#include <iterator>

#include "Agent.h"
#include "UDPSocket.h"

#ifdef _WIN32
#include "pthread.h"
#endif

const int MAX_NUM_AGENTS = 10000;
const int AGENTS_PER_BUCKET = 100;

const int MAX_PACKET_SIZE = 1500;
const unsigned int AGENT_SOCKET_LISTEN_PORT = 40103;
const int AGENT_SILENCE_THRESHOLD_USECS = 2 * 1000000;
extern const char SOLO_AGENT_TYPES[3];

extern char DOMAIN_HOSTNAME[];
extern char DOMAIN_IP[100];    //  IP Address will be re-set by lookup on startup
extern const int DOMAINSERVER_PORT;

class AgentListIterator;

class AgentList {
public:
    static AgentList* createInstance(char ownerType, unsigned int socketListenPort = AGENT_SOCKET_LISTEN_PORT);
    static AgentList* getInstance();
    
    typedef AgentListIterator iterator;
  
    AgentListIterator begin() const;
    AgentListIterator end() const;
    
    void(*linkedDataCreateCallback)(Agent *);
    void(*audioMixerSocketUpdate)(in_addr_t, in_port_t);
    
    int size() { return _numAgents; }
    
    UDPSocket& getAgentSocket();
    
    uint16_t getLastAgentId();
    void increaseAgentId();
    
    int updateList(unsigned char *packetData, size_t dataBytes);
    
    Agent* agentWithAddress(sockaddr *senderAddress);
    Agent* agentWithID(uint16_t agentID);
    
    bool addOrUpdateAgent(sockaddr *publicSocket, sockaddr *localSocket, char agentType, uint16_t agentId);
    
    void processAgentData(sockaddr *senderAddress, unsigned char *packetData, size_t dataBytes);
    void processBulkAgentData(sockaddr *senderAddress, unsigned char *packetData, int numTotalBytes);
   
    int updateAgentWithData(sockaddr *senderAddress, unsigned char *packetData, size_t dataBytes);
    int updateAgentWithData(Agent *agent, unsigned char *packetData, int dataBytes);
    
    void broadcastToAgents(unsigned char *broadcastData, size_t dataBytes, const char* agentTypes, int numAgentTypes);
    char getOwnerType();
    unsigned int getSocketListenPort();
    
    Agent* soloAgentOfType(char agentType);
    
    void startSilentAgentRemovalThread();
    void stopSilentAgentRemovalThread();
    void startDomainServerCheckInThread();
    void stopDomainServerCheckInThread();
    void startPingUnknownAgentsThread();
    void stopPingUnknownAgentsThread();
    
    friend class AgentListIterator;
private:
    static AgentList* _sharedInstance;
    
    AgentList(char ownerType, unsigned int socketListenPort);
    ~AgentList();
    AgentList(AgentList const&); // Don't implement, needed to avoid copies of singleton
    void operator=(AgentList const&); // Don't implement, needed to avoid copies of singleton
    
    void addAgentToList(Agent* newAgent);
    
    Agent** _agentBuckets[MAX_NUM_AGENTS / AGENTS_PER_BUCKET];
    int _numAgents;
    UDPSocket agentSocket;
    char ownerType;
    unsigned int socketListenPort;
    uint16_t lastAgentId;
    pthread_t removeSilentAgentsThread;
    pthread_t checkInWithDomainServerThread;
    pthread_t pingUnknownAgentsThread;
    
    void handlePingReply(sockaddr *agentAddress);
};

class AgentListIterator : public std::iterator<std::input_iterator_tag, Agent> {
public:
    AgentListIterator(const AgentList* agentList, int agentIndex);
    ~AgentListIterator() {};
    
    int getAgentIndex() { return _agentIndex; };
    
	AgentListIterator& operator=(const AgentListIterator& otherValue);
    
    bool operator==(const AgentListIterator& otherValue);
	bool operator!= (const AgentListIterator& otherValue);
    
    Agent& operator*();
    Agent* operator->();
    
	AgentListIterator& operator++();
    AgentListIterator operator++(int);
private:
    void skipDeadAndStopIncrement();
    
    const AgentList* _agentList;
    int _agentIndex;
};

#endif /* defined(__hifi__AgentList__) */