/**
 * Copyright (c) 2015-present, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under the BSD-style license found in the
 * LICENSE file in the root directory of this source tree. An additional grant
 * of patent rights can be found in the PATENTS file in the same directory.
 */

#import "RCTImageStoreManager.h"

#import "RCTAssert.h"
#import "RCTUtils.h"

@implementation RCTImageStoreManager
{
  NSMutableDictionary *_store;
}

@synthesize methodQueue = _methodQueue;

RCT_EXPORT_MODULE()

- (id)init
{
  if ((self = [super init])) {

    // TODO: need a way to clear this store
    _store = [[NSMutableDictionary alloc] init];
  }
  return self;
}

- (NSString *)storeImage:(UIImage *)image
{
  RCTAssertMainThread();
  NSString *tag = [NSString stringWithFormat:@"rct-image-store://%tu", [_store count]];
  _store[tag] = image;
  return tag;
}

- (UIImage *)imageForTag:(NSString *)imageTag
{
  RCTAssertMainThread();
  return _store[imageTag];
}

- (void)storeImage:(UIImage *)image withBlock:(void (^)(NSString *imageTag))block
{
  dispatch_async(dispatch_get_main_queue(), ^{
    NSString *imageTag = [self storeImage:image];
    if (block) {
      block(imageTag);
    }
  });
}

- (void)getImageForTag:(NSString *)imageTag withBlock:(void (^)(UIImage *image))block
{
  RCTAssert(block != nil, @"block must not be nil");
  dispatch_async(dispatch_get_main_queue(), ^{
    block([self imageForTag:imageTag]);
  });
}

// TODO (#5906496): Name could be more explicit - something like getBase64EncodedJPEGDataForTag:?
RCT_EXPORT_METHOD(getBase64ForTag:(NSString *)imageTag
                  successCallback:(RCTResponseSenderBlock)successCallback
                  errorCallback:(RCTResponseErrorBlock)errorCallback)
{
  [self getImageForTag:imageTag withBlock:^(UIImage *image) {
    if (!image) {
      errorCallback(RCTErrorWithMessage([NSString stringWithFormat:@"Invalid imageTag: %@", imageTag]));
      return;
    }
    dispatch_async(_methodQueue, ^{
      NSData *imageData = UIImageJPEGRepresentation(image, 1.0);
      NSString *base64 = [imageData base64EncodedStringWithOptions:NSDataBase64EncodingEndLineWithLineFeed];
      successCallback(@[[base64 stringByReplacingOccurrencesOfString:@"\n" withString:@""]]);
    });
  }];
}

RCT_EXPORT_METHOD(addImageFromBase64:(NSString *)base64String
                  successCallback:(RCTResponseSenderBlock)successCallback
                  errorCallback:(RCTResponseErrorBlock)errorCallback)

{
  NSData *imageData = [[NSData alloc] initWithBase64EncodedString:base64String options:0];
  if (imageData) {
    UIImage *image = [[UIImage alloc] initWithData:imageData];
    [self storeImage:image withBlock:^(NSString *imageTag) {
      successCallback(@[imageTag]);
    }];
  } else {
    errorCallback(RCTErrorWithMessage(@"Failed to add image from base64String"));
  }
}

#pragma mark - RCTURLRequestHandler

- (BOOL)canHandleRequest:(NSURLRequest *)request
{
  return [@[@"rct-image-store"] containsObject:[request.URL.scheme lowercaseString]];
}

- (id)sendRequest:(NSURLRequest *)request
     withDelegate:(id<RCTURLRequestDelegate>)delegate
{
  NSString *imageTag = [request.URL absoluteString];
  [self getImageForTag:imageTag withBlock:^(UIImage *image) {
    if (!image) {
      NSError *error = RCTErrorWithMessage([NSString stringWithFormat:@"Invalid imageTag: %@", imageTag]);
      [delegate URLRequest:request 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:request didReceiveResponse:response];
    [delegate URLRequest:request didReceiveData:imageData];
    [delegate URLRequest:request didCompleteWithError:nil];
  }];
  return request;
}

@end

@implementation RCTBridge (RCTImageStoreManager)

- (RCTImageStoreManager *)imageStoreManager
{
  return self.modules[RCTBridgeModuleNameForClass([RCTImageStoreManager class])];
}

@end