#INCLUDE "winhttp.bi"
#INCLUDE "http-client.bi"

'Implement friendly interfaces to interact with WinHTTP api

CONST AS STRING IguanaUserAgent = "IguanaCMS HTTP Client/1.0"

DIM SHARED AS ULONG PTR SessionHandler, ConnectionHandler, RequestHandler
DIM SHARED RequestMethod AS STRING, ResponseReceived AS LONG
DIM SHARED RequestMethod AS STRING, ResponseHeaders() AS STRING, ResponseReceived AS LONG

FUNCTION IsValidMethod(CurrentMethod AS STRING) AS BYTE
  DIM ArrayMethods(HTTP_Methods.Count - 1) AS STRING = { _
    "HEAD", "GET", "POST", "PUT", "DELETE" _
  }
  FOR i AS ULONG = 0 TO HTTP_Methods.Count - 1
    IF UCASE(CurrentMethod) = ArrayMethods(i) THEN
      IsValidMethod = -1
      EXIT FOR
    END IF
  NEXT
END FUNCTION


FUNCTION SplitUrl(FullUrl AS STRING, UrlPart AS URL_Properties) AS STRING
  IF FullUrl <> "" AND UrlPart < URL_Properties.Count THEN
    DIM Result AS STRING
    DIM AS ULONG ColonPos, DoubleSlashPos, PathStartPos
    ColonPos = INSTR(FullUrl, ":")
    DoubleSlashPos = INSTR(FullUrl, "//")

      CASE URL_URI
        PathStartPos = INSTR(DoubleSlashPos + 2, FullUrl, "/")
        Result = MID(FullUrl, PathStartPos)
      'CASE URL_QueryString
      'CASE URL_Fragment
    END SELECT
    SplitUrl = Result
  END IF
END FUNCTION

SUB ParseResponseHeaders(HeadersContent AS STRING)
  IF HeadersContent <> "" THEN
    DIM Char AS UBYTE, LineCount AS ULONG
    DIM AS STRING CleanHeaders, NewLine = CHR(13, 10)
    FOR i AS ULONG = 1 TO LEN(HeadersContent)
      Char = ASC(MID(HeadersContent, i, 1))
      SELECT CASE Char
        CASE 0
          i += 2
        CASE 10
          CleanHeaders += CHR(Char)
          i += 1
          LineCount += 1
        CASE ELSE
          CleanHeaders += CHR(Char)
          i += 1
      END SELECT
    NEXT
    IF LineCount THEN
      DIM AS ULONG IndexPos, NextPos, StartPos = 1
      REDIM ResponseHeaders(LineCount - 1)
      NextPos = StartPos
      WHILE NextPos
        NextPos = INSTR(StartPos, CleanHeaders, NewLine)
        'PRINT MID(CleanHeaders, StartPos, NextPos - StartPos)
        ResponseHeaders(IndexPos) = MID(CleanHeaders, StartPos, NextPos - StartPos)
        IndexPos += 1
        StartPos = NextPos + 2
      WEND
    END IF
  END IF
END SUB

SUB GetResponseHeaders()
  'Partially based on http://www.jose.it-berater.org/winhttp/winhttpqueryheaders.htm
  IF RequestHandler THEN
      DIM ResponseContent AS STRING, ResponseLen AS ULONG, QueryFlag AS ULONG
      'Fail the first attemtp, but gain the buffer length
      IF NOT WinHttpQueryHeaders( _
        RequestHandler, _
        WINHTTP_QUERY_RAW_HEADERS_CRLF, _
        WINHTTP_HEADER_NAME_BY_INDEX, _
        WINHTTP_NO_OUTPUT_BUFFER, _
        VarPtr(ResponseLen), _
        WINHTTP_NO_HEADER_INDEX _
      ) THEN
        LastError = GetLastError()
        IF LastError = ERROR_INSUFFICIENT_BUFFER THEN
          ResponseContent = SPACE(ResponseLen)
          'Now we know the size, get full headers
          IF WinHttpQueryHeaders( _
            RequestHandler, _
            WINHTTP_QUERY_RAW_HEADERS_CRLF, _
            WINHTTP_HEADER_NAME_BY_INDEX, _
            ResponseContent, _
            VarPtr(ResponseLen), _
            WINHTTP_NO_HEADER_INDEX _
          ) THEN
            ParseResponseHeaders(ResponseContent)
          END IF
        END IF
      END IF
  END IF
END SUB

