#pragma once

#include <wininet.h>
#include "net/INetRequest.h"
#include "logging/RhoLog.h"

namespace rho {
namespace common {
    class CRhoFile;
}

namespace net {

class CNetResponseImpl : public INetResponse
{
    bool   m_bValid;
    String m_data;
    int    m_nRespCode;
    String m_cookies;
public:
    CNetResponseImpl() : m_bValid(false),m_nRespCode(-1){}

    bool isValid()const{ return m_bValid; }
    void setValid(bool b){ m_bValid = b; }

    virtual const char* getCharData()
    {
        return m_bValid ? m_data.c_str() : "";
    }

    virtual unsigned int getDataSize()
    {
        return m_bValid ? m_data.size() : 0;
    }

    String& getRawData(){ return m_data; }

    boolean isResponseRecieved(){ return m_nRespCode!=-1;}
    void setResponseCode(int nRespCode){ m_nRespCode = nRespCode;}

    int getRespCode(){ return m_nRespCode;}

	boolean isOK()
	{
		return m_nRespCode == 200 || m_nRespCode == 206;
	}

    boolean isUnathorized()
    {
        return m_nRespCode == 401;
    }

    void setCookies(String s)
    {
        m_cookies = s;
    }
    
    String getCookies()
    {
        return m_cookies;
    }
};

class CNetRequest;
class CNetRequestImpl
{
    DEFINE_LOGCLASS;

    static common::CMutex m_mxInternet;
    static HINTERNET m_hInternet;
    static HANDLE    m_hWceConnMgrConnection;
    HINTERNET  m_hConnection, m_hRequest;

    LPCTSTR  m_pszErrFunction;

    URL_COMPONENTS m_uri;
    CAtlStringW m_strReqUrlW;
    String      m_strUrl;
    CNetRequest* m_pParent;
    Hashtable<String,String>* m_pHeaders;
    boolean m_bCancel;
    IRhoSession* m_pSession;
    boolean m_sslVerifyPeer;

public :
    CNetRequestImpl(CNetRequest* pParent, const char* method, const String& strUrl, 
        IRhoSession* oSession, Hashtable<String,String>* pHeaders, boolean sslVerifyPeer);
    ~CNetRequestImpl();

    void close();
    void cancel();
    bool isError(){ return m_pszErrFunction!= null; }
    CNetResponseImpl* sendString(const String& strBody);

    CNetResponseImpl* sendMultipartData(VectorPtr<CMultipartItem*>& arItems);

    CNetResponseImpl* downloadFile(common::CRhoFile& oFile);

    static void deinitConnection();
private:
    String makeClientCookie();

    void ErrorMessage(LPCTSTR pszFunction);
    void readInetFile( void* hRequest, CNetResponseImpl* pNetData, common::CRhoFile* pFile = NULL, char* pBuf = NULL, DWORD dwBufSize = 0);
    void alloc_url_components(URL_COMPONENTS *uri, const wchar_t *url);
    void free_url_components(URL_COMPONENTS *uri);
    void readResponse(CNetResponseImpl* pNetData);
    bool SetupInternetConnection(LPCTSTR url);

    boolean readHeaders(Hashtable<String,String>& oHeaders);
    void    writeHeaders(Hashtable<String,String>* pHeaders);
    boolean checkSslCertError();

    String getBodyContentType();
    bool internetWriteHeader( const char* szPrefix, const char* szBody, const char* szPrefixEnd);

    int processMultipartItems( VectorPtr<CMultipartItem*>& arItems );

    bool initConnection(boolean bLocalHost, LPCTSTR url);
};

}
}