import "Vector"
import "File"
import "HTTP/Form"
import "HTTP/MultiPart"
import "HTTP/MediaType"
import "Future"

type ResponseHandler = Response -> <Proc> ()
type FailureHandler = Throwable -> <Proc> ()

NO_RESPONSE_HANDLER = (\response -> ()) :: ResponseHandler 
NO_FAILURE_HANDLER = (\failure -> ()) :: FailureHandler 

importJava "javax.ws.rs.client.Invocation" where
    data Invocation

    @JavaName invoke
    syncInvoke :: Invocation -> <Proc, Exception> Response

importJava "javax.ws.rs.core.Response" where
    data Response
    @JavaName getStatus
    statusCodeOf :: Response -> <Proc> Integer
    @Private
    @JavaName readEntity
    readEntity_ :: Response -> Class a -> <Proc, Exception> a
    @JavaName getHeaderString
    possibleHeaderOf :: Response -> String -> <Proc> Maybe String

readEntity :: VecComp a => Response -> <Proc, Exception> a
readEntity response = readEntity_ response classObject

importJava "javax.ws.rs.client.Invocation$Builder" where
    data Builder
    
    header :: Builder -> String -> String -> <Proc> Builder
    @private
    @JavaName accept
    acceptMediaType_ :: Builder -> Vector MediaType -> <Proc> Builder
    @private
    @JavaName acceptEncoding
    acceptEncoding_ :: Builder -> Vector String -> <Proc> Builder
    @private
    @JavaName acceptLanguage
    acceptLanguage_ :: Builder -> Vector String -> <Proc> Builder
    cookie :: Builder -> String -> String -> <Proc> Builder
    build :: Builder -> String -> <Proc> Invocation
    buildGet :: Builder -> <Proc> Invocation
    buildPost :: Builder -> Entity -> <Proc> Invocation
    buildDelete :: Builder -> <Proc> Invocation
    buildPut :: Builder -> Entity -> <Proc> Invocation

acceptMediaType :: Builder -> [MediaType] -> <Proc> Builder
acceptMediaType builder mediaTypes =
    acceptMediaType_ builder $ vector mediaTypes

acceptEncoding :: Builder -> [String] -> <Proc> Builder
acceptEncoding builder encodings =
    acceptEncoding_ builder $ vector encodings

acceptLanguage :: Builder -> [String] -> <Proc> Builder
acceptLanguage builder languages =
    acceptEncoding_ builder $ vector languages

importJava "javax.ws.rs.client.WebTarget" where
    data WebTarget

    path :: WebTarget -> String -> <Proc> WebTarget
    request :: WebTarget -> <Proc> Builder

importJava "javax.ws.rs.client.ClientBuilder" where
    data ClientBuilder
    
    @JavaName newBuilder
    clientBuilder :: <Proc> ClientBuilder

importJava "org.simantics.scl.rest.HttpClientUtils" where
    buildClient :: ClientBuilder -> <Proc> Client
    statusMessageOf :: Response -> <Proc> String
    asyncInvoke :: Invocation -> ResponseHandler -> FailureHandler -> <Proc> Future Response
    trustAllClientBuilder :: <Proc> ClientBuilder
    onReadProgress :: WebTarget -> (Long -> <Proc> ()) -> <Proc> ()
    onWriteProgress :: WebTarget -> (Long -> <Proc> ()) -> <Proc> ()
    possibleContentLengthOf :: Response -> <Proc> Maybe Long

importJava "javax.ws.rs.client.Entity" where
    data Entity
    
    @JavaName form
    formEntity :: Form -> <Proc> Entity
    entity :: a -> MediaType -> <Proc> Entity

importJava "javax.ws.rs.client.Client" where
    data Client

    target :: Client -> String -> <Proc> WebTarget
    close :: Client -> <Proc> ()

/*
Example usage:

getExample :: String -> <Proc, Exception> String
getExample uri = do
    httpClient = buildClient clientBuilder
    webTarget = target httpClient uri
    builder = request webTarget
    acceptMediaType builder [WILDCARD_TYPE]
    invocation = buildGet builder
    response = syncInvoke invocation
    print $ statusCodeOf response
    print $ statusMessageOf response
    res = readEntity response
    close httpClient
    res

postMultiPartExample :: String -> [File] -> <Proc, Exception> String
postMultiPartExample uri files = do  
    httpClient = buildClient clientBuilder
    webTarget = target httpClient uri
    builder = request webTarget
    acceptMediaType builder [WILDCARD_TYPE]
    mp = formDataMultiPart
    addBodyPart mp $ formDataBodyPart "Name" "Test" $ withCharset TEXT_PLAIN_TYPE "utf-8"
    iterI (\i f ->
      addBodyPart mp $ fileDataBodyPart ("file" + show i) f APPLICATION_OCTET_STREAM_TYPE
    ) files
    invocation = buildPost builder $ entity mp MULTIPART_FORM_DATA_TYPE
    response = syncInvoke invocation
    print $ statusCodeOf response
    print $ statusMessageOf response
    res = readEntity response
    close httpClient
    res

postFileExample :: String -> File -> <Proc, Exception> String
postFileExample uri f = do  
    httpClient = buildClient clientBuilder
    webTarget = target httpClient uri
    builder = request webTarget
    acceptMediaType builder [WILDCARD_TYPE]
    invocation = buildPost builder $ entity f APPLICATION_OCTET_STREAM_TYPE
    response = syncInvoke invocation
    print $ statusCodeOf response
    print $ statusMessageOf response
    res = readEntity response
    close httpClient
    res

postFileAsyncExample :: String -> File -> <Proc, Exception> ()
postFileAsyncExample uri f = do  
    httpClient = buildClient clientBuilder
    webTarget = target httpClient uri
    builder = request webTarget
    acceptMediaType builder [WILDCARD_TYPE]
    mp = formDataMultiPart
    invocation = buildPost builder $ entity f APPLICATION_OCTET_STREAM_TYPE
    f = asyncInvoke invocation 
      (\response -> do
        print $ statusCodeOf response
        print $ statusMessageOf response
        catch (do
          content = ((readEntity response) :: String)
          print $ length content
        ) (\(t :: Throwable) -> print $ show t) 
        close httpClient
        ()
      )
      (\throwable -> do
        print $ show throwable
        close httpClient
        ()
      )
    print $ isFutureDone f
*/