FUNCTION HTTP_GetResponseHeader(HeaderName AS STRING) AS STRING
  IF HeaderName <> "" THEN
    FOR i AS ULONG = 0 TO UBOUND(ResponseHeaders)
      IF LCASE(HeaderName) = "status" THEN
        IF INSTR(ResponseHeaders(i), "HTTP") THEN
          HTTP_GetResponseHeader = MID(ResponseHeaders(i), LEN("HTTP/1.1") + 1)
          EXIT FOR
        END IF
      ELSE
        IF LCASE(HeaderName) = LCASE(RTRIM(MID(ResponseHeaders(i), 1, INSTR(ResponseHeaders(i), ":") - 1))) THEN
          HTTP_GetResponseHeader = LTRIM(MID(ResponseHeaders(i), INSTR(ResponseHeaders(i), ":") + 1))
          EXIT FOR
        END IF
      END IF
    NEXT
  END IF
END FUNCTION

SUB HTTP_Open(Method AS STRING, Url AS STRING)
  DIM AS ULONG ServerPort, RequestFlags
  DIM AS STRING ServerProtocol
  IF Method <> "" AND IsValidMethod(Method) THEN

        WINHTTP_NO_REFERER, _
        WINHTTP_DEFAULT_ACCEPT_TYPES, _
        RequestFlags _
      )
    END IF
  END IF
END SUB

SUB HTTP_SetRequestHeader(HeaderName AS STRING, HeaderValue AS STRING)
  IF RequestHandler THEN
    IF HeaderName <> "" AND HeaderValue <> "" THEN
      DIM CompleteHeader AS STRING
      CompleteHeader = HeaderName + ": " + HeaderValue
      IF NOT WinHttpAddRequestHeaders( _
          RequestHandler, _
          WSTR(CompleteHeader), _
          LEN(WSTR(CompleteHeader)), _
          0 _
        ) THEN
        LastError = GetLastError()
      END IF
    END IF
  END IF
END SUB

SUB HTTP_Send(RequestBody AS STRING = "")
  IF RequestHandler THEN
    DIM RequestFlags AS ULONG, ReTry AS LONG, RequestLen AS ULONG, RequestData AS WSTRING PTR
    DIM RequestFlags AS ULONG, ReTry AS LONG, RequestLen AS ULONG, RequestData AS STRING
    IF RequestBody <> "" AND RequestMethod = "POST" OR RequestMethod = "PUT" THEN
      RequestData[0] = WSTR(RequestBody)
      RequestData = RequestBody
      RequestLen = LEN(RequestData)
    ELSE
      RequestData = WINHTTP_NO_REQUEST_DATA
    END IF
    DO
      ReTry = 0
      IF WinHttpSendRequest( _
        RequestHandler, _
        WINHTTP_NO_ADDITIONAL_HEADERS, _
        0, _
        VarPtr(RequestData), _
        StrPtr(RequestData), _
        RequestLen, _
        RequestLen, _
        0 _
      ) THEN
        ResponseReceived = WinHttpReceiveResponse(RequestHandler)
        IF ResponseReceived THEN
          GetResponseHeaders()
        END IF
      ELSE
        LastError = GetLastError()
        IF LastError THEN
          SELECT CASE LastError
            CASE ERROR_WINHTTP_SECURE_INVALID_CA
              RequestFlags = _
                SECURITY_FLAG_IGNORE_UNKNOWN_CA OR _

      END IF
    LOOP WHILE ReTry
  END IF
END SUB

FUNCTION HTTP_GetResponse() AS STRING
  IF ResponseReceived THEN
    DIM AS STRING ResponseText, TempResponse
    DIM ResponseLen() AS UBYTE, ResponseText AS STRING, SavedLen AS ULONG, CharsLeft AS ULONG
    DIM AS ULONG SavedLen, CharsLeft
    DO
      CharsLeft = 0
      IF WinHttpQueryDataAvailable(RequestHandler, VarPtr(CharsLeft)) THEN
        REDIM ResponseLen(CharsLeft + 1)
        IF WinHttpReadData(RequestHandler, VarPtr(ResponseLen(0)), CharsLeft, VarPtr(SavedLen)) THEN
          FOR i AS ULONG = 0 TO UBOUND(ResponseLen)
            ResponseText += CHR(ResponseLen(i))
        TempResponse = SPACE(CharsLeft)
        IF WinHttpReadData( _
            RequestHandler, _
            StrPtr(TempResponse), _
            CharsLeft, _
            VarPtr(SavedLen) _
          ) THEN
          ResponseText += TempResponse
          NEXT
        END IF
      END IF
    LOOP UNTIL CharsLeft = 0
    HTTP_GetResponse = ResponseText
  END IF
