/*
  Licensed to the Apache Software Foundation (ASF) under one
  or more contributor license agreements.  See the NOTICE file
  distributed with this work for additional information
  regarding copyright ownership.  The ASF licenses this file
  to you under the Apache License, Version 2.0 (the
  "License"); you may not use this file except in compliance
  with the License.  You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

  Unless required by applicable law or agreed to in writing,
  software distributed under the License is distributed on an
  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  KIND, either express or implied.  See the License for the
  specific language governing permissions and limitations
  under the License.
 */

#import "CMISHttpUtil.h"
#import "CMISAuthenticationProvider.h"
#import "CMISErrors.h"
#import "CMISHttpRequest.h"
#import "CMISHttpDownloadRequest.h"
#import "CMISHttpUploadRequest.h"
#import "CMISRequest.h"


@implementation HttpUtil

#pragma mark block based methods

+ (void)invoke:(NSURL *)url
withHttpMethod:(CMISHttpRequestMethod)httpRequestMethod
   withSession:(CMISBindingSession *)session
          body:(NSData *)body
       headers:(NSDictionary *)additionalHeaders
completionBlock:(void (^)(CMISHttpResponse *httpResponse, NSError *error))completionBlock
{
    NSMutableURLRequest *urlRequest = [self createRequestForUrl:url
                                                 withHttpMethod:httpRequestMethod
                                                   usingSession:session];
    
    [CMISHttpRequest startRequest:urlRequest
                   withHttpMethod:httpRequestMethod
                      requestBody:body
                          headers:additionalHeaders
           authenticationProvider:session.authenticationProvider
                  completionBlock:completionBlock];
}

+ (void)invoke:(NSURL *)url
withHttpMethod:(CMISHttpRequestMethod)httpRequestMethod
   withSession:(CMISBindingSession *)session
   inputStream:(NSInputStream *)inputStream
       headers:(NSDictionary *)additionalHeaders
completionBlock:(void (^)(CMISHttpResponse *httpResponse, NSError *error))completionBlock
{
    NSMutableURLRequest *urlRequest = [self createRequestForUrl:url
                                                 withHttpMethod:httpRequestMethod
                                                   usingSession:session];
    
    [CMISHttpUploadRequest startRequest:urlRequest
                         withHttpMethod:httpRequestMethod
                            inputStream:inputStream
                                headers:additionalHeaders
                          bytesExpected:0
                 authenticationProvider:session.authenticationProvider
                        completionBlock:completionBlock
                          progressBlock:nil];
}

+ (void)invoke:(NSURL *)url
withHttpMethod:(CMISHttpRequestMethod)httpRequestMethod
   withSession:(CMISBindingSession *)session
   inputStream:(NSInputStream *)inputStream
       headers:(NSDictionary *)additionalHeaders
 bytesExpected:(unsigned long long)bytesExpected
completionBlock:(void (^)(CMISHttpResponse *httpResponse, NSError *error))completionBlock
 progressBlock:(void (^)(unsigned long long bytesDownloaded, unsigned long long bytesTotal))progressBlock
 requestObject:(CMISRequest *)requestObject
{
    if (!requestObject.isCancelled) {
        NSMutableURLRequest *urlRequest = [self createRequestForUrl:url
                                                     withHttpMethod:httpRequestMethod
                                                       usingSession:session];
        
        CMISHttpUploadRequest *uploadRequest = [CMISHttpUploadRequest startRequest:urlRequest
                                                                    withHttpMethod:httpRequestMethod
                                                                       inputStream:inputStream
                                                                           headers:additionalHeaders
                                                                     bytesExpected:bytesExpected
                                                            authenticationProvider:session.authenticationProvider
                                                                   completionBlock:completionBlock
                                                                     progressBlock:progressBlock];
        requestObject.httpRequest = uploadRequest;
    } else {
        if (completionBlock) {
            completionBlock(nil, [CMISErrors createCMISErrorWithCode:kCMISErrorCodeCancelled
                                             withDetailedDescription:@"Request was cancelled"]);
        }
    }
}

