Skip to content

Instantly share code, notes, and snippets.

@ntdiary
Created July 27, 2022 07:02
Show Gist options
  • Save ntdiary/6ce68557621b5b3fd25249220aee658d to your computer and use it in GitHub Desktop.
Save ntdiary/6ce68557621b5b3fd25249220aee658d to your computer and use it in GitHub Desktop.
Copy from RCTFileRequestHandler.mm
//
// OriginImageRequestHandler.mm
//
#import <Foundation/Foundation.h>
#import <ReactCommon/RCTTurboModule.h>
#import <MobileCoreServices/MobileCoreServices.h>
#import <React/RCTUtils.h>
#import "OriginImageRequestHandler.h"
@interface OriginImageRequestHandler() <RCTTurboModule>
@end
@implementation OriginImageRequestHandler
{
NSOperationQueue *_fileQueue;
}
RCT_EXPORT_MODULE()
- (void)invalidate
{
[_fileQueue cancelAllOperations];
_fileQueue = nil;
}
- (BOOL)canHandleRequest:(NSURLRequest *)request
{
NSLog(@"home path is %@", NSHomeDirectory());
return [request.URL.scheme caseInsensitiveCompare:@"file"] == NSOrderedSame
&& RCTIsLocalAssetURL(request.URL)
&& !RCTIsBundleAssetURL(request.URL);
}
- (NSOperation *)sendRequest:(NSURLRequest *)request
withDelegate:(id<RCTURLRequestDelegate>)delegate
{
// Lazy setup
if (!_fileQueue) {
_fileQueue = [NSOperationQueue new];
_fileQueue.maxConcurrentOperationCount = 4;
}
__weak __block NSBlockOperation *weakOp;
__block NSBlockOperation *op = [NSBlockOperation blockOperationWithBlock:^{
// Get content length
NSError *error = nil;
NSFileManager *fileManager = [NSFileManager new];
NSDictionary<NSString *, id> *fileAttributes = [fileManager attributesOfItemAtPath:request.URL.path error:&error];
if (!fileAttributes) {
[delegate URLRequest:weakOp didCompleteWithError:error];
return;
}
// Get mime type
NSString *fileExtension = [request.URL pathExtension];
NSString *UTI = (__bridge_transfer NSString *)UTTypeCreatePreferredIdentifierForTag(
kUTTagClassFilenameExtension, (__bridge CFStringRef)fileExtension, NULL);
NSString *contentType = (__bridge_transfer NSString *)UTTypeCopyPreferredTagWithClass(
(__bridge CFStringRef)UTI, kUTTagClassMIMEType);
// Send response
NSURLResponse *response = [[NSURLResponse alloc] initWithURL:request.URL
MIMEType:contentType
expectedContentLength:[fileAttributes[NSFileSize] ?: @-1 integerValue]
textEncodingName:nil];
[delegate URLRequest:weakOp didReceiveResponse:response];
// Load data
NSData *data = [NSData dataWithContentsOfURL:request.URL
options:NSDataReadingMappedIfSafe
error:&error];
if (data) {
[delegate URLRequest:weakOp didReceiveData:data];
}
[delegate URLRequest:weakOp didCompleteWithError:error];
}];
weakOp = op;
[_fileQueue addOperation:op];
return op;
}
- (void)cancelRequest:(NSOperation *)op
{
[op cancel];
}
- (std::shared_ptr<facebook::react::TurboModule>)getTurboModule:
(const facebook::react::ObjCTurboModule::InitParams &)params
{
return nullptr;
}
- (float)handlerPriority
{
return 4;
}
@end
Class OriginImageRequestHandlerCls(void) {
return OriginImageRequestHandler.class;
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment