
#ifndef QWON_HTTP_CLIENT_UTIL
#define QWON_HTTP_CLIENT_UTIL

#include <vcl.h>
#include <windows.h>
#include <wininet.h>
#pragma link "wininet.lib"

namespace QwonWebDav{

	class ParsedURL{
  private:
  public:
		String serverAddress;
		String remoteDirectory;
		String remoteFilename;
		String myString;
		String Header;
		int Port;
		String ID, PW;
  public:
		parseURL(wchar_t* URL){
      myString = URL;
      int pos;
			const String protocolIdentifier = "http://";
      pos = myString.Pos(protocolIdentifier); //없는 문자열일 경우 리턴값이 0이다.
      if(pos==0) return false; //http프로토콜로 시작하지 않으면 그만둔다.
      pos = pos + protocolIdentifier.Length();
      myString = myString.SubString(pos, myString.Length()-pos+1); //프로토콜 식별자 제거
      pos = myString.Pos("/")+1;
      serverAddress = myString.SubString(1, pos - 2);

      Port = 80; // http default
      for(int i=0; i<serverAddress.Length(); i++) {
        if (serverAddress.c_str()[i]==':') {
          Port = StrToInt(serverAddress.SubString(i+2, serverAddress.Length()-i-1));
          serverAddress = myString.SubString(1, i);
          break;
        }
      }

      remoteDirectory = "";
      remoteFilename = myString.SubString(pos, myString.Length()-pos+1);
      for(int i=myString.Length(); i>=pos ; i-- )
      {        //char a = myString[i];
        if( myString[i] =='/'){
          int pos2 = i;
          remoteDirectory = myString.SubString(pos,pos2-pos+1);
          remoteFilename = myString.SubString(pos2+1, myString.Length()-(pos2+1)+1);
          break;
        }
      }
      return true;
    }

    Copy(ParsedURL &serverParsedURL) {
      serverParsedURL.serverAddress = serverAddress;
      serverParsedURL.remoteDirectory = remoteDirectory;
      serverParsedURL.remoteFilename = remoteFilename;
      serverParsedURL.myString = myString;
      serverParsedURL.Port = Port;
      serverParsedURL.ID = ID;
      serverParsedURL.PW = PW;

      return true;
    }

    Clear() {
      serverAddress = "";
      remoteDirectory = "";
      remoteFilename = "";
      myString = "";
      Header = "";
      Port = 80;
      ID = "", PW = "";
      return 0;
    }
  };

  //---------------------------------------------------------------------------
	bool __fastcall parseURL(wchar_t *URL, ParsedURL &serverParsedURL)
  {
    return serverParsedURL.parseURL(URL);
  }