+ (void)invoke:(NSURL *)url
   withHttpMethod:(CMISHttpRequestMethod)httpRequestMethod
      withSession:(CMISBindingSession *)session
     outputStream:(NSOutputStream *)outputStream
    bytesExpected:(unsigned long long)bytesExpected
  completionBlock:(void (^)(CMISHttpResponse *httpResponse, NSError *error))completionBlock
    progressBlock:(void (^)(unsigned long long bytesDownloaded, unsigned long long bytesTotal))progressBlock
    requestObject:(CMISRequest *)requestObject
{
    if (!requestObject.isCancelled) {
        NSMutableURLRequest *urlRequest = [self createRequestForUrl:url
                                                     withHttpMethod:HTTP_GET
                                                       usingSession:session];
    
        CMISHttpDownloadRequest *downloadRequest = [CMISHttpDownloadRequest startRequest:urlRequest
                                                                          withHttpMethod:httpRequestMethod
                                                                            outputStream:outputStream
                                                                           bytesExpected:bytesExpected
                                                                  authenticationProvider:session.authenticationProvider
                                                                         completionBlock:completionBlock
                                                                           progressBlock:progressBlock];
        requestObject.httpRequest = downloadRequest;
    } else {
        if (completionBlock) {
            completionBlock(nil, [CMISErrors createCMISErrorWithCode:kCMISErrorCodeCancelled
                                             withDetailedDescription:@"Request was cancelled"]);

        }
    }
}

+ (void)invokeGET:(NSURL *)url
      withSession:(CMISBindingSession *)session
  completionBlock:(void (^)(CMISHttpResponse *httpResponse, NSError *error))completionBlock
{
    return [self invoke:url
         withHttpMethod:HTTP_GET
            withSession:session
                   body:nil
                headers:nil
        completionBlock:completionBlock];
}

+ (void)invokePOST:(NSURL *)url
       withSession:(CMISBindingSession *)session
              body:(NSData *)body
           headers:(NSDictionary *)additionalHeaders
   completionBlock:(void (^)(CMISHttpResponse *httpResponse, NSError *error))completionBlock
{
    return [self invoke:url
         withHttpMethod:HTTP_POST
            withSession:session
                   body:body
                headers:additionalHeaders
        completionBlock:completionBlock];
}

+ (void)invokePUT:(NSURL *)url
      withSession:(CMISBindingSession *)session
             body:(NSData *)body
          headers:(NSDictionary *)additionalHeaders
  completionBlock:(void (^)(CMISHttpResponse *httpResponse, NSError *error))completionBlock
{
    return [self invoke:url
         withHttpMethod:HTTP_PUT
            withSession:session
                   body:body
                headers:additionalHeaders
        completionBlock:completionBlock];
}

+ (void)invokeDELETE:(NSURL *)url
         withSession:(CMISBindingSession *)session
     completionBlock:(void (^)(CMISHttpResponse *httpResponse, NSError *error))completionBlock
{
    return [self invoke:url
         withHttpMethod:HTTP_DELETE
            withSession:session
                   body:nil
                headers:nil
        completionBlock:completionBlock];
}

#pragma mark Helper methods

+ (NSMutableURLRequest *)createRequestForUrl:(NSURL *)url
                              withHttpMethod:(CMISHttpRequestMethod)httpRequestMethod
                                usingSession:(CMISBindingSession *)session
{
    NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url
                                                           cachePolicy:NSURLRequestReloadIgnoringCacheData
                                                       timeoutInterval:60];
    NSString *httpMethod;
    switch (httpRequestMethod) {
        case HTTP_GET:
            httpMethod = @"GET";
            break;
        case HTTP_POST:
            httpMethod = @"POST";
            break;
        case HTTP_DELETE:
            httpMethod = @"DELETE";
            break;
        case HTTP_PUT:
            httpMethod = @"PUT";
            break;
        default:
            log(@"Invalid http request method: %d", httpRequestMethod);
            return nil;
    }

    [request setHTTPMethod:httpMethod];
    log(@"HTTP %@: %@", httpMethod, [url absoluteString]);

    return request;
}

@end