// // VydiaRNFileUploader.m // Vydia // // Created by Kenneth Leland on 12/8/16. // Copyright © 2016 Vydia. All rights reserved. // #import #import #import #import @interface VydiaRNFileUploader : RCTEventEmitter { NSMutableDictionary *_responsesData; } @end @implementation VydiaRNFileUploader RCT_EXPORT_MODULE(); @synthesize bridge = _bridge; static int uploadId = 0; static RCTEventEmitter* staticEventEmitter = nil; static NSString *BACKGROUND_SESSION_ID = @"VydiaRNFileUploader"; NSURLSession *_urlSession = nil; -(id) init { self = [super init]; if (self) { staticEventEmitter = self; _responsesData = [NSMutableDictionary dictionary]; } return self; } - (void)_sendEventWithName:(NSString *)eventName body:(id)body { if (staticEventEmitter == nil) return; [staticEventEmitter sendEventWithName:eventName body:body]; } - (NSArray *)supportedEvents { return @[@"RNFileUploader-progress", @"RNFileUploader-error", @"RNFileUploader-completed"]; } /* Gets file information for the path specified. Example valid path is: file:///var/mobile/Containers/Data/Application/3C8A0EFB-A316-45C0-A30A-761BF8CCF2F8/tmp/trim.A5F76017-14E9-4890-907E-36A045AF9436.MOV Returns an object such as: {mimeType: "video/quicktime", size: 2569900, exists: true, name: "trim.AF9A9225-FC37-416B-A25B-4EDB8275A625.MOV", extension: "MOV"} */ RCT_EXPORT_METHOD(getFileInfo:(NSString *)path resolve:(RCTPromiseResolveBlock)resolve reject:(RCTPromiseRejectBlock)reject) { @try { NSURL *fileUri = [NSURL URLWithString: path]; NSString *pathWithoutProtocol = [fileUri path]; NSString *name = [fileUri lastPathComponent]; NSString *extension = [name pathExtension]; bool exists = [[NSFileManager defaultManager] fileExistsAtPath:pathWithoutProtocol]; NSMutableDictionary *params = [NSMutableDictionary dictionaryWithObjectsAndKeys: name, @"name", nil]; [params setObject:extension forKey:@"extension"]; [params setObject:[NSNumber numberWithBool:exists] forKey:@"exists"]; if (exists) { [params setObject:[self guessMIMETypeFromFileName:name] forKey:@"mimeType"]; NSError* error; NSDictionary *attributes = [[NSFileManager defaultManager] attributesOfItemAtPath:pathWithoutProtocol error:&error]; if (error == nil) { unsigned long long fileSize = [attributes fileSize]; [params setObject:[NSNumber numberWithLong:fileSize] forKey:@"size"]; } } resolve(params); } @catch (NSException *exception) { reject(@"RN Uploader", exception.name, nil); } } // Borrowed from http://stackoverflow.com/questions/2439020/wheres-the-iphone-mime-type-database - (NSString *)guessMIMETypeFromFileName: (NSString *)fileName { CFStringRef UTI = UTTypeCreatePreferredIdentifierForTag(kUTTagClassFilenameExtension, (__bridge CFStringRef)[fileName pathExtension], NULL); CFStringRef MIMEType = UTTypeCopyPreferredTagWithClass(UTI, kUTTagClassMIMEType); CFRelease(UTI); if (!MIMEType) { return @"application/octet-stream"; } return (__bridge NSString *)(MIMEType); } /* * Starts a file upload. * Options are passed in as the first argument as a js hash: * { * url: string. url to post to. * path: string. path to the file on the device * headers: hash of name/value header pairs * } * * Returns a promise with the string ID of the upload. */ RCT_EXPORT_METHOD(startUpload:(NSDictionary *)options resolve:(RCTPromiseResolveBlock)resolve reject:(RCTPromiseRejectBlock)reject) { int thisUploadId; @synchronized(self.class) { thisUploadId = uploadId++; } NSString *uploadUrl = options[@"url"]; NSString *fileURI = options[@"path"]; NSString *method = options[@"method"] ?: @"POST"; NSString *uploadType = options[@"type"] ?: @"raw"; NSString *fieldName = options[@"field"]; NSString *customUploadId = options[@"customUploadId"]; NSDictionary *headers = options[@"headers"]; NSDictionary *parameters = options[@"parameters"]; @try { NSURL *requestUrl = [NSURL URLWithString: uploadUrl]; if (requestUrl == nil) { @throw @"Request cannot be nil"; } NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:requestUrl]; [request setHTTPMethod: method]; [headers enumerateKeysAndObjectsUsingBlock:^(id _Nonnull key, id _Nonnull val, BOOL * _Nonnull stop) { if ([val respondsToSelector:@selector(stringValue)]) { val = [val stringValue]; } if ([val isKindOfClass:[NSString class]]) { [request setValue:val forHTTPHeaderField:key]; } }]; NSURLSessionDataTask *uploadTask; if ([uploadType isEqualToString:@"multipart"]) { NSString *uuidStr = [[NSUUID UUID] UUIDString]; [request setValue:[NSString stringWithFormat:@"multipart/form-data; boundary=%@", uuidStr] forHTTPHeaderField:@"Content-Type"]; NSData *httpBody = [self createBodyWithBoundary:uuidStr path:fileURI parameters: parameters fieldName:fieldName]; [request setHTTPBody: httpBody]; // I am sorry about warning, but Upload tasks from NSData are not supported in background sessions. uploadTask = [[self urlSession] uploadTaskWithRequest:request fromData: nil]; } else { if (parameters.count > 0) { reject(@"RN Uploader", @"Parameters supported only in multipart type", nil); return; } uploadTask = [[self urlSession] uploadTaskWithRequest:request fromFile:[NSURL URLWithString: fileURI]]; } uploadTask.taskDescription = customUploadId ? customUploadId : [NSString stringWithFormat:@"%i", thisUploadId]; [uploadTask resume]; resolve(uploadTask.taskDescription); } @catch (NSException *exception) { reject(@"RN Uploader", exception.name, nil); } } - (NSData *)createBodyWithBoundary:(NSString *)boundary path:(NSString *)path parameters:(NSDictionary *)parameters fieldName:(NSString *)fieldName { NSMutableData *httpBody = [NSMutableData data]; // resolve path NSURL *fileUri = [NSURL URLWithString: path]; NSString *pathWithoutProtocol = [fileUri path]; NSData *data = [[NSFileManager defaultManager] contentsAtPath:pathWithoutProtocol]; NSString *filename = [path lastPathComponent]; NSString *mimetype = [self guessMIMETypeFromFileName:path]; [parameters enumerateKeysAndObjectsUsingBlock:^(NSString *parameterKey, NSString *parameterValue, BOOL *stop) { [httpBody appendData:[[NSString stringWithFormat:@"--%@\r\n", boundary] dataUsingEncoding:NSUTF8StringEncoding]]; [httpBody appendData:[[NSString stringWithFormat:@"Content-Disposition: form-data; name=\"%@\"\r\n\r\n", parameterKey] dataUsingEncoding:NSUTF8StringEncoding]]; [httpBody appendData:[[NSString stringWithFormat:@"%@\r\n", parameterValue] dataUsingEncoding:NSUTF8StringEncoding]]; }]; [httpBody appendData:[[NSString stringWithFormat:@"--%@\r\n", boundary] dataUsingEncoding:NSUTF8StringEncoding]]; [httpBody appendData:[[NSString stringWithFormat:@"Content-Disposition: form-data; name=\"%@\"; filename=\"%@\"\r\n", fieldName, filename] dataUsingEncoding:NSUTF8StringEncoding]]; [httpBody appendData:[[NSString stringWithFormat:@"Content-Type: %@\r\n\r\n", mimetype] dataUsingEncoding:NSUTF8StringEncoding]]; [httpBody appendData:data]; [httpBody appendData:[@"\r\n" dataUsingEncoding:NSUTF8StringEncoding]]; [httpBody appendData:[[NSString stringWithFormat:@"--%@--\r\n", boundary] dataUsingEncoding:NSUTF8StringEncoding]]; return httpBody; } - (NSURLSession *)urlSession { if(_urlSession == nil) { NSURLSessionConfiguration *sessionConfigurationt = [NSURLSessionConfiguration backgroundSessionConfigurationWithIdentifier:BACKGROUND_SESSION_ID]; _urlSession = [NSURLSession sessionWithConfiguration:sessionConfigurationt delegate:self delegateQueue:nil]; } return _urlSession; } #pragma NSURLSessionTaskDelegate - (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task didCompleteWithError:(NSError *)error { NSMutableDictionary *data = [NSMutableDictionary dictionaryWithObjectsAndKeys:task.taskDescription, @"id", nil]; NSURLSessionDataTask *uploadTask = (NSURLSessionDataTask *)task; NSHTTPURLResponse *response = (NSHTTPURLResponse *)uploadTask.response; if (response != nil) { [data setObject:[NSNumber numberWithInteger:response.statusCode] forKey:@"responseCode"]; } //Add data that was collected earlier by the didReceiveData method NSMutableData *responseData = _responsesData[@(task.taskIdentifier)]; if (responseData) { [_responsesData removeObjectForKey:@(task.taskIdentifier)]; NSString *response = [[NSString alloc] initWithData:responseData encoding:NSUTF8StringEncoding]; [data setObject:response forKey:@"responseBody"]; } else { [data setObject:[NSNull null] forKey:@"responseBody"]; } if (error == nil) { [self _sendEventWithName:@"RNFileUploader-completed" body:data]; } else { [data setObject:error.localizedDescription forKey:@"error"]; [self _sendEventWithName:@"RNFileUploader-error" body:data]; } } - (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task didSendBodyData:(int64_t)bytesSent totalBytesSent:(int64_t)totalBytesSent totalBytesExpectedToSend:(int64_t)totalBytesExpectedToSend { float progress = -1; if (totalBytesExpectedToSend > 0) //see documentation. For unknown size it's -1 (NSURLSessionTransferSizeUnknown) { progress = 100.0 * (float)totalBytesSent / (float)totalBytesExpectedToSend; } [self _sendEventWithName:@"RNFileUploader-progress" body:@{ @"id": task.taskDescription, @"progress": [NSNumber numberWithFloat:progress] }]; } - (void)URLSession:(NSURLSession *)session dataTask:(NSURLSessionDataTask *)dataTask didReceiveData:(NSData *)data { if (!data.length) { return; } //Hold returned data so it can be picked up by the didCompleteWithError method later NSMutableData *responseData = _responsesData[@(dataTask.taskIdentifier)]; if (!responseData) { responseData = [NSMutableData dataWithData:data]; _responsesData[@(dataTask.taskIdentifier)] = responseData; } else { [responseData appendData:data]; } } @end