RCTFileTransfer.m

//
//  RCTFileTransfer.m
//  react-native-file-transfer
//
//  Created by Kamil Pękala on 30.04.2015
//  Copyright (c) 2015 Kamil Pękala. All rights reserved.
//

#import "RCTBridgeModule.h"
#import "RCTUtils.h"
#import <AssetsLibrary/AssetsLibrary.h>
#import <UIKit/UIKit.h>

@interface FileTransfer : NSObject <RCTBridgeModule>
- (NSMutableURLRequest *)getMultiPartRequest:(NSData *)data serverUrl:(NSString *)server requestData:(NSDictionary *)requestData requestHeaders:(NSDictionary *)requestHeaders mimeType:(NSString *)mimeType fileName:(NSString *)fileName fileKey:(NSString *)fileKey;
- (void)uploadAssetsLibrary:(NSDictionary *)input callback:(RCTResponseSenderBlock)callback;
- (void)uploadUri:(NSDictionary *)input callback:(RCTResponseSenderBlock)callback;
- (void)uploadFile:(NSDictionary *)input callback:(RCTResponseSenderBlock)callback;
- (void)sendData:(NSData *)data withOptions:(NSDictionary *)input callback:(RCTResponseSenderBlock)callback;
@end

@implementation FileTransfer

RCT_EXPORT_MODULE();

RCT_EXPORT_METHOD(upload:(NSDictionary *)input callback:(RCTResponseSenderBlock)callback)
{
  NSString *uri = input[@"uri"];
  if([uri hasPrefix:@"assets-library"]){
    [self uploadAssetsLibrary:input callback:callback];
  }
  else if([uri hasPrefix:@"data:"]){
    [self uploadUri:input callback:callback];
  }
  else if([uri hasPrefix:@"file:"]){
    [self uploadUri:input callback:callback];
  }
  else if ([uri isAbsolutePath]) {
    [self uploadFile:input callback:callback];
  }
  else{
    callback(@[RCTMakeError(@"Unknown protocol for key: 'file'", nil, nil)]);
  }
}

- (void)uploadAssetsLibrary:(NSDictionary *)input callback:(RCTResponseSenderBlock)callback
{

  NSURL *assetUrl = [[NSURL alloc] initWithString:input[@"uri"]];
  ALAssetsLibrary *library = [[ALAssetsLibrary alloc] init];
  [library assetForURL:assetUrl resultBlock:^(ALAsset *asset) {

    ALAssetRepresentation *rep = [asset defaultRepresentation];

    CGImageRef fullScreenImageRef = [rep fullScreenImage];
    UIImage *image = [UIImage imageWithCGImage:fullScreenImageRef];
    NSData *fileData = UIImagePNGRepresentation(image);

    [self sendData:fileData withOptions:input callback:callback];

    //    Byte *buffer = (Byte*)malloc(rep.size);
    //    NSUInteger buffered = [rep getBytes:buffer fromOffset:0.0 length:rep.size error:nil];
    //
    //    NSData *fileData = [NSData dataWithBytesNoCopy:buffer length:buffered freeWhenDone:YES];
    // NSDictionary* requestData = [input objectForKey:@"data"];
    // NSMutableURLRequest* req = [self getMultiPartRequest:fileData serverUrl:uploadUrl requestData:requestData mimeType:mimeType fileName:fileName];
    //
    // NSHTTPURLResponse *response = nil;
    // NSData *returnData = [NSURLConnection sendSynchronousRequest:req returningResponse:&response error:nil];
    // NSInteger statusCode = [response statusCode];
    // NSString *returnString = [[NSString alloc] initWithData:returnData encoding:NSUTF8StringEncoding];
    //
    // NSDictionary *res=[[NSDictionary alloc] initWithObjectsAndKeys:[NSNumber numberWithInteger:statusCode],@"status",returnString,@"data",nil];
    //
    // callback(@[res]);
  } failureBlock:^(NSError *error) {
    callback(@[RCTMakeError(@"Error loading library asset", nil, nil)]);
  }];
}

- (void)uploadFile:(NSDictionary *)input callback:(RCTResponseSenderBlock)callback
{
  NSString *filePath = input[@"uri"];
  NSData *fileData = [NSData dataWithContentsOfFile:filePath];

  [self sendData:fileData withOptions:input callback:callback];
}

- (void)uploadUri:(NSDictionary *)input callback:(RCTResponseSenderBlock)callback
{
  NSString *dataUrlString = input[@"uri"];
  NSURL *dataUrl = [[NSURL alloc] initWithString:dataUrlString];
  NSData *fileData = [NSData dataWithContentsOfURL: dataUrl];

  [self sendData:fileData withOptions:input callback:callback];
}

- (void)sendData:(NSData *)data withOptions:(NSDictionary *)input callback:(RCTResponseSenderBlock)callback
{
  NSString *fileName = input[@"fileName"];
  NSString *mimeType = input[@"mimeType"];
  NSString *uploadUrl = input[@"uploadUrl"];
  NSString *fileKey = input[@"fileKey"];
    
  if (!fileKey) fileKey = @"file";

  NSDictionary* requestData = [input objectForKey:@"data"];
  NSDictionary* requestHeaders = [input objectForKey:@"headers"];
  NSMutableURLRequest* req = [self getMultiPartRequest:data serverUrl:uploadUrl requestData:requestData requestHeaders:requestHeaders mimeType:mimeType fileName:fileName fileKey:fileKey];

  NSHTTPURLResponse *response = nil;
  NSData *returnData = [NSURLConnection sendSynchronousRequest:req returningResponse:&response error:nil];
  NSInteger statusCode = [response statusCode];
  NSString *returnString = [[NSString alloc] initWithData:returnData encoding:NSUTF8StringEncoding];

  NSDictionary *res=[[NSDictionary alloc] initWithObjectsAndKeys:[NSNumber numberWithInteger:statusCode],@"status",returnString,@"data",nil];

  callback(@[[NSNull null], res]);
}

- (NSMutableURLRequest *)getMultiPartRequest:(NSData *)data serverUrl:(NSString *)server requestData:(NSDictionary *)requestData requestHeaders:(NSDictionary *)requestHeaders mimeType:(NSString *)mimeType fileName:(NSString *)fileName fileKey:(NSString *)fileKey
{
  NSURL* url = [NSURL URLWithString:server];
  NSMutableURLRequest* req = [NSMutableURLRequest requestWithURL:url];

  [req setHTTPMethod:@"POST"];

  NSString* formBoundaryString = @"----react.file.transfer.form.boundary";

  NSString* contentType = [NSString stringWithFormat:@"multipart/form-data; boundary=%@", formBoundaryString];
  [req setValue:contentType forHTTPHeaderField:@"Content-Type"];

  NSData* formBoundaryData = [[NSString stringWithFormat:@"--%@\r\n", formBoundaryString] dataUsingEncoding:NSUTF8StringEncoding];
  NSMutableData* requestBody = [NSMutableData data];

  for (NSString* key in requestHeaders) {
    id val = [requestHeaders objectForKey:key];
    if ([val respondsToSelector:@selector(stringValue)]) {
      val = [val stringValue];
    }
    if (![val isKindOfClass:[NSString class]]) {
      continue;
    }

    [req setValue:val forHTTPHeaderField:key];
  }

  for (NSString* key in requestData) {
    id val = [requestData objectForKey:key];
    if ([val respondsToSelector:@selector(stringValue)]) {
      val = [val stringValue];
    }

    if (![val isKindOfClass:[NSString class]]) {
      NSError *error;
      NSData *jsonData = [NSJSONSerialization dataWithJSONObject:val
                                              options:(NSJSONWritingOptions)0
                                              error:&error];
      val = [[NSString alloc] initWithData:jsonData encoding:NSUTF8StringEncoding];
    }

    [requestBody appendData:formBoundaryData];
    [requestBody appendData:[[NSString stringWithFormat:@"Content-Disposition: form-data; name=\"%@\"\r\n\r\n", key] dataUsingEncoding:NSUTF8StringEncoding]];
    [requestBody appendData:[val dataUsingEncoding:NSUTF8StringEncoding]];
    [requestBody appendData:[@"\r\n" dataUsingEncoding : NSUTF8StringEncoding]];
  }

  [requestBody appendData:formBoundaryData];
  [requestBody appendData:[[NSString stringWithFormat:@"Content-Disposition: form-data; name=\"%@\"; filename=\"%@\"\r\n", fileKey, fileName] dataUsingEncoding:NSUTF8StringEncoding]];
  if (mimeType != nil) {
    [requestBody appendData:[[NSString stringWithFormat:@"Content-Type: %@\r\n", mimeType] dataUsingEncoding:NSUTF8StringEncoding]];
  }
  [requestBody appendData:[[NSString stringWithFormat:@"Content-Length: %ld\r\n\r\n", (long)[data length]] dataUsingEncoding:NSUTF8StringEncoding]];

  NSData* afterFile = [[NSString stringWithFormat:@"\r\n--%@--\r\n", formBoundaryString] dataUsingEncoding:NSUTF8StringEncoding];

  long long totalPayloadLength = [requestBody length] + [data length] + [afterFile length];
  [req setValue:[[NSNumber numberWithLongLong:totalPayloadLength] stringValue] forHTTPHeaderField:@"Content-Length"];

  [requestBody appendData:data];
  [requestBody appendData:afterFile];
  [req setHTTPBody:requestBody];
  return req;
}

@end