//
//  RCTImageRequestHandler.m
//  RCTImage
//
//  Created by Nick Lockwood on 09/06/2015.
//  Copyright (c) 2015 Facebook. All rights reserved.
//

#import "RCTImageRequestHandler.h"

#import <UIKit/UIKit.h>

#import "RCTBridge.h"
#import "RCTImageLoader.h"
#import "RCTUtils.h"

@implementation RCTImageRequestHandler

RCT_EXPORT_MODULE()

@synthesize bridge = _bridge;

- (BOOL)canHandleRequest:(NSURLRequest *)request
{
  return [@[@"assets-library", @"ph"] containsObject:request.URL.scheme.lowercaseString];
}

- (id)sendRequest:(NSURLRequest *)request
     withDelegate:(id<RCTURLRequestDelegate>)delegate
{
  NSString *URLString = request.URL.absoluteString;

  __block RCTImageLoaderCancellationBlock requestToken = nil;
  requestToken = [_bridge.imageLoader loadImageWithTag:URLString callback:^(NSError *error, UIImage *image) {
    if (error) {
      [delegate URLRequest:requestToken didCompleteWithError:error];
      return;
    }

    NSString *mimeType = nil;
    NSData *imageData = nil;
    if (RCTImageHasAlpha(image.CGImage)) {
      mimeType = @"image/png";
      imageData = UIImagePNGRepresentation(image);
    } else {
      mimeType = @"image/jpeg";
      imageData = UIImageJPEGRepresentation(image, 1.0);
    }

    NSURLResponse *response = [[NSURLResponse alloc] initWithURL:request.URL
                                                        MIMEType:mimeType
                                           expectedContentLength:imageData.length
                                                textEncodingName:nil];

    [delegate URLRequest:requestToken didReceiveResponse:response];
    [delegate URLRequest:requestToken didReceiveData:imageData];
    [delegate URLRequest:requestToken didCompleteWithError:nil];
  }];

  return requestToken;
}

- (void)cancelRequest:(id /* RCTImageLoaderCancellationBlock */)requestToken
{
  if (requestToken) {
    ((RCTImageLoaderCancellationBlock)requestToken)();
  }
}

@end