#pragma once
#include <afxmt.h>
#include <afxsock.h>

#include "ConcurrentQueue.h"

#ifdef BUILD_COMMDLL
#define COMMDLL_DLLCLASS __declspec(dllexport)
#else
#define COMMDLL_DLLCLASS __declspec(dllimport)
#endif

class IBufferSocketCallBack
{
public:
    virtual void OnReceive(const char* buffer, int len) = 0;
    virtual void OnDisconnect(int errorCode)            = 0;
};

class COMMDLL_DLLCLASS CBufferedSocket
{
    // Forward declaration of internal classes
public:
    class CBufferedSocketMessage;
    class CBufferMessageThread;
    class CReceivingThread;
    class CSenderThread;
    class CInternalMessageThread;
    friend class CInternalMessageThread;
    friend class CReceivingThread;
    friend class CSenderThread;

    // Private fields
private:
    CConcurrentQueue<CBufferedSocketMessage>* m_pMessageQueue;
    CConcurrentQueue<CBufferedSocketMessage>* m_pSendQueue;
    CBufferMessageThread*                     m_pMessageThread;
    CBufferMessageThread*                     m_PSenderThread;
    IBufferSocketCallBack*                    m_pCallBack;
    CReceivingThread*                         m_pReceivingThread;
    CEvt*                                     m_hConnectedEvent;
    SOCKET                                    s;
    bool                                      bIsConnected;

    // Private methods
protected:
    void OnReceive(const char* buf, size_t len);
    void OnDisconnect(int errorCode);
    int  Read(char* buf, size_t size);
    int  Write(const char* buf, size_t len);

private:
    CBufferedSocket(const CBufferedSocket&);
    // Public methods
public:
    CBufferedSocket(IBufferSocketCallBack* pCallBack);
    ~CBufferedSocket(void);

    void ProcessMessage(CBufferedSocketMessage& message);
    void Close();
    bool Connect(const char* IpAddress, int port);
    bool Send(const char* buf, size_t len, int milliSeconds = INFINITE);
    bool IsConnected()
    {
        return bIsConnected;
    }

    bool WaitConnected();
    bool WaitConnected(int milliseconds);
    /* returns the percentage representing how full the send queue is */
    double SendQueueFullState();
};