END FUNCTION

SUB HTTP_Clean()
  WinHttpCloseHandle(RequestHandler)
  WinHttpCloseHandle(ConnectionHandler)
  WinHttpCloseHandle(SessionHandler)
END SUB

  HTTP_DELETE
  Count
END ENUM

DECLARE FUNCTION SplitUrl(FullUrl AS STRING, UrlPart AS URL_Properties) AS STRING
DECLARE SUB HTTP_Open(Method AS STRING, Url AS STRING)
DECLARE SUB HTTP_Send(RequestBody AS STRING = "")
DECLARE SUB HTTP_SetRequestHeader(HeaderName AS STRING, HeaderValue AS STRING)
DECLARE FUNCTION HTTP_GetResponseHeader(HeaderName AS STRING) AS STRING
DECLARE FUNCTION HTTP_GetResponse() AS STRING
DECLARE SUB HTTP_Clean()

COMMON SHARED LastError AS ULONG

CONST WINHTTP_OPTION_SECURITY_FLAGS = 31

CONST SECURITY_FLAG_IGNORE_UNKNOWN_CA = &H100
CONST SECURITY_FLAG_IGNORE_CERT_WRONG_USAGE = &H200
CONST SECURITY_FLAG_IGNORE_CERT_CN_INVALID = &H1000
CONST SECURITY_FLAG_IGNORE_CERT_DATE_INVALID = &H2000

CONST WINHTTP_HEADER_NAME_BY_INDEX = 0
CONST WINHTTP_NO_OUTPUT_BUFFER = 0
CONST WINHTTP_NO_HEADER_INDEX = 0

CONST WINHTTP_QUERY_RAW_HEADERS = 21
CONST WINHTTP_QUERY_RAW_HEADERS_CRLF = 22

CONST ERROR_INSUFFICIENT_BUFFER = 122

ENUM
  WINHTTP_ERROR_BASE = 12000
  ERROR_WINHTTP_OUT_OF_HANDLES = WINHTTP_ERROR_BASE + 1
  ERROR_WINHTTP_TIMEOUT = WINHTTP_ERROR_BASE + 2
  ERROR_WINHTTP_INTERNAL_ERROR = WINHTTP_ERROR_BASE + 4
  ERROR_WINHTTP_INVALID_URL = WINHTTP_ERROR_BASE + 5
  ERROR_WINHTTP_UNRECOGNIZED_SCHEME = WINHTTP_ERROR_BASE + 6

    dwFlags AS ULONG _
  ) AS ULONG PTR
  
  DECLARE FUNCTION WinHttpQueryDataAvailable( _
    hRequest AS ULONG PTR, _
    lpdwNumberOfBytesAvailable AS ULONG PTR _
  ) AS LONG
  
  DECLARE FUNCTION WinHttpQueryHeaders( _
    hRequest AS ULONG PTR, _
    dwInfoLevel AS ULONG, _
    pwszName AS CONST ZSTRING PTR, _
    lpBuffer AS ZSTRING PTR, _
    lpdwBufferLength AS ULONG PTR, _
    lpdwIndex AS ULONG PTR _
  ) AS LONG
  
  DECLARE FUNCTION WinHttpReadData( _
    hRequest AS ULONG PTR, _
    lpBuffer AS ULONG PTR, _
    lpBuffer AS ZSTRING PTR, _
    dwNumberOfBytesToRead AS ULONG, _
    lpdwNumberOfBytesRead AS ULONG PTR _
  ) AS LONG
  
  DECLARE FUNCTION WinHttpReceiveResponse( _
    hRequest AS ULONG PTR, _
    lpReserved AS LONG = 0 _
  ) AS LONG
  
  DECLARE FUNCTION WinHttpSendRequest( _
    hRequest AS ULONG PTR, _
    lpszHeaders AS CONST WSTRING PTR, _
    lpszHeaders AS CONST ZSTRING PTR, _
    dwHeadersLength AS ULONG, _
    lpOptional AS ULONG PTR, _
    lpOptional AS CONST ZSTRING PTR, _
    dwOptionalLength AS ULONG, _
    dwTotalLength AS ULONG, _
    dwContext AS ULONG PTR _
  ) AS LONG
  
  DECLARE FUNCTION WinHttpSetOption( _
    hInternet AS ULONG PTR, _