Add RCTLibraryPathForURL in RCTUtil

Reviewed By: fromcelticpark

Differential Revision: D6445626

fbshipit-source-id: aa37c87f019eea85d76365b6be919adfafc3c27a

Libraries/Network/RCTFileRequestHandler.m

@@ -30,7 +30,7 @@ - (BOOL)canHandleRequest:(NSURLRequest *)request
 {
   return
   [request.URL.scheme caseInsensitiveCompare:@"file"] == NSOrderedSame
-  && !RCTIsLocalAssetURL(request.URL);
+  && !RCTIsBundleAssetURL(request.URL);
 }
 
 - (NSOperation *)sendRequest:(NSURLRequest *)request

RNTester/RNTesterUnitTests/RCTURLUtilsTests.m

@@ -90,4 +90,17 @@ - (void)testNilURLAppendQueryParam
   XCTAssertNil(result);
 }
 
+- (void)testIsLocalAssetsURLParam
+{
+  NSString *libraryAssetsPath = [RCTLibraryPath() stringByAppendingPathComponent:@"assets/foo.png"];
+  NSURL *libraryAssetsURL = [NSURL fileURLWithPath:libraryAssetsPath];
+  XCTAssertTrue(RCTIsLocalAssetURL(libraryAssetsURL));
+  NSString *bundleAssetsPath = [[[NSBundle mainBundle] resourcePath] stringByAppendingPathComponent:@"assets/foo.png"];
+  NSURL *bundleAssetsURL = [NSURL fileURLWithPath:bundleAssetsPath];
+  XCTAssertTrue(RCTIsLocalAssetURL(bundleAssetsURL));
+  NSString *otherAssetsPath = @"/assets/foo.png";
+  NSURL *otherAssetsURL = [NSURL fileURLWithPath:otherAssetsPath];
+  XCTAssertFalse(RCTIsLocalAssetURL(otherAssetsURL));
+}
+
 @end

React/Base/RCTUtils.h

@@ -115,6 +115,19 @@ RCT_EXTERN NSData *__nullable RCTGzipData(NSData *__nullable data, float level);
 // (or nil, if the URL does not specify a path within the main bundle)
 RCT_EXTERN NSString *__nullable RCTBundlePathForURL(NSURL *__nullable URL);
 
+// Returns the Path of Library directory
+RCT_EXTERN NSString *__nullable RCTLibraryPath(void);
+
+// Returns the relative path within the library for an absolute URL
+// (or nil, if the URL does not specify a path within the Library directory)
+RCT_EXTERN NSString *__nullable RCTLibraryPathForURL(NSURL *__nullable URL);
+
+// Determines if a given image URL refers to a image in bundle
+RCT_EXTERN BOOL RCTIsBundleAssetURL(NSURL *__nullable imageURL);
+
+// Determines if a given image URL refers to a image in library
+RCT_EXTERN BOOL RCTIsLibraryAssetURL(NSURL *__nullable imageURL);
+
 // Determines if a given image URL refers to a local image
 RCT_EXTERN BOOL RCTIsLocalAssetURL(NSURL *__nullable imageURL);
 

React/Base/RCTUtils.m

@@ -596,36 +596,66 @@ BOOL RCTIsGzippedData(NSData *__nullable data)
   return output;
 }
 
-NSString *__nullable RCTBundlePathForURL(NSURL *__nullable URL)
+static NSString *RCTRelativePathForURL(NSString *basePath, NSURL *__nullable URL)
 {
   if (!URL.fileURL) {
     // Not a file path
     return nil;
   }
   NSString *path = [NSString stringWithUTF8String:[URL fileSystemRepresentation]];
-  NSString *bundlePath = [[NSBundle mainBundle] resourcePath];
-  if (![path hasPrefix:bundlePath]) {
+  if (![path hasPrefix:basePath]) {
     // Not a bundle-relative file
     return nil;
   }
-  path = [path substringFromIndex:bundlePath.length];
+  path = [path substringFromIndex:basePath.length];
   if ([path hasPrefix:@"/"]) {
     path = [path substringFromIndex:1];
   }
   return path;
 }
 
-BOOL RCTIsLocalAssetURL(NSURL *__nullable imageURL)
+NSString *__nullable RCTLibraryPath(void)
 {
-  NSString *name = RCTBundlePathForURL(imageURL);
-  if (!name) {
-    return NO;
-  }
+    static NSString *libraryPath = nil;
+    static dispatch_once_t onceToken;
+    dispatch_once(&onceToken, ^{
+        libraryPath = [NSSearchPathForDirectoriesInDomains(NSLibraryDirectory, NSUserDomainMask, YES) lastObject];
+    });
+    return libraryPath;
+}
 
-  NSString *extension = [name pathExtension];
+NSString *__nullable RCTBundlePathForURL(NSURL *__nullable URL)
+{
+  return RCTRelativePathForURL([[NSBundle mainBundle] resourcePath], URL);
+
+}
+
+NSString *__nullable RCTLibraryPathForURL(NSURL *__nullable URL)
+{
+  return RCTRelativePathForURL(RCTLibraryPath(), URL);
+}
+
+static BOOL RCTIsImageAssetsPath(NSString *path)
+{
+  NSString *extension = [path pathExtension];
   return [extension isEqualToString:@"png"] || [extension isEqualToString:@"jpg"];
 }
 
+BOOL RCTIsBundleAssetURL(NSURL *__nullable imageURL)
+{
+  return RCTIsImageAssetsPath(RCTBundlePathForURL(imageURL));
+}
+
+BOOL RCTIsLibraryAssetURL(NSURL *__nullable imageURL)
+{
+  return RCTIsImageAssetsPath(RCTLibraryPathForURL(imageURL));
+}
+
+BOOL RCTIsLocalAssetURL(NSURL *__nullable imageURL)
+{
+  return RCTIsBundleAssetURL(imageURL) || RCTIsLibraryAssetURL(imageURL);
+}
+
 static NSString *bundleName(NSBundle *bundle)
 {
   NSString *name = bundle.infoDictionary[@"CFBundleName"];