131 lines
4.7 KiB
Objective-C
131 lines
4.7 KiB
Objective-C
#import "Downloader.h"
|
|
|
|
@implementation RNFSDownloadParams
|
|
|
|
@end
|
|
|
|
@interface RNFSDownloader()
|
|
|
|
@property (copy) RNFSDownloadParams* params;
|
|
|
|
@property (retain) NSURLSession* session;
|
|
@property (retain) NSURLSessionTask* task;
|
|
@property (retain) NSNumber* statusCode;
|
|
@property (retain) NSNumber* lastProgressValue;
|
|
@property (retain) NSNumber* contentLength;
|
|
@property (retain) NSNumber* bytesWritten;
|
|
|
|
@property (retain) NSFileHandle* fileHandle;
|
|
|
|
@end
|
|
|
|
@implementation RNFSDownloader
|
|
|
|
- (void)downloadFile:(RNFSDownloadParams*)params
|
|
{
|
|
_params = params;
|
|
|
|
_bytesWritten = 0;
|
|
|
|
NSURL* url = [NSURL URLWithString:_params.fromUrl];
|
|
|
|
[[NSFileManager defaultManager] createFileAtPath:_params.toFile contents:nil attributes:nil];
|
|
_fileHandle = [NSFileHandle fileHandleForWritingAtPath:_params.toFile];
|
|
|
|
if (!_fileHandle) {
|
|
NSError* error = [NSError errorWithDomain:@"Downloader" code:NSURLErrorFileDoesNotExist
|
|
userInfo:@{NSLocalizedDescriptionKey: [NSString stringWithFormat: @"Failed to create target file at path: %@", _params.toFile]}];
|
|
|
|
return _params.errorCallback(error);
|
|
} else {
|
|
[_fileHandle closeFile];
|
|
}
|
|
|
|
NSURLSessionConfiguration *config;
|
|
if (_params.background) {
|
|
NSString *uuid = [[NSUUID UUID] UUIDString];
|
|
config = [NSURLSessionConfiguration backgroundSessionConfigurationWithIdentifier:uuid];
|
|
config.discretionary = _params.discretionary;
|
|
} else {
|
|
config = [NSURLSessionConfiguration defaultSessionConfiguration];
|
|
}
|
|
|
|
config.HTTPAdditionalHeaders = _params.headers;
|
|
|
|
_session = [NSURLSession sessionWithConfiguration:config delegate:self delegateQueue:nil];
|
|
_task = [_session downloadTaskWithURL:url];
|
|
[_task resume];
|
|
}
|
|
|
|
- (void)URLSession:(NSURLSession *)session downloadTask:(NSURLSessionDownloadTask *)downloadTask didWriteData:(int64_t)bytesWritten totalBytesWritten:(int64_t)totalBytesWritten totalBytesExpectedToWrite:(int64_t)totalBytesExpectedToWrite
|
|
{
|
|
NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse *)downloadTask.response;
|
|
if (!_statusCode) {
|
|
_statusCode = [NSNumber numberWithLong:httpResponse.statusCode];
|
|
_contentLength = [NSNumber numberWithLong:httpResponse.expectedContentLength];
|
|
return _params.beginCallback(_statusCode, _contentLength, httpResponse.allHeaderFields);
|
|
}
|
|
|
|
if ([_statusCode isEqualToNumber:[NSNumber numberWithInt:200]]) {
|
|
_bytesWritten = @(totalBytesWritten);
|
|
|
|
if (_params.progressDivider.integerValue <= 0) {
|
|
return _params.progressCallback(_contentLength, _bytesWritten);
|
|
} else {
|
|
double doubleBytesWritten = (double)[_bytesWritten longValue];
|
|
double doubleContentLength = (double)[_contentLength longValue];
|
|
double doublePercents = doubleBytesWritten / doubleContentLength * 100;
|
|
NSNumber* progress = [NSNumber numberWithUnsignedInt: floor(doublePercents)];
|
|
if ([progress unsignedIntValue] % [_params.progressDivider integerValue] == 0) {
|
|
if (([progress unsignedIntValue] != [_lastProgressValue unsignedIntValue]) || ([_bytesWritten unsignedIntegerValue] == [_contentLength longValue])) {
|
|
NSLog(@"---Progress callback EMIT--- %zu", [progress unsignedIntValue]);
|
|
_lastProgressValue = [NSNumber numberWithUnsignedInt:[progress unsignedIntValue]];
|
|
return _params.progressCallback(_contentLength, _bytesWritten);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
- (void)URLSession:(NSURLSession *)session downloadTask:(NSURLSessionDownloadTask *)downloadTask didFinishDownloadingToURL:(NSURL *)location
|
|
{
|
|
NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse *)downloadTask.response;
|
|
if (!_statusCode) {
|
|
_statusCode = [NSNumber numberWithLong:httpResponse.statusCode];
|
|
}
|
|
NSURL *destURL = [NSURL fileURLWithPath:_params.toFile];
|
|
NSFileManager *fm = [NSFileManager defaultManager];
|
|
NSError *error = nil;
|
|
[fm removeItemAtURL:destURL error:nil]; // Remove file at destination path, if it exists
|
|
[fm moveItemAtURL:location toURL:destURL error:&error];
|
|
if (error) {
|
|
NSLog(@"RNFS download: unable to move tempfile to destination. %@, %@", error, error.userInfo);
|
|
}
|
|
|
|
return _params.completeCallback(_statusCode, _bytesWritten);
|
|
}
|
|
|
|
- (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task didCompleteWithError:(NSError *)error
|
|
{
|
|
if (error && error.code != -999) {
|
|
_params.errorCallback(error);
|
|
}
|
|
}
|
|
|
|
- (void)stopDownload
|
|
{
|
|
if (_task.state == NSURLSessionTaskStateRunning) {
|
|
[_task cancel];
|
|
|
|
NSError *error = [NSError errorWithDomain:@"RNFS"
|
|
code:@"Aborted"
|
|
userInfo:@{
|
|
NSLocalizedDescriptionKey: @"Download has been aborted"
|
|
}];
|
|
|
|
return _params.errorCallback(error);
|
|
}
|
|
}
|
|
|
|
@end
|