  //---------------------------------------------------------------------------
  // Header를 확인해서 정상적으로 작동되었는지 확인하는 부분
	bool __fastcall CheckSrvData(HINTERNET hURL, String &returnValue, ParsedURL &srvURL)
	{
    LPVOID lpOutBuffer=NULL;
    DWORD dwSize = 0;

  retry:

    // This call will fail on the first pass, because
    // no buffer is allocated.
    if(!HttpQueryInfo(hURL,HTTP_QUERY_RAW_HEADERS,
      (LPVOID)lpOutBuffer,&dwSize,NULL))
    {
      if (GetLastError()==ERROR_HTTP_HEADER_NOT_FOUND)
      {
        // Code to handle the case where the header isn't available.
        return TRUE;
      }
      else
      {
        // Check for an insufficient buffer.
        if (GetLastError()==ERROR_INSUFFICIENT_BUFFER)
        {
            // Allocate the necessary buffer.
            lpOutBuffer = new wchar_t[dwSize];

            // Retry the call.
            goto retry;
        }
        else
        {
            // Error handling code.
            if (lpOutBuffer) delete[] lpOutBuffer;
            return FALSE;
        }
      }
    }
    srvURL.Header = (wchar_t*)lpOutBuffer;
    returnValue = srvURL.Header.SubString(10, 3);
    if (lpOutBuffer) delete[] lpOutBuffer;    
    return true;
  }
  //---------------------------------------------------------------------------
  // \\61.251.191.50\webdav
	bool __fastcall PProcessHTTPUp(HINTERNET lngServer, ParsedURL &myParsedServerURL,
		wchar_t * strLocalFile)
  {//SendRequestEx를 사용하지 않는 함수, 큰 파일을 전송하거나 중간중간 상태를 보고자 할 때는 부적합함.
    DWORD dwRead, dwWritten, Recv=0;
    HANDLE hRequest=NULL, hLocalFile=NULL;
    int nFilesize;
		bool bHttpSendRequest;
    bool bInternetWriteFile;
		wchar_t *cBuffer;
		String Header;
		String strRemoteFile;
    strRemoteFile = myParsedServerURL.remoteDirectory+myParsedServerURL.remoteFilename;

    //인터넷 파일의 핸들을 얻는다.
//    hRequest = HttpOpenRequest(lngServer, "PUT", strRemoteFile.c_str(), NULL, NULL,
//      0, INTERNET_FLAG_KEEP_CONNECTION | INTERNET_FLAG_RELOAD | INTERNET_FLAG_NO_CACHE_WRITE, 0);
		hRequest = HttpOpenRequest(lngServer, L"PUT", strRemoteFile.c_str(), NULL, NULL,
			0, INTERNET_FLAG_NO_CACHE_WRITE, 0);
    if (!hRequest) goto exit;

    //로컬 파일의 핸들을 얻고, 사이즈를 저장한다.
		hLocalFile=CreateFile(strLocalFile, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    if(!hLocalFile) goto exit;
    nFilesize = GetFileSize(hLocalFile, NULL);

    cBuffer = (wchar_t *)malloc(nFilesize); //메모리 할당 사용.
    if(!ReadFile(hLocalFile, cBuffer, nFilesize, &dwRead, NULL)) goto exit;

    //리퀘스트를 보낸다.
    bHttpSendRequest = HttpSendRequest(hRequest, NULL, NULL, cBuffer, dwRead);

    // 200: OK
    // 201: Created
    // 401: Unauthorized
    // 404: Not Found
		if (!CheckSrvData(hRequest, Header, myParsedServerURL) || !(Header=="200" || Header=="201"))
      goto exit;

    //로컬파일 핸들을 마무리한다.
    if (hLocalFile)
      CloseHandle(hLocalFile);
    //인터넷파일 핸들을 마무리한다.
    if (hRequest)
      InternetCloseHandle(hRequest);

    return bHttpSendRequest; //bHttpSendRequest와 동일한 결과값을 리턴

  exit:
    //로컬파일 핸들을 마무리한다.
    if (hLocalFile)
      CloseHandle(hLocalFile);
    //인터넷파일 핸들을 마무리한다.
    if (hRequest)
      InternetCloseHandle(hRequest);

    return false; //bHttpSendRequest와 동일한 결과값을 리턴

  }

  //---------------------------------------------------------------------------
  BOOL UseHttpSendReqEx(HINTERNET hRequest, DWORD dwPostSize, wchar_t *pBuffer )
  {
    Wininet::INTERNET_BUFFERS BufferIn;
    DWORD dwBytesWritten;
    int sum=0;
    const int BUFFERSIZE = 1024;
    int dwdwBytesToWrite;
//    BYTE pBuffer[1024];
    BOOL bRet;
    BufferIn.dwStructSize = sizeof( Wininet::INTERNET_BUFFERS ); // Must be set or error will occur
    BufferIn.Next = NULL;
    BufferIn.lpcszHeader = NULL;
    BufferIn.dwHeadersLength = 0;
    BufferIn.dwHeadersTotal = 0;
    BufferIn.lpvBuffer = NULL;
    BufferIn.dwBufferLength = 0;
    BufferIn.dwBufferTotal = dwPostSize; // This is the only member used other than dwStructSize
    BufferIn.dwOffsetLow = 0;
    BufferIn.dwOffsetHigh = 0;

    if(!HttpSendRequestEx( hRequest, &BufferIn, NULL, 0, 0))
    {
      //printf( "Error on HttpSendRequestEx %d\n",GetLastError() );
      return FALSE;
    }

//    FillMemory(pBuffer, 1024, 'D'); // Fill buffer with data
    bRet=TRUE;
/*
    for(n=1; n<=(int)dwPostSize/1024 && bRet; n++)
    {
      if(bRet=InternetWriteFile( hRequest, pBuffer, 1024, &dwBytesWritten))
      ;//  printf( "\r%d bytes sent.", n*1024);
    }
    if(!bRet)
    {
      //printf( "\nError on InternetWriteFile %lu\n",GetLastError() );
      return FALSE;
    }
*/
    dwdwBytesToWrite = BUFFERSIZE;
    do {
      sum+=BUFFERSIZE;
      if (sum>dwPostSize) {
        dwdwBytesToWrite=dwPostSize%BUFFERSIZE;
        sum=dwPostSize;
      }
      if (dwdwBytesToWrite==0)
        dwdwBytesToWrite=BUFFERSIZE;
      if(bRet=InternetWriteFile( hRequest, pBuffer, dwdwBytesToWrite, &dwBytesWritten))
      ;//  printf( "\r%d bytes sent.", n*1024);
      if(!bRet)
      {
        //printf( "\nError on InternetWriteFile %lu\n",GetLastError() );
        return FALSE;
      }
    } while(sum<dwPostSize);

    return TRUE;
  }
  //---------------------------------------------------------------------------
  bool __fastcall PProcessHTTPUp2(HINTERNET lngServer, ParsedURL &myParsedServerURL,
		wchar_t * strLocalFile)
  {//SendRequestEx를 사용하지 않는 함수, 큰 파일을 전송하거나 중간중간 상태를 보고자 할 때는 부적합함.
    DWORD dwRead, dwWritten, Recv=0;
    HANDLE hRequest=NULL, hLocalFile=NULL;
    int nFilesize;
    bool bHttpSendRequest;
    bool bInternetWriteFile;
    wchar_t *cBuffer;
    String Header;
		String strRemoteFile;
		strRemoteFile = myParsedServerURL.remoteDirectory+myParsedServerURL.remoteFilename;

    //인터넷 파일의 핸들을 얻는다.
//    hRequest = HttpOpenRequest(lngServer, "PUT", strRemoteFile.c_str(), NULL, NULL,
//      0, INTERNET_FLAG_KEEP_CONNECTION | INTERNET_FLAG_RELOAD | INTERNET_FLAG_NO_CACHE_WRITE, 0);
    hRequest = HttpOpenRequest(lngServer, L"PUT", strRemoteFile.c_str(), NULL, NULL,
      0, INTERNET_FLAG_NO_CACHE_WRITE, 0);
    if (!hRequest) goto exit;

    //로컬 파일의 핸들을 얻고, 사이즈를 저장한다.
    hLocalFile=CreateFile(strLocalFile, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    if(!hLocalFile) goto exit;
    nFilesize = GetFileSize(hLocalFile, NULL);
    cBuffer = (wchar_t *)malloc(nFilesize); //메모리 할당 사용.
    if(!ReadFile(hLocalFile, cBuffer, nFilesize, &dwRead, NULL)) goto exit;
    //로컬파일 핸들을 마무리한다.
    if (hLocalFile)
      CloseHandle(hLocalFile);

    UseHttpSendReqEx(hRequest, dwRead, cBuffer);

    if(!HttpEndRequest(hRequest, NULL, 0, 0))
    {
      //printf( "Error on HttpEndRequest %lu \n", GetLastError());
      return FALSE;
    }
    //리퀘스트를 보낸다.
//    bHttpSendRequest = HttpSendRequest(hRequest, NULL, NULL, cBuffer, dwRead);

    // 200: OK
    // 201: Created
    // 401: Unauthorized
    // 404: Not Found
    if (!CheckSrvData(hRequest, Header, myParsedServerURL) || !(Header=="200" || Header=="201"))
      goto exit;

    //인터넷파일 핸들을 마무리한다.
    if (hRequest)
      InternetCloseHandle(hRequest);

//    return bHttpSendRequest; //bHttpSendRequest와 동일한 결과값을 리턴
    return true;

  exit:
    //로컬파일 핸들을 마무리한다.
    if (hLocalFile)
      CloseHandle(hLocalFile);
    //인터넷파일 핸들을 마무리한다.
    if (hRequest)
      InternetCloseHandle(hRequest);

    return false; //bHttpSendRequest와 동일한 결과값을 리턴

  }

	//---------------------------------------------------------------------------
	bool __fastcall ProcessHTTPUp(HINTERNET lngServer, wchar_t *strRemoteFile,
		wchar_t * strLocalFile)
	//                 HWND hWnd, char *URL, char *Path)
  {//SendRequestEx를 사용하는 함수.
    #define BUFFER_SIZE 10000
    DWORD dwRead, dwWritten, Recv=0;
    wchar_t cBuffer[BUFFER_SIZE];
    HANDLE hInternetFile, hLocalFile;
    int nFilesize;
    bool bHttpSendRequestEx;
    bool bInternetWriteFile;
    //unsigned long nRet;

    //다운로드 중임과 취소는 하지 않음을 전역변수에 기록한다.
    /*bStopHttpDownLoad=false;
    bDownLoading=false;*/

    //인터넷 파일의 핸들을 얻는다.
		hInternetFile = HttpOpenRequest(lngServer, L"PUT", strRemoteFile , NULL, NULL,
      0, INTERNET_FLAG_KEEP_CONNECTION | INTERNET_FLAG_RELOAD | INTERNET_FLAG_NO_CACHE_WRITE, 0);

    //로컬 파일의 핸들을 얻고, 사이즈를 저장한다.
    hLocalFile=CreateFile(strLocalFile, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    nFilesize = GetFileSize(hLocalFile, NULL);

    //리퀘스트를 보낸다.
    Wininet::INTERNET_BUFFERS *BufferIn = new Wininet::INTERNET_BUFFERS();
    //LPINTERNET_BUFFERS BufferIn = new Wininet::INTERNET_BUFFERS();
    BufferIn->dwStructSize = sizeof(Wininet::INTERNET_BUFFERS);
    BufferIn->dwBufferTotal = nFilesize;
    bHttpSendRequestEx = HttpSendRequestEx(hInternetFile, BufferIn, NULL, WININET_API_FLAG_SYNC|HSR_INITIATE, 0);
    //bHttpSendRequestEx = HttpSendRequestEx(hInternetFile, BufferIn, NULL, 0, 0);
    //WININET_API_FLAG_SYNC|HSR_INITIATE 과 0을 쓰는 것의 차이점을 알기 위해선 MSDN을 조사해야함.
    if(bHttpSendRequestEx == 0) //실패했으면
    {
      DWORD error;
      error = GetLastError();
      //로그를 작성한 뒤
      //로컬 파일을 생성한다.
			HANDLE hFile=CreateFile(L"C:\\kyuwonlog.txt", GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
      do {
        //다운로드를 받아 로컬 파일에 기록한다.
        DWORD Size;
        wchar_t buf[BUFFER_SIZE];
        InternetQueryDataAvailable(hInternetFile, &Size, 0, 0);
        InternetReadFile(hInternetFile, buf, Size, &dwRead);
        WriteFile(hFile, buf, dwRead, &dwWritten, NULL);
        //다운로드 받은 양을 메세지로 전달
        Recv+=dwWritten;
        //SendMessage(hWnd, WM_USER+10, 0, (LPARAM)Recv);
        //취소 처리
        /*if (bStopHttpDownLoad) {
          break;
        }*/
      } while(dwRead!=0);
      CloseHandle(hFile);
      if (BufferIn) delete BufferIn;
      return false; //실패값을 리턴
    }

    //쓰기 작업을 수행한다.
    for(int i=0; i<nFilesize; i+=BUFFER_SIZE){
      /*해당 코드*///파일에서 버퍼로 버퍼사이즈만큼 읽기
      ReadFile(hLocalFile, cBuffer, BUFFER_SIZE, &dwRead, NULL);
      bInternetWriteFile = InternetWriteFile(hInternetFile, cBuffer, dwRead, &dwWritten);
      DWORD error;
      error = GetLastError();
      Recv+=dwWritten;
    }

    //로컬파일 핸들을 마무리한다.
    CloseHandle(hLocalFile);
    //인터넷파일 핸들을 마무리한다.
    InternetCloseHandle(hInternetFile);
    if (BufferIn) delete BufferIn;    
    return true; //성공값을 리턴
  }

  //---------------------------------------------------------------------------
	bool __fastcall HTTPUp(HWND hWnd, ParsedURL &myParsedServerURL, wchar_t *strLocalFile)
  {
    HINTERNET hInternet, hURL;

    //인터넷 연결
		hInternet=InternetOpen(L"HTTPClient"/*User-Agent 헤더값*/, INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);
    if (hInternet==NULL) return false;

    //URL을 연다.
    //hURL = InternetConnect(hInternet, myParsedServerURL.serverAddress.c_str() /*Host 헤더값, 서버 주소*/,
    //  INTERNET_DEFAULT_HTTP_PORT, NULL/*strID*/, NULL/*strPassword*/,
    //  INTERNET_SERVICE_HTTP, 0, 0);
    hURL = InternetConnect(hInternet, myParsedServerURL.serverAddress.c_str() /*Host 헤더값, 서버 주소*/,
      myParsedServerURL.Port, myParsedServerURL.ID.c_str()/*strID*/,
      myParsedServerURL.PW.c_str()/*strPassword*/,
      INTERNET_SERVICE_HTTP, 0, 0);
    //hURL=InternetOpenUrl(hInternet, URL, NULL, 0, INTERNET_FLAG_RELOAD, 0);
    if (hURL==NULL) {
      InternetCloseHandle(hInternet);
      return false;
    }

    //동작수행시작
    //.
    //.
    bool bHTTPUp;
//    bHTTPUp = PProcessHTTPUp(hURL, myParsedServerURL, strLocalFile);
    bHTTPUp = PProcessHTTPUp2(hURL, myParsedServerURL, strLocalFile);
    //.
    //.
    //동작수행끝
    if (hURL)
      InternetCloseHandle(hURL);
    if (hInternet)
      InternetCloseHandle(hInternet);

    //다운로드 완료 처리 (최소한 경우는 제외)
  /*
    if (bStopHttpDownLoad==false)
      SendMessage(hWnd, WM_USER+11, 0, (LPARAM)Recv);
  */
    return bHTTPUp;//true;
  }

  //---------------------------------------------------------------------------
	bool __fastcall HTTPUp(HWND hWnd, wchar_t *URL, wchar_t *strLocalFile)
  {
    ParsedURL myParsedServerURL;
    parseURL(URL, myParsedServerURL);
    return HTTPUp(hWnd, myParsedServerURL, strLocalFile);
  }

  //---------------------------------------------------------------------------
	bool __fastcall HTTPDown(HWND hWnd, wchar_t * URL, wchar_t * strLocalFile,
		ParsedURL &myURL)
  {
    HINTERNET hInternet=NULL, hURL=NULL;
    HANDLE hFile=NULL;
    DWORD Size;
    DWORD dwRead, dwWritten, Recv=0;
    wchar_t buf[100000];
    MSG Message;
    String Header;

    //인터넷 연결
    hInternet=InternetOpen(L"HTTPClient", INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0);
    if (hInternet==NULL) return false;

    //URL을 연다.
    hURL=InternetOpenUrl(hInternet, URL, NULL, 0, INTERNET_FLAG_RELOAD, 0);
  //    GetLastError(hURL);
    if (hURL==NULL) {
      InternetCloseHandle(hInternet);
      return false;
    }

    if (!CheckSrvData(hURL, Header, myURL) || Header!="200") goto exit;

    //다운로드 중임과 취소는 하지 않음을 전역변수에 기록한다.
    /*bStopHttpDownLoad=false;
    bDownLoading=false;*/

    //로컬 파일을 생성한다.
    hFile=CreateFile(strLocalFile, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);

    do {
      //다운로드를 받아 로컬 파일에 기록한다.
      if (!InternetQueryDataAvailable(hURL, &Size, 0, 0)) goto exit;
      if (!InternetReadFile(hURL, buf, Size, &dwRead)) goto exit;
      if (!WriteFile(hFile, buf, dwRead, &dwWritten, NULL)) goto exit;

      //다운로드 받은 양을 메세지로 전달
      Recv+=dwWritten;
  //    SendMessage(hWnd, WM_USER+10, 0, (LPARAM)Recv);

      //취소 처리
      /*if (bStopHttpDownLoad) {
        break;
      }*/
    } while(dwRead!=0);

    if (hURL)
      InternetCloseHandle(hURL);
    if (hInternet)
      InternetCloseHandle(hInternet);
    if (hFile)
      CloseHandle(hFile);
    //bDownLoading=false;

    //다운로드 완료 처리 (최소한 경우는 제외)
  /*
    if (bStopHttpDownLoad==false)
      SendMessage(hWnd, WM_USER+11, 0, (LPARAM)Recv);
  */
    return true;

  exit:
    if (hURL)
      InternetCloseHandle(hURL);
    if (hInternet)
      InternetCloseHandle(hInternet);
    if (hFile)
      CloseHandle(hFile);
    return false;
  }
//---------------------------------------------------------------------------
}



#endif //QWON_HTTP_CLIENT_UTIL